Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorListResizeOp (0.28 sec)

  1. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

    // elements, and append it to the end of the input tensorlist.
    struct ConvertTensorListResize
        : public OpConversionPattern<TF::TensorListResizeOp> {
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          TF::TensorListResizeOp op, OpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        Value input_handle = adaptor.getOperands()[0];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          }
          if (auto pop = llvm::dyn_cast<TensorListPopBackOp>(use.getOwner())) {
            add_to_worklist(pop.getOutputHandle());
            continue;
          }
          if (auto resize = llvm::dyn_cast<TensorListResizeOp>(use.getOwner())) {
            add_to_worklist(resize.getOutputHandle());
            continue;
          }
          // WhileRegionOp can explicitly capture TensorList value to be used inside
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top