Searched refs:IsComplexType (Results 1 – 6 of 6) sorted by relevance
44 } else if (xla::primitive_util::IsComplexType(src_type_) && in Compile()45 !xla::primitive_util::IsComplexType(dst_type_)) { in Compile()
27 bool IsComplexType(PrimitiveType type) { return type == C64; } in IsComplexType() function
128 bool IsComplexType(PrimitiveType type);
313 return primitive_util::IsComplexType(shape.element_type()); in ElementIsComplex()
241 if (primitive_util::IsComplexType(to_type)) { in EmitIntegerUnaryOp()346 if (primitive_util::IsComplexType(to_type)) { in EmitFloatUnaryOp()466 primitive_util::IsComplexType(input_type) in EmitComplexUnaryOp()485 TF_RET_CHECK(primitive_util::IsComplexType(from_type)); in EmitComplexUnaryOp()487 TF_RET_CHECK(primitive_util::IsComplexType(to_type)); in EmitComplexUnaryOp()668 } else if (primitive_util::IsComplexType(operand_type)) { in EmitBinaryOp()1820 if (primitive_util::IsComplexType(primitive_type)) { in MakeElementGenerator()
452 if (primitive_util::IsComplexType(old_element_type) && in InferConvertShape()453 !primitive_util::IsComplexType(new_element_type)) { in InferConvertShape()475 if (primitive_util::IsComplexType(old_element_type) != in InferBitcastConvertShape()476 primitive_util::IsComplexType(new_element_type)) { in InferBitcastConvertShape()