virtual const FunctionDef* FindFunctionDef(const string& name) const = 0;

  // Find and return a function record added by its name.
  virtual core::RefCountPtr<FunctionRecord> FindRecord(
      const string& name) const = 0;

  // Return the ParsedName of Host CPU device.
  virtual const DeviceNameUtils::ParsedName& HostCPUParsedName() const = 0;
  virtual const string& HostCPUName() const = 0;

}

void TFE_ContextGetGraphDebugInfo(TFE_Context* ctx, const char* function_name,
                                  TF_Buffer* buf, TF_Status* status) {
  auto function_record = tensorflow::unwrap(ctx)->FindRecord(function_name);
  if (function_record == nullptr) {
    status->status = tensorflow::errors::NotFound(
        "Unable to find function with name: ", function_name);
    return;
  }

TF_Function* TFE_ContextGetFunction(TFE_Context* ctx, const char* name,
                                    TF_Status* status) {
  tensorflow::core::RefCountPtr<tensorflow::FunctionRecord> record =
      tensorflow::unwrap(ctx)->FindRecord(name);

  if (record == nullptr) {
    status->status = tensorflow::errors::NotFound(
        "Unable to find Function with name: ", name);
    return nullptr;
  }