Searched defs:DLPackTensor (Results 1 – 1 of 1) sorted by relevance
43 struct DLPackTensor { struct47 py::object buffer_reference;50 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference;52 std::vector<int64> shape;53 std::vector<int64> strides;54 DLManagedTensor tensor;