Searched refs:dim_callback (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 497 DimCallback dim_callback, in CAPICustomDeviceTensorHandle() argument
502 dim_callback_(dim_callback), in CAPICustomDeviceTensorHandle()
531 int64_t (*dim_callback)(void* data, int dim_index, void* arg, in TFE_NewCustomDeviceTensorHandle()
551 [dim_callback, data, arg](int dim_index, TF_Status* status) { in TFE_NewCustomDeviceTensorHandle()
552 return dim_callback(data, dim_index, arg, status); in TFE_NewCustomDeviceTensorHandle()
|
| D | c_api_experimental.h | 539 int64_t (*dim_callback)(void* data, int dim_index, void* arg,
|