Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for low_padding (0.18 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

          llvm::SmallVector<int64_t, 4> low_padding(rank, 0);
          llvm::SmallVector<int64_t, 4> high_padding(rank, 0);
          llvm::SmallVector<int64_t, 4> interior_padding(rank, 0);
          low_padding[axis] = 1;
          high_padding[axis] = -1;
          result = rewriter.create<PadOp>(
              op.getLoc(), result, init, GetI64ElementsAttr(low_padding, &rewriter),
              GetI64ElementsAttr(high_padding, &rewriter),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc

        if (padding.has_value()) {
          SmallVector<int64_t> new_padding(rank * 2, 0);
          auto old_padding = (*padding).getValues<int64_t>();
          for (int64_t idx = 0; idx < rank; ++idx) {
            new_padding[2 * transpose_perm[idx]] = old_padding[2 * idx];
            new_padding[2 * transpose_perm[idx] + 1] = old_padding[2 * idx + 1];
          }
          new_padding_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 21:59:06 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        // Emulate negative padding with a slice and remove negative values from the
        // padding vector.
        Value sliced_lhs = SliceNegativePadding(lhs, explicit_padding, rewriter);
        auto new_padding =
            llvm::to_vector<(num_spatial_dims * 2) + 4>(llvm::map_range(
                explicit_padding, [](int64_t dim) { return dim > 0 ? dim : 0; }));
    
    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