Searched defs:on_delete_callback (Results 1 – 5 of 5) sorted by relevance
27 std::function<void()> on_delete_callback) in TrackedTfrtCpuDeviceBuffer()
169 std::function<void()> on_delete_callback) in TrackedDeviceBuffer()
404 std::function<void()> on_delete_callback) { in CreateViewOfDeviceBuffer()452 std::function<void()> on_delete_callback; in BufferFromHostBuffer() local
731 std::function<void()> on_delete_callback; in BufferFromHostBuffer() local1077 std::function<void()> on_delete_callback) { in CreateViewOfDeviceBuffer()
375 std::function<void()> on_delete_callback; in DLPackManagedTensorToBuffer() local