Home
last modified time | relevance | path

Searched refs:Slice3D (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultidimensional_slice_test.cc45 XLA_TEST_F(SliceTest, Slice3D) { in XLA_TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc166 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()
Dreference_util.h365 static std::unique_ptr<Array3D<T>> Slice3D(const Array3D<T>& input, in Slice3D() function