- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for 10x8xi1 (0.25 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK: return %[[VAL_2]] : tensor<10x8xi64> // CHECK: } func.func @convert_remainder_for_int64(%arg0: tensor<10x8xi64>, %arg1: tensor<10x8xi64>) -> tensor<10x8xi64> { %0 = mhlo.remainder %arg0, %arg1 : tensor<10x8xi64> func.return %0 : tensor<10x8xi64> } // CHECK-LABEL: func @not_convert_remainder_for_int16( // CHECK-NOT: "tf.Mod"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-binary-elementwise.mlir
func.func @equal_broadcast(%arg0: tensor<1xi32>, %arg1: tensor<1x2xi32>) -> tensor<1x2xi1> { // CHECK-DAG: %[[LHS_BCAST:.+]] = "mhlo.broadcast_in_dim"(%arg0) <{broadcast_dimensions = dense<1> : tensor<1xi64>}> // CHECK-NEXT: mhlo.compare EQ, %[[LHS_BCAST]], %arg1 %0 = "tf.Equal"(%arg0, %arg1) : (tensor<1xi32>, tensor<1x2xi32>) -> tensor<1x2xi1> func.return %0: tensor<1x2xi1> } // CHECK-LABEL: func @equal_broadcast_no_incompatible_shapes_error
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-stablehlo-tfl-composite.mlir
%1 = stablehlo.constant dense<0> : tensor<100xi64> %2 = stablehlo.compare LT, %arg2, %1 : (tensor<100xi64>, tensor<100xi64>) -> tensor<100xi1> %3 = stablehlo.add %arg2, %0 : tensor<100xi64> %4 = stablehlo.select %2, %3, %arg2 : tensor<100xi1>, tensor<100xi64> %5 = stablehlo.reshape %4 : (tensor<100xi64>) -> tensor<100x1xi64> %6 = "stablehlo.scatter"(%arg0, %5, %arg3) ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize/quantize_same_scale.mlir
// CHECK-SAME: %[[ARG0:.*]]: tensor<1x2xf32> // CHECK-SAME: %[[ARG1:.*]]: tensor<2x3xf32> // CHECK-SAME: %[[ARG2:.*]]: tensor<1x3xi1> // CHECK-SAME: %[[ARG3:.*]]: tensor<1x3xf32> func.func private @composite_and_select(%arg0: tensor<1x2xf32>, %arg1: tensor<2x3xf32>, %arg2: tensor<1x3xi1>, %arg3: tensor<1x3xf32>) -> tensor<1x3xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 35.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
func.func @all(%input: tensor<4x8xi1>) -> tensor<4xi1> { %dims = "tf.Const"() { value = dense<1> : tensor<1xi32>} : () -> tensor<1xi32> // CHECK: %[[INIT:.*]] = mhlo.constant dense<true> : tensor<i1> // CHECK: %[[REDUCED:.*]] = mhlo.reduce(%{{.*}} init: %[[INIT]]) applies mhlo.and across dimensions = [1] : (tensor<4x8xi1>, tensor<i1>) -> tensor<4xi1> %0 = "tf.All"(%input, %dims) : (tensor<4x8xi1>, tensor<1xi32>) -> tensor<4xi1>
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/lower_tf.mlir
// CHECK: return %[[ADD]] : tensor<14xf32> } func.func @scatter_nd_updates_bool(%arg0: tensor<1x24xi1>, %arg1: tensor<1x2x2xi32>, %arg2: tensor<1x2xi1>) -> tensor<1x24xi1> { %0 = "tf.TensorScatterUpdate"(%arg0, %arg1, %arg2) : (tensor<1x24xi1>, tensor<1x2x2xi32>, tensor<1x2xi1>) -> tensor<1x24xi1> func.return %0 : tensor<1x24xi1> // CHECK-LABEL: scatter_nd_updates_bool(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
func.func @select( %arg0: tensor<1x3xi1>, %arg1: tensor<1x3x!quant.uniform<i8:f32, 2.000000e+00:-1>>, %arg2: tensor<1x3x!quant.uniform<i8:f32, 2.000000e+00:-1>> ) -> tensor<1x3x!quant.uniform<i8:f32, 2.000000e+00:-1>> { %0 = "stablehlo.select"(%arg0, %arg1, %arg2) : ( tensor<1x3xi1>, tensor<1x3x!quant.uniform<i8:f32, 2.000000e+00:-1>>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/optimize.mlir
%1 = "tfl.select"(%0, %cst_0, %arg0) : (tensor<10xi1>, tensor<10xf32>, tensor<10xf32>) -> tensor<10xf32> %2 = "tfl.exp"(%1) : (tensor<10xf32>) -> tensor<10xf32> %3 = tfl.sub(%2, %cst) {fused_activation_function = "NONE"} : (tensor<10xf32>, tensor<f32>) -> tensor<10xf32> %4 = "tfl.select"(%0, %arg0, %3) : (tensor<10xi1>, tensor<10xf32>, tensor<10xf32>) -> tensor<10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 284.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.return %0: tensor<2xi32> } // ----- func.func @invalidSelect(%arg0: tensor<1x8xi1>, %arg1: tensor<1x8x8xi32>, %arg2: tensor<1x8x8xi32>) -> tensor<1x8x8xi32> { // expected-error @+1 {{requires that pred is a scalar OR has the same rank as t and e OR is a vector}} %0 = "tf.Select"(%arg0, %arg1, %arg2) : (tensor<1x8xi1>, tensor<1x8x8xi32>, tensor<1x8x8xi32>) -> tensor<1x8x8xi32> func.return %0: tensor<1x8x8xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)