Searched refs:kDlTensorCapsuleName (Results 1 – 3 of 3) sorted by relevance
38 const char* const kDlTensorCapsuleName = "dltensor"; variable303 py::capsule capsule(&pack.release()->tensor, kDlTensorCapsuleName, in BufferToDLPackManagedTensor()306 PyCapsule_GetPointer(obj, kDlTensorCapsuleName)); in BufferToDLPackManagedTensor()320 if (absl::string_view(tensor.name()) != kDlTensorCapsuleName) { in DLPackManagedTensorToBuffer()
24 const char* const kDlTensorCapsuleName = "dltensor"; variable
1387 dlm_ptr, tensorflow::kDlTensorCapsuleName, [](PyObject* capsule) { in PYBIND11_MODULE()1388 if (PyCapsule_IsValid(capsule, tensorflow::kDlTensorCapsuleName)) { in PYBIND11_MODULE()1390 PyCapsule_GetPointer(capsule, tensorflow::kDlTensorCapsuleName); in PYBIND11_MODULE()1405 tensorflow::kDlTensorCapsuleName) { in PYBIND11_MODULE()