Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorListFromTensorOp (0.44 sec)

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

    // TensorListGetItemOp
    //===----------------------------------------------------------------------===//
    
    namespace {
    // If the input of TensorListGetItemOp is TensorListFromTensorOp and the
    // TensorListFromTensorOp is only used by TensorListGetItemOp (not modified by
    // other TensorList ops), we can convert it to a GatherOp.
    class ConvertTensorListGetItemOpOfTensorListFromTensorOpToGather
    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/tensorflow/transforms/shape_inference.cc

    }
    
    // Returns true if the op creates a TensorList.
    bool IsTensorListInitOp(Operation* op) {
      return isa<TensorListReserveOp>(op) || isa<EmptyTensorListOp>(op) ||
             isa<TensorListFromTensorOp>(op);
    }
    
    // Returns the `element_shape` operand of the ops that create a TensorList.
    Value GetElementShapeOperand(Operation* op) {
      if (auto empty_tl = dyn_cast<EmptyTensorListOp>(op))
    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