HomeSort by relevance Sort by last modified time
    Searched defs:use_bfloat16 (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
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...]
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);

Completed in 341 milliseconds