- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for PadOp (0.05 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
class RewriteQuantizedPadOp : public OpRewritePattern<stablehlo::PadOp> { public: using OpRewritePattern<stablehlo::PadOp>::OpRewritePattern; LogicalResult match(stablehlo::PadOp op) const override { return success(IsOpFullyQuantized(op)); } void rewrite(stablehlo::PadOp op, PatternRewriter& rewriter) const override { Value input = op.getOperand();
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/stablehlo/transforms/legalize_hlo.cc
auto new_padding = llvm::to_vector<(num_spatial_dims * 2) + 4>(llvm::map_range( explicit_padding, [](int64_t dim) { return dim > 0 ? dim : 0; })); // Add an TF.PadOp before the LHS if there is explicit_padding on a 3D // Convolution. This is needed because TF.Conv3DOp doesn't support EXPLICIT. if (padding == "EXPLICIT" && num_spatial_dims == 3) { auto lhs_type =
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/tensorflow/ir/tf_ops_n_z.cc
results.add<ConvertPackToReshape>(context); } //===----------------------------------------------------------------------===// // PadOp //===----------------------------------------------------------------------===// LogicalResult PadOp::FoldOperandsPermutation(ArrayRef<int64_t> permutation) { // Paddings must be defined by a constant operation. auto paddings_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
const std::vector<int32_t>& operands, const std::vector<int32_t>& results); std::optional<BufferOffset<tflite::Operator>> BuildStablehloPadOp( mlir::stablehlo::PadOp pad_op, const std::vector<int32_t>& operands, const std::vector<int32_t>& results); std::optional<BufferOffset<tflite::Operator>> BuildVhloGatherV1Op(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
return 3 * count; } return -1; } //===----------------------------------------------------------------------===// // PadOp //===----------------------------------------------------------------------===// OpFoldResult PadOp::fold(FoldAdaptor) { if (InputOutputHasSameShape(getInput().getType(), getOutput().getType())) return getInput(); return {}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)