Searched defs:DLPackTensor (Results 1 – 2 of 2) sorted by relevance
42 struct DLPackTensor { struct46 py::object buffer_reference;49 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference;51 std::vector<int64_t> shape;52 std::vector<int64_t> strides;53 DLManagedTensor tensor;
175 class DLPackTensor: class