Home
last modified time | relevance | path

Searched defs:custom_device (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc157 TFE_CustomDevice custom_device; in RegisterLoggingDevice() local
181 TFE_CustomDevice* custom_device = new TFE_CustomDevice; in AllocateLoggingDevice() local
Dc_api.cc566 tensorflow::CustomDevice* custom_device = nullptr; in TFE_NewTensorHandleFromDeviceMemory() local
1153 auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( in TFE_RegisterCustomDevice() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc57 tensorflow::CustomDevice* custom_device = nullptr; in Execute() local
Dcustom_device_test.cc120 TestCustomDevice custom_device(custom_device_name); in TEST() local