Searched defs:on_delete_callback (Results 1 – 6 of 6) sorted by relevance
77 std::function<void()> on_delete_callback) in TrackedTfrtCpuDeviceBuffer()86 std::function<void()> on_delete_callback) in TrackedTfrtCpuDeviceBuffer()
169 std::function<void()> on_delete_callback) in TrackedDeviceBuffer()
220 std::function<void()> on_delete_callback) override { in CreateViewOfDeviceBuffer()
465 std::function<void()> on_delete_callback) { in CreateViewOfDeviceBuffer()513 std::function<void()> on_delete_callback; in BufferFromHostBuffer() local
741 std::function<void()> on_delete_callback; in BufferFromHostBuffer() local1076 std::function<void()> on_delete_callback) { in CreateViewOfDeviceBuffer()
388 std::function<void()> on_delete_callback; in DLPackManagedTensorToBuffer() local