Searched refs:DynamicUpdateSliceTest (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | dynamic_ops_test.cc | 304 class DynamicUpdateSliceTest : public ClientLibraryTestBase { class
568 XLA_TEST_F(DynamicUpdateSliceTest, Int32R0BF16) { TestR0<int32_t, bfloat16>(); } in XLA_TEST_F() argument
569 XLA_TEST_F(DynamicUpdateSliceTest, Int32R0) { TestR0<int32_t, float>(); } in XLA_TEST_F() argument
570 XLA_TEST_F(DynamicUpdateSliceTest, Int64R0) { TestR0<int64_t, float>(); } in XLA_TEST_F() argument
571 XLA_TEST_F(DynamicUpdateSliceTest, UInt64R0) { TestR0<uint64_t, float>(); } in XLA_TEST_F() argument
573 XLA_TEST_F(DynamicUpdateSliceTest, Int32R1BF16) { TestR1<int32_t, bfloat16>(); } in XLA_TEST_F() argument
574 XLA_TEST_F(DynamicUpdateSliceTest, Int32R1) { TestR1<int32_t, float>(); } in XLA_TEST_F() argument
575 XLA_TEST_F(DynamicUpdateSliceTest, Int64R1) { TestR1<int64_t, float>(); } in XLA_TEST_F() argument
576 XLA_TEST_F(DynamicUpdateSliceTest, UInt64R1) { TestR1<uint64_t, float>(); } in XLA_TEST_F() argument
577 XLA_TEST_F(DynamicUpdateSliceTest, UInt32R1OOB) { in XLA_TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_update_slice_test.cc | 27 class DynamicUpdateSliceTest : public HloTestBase {}; class
29 XLA_TEST_F(DynamicUpdateSliceTest, ShardedInPlaceDUS) { in XLA_TEST_F() argument
86 XLA_TEST_F(DynamicUpdateSliceTest, ExpandedScatter) { in XLA_TEST_F() argument
|