Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 18 of 18 for PadOp (0.05 sec)

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

            TypeID::get<TF::NegOp>(),
            TypeID::get<TF::NextAfterOp>(),
            TypeID::get<TF::NonMaxSuppressionV4Op>(),
            TypeID::get<TF::NotEqualOp>(),
            TypeID::get<TF::PadOp>(),
            TypeID::get<TF::ParameterizedTruncatedNormalOp>(),
            TypeID::get<TF::PlaceholderWithDefaultOp>(),
            TypeID::get<TF::PolygammaOp>(),
            TypeID::get<TF::PopulationCountOp>(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 21.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

                               ConstOp::getOperationName(),
                               ConcatV2Op::getOperationName(),
                               AddV2Op::getOperationName(),
                               PadOp::getOperationName(),
                               SplitOp::getOperationName(),
                               UnpackOp::getOperationName(),
                               DivOp::getOperationName(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

          // Update `out_grad_shape` to result shape of following `PadOp`.
          out_grad_shape[dim] = low_padding[dim] + high_padding[dim] +
                                (out_grad_shape[dim] - 1) * strides[dim] + 1;
        }
        Value reduce_window_input = rewriter.create<PadOp>(
            loc, tensorflow::GetTypeFromTFTensorShape(out_grad_shape, element_type),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        auto new_padding =
            llvm::to_vector<(num_spatial_dims * 2) + 4>(llvm::map_range(
                explicit_padding, [](int64_t dim) { return dim > 0 ? dim : 0; }));
    
        // Add an TF.PadOp before the LHS if there is explicit_padding on a 3D
        // Convolution. This is needed because TF.Conv3DOp doesn't support EXPLICIT.
        if (padding == "EXPLICIT" && num_spatial_dims == 3) {
          auto lhs_type =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

          auto padding_const =
              rewriter.create<TF::ConstOp>(op->getLoc(), padding_attr);
    
          // Add Pad op.
          auto pad_output_type = UnrankedTensorType::get(elem_type);
          input = rewriter.create<TF::PadOp>(op->getLoc(), pad_output_type, input,
                                             padding_const);
    
          // Set Conv padding to `VALID` since padding has been handled by Pad op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      results.add<ConvertPackToReshape>(context);
    }
    
    //===----------------------------------------------------------------------===//
    // PadOp
    //===----------------------------------------------------------------------===//
    
    LogicalResult PadOp::FoldOperandsPermutation(ArrayRef<int64_t> permutation) {
      // Paddings must be defined by a constant operation.
      auto paddings_op =
    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/lite/flatbuffer_export.cc

          const std::vector<int32_t>& operands,
          const std::vector<int32_t>& results);
    
      std::optional<BufferOffset<tflite::Operator>> BuildStablehloPadOp(
          mlir::stablehlo::PadOp pad_op, const std::vector<int32_t>& operands,
          const std::vector<int32_t>& results);
    
      std::optional<BufferOffset<tflite::Operator>> BuildVhloGatherV1Op(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

        return 3 * count;
      }
    
      return -1;
    }
    
    //===----------------------------------------------------------------------===//
    // PadOp
    //===----------------------------------------------------------------------===//
    
    OpFoldResult PadOp::fold(FoldAdaptor) {
      if (InputOutputHasSameShape(getInput().getType(), getOutput().getType()))
        return getInput();
    
      return {};
    }
    
    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