Searched refs:TFE_CustomDevice (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | pywrap_parallel_device.cc | 32 delete reinterpret_cast<TFE_CustomDevice*>( in CallDelete_Device()
51 TFE_CustomDevice* device = new TFE_CustomDevice; in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | custom_device_testutil.cc | 40 delete reinterpret_cast<TFE_CustomDevice*>( in CallDelete_Device()
59 TFE_CustomDevice* device; in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | custom_device_testutil.cc | 157 TFE_CustomDevice custom_device; in RegisterLoggingDevice()
179 bool* executed_flag, TFE_CustomDevice** device, in AllocateLoggingDevice()
181 TFE_CustomDevice* custom_device = new TFE_CustomDevice; in AllocateLoggingDevice()
|
| D | custom_device_testutil.h | 31 bool* executed_flag, TFE_CustomDevice** device,
|
| D | c_api_experimental.h | 441 typedef struct TFE_CustomDevice { struct
481 } TFE_CustomDevice; typedef
515 TFE_CustomDevice device,
|
| D | c_api.cc | 411 CustomDeviceAPI(TFE_Context* context, TFE_CustomDevice device, void* info, in CustomDeviceAPI()
479 TFE_CustomDevice device_;
1146 void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, in TFE_RegisterCustomDevice()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device.h | 60 TFE_CustomDevice* device, void** device_info);
|
| D | parallel_device_testlib.h | 157 TFE_CustomDevice device; in RegisterParallelDevice()
|
| D | parallel_device.cc | 367 TFE_CustomDevice* device, void** device_info) { in AllocateParallelDevice()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1450 *reinterpret_cast<TFE_CustomDevice*>( in PYBIND11_MODULE()
|