Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetPaddingValue (0.27 sec)

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

      // Due to this, if the padding values in the input are equal, they will become
      // the same constant operator and the following check (which compares memory
      // addresses) works.
      if (pad_op.getPaddingValue() != parent_pad.getPaddingValue()) {
        return rewriter.notifyMatchFailure(
            pad_op, "parent and child pad have different padding values");
      }
    
      // NOTE: Because negative paddings are allowed, we assert that if
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 26.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

      const int dilation_h = mlir::cast<IntegerAttr>(dilations[1]).getInt();
      const int dilation_w = mlir::cast<IntegerAttr>(dilations[2]).getInt();
      return rewriter.getI64ArrayAttr({dilation_h, dilation_w});
    }
    
    Attribute GetPaddingValue(PatternRewriter& rewriter,
                              llvm::StringMap<Attribute>& identifier_to_attr) {
      llvm::StringRef padding =
          mlir::dyn_cast<StringAttr>(identifier_to_attr["padding"]).getValue();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

        return false;
      if (ExtractSingleElementAsInteger(num_cols).getInt() != -1) return false;
    
      // Verify padding_value is a tensor with all 0s.
      mlir::Value padding_value = tf_matrix_diag_v2_or_v3_op.getPaddingValue();
      mlir::Type element_type =
          mlir::cast<ShapedType>(padding_value.getType()).getElementType();
      if (mlir::isa<FloatType>(element_type)) {
        DenseFPElementsAttr padding_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
Back to top