Searched refs:ReduceLayout (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_hlo_test.cc | 31 struct ReduceLayout { struct
42 ::testing::TestParamInfo<ReduceLayout> reduce_layout_param) { in PrintReduceLayout()
46 void PrintTo(const ReduceLayout& reduce_layout, ::std::ostream* os) { in PrintTo()
52 public ::testing::WithParamInterface<ReduceLayout> {
82 const ReduceLayout& reduce_layout = GetParam(); in XLA_TEST_P()
118 ReduceLayout{{3, 2, 1, 0}, {0, 1, 2}}, //
119 ReduceLayout{{3, 2, 1, 0}, {0, 2, 1}}, //
120 ReduceLayout{{3, 2, 1, 0}, {1, 2, 0}}, //
121 ReduceLayout{{3, 2, 1, 0}, {1, 0, 2}}, //
122 ReduceLayout{{3, 2, 1, 0}, {2, 0, 1}}, //
[all …]
|