- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for PADV2 (0.08 sec)
-
tensorflow/compiler/mlir/lite/tests/ops.mlir
^bb0(%arg0: tensor<*xf32>, %arg1: tensor<5x3xi32>): %cst = arith.constant dense<2.0> : tensor<f32> // CHECK: "tfl.padv2"(%arg0, %arg1, %cst) %0 = "tfl.padv2"(%arg0, %arg1, %cst) : (tensor<*xf32>, tensor<5x3xi32>, tensor<f32>) -> tensor<? x f32> func.return %0#0 : tensor<? x f32> } // ----- // test PadV2 with invalid paddings size
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/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// CHECK: %[[PAD:.+]] = "tfl.padv2"(%[[ARG0]], %[[PADDING]], %[[ARG1]]) : (tensor<2x3x!quant.uniform<i8:f32, 2.000000e+00:-1>>, tensor<2x2xi32>, tensor<!quant.uniform<i8:f32, 2.000000e+00:-1>>) -> tensor<4x5x!quant.uniform<i8:f32, 2.000000e+00:-1>> // CHECK: return %[[PAD]] // ----- // Tests that a quantized `stablehlo.pad` with interior padding is converted to // `tfl.dilate` and `tfl.padv2`. func.func @pad_with_interior_padding(
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/legalize-tf.mlir
// CHECK-LABEL: padv2 // CHECK: "tfl.padv2"(%arg0, %arg1, %cst) : (tensor<2x1x3xf32>, tensor<3x2xi32>, tensor<f32>) -> tensor<?xf32> // CHECK: return } func.func @padv2_5D(tensor<2x1x3x1x1xf32>, tensor<5x2xi32>) -> tensor<? x f32> { ^bb0(%arg0: tensor<2x1x3x1x1xf32>, %arg1: tensor<5x2xi32>): %cst = "tf.Const"() { value = dense<2.0> : tensor<f32> } : () -> tensor<f32>
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/stablehlo/transforms/legalize_hlo.cc
ReplaceReturnOp(new_op.getElseBranch(), rewriter); rewriter.replaceOp(op, new_op.getResults()); return success(); } }; // Converts mhlo.pad to tf.PadV2 Value ConvertPadOp(PatternRewriter& rewriter, Operation* old_op) { auto pad_op = cast<mhlo::PadOp>(old_op); mlir::Location loc = pad_op.getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
op, output_type, op.getOperands(), axis, /*fused_activation_function=*/rewriter.getStringAttr("NONE")); } }; // Rewrites quantized stablehlo.pad to tfl.padv2. // tfl.dilate is introduced in between when interior padding exists. // TODO: b/322428814 - Add StableHLO quantizer integration tests for ODML. class RewriteQuantizedPadOp : public OpRewritePattern<stablehlo::PadOp> {
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/ir/tfl_ops.td
let results = (outs TFL_TensorOf<[F32, I32, I64, QI8, QUI8, TFL_Quint8, QI16]>:$output); let hasOptions = 1; let hasFolder = 1; } def TFL_PadV2Op : TFL_Op<"padv2", [ PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>, Pure, QuantizableResult, SameOperandsAndResultsScale, TFL_OperandHasRankAtMost<0, 5>,
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/tf-ops.mlir
%paddings = "tf.Const"() {value = dense<[0, 1]> : tensor<2xi64>} : () -> tensor<2xi64> // expected-error @+1 {{failed to verify that operand 1 is 2-D}} %0 = "tf.PadV2"(%input, %paddings, %constant) : (tensor<2xi64>, tensor<2xi64>, tensor<i64>) -> tensor<3xi64> func.return %0 : tensor<3xi64> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)