Searched defs:DLPackTensor (Results 1 – 1 of 1) sorted by relevance
40 struct DLPackTensor { struct44 py::object buffer_reference;47 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference;49 std::vector<int64> shape;50 std::vector<int64> strides;51 DLManagedTensor tensor;