Searched refs:num_dims_callback (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 496 NumDimsCallback num_dims_callback, in CAPICustomDeviceTensorHandle() argument
501 num_dims_callback_(num_dims_callback), in CAPICustomDeviceTensorHandle()
530 int (*num_dims_callback)(void* data, void* arg, TF_Status* status), in TFE_NewCustomDeviceTensorHandle()
547 [num_dims_callback, data, arg](TF_Status* status) { in TFE_NewCustomDeviceTensorHandle()
548 return num_dims_callback(data, arg, status); in TFE_NewCustomDeviceTensorHandle()
|
| D | c_api_experimental.h | 538 int (*num_dims_callback)(void* data, void* arg, TF_Status* status),
|