Home
last modified time | relevance | path

Searched refs:BF16 (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc137 if (comp->root_instruction()->shape().element_type() == BF16) { in ConvertCalledComputations()
142 if (param->shape().element_type() == BF16) { in ConvertCalledComputations()
165 } else if (operand_types[i] == BF16) { in HandleMultipleOutputs()
173 } else if (output_types[i] == BF16) { in HandleMultipleOutputs()
186 if (operand_types[i] != BF16) { in HandleMultipleOutputs()
218 } else if (comp->root_instruction()->shape().element_type() == BF16) { in HandleMultipleOutputs()
225 } else if (param->shape().element_type() == BF16) { in HandleMultipleOutputs()
240 if (output_types[i] != BF16) { in HandleMultipleOutputs()
250 ShapeUtil::ChangeElementType(*subshape, BF16), gte)); in HandleMultipleOutputs()
276 } else if (hlo->operand(i)->shape().element_type() == BF16) { in HandleInstruction()
[all …]
Dbfloat16_propagation.cc57 if (OutputTypeAfterChange(fusion, index) == BF16) { in DetermineFusionComputationPrecision()
58 AddToOrRemoveFromBF16ChangeSet(root, index, BF16); in DetermineFusionComputationPrecision()
182 if (OutputTypeAfterChange(while_hlo, index) == BF16) { in DetermineWhileComputationsPrecision()
183 AddToOrRemoveFromBF16ChangeSet(body_root, index, BF16); in DetermineWhileComputationsPrecision()
210 if (subshape.element_type() != BF16 && subshape.element_type() != F32) { in AllUsersConsumeBF16()
224 if (value->shape().element_type() == BF16) { in AllUsersConsumeBF16()
250 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
258 if (OutputTypeAfterChange(cond_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
264 if (OutputTypeAfterChange(body_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
286 BF16) { in AllUsersConsumeBF16()
[all …]
Dhlo_element_type_converter_test.cc43 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
59 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
78 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
106 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
130 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
136 return inst->shape().element_type() == BF16 && in TEST_F()
157 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
188 HloElementTypeConverter converter(BF16, F32); in TEST_F()
Dbfloat16_conversion_folding_test.cc86 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
112 EXPECT_EQ(add0->shape().element_type(), BF16); in TEST_F()
113 EXPECT_EQ(add1->shape().element_type(), BF16); in TEST_F()
120 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
155 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
190 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
230 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
260 EXPECT_EQ(gte_b->shape().element_type(), BF16); in TEST_F()
263 EXPECT_EQ(a->shape().element_type(), BF16); in TEST_F()
266 EXPECT_EQ(ShapeUtil::GetSubshape(crs->shape(), {1}).element_type(), BF16); in TEST_F()
Dbfloat16_normalization_test.cc95 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
116 EXPECT_EQ(add0->shape().element_type(), BF16); in TEST_F()
123 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
153 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
184 Shape bf16_scalar_shape = ShapeUtil::MakeShape(BF16, {}); in TEST_F()
250 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
269 EXPECT_EQ(gte->shape().element_type(), BF16); in TEST_F()
278 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {1024}); in TEST_F()
279 Shape s32_shape = ShapeUtil::MakeShape(BF16, {1024}); in TEST_F()
299 EXPECT_EQ(gte->shape().element_type(), BF16); in TEST_F()
[all …]
Dbfloat16_conversion_folding.cc70 hlo->mutable_shape()->set_element_type(BF16); in FoldOutputConversions()
99 user->shape().element_type() == BF16) { in AllUsersAreF32ToBF16Converts()
117 operand->operand(0)->shape().element_type() == BF16 && in TryFoldBF16Conversions()
230 ->set_element_type(BF16); in HandleAllReduce()
Dbfloat16_support.cc36 return hlo.operand(0)->shape().element_type() == BF16; in SupportsBF16Operand()
54 return hlo.shape().element_type() == BF16; in SupportsBF16Output()
Dbfloat16_propagation_test.cc77 if (inst->shape().element_type() == BF16) { in OutputsBF16()
82 inst->users()[0]->shape().element_type() == BF16; in OutputsBF16()
902 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {4, 4}); in TEST_F()
931 EXPECT_EQ(add0->shape().element_type(), BF16); in TEST_F()
932 EXPECT_EQ(add1->shape().element_type(), BF16); in TEST_F()
967 BF16); in TEST_F()
969 BF16); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants.cc50 case BF16: in Epsilon()
72 case BF16: in MinFiniteValue()
88 case BF16: in MinPositiveNormalValue()
110 case BF16: in MaxFiniteValue()
127 case BF16: in NanValue()
Dcomparators_test.cc94 BuildComparatorAndComparisons<BF16>(this, /*compare_less_than=*/true, in XLA_TEST_F()
101 BuildComparatorAndComparisons<BF16>(this, /*compare_less_than=*/false, in XLA_TEST_F()
Dmath.cc121 case BF16: in IsNegZero()
216 return DoWithUpcastToF32(x, {BF16}, [](XlaOp x) { in Erfc()
233 return DoWithUpcastToF32(x, {BF16}, [](XlaOp x) { in Erf()
414 return DoWithUpcastToF32(input, {BF16, F16}, do_it); in Lgamma()
485 return DoWithUpcastToF32(input, {BF16, F16}, do_it); in Digamma()
Dcomparators.cc129 if (compare_type == BF16) { in CreateScalarComparisonComputation()
Dconstants.h51 case BF16: in ConstantR0WithType()
/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.cc33 case BF16: in SignificandWidth()
43 return type == F16 || type == F32 || type == F64 || type == BF16; in IsFloatingPointType()
72 case BF16: in BitWidth()
Dliteral_util.cc125 case BF16: in Zero()
162 case BF16: in One()
218 case BF16: in MinValue()
258 case BF16: in MaxValue()
380 case BF16: in GetFirstScalarLiteral()
Dprimitive_util.h124 return BF16;
229 struct PrimitiveTypeToNative<BF16> {
Dliteral.cc411 COPY_ELEMENTS(BF16, bfloat16); in CopyFrom()
541 case BF16: in CopySliceFrom()
766 case BF16: in Slice()
815 case BF16: in GetAsString()
871 case BF16: in GetSparseElementAsString()
1021 case BF16: in SortSparseElements()
1352 CONVERT_IF_TYPES_MATCH(BF16) in ConvertIfDestTypeMatches()
1397 CONVERT_IF_DEST_TYPE_MATCHES(BF16) in ConvertSwitch()
1509 case BF16: in EqualElements()
1599 case BF16: in IsAll()
[all …]
Dshape_util_test.cc144 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST()
150 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST()
211 {ShapeUtil::MakeShape(BF16, {3, 2}), ShapeUtil::MakeShape(F32, {4, 5})}); in TEST()
213 {ShapeUtil::MakeShape(F64, {3, 2}), ShapeUtil::MakeShape(BF16, {4, 5})}); in TEST()
228 {ShapeUtil::MakeShape(BF16, {4, 5}), ShapeUtil::MakeShape(F32, {3, 2})}); in TEST()
230 {ShapeUtil::MakeShape(F32, {3, 2}), ShapeUtil::MakeShape(BF16, {4, 5})}); in TEST()
Dlayout_util_test.cc342 shape = ShapeUtil::MakeShapeWithLayout(BF16, {2, 3, 4}, {1, 2, 0}); in TEST_F()
376 shape = ShapeUtil::MakeShapeWithLayout(BF16, {2, 3, 1004}, {2, 1, 0}); in TEST_F()
385 shape = ShapeUtil::MakeShapeWithLayout(BF16, {8, 2, 3, 1004}, {3, 2, 1, 0}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dexhaustive_op_test.cc217 case BF16: in Run()
504 } else if (platform_ == "Host" && ty_ == BF16) { in XLA_TEST_P()
523 } else if (platform_ == "Host" && ty_ == BF16) { in XLA_TEST_P()
640 BF16, ExhaustiveOpTest,
641 ::testing::Combine(::testing::Values(BF16),
Dclient_library_test_base.cc334 subshape->set_element_type(BF16); in ComputeAndCompareLiteralWithStatus()
391 subshape->set_element_type(BF16); in ComputeAndCompareLiteralWithStatus()
530 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarRelu()
543 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarMax()
554 auto shape = ShapeUtil::MakeShape(use_bfloat16_ ? BF16 : F32, {}); in CreateScalarReluSensitivity()
626 subshape->set_element_type(BF16); in MaybeConvertShapeToBfloat16()
Diota_test.cc83 ::testing::Combine(::testing::Values(F32, S32, BF16),
/external/tensorflow/tensorflow/compiler/xla/python_api/
Dtypes.py44 xla_data_pb2.BF16:
46 primitive_type=xla_data_pb2.BF16,
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dutil.cc45 case xla::BF16: in FloatLiteral()
104 case xla::BF16: in IntegerLiteral()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtype_util.cc58 *type = xla::BF16; in DataTypeToPrimitiveType()

123