Searched refs:use_bfloat16_ (Results 1 – 3 of 3) sorted by relevance
326 if (use_bfloat16_) { in ComputeAndCompareLiteralWithStatus()383 if (use_bfloat16_) { in ComputeAndCompareLiteralWithStatus()530 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarRelu()532 auto zero = use_bfloat16_ in CreateScalarRelu()543 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarMax()554 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarReluSensitivity()557 auto zero = use_bfloat16_ in CreateScalarReluSensitivity()603 return ConstantLiteral(builder, use_bfloat16_ in CreateConstantFromLiteral()619 if (!use_bfloat16_) { in MaybeConvertShapeToBfloat16()634 if (use_bfloat16_) { in MaybeConvertLiteralToBfloat16()
376 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16()377 void set_use_bfloat16(bool value) { use_bfloat16_ = value; } in set_use_bfloat16()380 PrimitiveType FloatType() const { return use_bfloat16_ ? BF16 : F32; } in FloatType()411 bool use_bfloat16_ = false; variable553 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR0Parameter()567 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR1Parameter()581 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR2Parameter()595 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR3Parameter()
59 HloEvaluatorTest() : use_bfloat16_(false) {} in HloEvaluatorTest()63 if (use_bfloat16_) {74 if (use_bfloat16_) {133 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) {} in HloEvaluatorTest()136 const bool use_bfloat16_; member in xla::__anon67d5331b0111::HloEvaluatorTest353 use_bfloat16_ ? 0.031250 : 9.5367431640625E-7); in TEST_P()359 use_bfloat16_ ? 0.031250 : 9.5367431640625E-7); in TEST_P()1079 use_bfloat16_ ? expected_array_bf16 : expected_array); in TEST_P()1157 use_bfloat16_ ? expected_array_bf16 : expected_array); in TEST_P()2983 if (use_bfloat16_) { in TEST_P()[all …]