Searched defs:use_bfloat16 (Results 1 – 7 of 7) sorted by relevance
30 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()57 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextDepthwiseConvolution2D()
30 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()139 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16, in BuildHloTextBatchGroupedConvolution2D()171 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
35 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()148 bool use_bfloat16) { in BuildHloTextGroupedConvolution2D()231 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
88 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
45 bool use_bfloat16; member
56 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local405 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
190 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) { in HloEvaluatorTest()