Home
last modified time | relevance | path

Searched defs:use_bfloat16 (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconv_depthwise_backprop_filter_test.cc31 std::string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()
140 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextBatchGroupedConvolution2D()
172 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
H A Dgrouped_convolution_test.cc35 std::string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()
149 const GroupedConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextGroupedConvolution2D()
232 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
H A Dconv_depthwise_common.cc31 std::string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()
58 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextDepthwiseConvolution2D()
H A Dclient_library_test_base.h54 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local
403 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
H A Dconv_depthwise_test.cc88 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
H A Dreverse_test.cc44 bool use_bfloat16; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc215 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) { in HloEvaluatorTest()