Searched refs:BoundsLayout (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 546 struct BoundsLayout { struct
552 void PrintTo(const BoundsLayout& spec, std::ostream* os) { in PrintTo()
809 public ::testing::WithParamInterface<BoundsLayout> {};
841 ::testing::Values(BoundsLayout{{4, 8, 128}, {2, 1, 0}, {0}},
842 BoundsLayout{{4, 8, 128}, {2, 1, 0}, {1}},
843 BoundsLayout{{4, 8, 128}, {2, 1, 0}, {2}},
845 BoundsLayout{{1, 21, 43}, {2, 1, 0}, {0}},
846 BoundsLayout{{1, 1, 1}, {2, 1, 0}, {0}},
847 BoundsLayout{{1, 1, 1}, {2, 1, 0}, {1}},
848 BoundsLayout{{1, 1, 1}, {2, 1, 0}, {2}},
[all …]
|