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()
405 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16()406 void set_use_bfloat16(bool value) { use_bfloat16_ = value; } in set_use_bfloat16()409 PrimitiveType FloatType() const { return use_bfloat16_ ? BF16 : F32; } in FloatType()442 bool use_bfloat16_ = false; variable609 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR0Parameter()623 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR1Parameter()637 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR2Parameter()651 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR3Parameter()665 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateR4Parameter()679 if (use_bfloat16_ && literal.shape().element_type() == F32) { in CreateParameter()
60 HloEvaluatorTest() : use_bfloat16_(false) { InitializeFftData(); } in HloEvaluatorTest()64 if (use_bfloat16_) {75 if (use_bfloat16_) {190 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) { in HloEvaluatorTest()199 const bool use_bfloat16_; member in xla::__anona47f93e20111::HloEvaluatorTest430 use_bfloat16_ ? 0.031250 : 9.5367431640625E-7); in TEST_P()436 use_bfloat16_ ? 0.031250 : 9.5367431640625E-7); in TEST_P()1156 use_bfloat16_ ? expected_array_bf16 : expected_array); in TEST_P()1234 use_bfloat16_ ? expected_array_bf16 : expected_array); in TEST_P()4210 if (use_bfloat16_) { in TEST_P()[all …]