Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for PadOp (0.05 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top