Searched refs:use_bfloat16_params (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 35 static std::array<bool, 2> use_bfloat16_params{false, true}; variable
38 static std::array<bool, 1> use_bfloat16_params{false};
56 use_bfloat16_params, in GetTestCases()
|
| D | reduce_window_test.cc | 50 static std::array<bool, 2> use_bfloat16_params{false, true}; variable
53 static std::array<bool, 1> use_bfloat16_params{false};
559 ::testing::ValuesIn(use_bfloat16_params));
845 ::testing::ValuesIn(use_bfloat16_params)),
931 ::testing::ValuesIn(use_bfloat16_params)),
1052 ::testing::ValuesIn(use_bfloat16_params)),
1206 ::testing::ValuesIn(use_bfloat16_params)),
1399 ::testing::ValuesIn(use_bfloat16_params)),
|
| D | pad_test.cc | 38 static std::array<bool, 2> use_bfloat16_params{false, true}; variable
41 static std::array<bool, 1> use_bfloat16_params{false};
469 ::testing::ValuesIn(use_bfloat16_params));
|
| D | client_library_test_base.h | 52 absl::Span<const bool> use_bfloat16_params, in ExpandUseBfloat16() argument
55 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16()
|