- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for 4x6xi32 (0.11 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/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/tensorflow/tests/tpu-resource-read-for-write.mlir
func.return } func.func private @write_chain_func(%arg0: tensor<1xf32>) -> (tensor<1xf32>) { %cst = "tf.Const"() <{value = dense<[[0, 1]]> : tensor<1x2xi32>}> : () -> tensor<1x2xi32> %0 = "tf.XlaAllReduce"(%arg0, %cst) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<1xf32>, tensor<1x2xi32>) -> tensor<1xf32> return %0 : tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// CHECK-NEXT: %[[GROUP:.*]] = "tf.Const"() // CHECK-SAME: [0, 1, 2, 3] // CHECK-NEXT: %[[REDUCED:.*]] = "tf.XlaAllReduce"(%[[ID]], %[[GROUP]]) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<f32>, tensor<1x4xi32>) -> tensor<f32> // CHECK-NEXT: "tf.OpA"(%[[REDUCED]]) : (tensor<f32>) -> () tf_device.replicate {n = 4 : i32} { "tf_device.cluster"() ({ "tf.OpA"(%arg0) : (tensor<f32>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/post_calibration_component.mlir
// Note: Argument quantization sequence omitted. // CHECK: stablehlo.dot_general %{{.+}}, %{{.+}}, contracting_dims = [1] x [0] : (tensor<1x1024xi8>, tensor<1024x3xi8>) -> tensor<1x3xi32> // Note: Result dequantization sequence omitted. // CHECK: return %{{.+}} : tensor<1x3xf32> // ----- // Tests that a simple dot_general (lifted as a function) with CustomAggregators
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 6.7K bytes - Viewed (0)