Home
last modified time | relevance | path

Searched refs:test_transpose (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/bazelbuild-rules_rust/examples/ffi/rust_calling_c/c/
H A Dmatrix_test.c64 void test_transpose() { in test_transpose() function
86 test_transpose(); in main()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/
H A DMetalOpTestRunner.mm58 REG_TEST("test_transpose", test_transpose);
H A DMPSCNNTests.h26 bool test_transpose();
H A DMPSCNNTests.mm586 bool test_transpose() { function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp5180 void test_transpose(const at::IntArrayRef input_shape, int64_t index0, int64_t index1) { in test_transpose() function
5196 test_transpose({1, 1}, 0, 1); in TEST_F()
5200 test_transpose({7, 5}, 0, 1); in TEST_F()
5204 test_transpose({53, 117}, 0, 1); in TEST_F()
5208 test_transpose({53, 117}, 0, 0); in TEST_F()
5212 test_transpose({53, 117}, 1, 1); in TEST_F()
5216 test_transpose({1, 1, 1}, 1, 2); in TEST_F()
5220 test_transpose({3, 2, 5}, 1, 2); in TEST_F()
5224 test_transpose({100, 1, 144}, 1, 2); in TEST_F()
5228 test_transpose({100, 1, 144}, 2, 2); in TEST_F()
[all …]
/aosp_15_r20/external/bazelbuild-rules_rust/examples/ffi/rust_calling_c/src/
H A Dmatrix.rs133 fn test_transpose() { in test_transpose() function
/aosp_15_r20/external/executorch/backends/arm/test/quantizer/
H A Dtest_generic_annotater.py72 def test_transpose(self): member in TestGenericAnnotator
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Darray_test.py319 def test_transpose(self): member in ArrayTest
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
H A Dtf-to-tosa-pipeline.mlir477 // CHECK-LABEL: test_transpose
480 func.func @test_transpose(%arg0: tensor<13x21x3xf32>) -> tensor<3x13x21xf32> {
H A Dtfl-to-tosa-pipeline.mlir673 // CHECK-LABEL: test_transpose
676 func.func @test_transpose(%arg0: tensor<13x21x3xf32>) -> tensor<*xf32> {
684 // CHECK-LABEL: test_transpose
687 func.func @test_transpose(%arg0: tensor<13x?x3xf32>) -> tensor<*xf32> {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dimport.hlotxt1324 // CHECK-LABEL: func private @test_transpose(%arg0: tensor<1x2x3x4xi32>) -> tensor<2x1x4x3xi32>
1325 %test_transpose {
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_operators.py214 def test_transpose(self): member in TestOperators
/aosp_15_r20/external/pytorch/test/
H A Dtest_type_promotion.py722 def test_transpose(self, device): member in TestTypePromotion
H A Dtest_tensorexpr.py1353 def test_transpose(self): member in TestTensorExprFuser
H A Dtest_mkldnn.py1234 def test_transpose(self): member in TestMkldnn
H A Dtest_legacy_vmap.py2304 def test_transpose(self): member in TestVmapOperatorsLegacy
H A Dtest_sparse_csr.py2984 def test_transpose(self, device, dtype, layout): member in TestSparseCSR
H A Dtest_sparse.py929 def test_transpose(self, device, dtype, coalesced): member in TestSparse
H A Dtest_nestedtensor.py2288 def test_transpose(self, device, dtype): member in TestNestedTensorDeviceType
H A Dtest_jit.py817 def test_transpose(self): member in TestJit
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_z3_gradual_types.py200 def test_transpose(self): member in HFOperations
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/
H A Dtest_numeric.py330 def test_transpose(self): member in TestNonarrayArgs
H A Dtest_multiarray.py1586 def test_transpose(self): member in TestMethods
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_vmap.py3110 def test_transpose(self): member in TestVmapOperators
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py2716 def test_transpose(self): member in CommonTemplate