| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| resnet_preprocessing.py | 135 def preprocess_for_train(image_bytes, use_bfloat16, image_size=IMAGE_SIZE):
140 use_bfloat16: `bool` for whether to use bfloat16.
150 image, dtype=tf.bfloat16 if use_bfloat16 else tf.float32)
154 def preprocess_for_eval(image_bytes, use_bfloat16, image_size=IMAGE_SIZE):
159 use_bfloat16: `bool` for whether to use bfloat16.
168 image, dtype=tf.bfloat16 if use_bfloat16 else tf.float32)
174 use_bfloat16=False,
181 use_bfloat16: `bool` for whether to use bfloat16.
188 return preprocess_for_train(image_bytes, use_bfloat16, image_size)
190 return preprocess_for_eval(image_bytes, use_bfloat16, image_size
[all...] |
| imagenet_input.py | 68 use_bfloat16: If True, use bfloat16 precision; else use float32.
74 use_bfloat16,
83 self.use_bfloat16 = use_bfloat16
129 use_bfloat16=self.use_bfloat16)
230 if self.use_bfloat16 else tf.float32)
|
| main_estimator_tpu.py | 216 use_bfloat16=False) for is_training in [True, False]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| conv_depthwise_backprop_filter_test.cc | 31 string GetFloatDataType(bool use_bfloat16) {
32 return use_bfloat16 ? "bf16" : "f32";
127 const BatchGroupedConvolution2DSpec& spec, bool use_bfloat16) {
128 const string data_type = GetFloatDataType(use_bfloat16);
157 bool use_bfloat16 = ::testing::get<1>(GetParam()); local
159 BuildHloTextBatchGroupedConvolution2D(spec, use_bfloat16);
|
| conv_depthwise_test.cc | 31 string GetFloatDataType(bool use_bfloat16) {
32 return use_bfloat16 ? "bf16" : "f32";
134 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16) {
135 const string data_type = GetFloatDataType(use_bfloat16);
214 bool use_bfloat16 = ::testing::get<1>(GetParam()); local
216 BuildHloTextDepthwiseConvolution2D(spec, use_bfloat16);
|
| grouped_convolution_test.cc | 31 string GetFloatDataType(bool use_bfloat16) {
32 return use_bfloat16 ? "bf16" : "f32";
143 bool use_bfloat16) {
144 const string data_type = GetFloatDataType(use_bfloat16);
226 bool use_bfloat16 = ::testing::get<1>(GetParam()); local
227 const string hlo_text = BuildHloTextGroupedConvolution2D(spec, use_bfloat16);
|
| reverse_test.cc | 44 bool use_bfloat16; member in struct:xla::__anon44652::ReverseSpec
49 absl::StrJoin(reversal, "x"), use_bfloat16 ? "bf16" : "f32");
77 FloatReverseTest() { set_use_bfloat16(GetParam().use_bfloat16); }
|
| client_library_test_base.h | 47 // Sets the use_bfloat16 on a container of test cases according to the values in
55 for (bool use_bfloat16 : use_bfloat16_params) {
58 expanded.back().use_bfloat16 = use_bfloat16;
269 // parameter. When the use_bfloat16 flag is set but the literal has F32
296 // use_bfloat16 flag is set but the literal has F32 elements, the elements
300 // Creates a constant instruction with the given array. When the use_bfloat16
324 // When the use_bfloat16 flag is set but NativeT is float, the data will be
339 // When the use_bfloat16 flag is set but NativeT is float, the data will be
353 // When the use_bfloat16 flag is set but NativeT is float, the data will b
376 bool use_bfloat16() const { return use_bfloat16_; } function in class:xla::ClientLibraryTestBase
[all...] |
| reshape_test.cc | 620 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8},
628 if (use_bfloat16()) {
[all...] |
| pad_test.cc | 78 if (use_bfloat16()) {
|
| reduce_window_test.cc | 59 if (use_bfloat16()) {
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| keras_mnist.py | 30 def get_input_datasets(use_bfloat16=False):
34 use_bfloat16: Boolean to determine if input should be cast to bfloat16
42 cast_dtype = tf.bfloat16 if use_bfloat16 else tf.float32
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | 133 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) {}
144 HloEvaluatorBf16Test() : HloEvaluatorTest(/*use_bfloat16=*/GetParam()) {}
[all...] |
