Searched refs:Slice2D (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multidimensional_slice_test.cc | 34 XLA_TEST_F(SliceTest, Slice2D) { in XLA_TEST_F() argument
|
| D | slice_test.cc | 403 std::unique_ptr<Array2D<int32>> expected = ReferenceUtil::Slice2D( in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 146 auto result = ReferenceUtil::Slice2D(*matrix_, {{0, 0}}, {{2, 2}}, {{1, 1}}); in TEST_F()
154 auto result = ReferenceUtil::Slice2D(*matrix_, {{0, 0}}, {{2, 3}}, {{1, 2}}); in TEST_F()
|
| D | reference_util.h | 342 static std::unique_ptr<Array2D<T>> Slice2D(const Array2D<T>& input, in Slice2D() function
|