Searched defs:use_bfloat16 (Results 1 – 9 of 9) sorted by relevance
135 def preprocess_for_train(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument154 def preprocess_for_eval(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument174 use_bfloat16=False, argument
74 use_bfloat16, argument
31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()134 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextDepthwiseConvolution2D()214 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()127 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextBatchGroupedConvolution2D()157 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType()143 bool use_bfloat16) { in BuildHloTextGroupedConvolution2D()226 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
44 bool use_bfloat16; member
55 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local376 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
30 def get_input_datasets(use_bfloat16=False): argument
133 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) {} in HloEvaluatorTest()