Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for getBaseDilations (0.19 sec)

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

        DenseI64ArrayAttr new_window_strides =
            getInvertPermutedAttr(reduce_op.getWindowStrides());
        DenseI64ArrayAttr new_base_dilations =
            getInvertPermutedAttr(reduce_op.getBaseDilations());
        DenseI64ArrayAttr new_win_dilations =
            getInvertPermutedAttr(reduce_op.getWindowDilations());
    
        auto padding = reduce_op.getPadding();
        int64_t rank = transpose_perm.size();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 21:59:06 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc

                PermuteI64ArrayAttr(rewriter, op.getWindowStrides(),
                                    kNchwToNhwcPermutation),
                /*base_dilations=*/
                PermuteI64ArrayAttr(rewriter, op.getBaseDilations(),
                                    kNchwToNhwcPermutation),
                /*window_dilations=*/
                PermuteI64ArrayAttr(rewriter, op.getWindowDilations(),
                                    kNchwToNhwcPermutation),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          !(rw.getWindowDilations()->isSplat() &&
            rw.getWindowDilations()->getSplatValue<APInt>() == 1))
        return false;
    
      if (rw.getBaseDilations().has_value() &&
          !(rw.getBaseDilations()->isSplat() &&
            rw.getBaseDilations()->getSplatValue<APInt>() == 1))
        return false;
    
      if (llvm::all_of(padding, [](int64_t i) { return i == 0; })) {
        *padding_mode = "VALID";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          mlir::GetOptionalVector<int64_t>(reduce_window_op.getWindowStrides()));
      auto base_dilations = builder_.CreateVector(
          mlir::GetOptionalVector<int64_t>(reduce_window_op.getBaseDilations()));
      auto window_dilations = builder_.CreateVector(
          mlir::GetOptionalVector<int64_t>(reduce_window_op.getWindowDilations()));
      auto padding = builder_.CreateVector(
    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/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

          return failure();
        }
    
        // reduce_window op with dilations or padding will supported later.
        // TODO: b/321099943 - Support reduce_window op with dilations and padding.
        if (op.getBaseDilations().has_value() ||
            op.getWindowDilations().has_value() || op.getPadding().has_value()) {
          return failure();
        }
    
        // Window_dimensions and window_strides should have batch and channel
    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/tensorflow/ir/tf_ops_n_z.cc

      };
    
      if (check(op.getWindowDimensions(), "window_dimensions").failed())
        return failure();
    
      if (check(op.getWindowStrides(), "window_strides").failed()) return failure();
    
      if (check(op.getBaseDilations(), "base_dilations").failed()) return failure();
    
      if (check(op.getWindowDilations(), "window_dilations").failed())
        return failure();
    
      ElementsAttr padding;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      if (input_ty.hasStaticShape() &&
          matchPattern(op.getWindowDimensions(), m_Constant(&window_dimensions)) &&
          matchPattern(op.getWindowStrides(), m_Constant(&window_strides)) &&
          matchPattern(op.getBaseDilations(), m_Constant(&base_dilations)) &&
          matchPattern(op.getWindowDilations(), m_Constant(&window_dilations)) &&
          matchPattern(op.getPadding(), m_Constant(&padding))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        if (!(matchPattern(op.getWindowDimensions(),
                           m_Constant(&window_dimensions)) &&
              matchPattern(op.getWindowStrides(), m_Constant(&window_strides)) &&
              matchPattern(op.getBaseDilations(), m_Constant(&base_dilations)) &&
              matchPattern(op.getWindowDilations(),
                           m_Constant(&window_dilations)) &&
              matchPattern(op.getPadding(), m_Constant(&padding))))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top