Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorListPopBack (0.15 sec)

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

        // op into an op with a single varidic output in `.td`.
        auto converted = rewriter.create<TFL::CustomOp>(
            op->getLoc(), op->getResultTypes(), op->getOperands(),
            "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), ""));
        rewriter.replaceOp(op, converted.getResults());
        return success();
      }
    };
    
    struct ConvertTensorListPushBack
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      auto buffer = pop.getInputHandle();
      auto it = buffer_to_size->find(buffer);
      if (it == buffer_to_size->end()) {
        pop.emitOpError("found tf.TensorListPopBack on unknown TensorList.");
        return failure();
      }
      if (it->getSecond().fixed) {
        return pop.emitError("cannot pop on a fixed-size tensor list");
      }
      auto size = it->getSecond().size;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
Back to top