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