Searched defs:convert_element_type (Results 1 – 3 of 3) sorted by relevance
159 def convert_element_type(operand, dtype): function
311 def convert_element_type(cls, harness: primitive_harness.Harness): member in Jax2TfLimitation
408 def convert_element_type(operand: Array, new_dtype: DType) -> Array: function