- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for 5x6x7xf32 (0.18 sec)
-
tensorflow/compiler/mlir/lite/tests/const-fold.mlir
// CHECK-LABEL: @ShapeOpI32 func.func @ShapeOpI32(%arg0 : tensor<576x72xf32>) -> tensor<2xi32> { %0 = "tfl.shape"(%arg0) : (tensor<576x72xf32>) -> tensor<2xi32> func.return %0 : tensor<2xi32> // CHECK: %[[CST:.*]] = arith.constant dense<[576, 72]> : tensor<2xi32> // CHECK: return %[[CST]] } // CHECK-LABEL: @ShapeOpI64 func.func @ShapeOpI64(%arg0 : tensor<576x72xf32>) -> tensor<2xi64> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 45.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tfl_legalize_hlo.mlir
// RUN: odml-to-stablehlo-opt %s -tfl-legalize-hlo -split-input-file | FileCheck %s --dump-input=fail func.func @main(%arg0: tensor<5x7xf32>) -> tensor<5x7xf32> { func.return %arg0: tensor<5x7xf32> // CHECK-LABEL: main // CHECK: return %arg0 : tensor<5x7xf32> } // - transpose // func.func @transpose_2d(%arg0: tensor<2x3xf32>) -> tensor<3x2xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 40.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir
%cst = arith.constant dense<0> : tensor<2xi32> %cst_0 = arith.constant dense<1> : tensor<2xi32> %0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : (tensor<21x15x7xf32>, tensor<2xi32>, tensor<2xi32>, tensor<2xi32>) -> tensor<5x15x7xf32> func.return %0 : tensor<5x15x7xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 59.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK: tf.Reshape // CHECK-SAME: tensor<1x3x7xf32> %1 = "tf.Reshape"(%arg1, %0) : (tensor<3x7xf32>, tensor<3xi32>) -> tensor<*xf32> return %1 : tensor<*xf32> } // CHECK-LABEL: func @gather_concat_reshape func.func @gather_concat_reshape(%arg0: tensor<?x1x7xf32>, %arg1: tensor<?x7xf32>) -> tensor<*xf32> { %0 = "tf.Shape"(%arg0) : (tensor<?x1x7xf32>) -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// ----- // CHECK-LABEL: matmul_notranspose // CHECK-SAME: (%[[A:.*]]: tensor<5x7xf32>, %[[B:.*]]: tensor<7x11xf32>) func.func @matmul_notranspose(%a: tensor<5x7xf32>, %b: tensor<7x11xf32>) -> tensor<5x11xf32> { // CHECK: "mhlo.dot"(%[[A]], %[[B]]) %0 = "tf.MatMul"(%a, %b) {transpose_a = false, transpose_b = false} : (tensor<5x7xf32>, tensor<7x11xf32>) -> tensor<5x11xf32> func.return %0 : tensor<5x11xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.func @testBatchMatMul(%lhs: tensor<2x?x2x?x3x5xf32>, %rhs: tensor<2x2x?x?x5x7xf32>) { %0 = "tf.BatchMatMul"(%lhs, %rhs) : (tensor<2x?x2x?x3x5xf32>, tensor<2x2x?x?x5x7xf32>) -> tensor<2x?x?x?x3x7xf32> func.return } // ----- // Mismatching batch dimensions. func.func @testBatchMatMul(%lhs: tensor<1x3x5xf32>, %rhs: tensor<2x5x7xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)