Home
last modified time | relevance | path

Searched refs:deallocator_arg (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/
Dtf_tensor_internal.h42 void* deallocator_arg, bool owns_memory) in TF_ManagedBuffer() argument
46 deallocator_arg_(deallocator_arg), in TF_ManagedBuffer()
Dtf_tensor.cc103 void* deallocator_arg) { in TF_NewTensor() argument
122 deallocator(data, len, deallocator_arg); in TF_NewTensor()
124 buf = new TF_ManagedBuffer(data, len, deallocator, deallocator_arg, in TF_NewTensor()
Dtf_tensor.h91 void* deallocator_arg);
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc561 void* deallocator_arg, TF_Status* status) { in TFE_NewTensorHandleFromDeviceMemory() argument
570 deallocator(data, len, deallocator_arg); in TFE_NewTensorHandleFromDeviceMemory()
593 [data, len, deallocator, deallocator_arg]() { in TFE_NewTensorHandleFromDeviceMemory()
594 deallocator(data, len, deallocator_arg); in TFE_NewTensorHandleFromDeviceMemory()
601 new TF_ManagedBuffer(data, len, deallocator, deallocator_arg, in TFE_NewTensorHandleFromDeviceMemory()
Dc_api_experimental.h394 void* deallocator_arg, TF_Status* status);