- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for 1024x512xf32 (0.09 sec)
-
tensorflow/compiler/mlir/lite/tests/optimize_batch_matmul.mlir
// CHECK-LABEL: FuseTransposeFCRhsToBatchMatmul func.func @FuseTransposeFCRhsToBatchMatmul(%arg0: tensor<16x1024xf32>, %arg1: tensor<1024x128xf32>, %arg2: none) -> tensor<16x128xf32> { %cst = arith.constant dense<[1, 0]> : tensor<2xi32> %0 = "tfl.transpose"(%arg1, %cst) : (tensor<1024x128xf32>, tensor<2xi32>) -> tensor<128x1024xf32> // CHECK: "tfl.batch_matmul"(%arg0, %arg1)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 9K bytes - Viewed (0)