Searched refs:SlicingTest (Results 1 – 1 of 1) sorted by relevance
28 using SlicingTest = xla::ClientLibraryTestBase; typedef57 XLA_TEST_F(SlicingTest, Simple2dLookup) { in XLA_TEST_F() argument71 XLA_TEST_F(SlicingTest, Simple3dLookup) { in XLA_TEST_F() argument86 XLA_TEST_F(SlicingTest, SimpleSliceUpdate) { in XLA_TEST_F() argument105 XLA_TEST_F(SlicingTest, TorchGather) { in XLA_TEST_F() argument119 XLA_TEST_F(SlicingTest, TorchIndexSelectOn0) { in XLA_TEST_F() argument138 XLA_TEST_F(SlicingTest, TorchIndexSelectOn1) { in XLA_TEST_F() argument