Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dresnet_preprocessing.py135 def preprocess_for_train(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument
154 def preprocess_for_eval(image_bytes, use_bfloat16, image_size=IMAGE_SIZE): argument
174 use_bfloat16=False, argument
Dimagenet_input.py74 use_bfloat16, argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc31 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
Dconv_depthwise_backprop_filter_test.cc31 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
Dgrouped_convolution_test.cc31 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
Dreverse_test.cc44 bool use_bfloat16; member
Dclient_library_test_base.h55 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16() local
376 bool use_bfloat16() const { return use_bfloat16_; } in use_bfloat16() function
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dkeras_mnist.py30 def get_input_datasets(use_bfloat16=False): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc133 explicit HloEvaluatorTest(bool use_bfloat16) : use_bfloat16_(use_bfloat16) {} in HloEvaluatorTest()