Searched defs:TransferManagerTest (Results 1 – 2 of 2) sorted by relevance
41 class TransferManagerTest : public LocalClientTestBase { class43 TransferManagerTest() in TransferManagerTest() function in xla::__anon74d045f30111::TransferManagerTest71 XLA_TEST_F(TransferManagerTest, TransferR0U32) { in XLA_TEST_F() argument86 XLA_TEST_F(TransferManagerTest, TransferR1F32) { in XLA_TEST_F() argument103 XLA_TEST_F(TransferManagerTest, TransferR1F32AwkwardSizes) { in XLA_TEST_F() argument125 XLA_TEST_F(TransferManagerTest, TransferR1LargeF32) { in XLA_TEST_F() argument142 XLA_TEST_F(TransferManagerTest, TransferR1LargeUnalignedF32) { in XLA_TEST_F() argument162 XLA_TEST_F(TransferManagerTest, TransferR1U8) { in XLA_TEST_F() argument178 XLA_TEST_F(TransferManagerTest, TransferR2F32) { in XLA_TEST_F() argument195 XLA_TEST_F(TransferManagerTest, in XLA_TEST_F() argument[all …]
40 class TransferManagerTest(unittest.TestCase): class