Home
last modified time | relevance | path

Searched defs:HloTestBase (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Diota_test.cc29 XLA_TEST_F(HloTestBase, IotaReshapeR1) { in XLA_TEST_F() argument
40 XLA_TEST_F(HloTestBase, IotaReshapeExtraDims) { in XLA_TEST_F() argument
Dreduce_window_test.cc1590 XLA_TEST_F(HloTestBase, ReduceWindowIdentity) { in XLA_TEST_F() argument
1610 XLA_TEST_F(HloTestBase, ReduceWindowIdentityNoPadding) { in XLA_TEST_F() argument
1630 XLA_TEST_F(HloTestBase, ReduceWindowS32) { in XLA_TEST_F() argument
1649 XLA_TEST_F(HloTestBase, ReduceWindowS64) { in XLA_TEST_F() argument
1668 XLA_TEST_F(HloTestBase, ReduceWindowF16) { in XLA_TEST_F() argument
1707 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport)) { in XLA_TEST_F() argument
1734 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport2)) { in XLA_TEST_F() argument
1760 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport3)) { in XLA_TEST_F() argument
1786 XLA_TEST_F(HloTestBase, DISABLED_ON_GPU(ReduceWindowVariadicSupport4)) { in XLA_TEST_F() argument
Dhlo_test_base.cc88 HloTestBase::HloTestBase(bool verifier_layout_sensitive, in HloTestBase() function in xla::HloTestBase
97 HloTestBase::HloTestBase(se::Platform* test_platform, in HloTestBase() function in xla::HloTestBase
Ddynamic_ops_test.cc737 XLA_TEST_F(HloTestBase, AddOfDUS) { in XLA_TEST_F() argument