Searched refs:IsFloatingPointType (Results 1 – 16 of 16) sorted by relevance
75 if (primitive_util::IsFloatingPointType(element_type)) { in TEST_P()102 if (primitive_util::IsFloatingPointType(element_type)) { in TEST_P()127 if (primitive_util::IsFloatingPointType(element_type)) { in TEST_P()
132 if (primitive_util::IsFloatingPointType(prim_type)) { in TestElementwise2D()
57 xla::primitive_util::IsFloatingPointType(src_type_) && in Compile()58 xla::primitive_util::IsFloatingPointType(dst_type_), in Compile()
124 if (primitive_util::IsFloatingPointType(operand_types[0])) { in CreateScalarComparisonComputation()
36 !(primitive_util::IsFloatingPointType(type) || in ConstantR0WithType()
209 EXPECT_EQ(b.first_error().ok(), primitive_util::IsFloatingPointType(ty)); in XLA_TEST_F()
414 if (!primitive_util::IsFloatingPointType(type)) { in SelfAdjointEig()
59 if (!primitive_util::IsFloatingPointType(elem_ty)) { in EnsureOperandIsRealFp()
138 bool IsFloatingPointType(PrimitiveType type);
42 bool IsFloatingPointType(PrimitiveType type) { in IsFloatingPointType() function
325 return primitive_util::IsFloatingPointType(shape.element_type()); in ElementIsFloating()
242 if (primitive_util::IsFloatingPointType(to_type)) { in EmitIntegerUnaryOp()341 CHECK(primitive_util::IsFloatingPointType(from_type)) << from_type; in EmitFloatUnaryOp()374 if (primitive_util::IsFloatingPointType(to_type)) { in EmitFloatUnaryOp()393 CHECK(primitive_util::IsFloatingPointType(from_type)); in EmitFloatUnaryOp()1693 if (primitive_util::IsFloatingPointType(prim_type)) { in EmitElementalClamp()2191 } else if (primitive_util::IsFloatingPointType(primitive_type)) { in EmitElementalDot()2342 primitive_util::IsFloatingPointType(component_element_type)) in MakeElementGenerator()
303 if (!primitive_util::IsFloatingPointType(element_type) && in HandleRng()315 if (!primitive_util::IsFloatingPointType(element_type)) { in HandleRng()
2307 } else if (primitive_util::IsFloatingPointType(shape.element_type())) { in ParseDenseLiteral()2379 } else if (primitive_util::IsFloatingPointType(shape.element_type())) { in ParseSparseLiteral()
266 CHECK(primitive_util::IsFloatingPointType(primitive_type) || in EmitTiledLlvmIrGemv()831 (primitive_util::IsFloatingPointType(element_type) || in GetDotImplementationStrategy()
1660 if (primitive_util::IsFloatingPointType(operand_types[0])) { in Sort()