Home
last modified time | relevance | path

Searched refs:use_bfloat16 (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_backprop_filter_test.cc30 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
31 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
139 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextBatchGroupedConvolution2D() argument
141 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextBatchGroupedConvolution2D()
171 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
174 if (use_bfloat16) { in XLA_TEST_P()
180 spec, use_bfloat16, /*scheduled=*/false); in XLA_TEST_P()
Dgrouped_convolution_test.cc35 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
36 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
148 bool use_bfloat16) { in BuildHloTextGroupedConvolution2D() argument
149 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextGroupedConvolution2D()
231 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
234 if (use_bfloat16) { in XLA_TEST_P()
239 const string hlo_text = BuildHloTextGroupedConvolution2D(spec, use_bfloat16); in XLA_TEST_P()
Dconv_depthwise_common.cc30 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
31 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
57 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextDepthwiseConvolution2D() argument
59 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextDepthwiseConvolution2D()
Dconv_depthwise_test.cc88 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
91 if (use_bfloat16) { in XLA_TEST_P()
97 BuildHloTextDepthwiseConvolution2D(spec, use_bfloat16); in XLA_TEST_P()
Dreverse_test.cc45 bool use_bfloat16; member
50 absl::StrJoin(reversal, "x"), use_bfloat16 ? "bf16" : "f32"); in ToTestCaseName()
78 FloatReverseTest() { set_use_bfloat16(GetParam().use_bfloat16); } in FloatReverseTest()
Dconv_depthwise_common.h31 string GetFloatDataType(bool use_bfloat16);
49 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16,
Dclient_library_test_base.h56 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local
59 expanded.back().use_bfloat16 = use_bfloat16; in ExpandUseBfloat16()
405 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
Dreshape_test.cc653 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8}, in XLA_TEST_P()
661 if (use_bfloat16()) { in XLA_TEST_P()
809 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {7, 2, 3, 5}, in XLA_TEST_P()
820 if (use_bfloat16()) { in XLA_TEST_P()
Dpad_test.cc78 if (use_bfloat16()) { in DefaultErrorSpec()
Dreduce_window_test.cc59 if (use_bfloat16()) { in DefaultErrorSpec()
1040 if (use_bfloat16()) { in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc190 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) { in HloEvaluatorTest() argument