Searched defs:TfDlManagedTensorCtx (Results 1 – 1 of 1) sorted by relevance
35 struct TfDlManagedTensorCtx { struct36 TensorReference reference;37 std::vector<int64_t> shape;38 std::vector<int64_t> strides;39 DLManagedTensor tensor;41 explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {} in TfDlManagedTensorCtx() argument