Searched refs:IsFloatingPointType (Results 1 – 24 of 24) sorted by relevance
140 bool IsFloatingPointType(PrimitiveType type);184 IsFloatingPointType(component_type) ? OverflowExponent(component_type) in HigherPrecisionType()188 IsFloatingPointType(component_type) ? SignificandWidth(component_type) in HigherPrecisionType()236 if (primitive_util::IsFloatingPointType(from_type) && in CastPreservesValues()237 primitive_util::IsFloatingPointType(to_type)) { in CastPreservesValues()259 if (primitive_util::IsFloatingPointType(to_type)) { in CastPreservesValues()
79 bool IsFloatingPointType(PrimitiveType type) { in IsFloatingPointType() function
144 return primitive_util::IsFloatingPointType(primitive_type_); in IsFloatingPointPrimitiveType()
544 return primitive_util::IsFloatingPointType(shape.element_type()); in ElementIsFloating()
2000 if (primitive_util::IsFloatingPointType(ty)) { in IsAll()
105 if (primitive_util::IsFloatingPointType(element_type)) { in XLA_TEST_P()137 if (primitive_util::IsFloatingPointType(element_type)) { in XLA_TEST_P()
129 if (primitive_util::IsFloatingPointType(prim_type)) { in TestElementwise2D()
62 xla::primitive_util::IsFloatingPointType(src_type_) && in Compile()63 xla::primitive_util::IsFloatingPointType(dst_type_), in Compile()
69 primitive_util::IsFloatingPointType(lhs->shape().element_type())) { in InstructionMatchesPattern()87 CHECK(primitive_util::IsFloatingPointType(compare_type)); in ExpandInstruction()
35 primitive_util::IsFloatingPointType(input->shape().element_type())) { in RunOnComputation()
102 if (primitive_util::IsFloatingPointType(result_shape.element_type())) { in GetComputationForRng()
65 if (!primitive_util::IsFloatingPointType(src_ty)) { in EmitReducePrecisionIR()276 if (primitive_util::IsFloatingPointType(to_type)) { in EmitIntegerUnaryOp()380 CHECK(primitive_util::IsFloatingPointType(from_type)) << from_type; in EmitFloatUnaryOp()420 if (primitive_util::IsFloatingPointType(to_type)) { in EmitFloatUnaryOp()470 CHECK(primitive_util::IsFloatingPointType(from_type)); in EmitFloatUnaryOp()1898 if (primitive_util::IsFloatingPointType(prim_type)) { in EmitElementalClamp()2580 primitive_util::IsFloatingPointType(component_element_type)) in MakeElementGenerator()2764 } else if (primitive_util::IsFloatingPointType(primitive_type)) { in EmitMulAdd()
828 if (!primitive_util::IsFloatingPointType(element_type) && in HandleRng()840 if (!primitive_util::IsFloatingPointType(element_type)) { in HandleRng()986 !primitive_util::IsFloatingPointType(primitive_type) && in HandleIota()2335 if (primitive_util::IsFloatingPointType(operand_element_type)) { in CheckElementwiseInstruction()
441 if (!primitive_util::IsFloatingPointType(type) && in BuildEigh()
217 if (!primitive_util::IsFloatingPointType(from_type) && in MaybeUpcast()
2898 (primitive_util::IsFloatingPointType(ty) && in HandleMaximum()2952 (primitive_util::IsFloatingPointType(ty) && in HandleMinimum()
3722 } else if (primitive_util::IsFloatingPointType(shape.element_type())) { in ParseDenseLiteral()
51 if (!primitive_util::IsFloatingPointType(type) && in SelfAdjointEig()
36 !(primitive_util::IsFloatingPointType(type) || in ConstantR0WithType()
221 EXPECT_EQ(b.first_error().ok(), primitive_util::IsFloatingPointType(ty)); in XLA_TEST_F()
91 if (!primitive_util::IsFloatingPointType(elem_ty)) { in EnsureOperandIsRealFp()1741 if (!primitive_util::IsFloatingPointType(shape.element_type())) { in RegularizedIncompleteBeta()
421 CHECK(primitive_util::IsFloatingPointType(primitive_type) || in EmitTiledLlvmIrGemv()1207 (primitive_util::IsFloatingPointType(element_type) || in GetDotImplementationStrategy()1221 (primitive_util::IsFloatingPointType(element_type) || in GetDotImplementationStrategy()
281 primitive_util::IsFloatingPointType(conv_ty) ? conv_ty : F32; in FuseBiasOrSideInput()
4163 primitive_util::IsFloatingPointType(operand_element_type) in CompareTotalOrder()