Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TfDlManagedTensorCtx (0.11 sec)

  1. tensorflow/c/eager/dlpack.cc

    // original framework of destruction, and this context will be deleted also.
    struct TfDlManagedTensorCtx {
      TensorReference reference;
      std::vector<int64_t> shape;
      std::vector<int64_t> strides;
      DLManagedTensor tensor;
    
      explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {}
    };
    
    // Gets tensor from eager tensor handle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
Back to top