Searched refs:MaybePinToCustomDevice (Results 1 – 3 of 3) sorted by relevance
142 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()
59 TF_RETURN_IF_ERROR(MaybePinToCustomDevice(&custom_device, *op)); in Execute()98 Status CustomDeviceOpHandler::MaybePinToCustomDevice( in MaybePinToCustomDevice() function in tensorflow::CustomDeviceOpHandler
41 Status MaybePinToCustomDevice(CustomDevice** device,