Searched refs:Slice3D (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | multidimensional_slice_test.cc | 45 XLA_TEST_F(SliceTest, Slice3D) { in XLA_TEST_F() argument
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util_test.cc | 166 ReferenceUtil::Slice3D(input, {{0, 0, 0}}, {{2, 2, 2}}, {{1, 1, 1}}); in TEST_F() 179 ReferenceUtil::Slice3D(input, {{0, 0, 0}}, {{2, 3, 4}}, {{1, 2, 2}}); in TEST_F()
|
D | reference_util.h | 365 static std::unique_ptr<Array3D<T>> Slice3D(const Array3D<T>& input, in Slice3D() function
|