| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | resnet_preprocessing.py | 135 def preprocess_for_train(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument
150 image, dtype=tf.bfloat16 if use_bfloat16 else tf.float32)
154 def preprocess_for_eval(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument
168 image, dtype=tf.bfloat16 if use_bfloat16 else tf.float32)
174 use_bfloat16=False, argument
188 return preprocess_for_train(image_bytes, use_bfloat16, image_size)
190 return preprocess_for_eval(image_bytes, use_bfloat16, image_size)
|
| D | imagenet_input.py | 74 use_bfloat16, argument
83 self.use_bfloat16 = use_bfloat16
129 use_bfloat16=self.use_bfloat16)
230 if self.use_bfloat16 else tf.float32)
|
| D | main_estimator_tpu.py | 216 use_bfloat16=False) for is_training in [True, False]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_backprop_filter_test.cc | 31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
32 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
127 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextBatchGroupedConvolution2D() argument
128 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextBatchGroupedConvolution2D()
157 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
159 BuildHloTextBatchGroupedConvolution2D(spec, use_bfloat16); in XLA_TEST_P()
|
| D | conv_depthwise_test.cc | 31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
32 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
134 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextDepthwiseConvolution2D() argument
135 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextDepthwiseConvolution2D()
214 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
216 BuildHloTextDepthwiseConvolution2D(spec, use_bfloat16); in XLA_TEST_P()
|
| D | grouped_convolution_test.cc | 31 string GetFloatDataType(bool use_bfloat16) { in GetFloatDataType() argument
32 return use_bfloat16 ? "bf16" : "f32"; in GetFloatDataType()
143 bool use_bfloat16) { in BuildHloTextGroupedConvolution2D() argument
144 const string data_type = GetFloatDataType(use_bfloat16); in BuildHloTextGroupedConvolution2D()
226 bool use_bfloat16 = ::testing::get<1>(GetParam()); in XLA_TEST_P() local
227 const string hlo_text = BuildHloTextGroupedConvolution2D(spec, use_bfloat16); in XLA_TEST_P()
|
| D | reverse_test.cc | 44 bool use_bfloat16; member
49 absl::StrJoin(reversal, "x"), use_bfloat16 ? "bf16" : "f32"); in ToTestCaseName()
77 FloatReverseTest() { set_use_bfloat16(GetParam().use_bfloat16); } in FloatReverseTest()
|
| D | client_library_test_base.h | 55 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local
58 expanded.back().use_bfloat16 = use_bfloat16; in ExpandUseBfloat16()
376 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
|
| D | reshape_test.cc | 620 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8}, in XLA_TEST_P()
628 if (use_bfloat16()) { in XLA_TEST_P()
770 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {7, 2, 3, 5}, in XLA_TEST_P()
781 if (use_bfloat16()) { in XLA_TEST_P()
|
| D | pad_test.cc | 78 if (use_bfloat16()) { in DefaultErrorSpec()
|
| D | reduce_window_test.cc | 59 if (use_bfloat16()) { in DefaultErrorSpec()
1024 if (use_bfloat16()) { in TEST_P()
|
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| D | keras_mnist.py | 30 def get_input_datasets(use_bfloat16=False): argument
42 cast_dtype = tf.bfloat16 if use_bfloat16 else tf.float32
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 133 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) {} in HloEvaluatorTest() argument
|