Searched refs:Slice4D (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 161 auto expected = ReferenceUtil::Slice4D( in TEST_F()
192 auto expected = ReferenceUtil::Slice4D(values, {{0, 0, 0, 0}}, {{2, 4, 6, 8}}, in XLA_TEST_F()
485 auto expected = ReferenceUtil::Slice4D( in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 191 auto result = ReferenceUtil::Slice4D(input, {{1, 0, 0, 0}}, {{2, 2, 2, 2}}, in TEST_F()
204 auto result = ReferenceUtil::Slice4D(input, {{1, 0, 0, 0}}, {{2, 3, 4, 5}}, in TEST_F()
|
| D | reference_util.h | 396 static std::unique_ptr<Array4D<T>> Slice4D(const Array4D<T>& input, in Slice4D() function
|