Searched defs:GatherOpTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_op_test.cc | 41 class GatherOpTest : public OpsTestBase { class
54 TEST_F(GatherOpTest, ScalarIndices) { in TEST_F() argument
69 TEST_F(GatherOpTest, ScalarIndices_Complex) { in TEST_F() argument
88 TEST_F(GatherOpTest, Simple_TwoD32_Axis0) { in TEST_F() argument
104 TEST_F(GatherOpTest, Simple_TwoD32_Axis1) { in TEST_F() argument
121 TEST_F(GatherOpTest, ZeroSize_TwoD32) { in TEST_F() argument
135 TEST_F(GatherOpTest, Simple_TwoD64) { in TEST_F() argument
151 TEST_F(GatherOpTest, HighRank) { in TEST_F() argument
166 TEST_F(GatherOpTest, Error_IndexOutOfRange) { in TEST_F() argument
180 TEST_F(GatherOpTest, Error_BatchDimsOutOfRange) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_test.cc | 75 TEST(GatherOpTest, Shuffle) { in TEST() argument
84 TEST(GatherOpTest, Test0DIndex) { in TEST() argument
94 TEST(GatherOpTest, Test0DIndexWith0DResult) { in TEST() argument
105 TEST(GatherOpTest, Test1DInput1DIndex) { in TEST() argument
114 TEST(GatherOpTest, Test2DIndexWith2DResult) { in TEST() argument
315 TEST(GatherOpTest, SimpleString) { in TEST() argument
|