- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for 2x3x2xi32 (0.12 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/fold_constant_transpose.mlir
// ----- // Tests that int constants are not folded. // CHECK-LABEL: transpose_int func.func @transpose_int() -> tensor<3x2xi32> { %0 = stablehlo.constant dense<0> : tensor<2x3xi32> %1 = stablehlo.transpose %0, dims = [1, 0] : (tensor<2x3xi32>) -> tensor<3x2xi32> return %1 : tensor<3x2xi32> } // CHECK: transpose // ----- // Tests that transposing an argument cannot be folded. // CHECK-LABEL: transpose_arg
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 08:06:02 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tf-tfl-translate-serialize-stablehlo.mlir
module { func.func @tfInplaceUpdate(%arg0: tensor<2x1x2xf32>) -> tensor<2x1x2xf32> { %1 = arith.constant dense<1> : tensor<1xi32> %2 = arith.constant dense<2.0> : tensor<1x1x2xf32> %3 = "tf.InplaceUpdate"(%arg0, %1, %2) {device = ""} : (tensor<2x1x2xf32>, tensor<1xi32>, tensor<1x1x2xf32>) -> tensor<2x1x2xf32> func.return %3 : tensor<2x1x2xf32> } } //CHECK: module attributes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tf-tfl-translate-tf-quantize.mlir
module { func.func @tfInplaceUpdate(%arg0: tensor<2x1x2xf32>) -> tensor<2x1x2xf32> { %1 = arith.constant dense<1> : tensor<1xi32> %2 = arith.constant dense<2.0> : tensor<1x1x2xf32> %3 = "tf.InplaceUpdate"(%arg0, %1, %2) {device = ""} : (tensor<2x1x2xf32>, tensor<1xi32>, tensor<1x1x2xf32>) -> tensor<2x1x2xf32> func.return %3 : tensor<2x1x2xf32> } } //CHECK: module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/batchmatmul_to_einsum.mlir
// CHECK-LABEL: test_batch_matmul_broadcast_to_einsum // CHECK: "tf.Einsum"(%arg0, %arg1) <{equation = "...mk,...kn->...mn"}> : (tensor<2x2x4xf32>, tensor<2x4x2xf32>) -> tensor<2x2x2xf32> %0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x2x4xf32>, tensor<2x4x2xf32>) -> tensor<2x2x2xf32> func.return %0: tensor<2x2x2xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/bucketize.mlir
func.func @main(%arg0: tensor<3x2xf32>) -> tensor<3x2xi32> { // CHECK-LABEL: @main // CHECK: "tfl.bucketize"(%arg0) <{boundaries = [0.000000e+00 : f32, 1.000000e+01 : f32, 1.000000e+02 : f32]}> : (tensor<3x2xf32>) -> tensor<3x2xi32> %0 = "tfl.bucketize"(%arg0) {boundaries = [0.0 : f32, 10.0 : f32, 100.0 : f32]} : (tensor<3x2xf32>) -> tensor<3x2xi32> func.return %0 : tensor<3x2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 571 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/bridge/convert_tf_quant_ops_to_mhlo.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batching_fallback.mlir
%ch0 = tfrt.new.chain %a1 = tfrt_fallback_async.const_dense_tensor dense<[[2, 2], [2, 2]]> : tensor<2x2xi32> %a2 = tfrt_fallback_async.const_dense_tensor dense<[[3, 3], [3, 3]]> : tensor<2x2xi32> %b = tfrt_fallback_async.const_dense_tensor dense<[[1, 1], [1, 1]]> : tensor<2x2xi32> // Two batch_size=2 batches get concatenated.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 18 22:58:56 UTC 2023 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-include-tf2xla-fallback.mlir
%0 = mhlo.constant dense<[[1, 1], [2, 2]]> : tensor<2x2xi32> // NO_FALLBACK: tf.MirrorPad // SUPPORTED_FALLBACK_DEVICE-NOT: tf.MirrorPad // UNSPECIFIED_FALLBACK_DEVICE: tf.MirrorPad // UNSUPPORTED_FALLBACK_DEVICE: tf.MirrorPad %1 = "tf.MirrorPad"(%arg0, %0) {mode = "SYMMETRIC"} : (tensor<2x3xcomplex<f64>>, tensor<2x2xi32>) -> tensor<4x7xcomplex<f64>> func.return %1 : tensor<4x7xcomplex<f64>> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 3.2K bytes - Viewed (0)