Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for AStride (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        MutableArrayRef<int64_t> end, MutableArrayRef<int64_t> stride) {
      assert(input_shape.size() <= 32);  // Only 32-bit masks are supported.
    
      // Make sure ranges' ranks are consistent with the input.
      assert(input_shape.size() == begin.size());
      assert(input_shape.size() == end.size());
      assert(input_shape.size() == stride.size());
    
      for (int i = 0, e = input_shape.size(); i < e; ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

              (((kernel_size - 1) != padding[2 * i]) ||
               (output_size != (stride * (input_size - 1)) + kernel_size))) {
            // padding[2 * i] == padding[2 * i + 1] means equal padding is applied
            // on both sides of a spatial dimension.
            // This happens when kernel_dim >= stride
            return false;
          } else if ((padding[2 * i] != padding[2 * i + 1]) &&
    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/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

            rewriter.create<arith::ConstantOp>(op.getLoc(), slice_size_attr);
    
        ArrayRef<int64_t> strides = op.getStrides();
        // If stride of every dimension is 1, create tfl.slice and return early.
        // Otherwise, create tfl.strided_slice instead.
        if (llvm::all_of(strides, [](int64_t stride) { return stride == 1; })) {
          rewriter.replaceOpWithNewOp<TFL::SliceOp>(
              op, output_type, op.getOperand(), start_idx, slice_size);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

      %0 = stablehlo.constant() {value = dense<3> : tensor<3x3x4x2xi8>} : () -> tensor<3x3x4x2x!quant.uniform<i8:f32:3, {2.000000e+2, 3.000000e+3}>>
      // The stride value is explicitly set to [1, 2].
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          llvm::errs() << "Window " << window.DebugString()
                       << " has a non-positive dimension.\n";
          return std::nullopt;
        }
        if (dim.stride() <= 0) {
          llvm::errs() << "Window " << window.DebugString()
                       << " has a non-positive stride.\n";
          return std::nullopt;
        }
        if (dim.base_dilation() < 1) {
          llvm::errs() << "Window " << window.DebugString()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

        // Checks the size of each of the output spatial dimensions.
        for (auto i : llvm::seq<int>(0, num_spatial_dims)) {
          const int64_t dim = GetTensorSpatialDimIndex(num_dims, format, i);
          int64_t stride = get_int(strides[dim]);
          int64_t expected_output_size;
          int64_t pad_low;
          int64_t pad_high;
          // Retrieve padding, if defined explicitly.
          if (padding == tensorflow::Padding::EXPLICIT) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

        int64_t input_size, int64_t filter_size, int64_t dilation_rate,
        int64_t stride, tensorflow::Padding padding, int64_t* output_size) {
      int64_t pad_low;
      int64_t pad_high;
    
      tensorflow::Status status = tensorflow::GetWindowedOutputSizeVerbose(
          input_size, filter_size, dilation_rate, stride, padding, output_size,
          &pad_low, &pad_high);
    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