onnxruntime/winml/lib/Common/inc/onnx.h
Brian Martin d738bdd968
Brianma/cpuwinml (#2466)
* allow building winml cpu without dml.
2019-11-22 16:13:28 -08:00

28 lines
No EOL
663 B
C

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
#pragma once
#include "common.h"
// Needed to work around the fact that OnnxRuntime defines ERROR
#ifdef ERROR
#undef ERROR
#endif
#include "core/session/inference_session.h"
// Restore ERROR define
#define ERROR 0
// the C++ ort api
#include "core/session/onnxruntime_cxx_api.h"
#ifdef USE_DML
#include <DirectML.h>
#endif USE_DML
#include "core/framework/customregistry.h"
#include "core/framework/allocatormgr.h"
#include "core/session/environment.h"
#include "core/session/IOBinding.h"
#include "core/common/logging/logging.h"
#include "core/common/logging/sinks/clog_sink.h"