- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for io32 (0.14 sec)
-
tensorflow/compiler/mlir/lite/tests/ops.mlir
func.return %0 : tensor<? x i32> } // CHECK-LABEL: testPow func.func @testPow(tensor<? x i32>, tensor<? x i32>) -> tensor<? x i32> { ^bb0(%arg0: tensor<? x i32>, %arg1: tensor<? x i32>): // CHECK: tfl.pow %arg0, %arg1 %0 = tfl.pow %arg0, %arg1 : tensor<? x i32> func.return %0#0 : tensor<? x i32> } // CHECK-LABEL: testAtan2
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
}, { // loop body ^bb0(%barg0: tensor<i32>, %barg1: tensor<i32>): %one = arith.constant dense<1> : tensor<i32> %sub0 = "tf.Sub"(%barg0, %one) : (tensor<i32>, tensor<i32>) -> tensor<i32> %sub1 = "tf.Sub"(%barg1, %one) : (tensor<i32>, tensor<i32>) -> tensor<i32> "tf.Yield"(%sub0, %sub1) : (tensor<i32>, tensor<i32>) -> () }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
%cluster = "tf_device.cluster_func"(%arg0) {func = @_func} : (tensor<i32>) -> tensor<i32> return %cluster : tensor<i32> } func @_func(%arg0: tensor<i32>) -> tensor<i32> { %identity = "tf.Identity"(%arg0) : (tensor<i32>) -> tensor<i32> return %identity : tensor<i32> } ``` }]; let constructor = "TFDevice::CreateClusterOutliningPass()"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
} func.func @argmin(%arg0: tensor<3xi32>, %arg1: tensor<i32>) -> tensor<i32> { %0 = "tf.ArgMin"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<i32> func.return %0 : tensor<i32> // CHECK-LABEL: argmin // CHECK: "tfl.arg_min"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<i32> } func.func @sigmoid(%arg0: tensor<?x88xf32>) -> tensor<?x88xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
}]; let arguments = (ins TFL_TensorOf<[F32, I16, I32, I64, I8, UI8, QI8, QUI8, QI16]>:$value, TFL_1DTensorOf<[I32], [I32]>:$size_splits, TFL_0DTensorOf<[I32], [I32]>:$split_dim, ConfinedAttr<I32Attr, [IntPositive]>:$num_splits ); let results = (outs TFL_VariadicTensorOf<[F32, I16, I32, I64, I8, UI8, QI8, QUI8, QI16]>:$outputs ); let hasVerifier = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
%read0 = "tf.ReadVariableOp"(%arg0) : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> %read1 = "tf.ReadVariableOp"(%arg1) : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> %partitioned_input = "tf.TPUPartitionedInputV2"(%read0, %read1) {N = 2 : i64, partition_dims = []} : (tensor<i32>, tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// CHECK: %[[CONV2D:.+]] = "tfl.conv_2d"(%[[ARG]], %[[QCONST_0]], %[[QCONST_1]]) <{dilation_h_factor = 1 : i32, dilation_w_factor = 1 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 1 : i32, stride_w = 1 : i32}> : (tensor<1x3x3x4x!quant.uniform<i8:f32, 3.000000e+00:-100>>,...
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/prepare-composite-functions-tf.mlir
%0 = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32> %1 = "tf.ExpandDims"(%arg1, %0) : (tensor<*xi32>, tensor<i32>) -> tensor<*xi32> %2 = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32> %3 = "tf.Const"() {value = dense<4096> : tensor<i32>} : () -> tensor<i32> %4 = "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32> %5 = "tf.Range"(%4, %3, %2) : (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<4096xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
ArrayRef<int64_t> shape(rank); TensorType permutation_type = operand_type.cloneWith(shape, rewriter.getI32Type()); // Cast permutation attribute from i64 to i32 as they are required to be i32 // in TFLite. SmallVector<int32_t> permutation_i32 = CastI64ArrayToI32(op.getPermutation()).value(); auto permutation_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// static shape and activation function is none. For example: // // // %input: tensor<1x3x2xf32> // %unpack:3 = "tfl.unpack"(%input) {axis = 1 : i32, num = 3 : i32} // %res = "tfl.concatenation"(%unpack#0, %unpack#1, %unpack#2) // {axis = -1 : i32, fused_activation_function = "NONE"} // // can be optimized to // // %cst = arith.constant dense<[1, 6]> : tensor<2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)