Searched refs:TestTranspose (Results 1 – 4 of 4) sorted by relevance
355 void TestTranspose(int parallelism) { in TestTranspose() function in xla::TransposeTest387 TEST_P(TransposeTest, TransposeInt8) { TestTranspose<int8_t>(1); } in TEST_P()388 TEST_P(TransposeTest, TransposeInt16) { TestTranspose<int16_t>(1); } in TEST_P()389 TEST_P(TransposeTest, TransposeInt32) { TestTranspose<int32_t>(1); } in TEST_P()390 TEST_P(TransposeTest, TransposeInt64) { TestTranspose<int64_t>(1); } in TEST_P()391 TEST_P(TransposeTest, TransposeInt128) { TestTranspose<absl::int128>(1); } in TEST_P()393 TEST_P(TransposeTest, ParallelTransposeInt8) { TestTranspose<int8_t>(16); } in TEST_P()394 TEST_P(TransposeTest, ParallelTransposeInt32) { TestTranspose<int32_t>(16); } in TEST_P()
13 func TestTranspose(t *testing.T) { func
97 class TestTranspose(TestCase): class
6901 TEST_F(LazyOpsTest, TestTranspose) { in TEST_F() argument