| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | transpose_test.cc | 67 TEST(TransposeTest, TestRefOps1D) { in TEST() argument
74 TEST(TransposeTest, TestRefOps2D) { in TEST() argument
84 TEST(TransposeTest, TestRefOps3D) { in TEST() argument
133 TEST(TransposeTest, TestRefOps3D_OneInDimension) { in TEST() argument
161 TEST(TransposeTest, TestRefOps4D) { in TEST() argument
206 TEST(TransposeTest, TestRefOps4DInt8) { TransposeTestTestRefOps4D<int8_t>(); } in TEST() argument
208 TEST(TransposeTest, TestRefOps4DInt16) { TransposeTestTestRefOps4D<int16_t>(); } in TEST() argument
270 TEST(TransposeTest, TestUnequalPermSize) { in TEST() argument
274 TEST(TransposeTest, TestPermOutOfBounds) { in TEST() argument
282 TEST(TransposeTest, Test1DInputConstTensor) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | transpose_test.cc | 31 class TransposeTest : public ClientLibraryTestBase { class
39 XLA_TEST_F(TransposeTest, Transpose0x0) { in XLA_TEST_F() argument
47 XLA_TEST_F(TransposeTest, Transpose0x42) { in XLA_TEST_F() argument
55 XLA_TEST_F(TransposeTest, Transpose7x0) { in XLA_TEST_F() argument
63 TEST_F(TransposeTest, Transpose2x2) { in TEST_F() argument
76 XLA_TEST_F(TransposeTest, Transpose0x2x3_2x3x0) { in XLA_TEST_F() argument
85 TEST_F(TransposeTest, Transpose1x2x3_2x3x1) { in TEST_F() argument
96 TEST_F(TransposeTest, Transpose1x2x3_3x2x1) { in TEST_F() argument
107 TEST_F(TransposeTest, Transpose1x2x3_1x2x3) { in TEST_F() argument
118 TEST_F(TransposeTest, MultiTranspose3x2) { in TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose_test.cc | 44 TEST(TransposeTest, RemoveTrivialDimensions) { in TEST() argument
68 TEST(TransposeTest, CoalesceDimensions) { in TEST() argument
118 TEST(TransposeTest, InvalidTilings) { in TEST() argument
352 class TransposeTest : public ::testing::TestWithParam<TransposeTestCase> { class
387 TEST_P(TransposeTest, TransposeInt8) { TestTranspose<int8_t>(1); } in TEST_P() argument
388 TEST_P(TransposeTest, TransposeInt16) { TestTranspose<int16_t>(1); } in TEST_P() argument
389 TEST_P(TransposeTest, TransposeInt32) { TestTranspose<int32_t>(1); } in TEST_P() argument
390 TEST_P(TransposeTest, TransposeInt64) { TestTranspose<int64_t>(1); } in TEST_P() argument
391 TEST_P(TransposeTest, TransposeInt128) { TestTranspose<absl::int128>(1); } in TEST_P() argument
393 TEST_P(TransposeTest, ParallelTransposeInt8) { TestTranspose<int8_t>(16); } in TEST_P() argument
[all …]
|
| /aosp_15_r20/external/libgav1/src/dsp/arm/ |
| H A D | common_neon_test.cc | 33 class TransposeTest : public testing::Test { class
35 TransposeTest() { in TransposeTest() function in libgav1::dsp::__anon3ea09b630111::TransposeTest
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | packedtensoraccessor_test.cpp | 14 TEST(PackedtensoraccessorTest, TransposeTest) { in TEST() argument
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | TransposeTestHelper.hpp | 87 void TransposeTest(std::vector<armnn::BackendId>& backends, in TransposeTest() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| H A D | reshape_transpose_test.py | 77 class TransposeTest(trt_test.TfTrtIntegrationTestBase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | transpose_test_util.cc | 115 absl::Status TransposeTest(TestExecutionEnvironment* env) { in TransposeTest() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | transpose_op_test.py | 33 class TransposeTest(test.TestCase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference_test.cc | 492 TEST_F(DynamicDimensionInferenceTest, TransposeTest) { in TEST_F() argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 2573 class TransposeTest(torch.nn.Module): class
|