Searched defs:new_element_type (Results 1 – 5 of 5) sorted by relevance
131 const DataType new_element_type) { in ConvertElementType()
362 Type new_element_type = result_element_type_fn(output); in InferReturnTypeComponentsForTFOp() local
433 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape()449 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape()
2202 PrimitiveType new_element_type) { in ConvertElementType()2216 PrimitiveType new_element_type) { in BitcastConvertType()4821 XlaOp ConvertElementType(const XlaOp operand, PrimitiveType new_element_type) { in ConvertElementType()4825 XlaOp BitcastConvertType(const XlaOp operand, PrimitiveType new_element_type) { in BitcastConvertType()
285 [&](RankedTensorType new_element_type) -> bool { in CanInferTensorListElementType()