Home
last modified time | relevance | path

Searched refs:testTranspose (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtranspose_test.mm36 - (void)testTranspose { method
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py121 def testTranspose(self): member in CSRSparseMatrixTest
Dcsr_sparse_matrix_ops_test.py865 def testTranspose(self): member in CSRSparseMatrixOpsTest
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py497 def testTranspose(self): member in XlaOpsNumericalTest
Dbinary_ops_test.py1508 def testTranspose(self): member in BinaryOpsTest
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py314 def testTranspose(self, num_parallel_calls): member in MapVectorizationTest
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf-ops.mlir2477 // CHECK-LABEL: testTranspose
2478 func @testTranspose(tensor<2x3xf32>) -> tensor<3x2xf32> {
2488 // CHECK-LABEL: testTranspose
2489 func @testTranspose(tensor<2x?xf32>) -> tensor<?x2xf32> {
2499 // CHECK-LABEL: testTranspose
2500 func @testTranspose(tensor<2x?x?xf32>) -> tensor<3x?x2xf32> {
2510 func @testTranspose(tensor<2x3xf32>, tensor<1x2xi32>) -> tensor<3x2xf32> {
2520 func @testTranspose(tensor<2x3xf32>) -> tensor<3x2xf32> {
2531 func @testTranspose(tensor<2x3xf32>) -> tensor<3x2x1xf32> {
2542 func @testTranspose(tensor<2x3x4xf32>) -> tensor<3x2x4xf32> {
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_ops_test.py1023 def testTranspose(self): member in SparseTransposeTest
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops_test.py956 def testTranspose(self): member in ArrayMethodsTest
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py938 def testTranspose(self): member in TestFactory.SingleOpTest
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_test.py1386 def testTranspose(self): member in FlipTransposeRotateTest