Searched defs:HloTestBase (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_window_test.cc | 1663 XLA_TEST_F(HloTestBase, ReduceWindowIdentity) { in XLA_TEST_F() argument
1683 XLA_TEST_F(HloTestBase, ReduceWindowIdentityNoPadding) { in XLA_TEST_F() argument
1703 XLA_TEST_F(HloTestBase, ReduceWindowS32) { in XLA_TEST_F() argument
1722 XLA_TEST_F(HloTestBase, ReduceWindowS64) { in XLA_TEST_F() argument
1741 XLA_TEST_F(HloTestBase, ReduceWindowF16) { in XLA_TEST_F() argument
1780 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport)) { in XLA_TEST_F() argument
1807 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport2)) { in XLA_TEST_F() argument
1833 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport3)) { in XLA_TEST_F() argument
1859 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport4)) { in XLA_TEST_F() argument
1885 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowS64Support)) { in XLA_TEST_F() argument
[all …]
|
| H A D | iota_test.cc | 29 XLA_TEST_F(HloTestBase, IotaReshapeR1) { in XLA_TEST_F() argument
40 XLA_TEST_F(HloTestBase, IotaReshapeExtraDims) { in XLA_TEST_F() argument
|
| H A D | hlo_test_base.cc | 76 HloTestBase::HloTestBase(bool verifier_layout_sensitive, in HloTestBase() function in xla::HloTestBase
84 HloTestBase::HloTestBase(se::Platform* test_platform, in HloTestBase() function in xla::HloTestBase
|
| H A D | dynamic_ops_test.cc | 740 XLA_TEST_F(HloTestBase, AddOfDUS) { in XLA_TEST_F() argument
|