Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for getPaddingAttr (0.19 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc

        Value new_conv = rewriter.create<mhlo::ConvolutionOp>(
            mul_op.getLoc(), conv_op.getType(), conv_op.getLhs(), new_filter,
            conv_op.getWindowStridesAttr(), conv_op.getPaddingAttr(),
            conv_op.getLhsDilationAttr(), conv_op.getRhsDilationAttr(),
            conv_op.getWindowReversalAttr(), conv_op.getDimensionNumbers(),
            conv_op.getFeatureGroupCount(), conv_op.getBatchGroupCount(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 22:21:19 UTC 2024
    - 8.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc

            op.getLoc(), /*resultType0=*/new_conv_output_tensor_type,
            /*lhs=*/input_transpose_op,
            /*rhs=*/filter_transpose_op,
            /*window_strides=*/op.getWindowStridesAttr(),
            /*padding=*/op.getPaddingAttr(),
            /*lhs_dilation=*/op.getLhsDilationAttr(),
            /*rhs_dilation=*/op.getRhsDilationAttr(),
            /*window_reversal=*/op.getWindowReversalAttr(),
            /*dimension_numbers=*/new_dimension_nums,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

    FailureOr<ElementsAttr> ConvertPaddingAttr(
        UniformQuantizedConvolutionOp op,
        const xla::ConvolutionDimensionNumbers &dnums, PatternRewriter &rewriter) {
      StringAttr conv_padding = op.getPaddingAttr();
      SmallVector<int64_t> padding_nums;
      ShapedType lhs_shape = mlir::cast<ShapedType>(op.getLhs().getType());
      ShapedType rhs_shape = mlir::cast<ShapedType>(op.getRhs().getType());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

    #include "tsl/platform/status.h"
    
    namespace {
    
    using ::absl::StatusOr;
    using ::tensorflow::Status;
    using ::tensorflow::errors::InvalidArgument;
    
    StatusOr<mlir::StringAttr> GetPaddingAttr(TfLitePadding pad_params,
                                              mlir::Builder builder,
                                              mlir::Location loc) {
      auto padding = tflite::Padding::Padding_VALID;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

            /*fuse_bias_constant=*/fuse_bias_constant);
    
        // Determine the attributes for the TFL::Conv2DOp.
        Value input_value = op.getOperand(0);
        if (const DenseIntElementsAttr padding_attr = op.getPaddingAttr();
            !HasProperPadding(op, dimension_numbers, padding_attr)) {
          // Add an extra tfl.pad_op if there are explicit padding values. This
          // extra pad op will allow us to always set the `padding` attribute of
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          auto conv_result = rewriter.create<mhlo::ConvolutionOp>(
              conv_op.getLoc(), new_output_type, sliced_input, sliced_kernel,
              conv_op.getWindowStridesAttr(), conv_op.getPaddingAttr(),
              conv_op.getLhsDilationAttr(), conv_op.getRhsDilationAttr(),
              conv_op.getWindowReversalAttr(), conv_op.getDimensionNumbers(), 1, 1,
              conv_op.getPrecisionConfigAttr());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top