Searched refs:testReshape (Results 1 – 7 of 7) sorted by relevance
94 def testReshape(self): member in ScalarTest
220 // CHECK-LABEL: func @testReshape(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>, %arg2: tensor<10000xf…221 func @testReshape(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>, %arg2: tensor<10000xf32>, %arg3: tens…236 func @testReshape(tensor<*xf32>, tensor<*xf32>) -> (tensor<100x100xf32>) {246 func @testReshape(%arg0: tensor<10x10x10xf32>) -> tensor<100x100xf32> {255 func @testReshape(%arg0: tensor<10x10x10x10xf32>) -> tensor<100x100xf32> {264 func @testReshape(%arg0: tensor<10x10x10x10xf32>) -> tensor<100x100xf32> {273 func @testReshape(%arg0: tensor<10x10x?xf32>) -> tensor<10x10xf32> {
282 def testReshape(self): member in XlaCompilationTest
1317 def testReshape(self): member in BinaryOpsTest
299 def testReshape(self, num_parallel_calls): member in MapVectorizationTest
1148 def testReshape(self): member in SingleOpTest
673 def testReshape(self): member in SaverTest