Home
last modified time | relevance | path

Searched refs:GetCustomDeviceOpHandler (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_test.cc142 TF_ASSERT_OK(ctx->GetCustomDeviceOpHandler().MaybePinToCustomDevice( in TEST()
153 TF_ASSERT_OK(ctx->GetCustomDeviceOpHandler().MaybePinToCustomDevice( in TEST()
164 TF_ASSERT_OK(ctx->GetCustomDeviceOpHandler().MaybePinToCustomDevice( in TEST()
176 TF_ASSERT_OK(ctx->GetCustomDeviceOpHandler().MaybePinToCustomDevice( in TEST()
Dcontext.h482 CustomDeviceOpHandler& GetCustomDeviceOpHandler() override { in GetCustomDeviceOpHandler() function
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_context.h157 virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0;
Dc_api.cc537 if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName(device_name, in TFE_NewCustomDeviceTensorHandle()
568 if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName( in TFE_NewTensorHandleFromDeviceMemory()
899 unwrapped_op->GetContext()->GetCustomDeviceOpHandler().Execute( in TFE_Execute()