Searched refs:use_bfloat16_params (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 36 static std::array<bool, 2> use_bfloat16_params{false, true}; variable
39 static std::array<bool, 1> use_bfloat16_params{false};
57 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};
574 ::testing::ValuesIn(use_bfloat16_params));
861 ::testing::ValuesIn(use_bfloat16_params)),
947 ::testing::ValuesIn(use_bfloat16_params)),
1068 ::testing::ValuesIn(use_bfloat16_params)),
1285 ::testing::ValuesIn(use_bfloat16_params)),
1493 ::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 | 53 absl::Span<const bool> use_bfloat16_params, in ExpandUseBfloat16() argument
56 for (bool use_bfloat16 : use_bfloat16_params) { in ExpandUseBfloat16()
|