fix headers for training apis (#14350)

### Description
Minor refactor PR for fixing header placement for training apis

cmake/onnxruntime_session.cmake

@@ -10,6 +10,7 @@ file(GLOB onnxruntime_session_srcs CONFIGURE_DEPENDS
 if (onnxruntime_ENABLE_TRAINING_APIS)
   file(GLOB_RECURSE training_api_srcs CONFIGURE_DEPENDS
     "${ORTTRAINING_SOURCE_DIR}/training_api/*.cc"
+    "${ORTTRAINING_SOURCE_DIR}/training_api/*.h"
     "${ORTTRAINING_SOURCE_DIR}/core/framework/checkpoint_common.cc"
     "${ORTTRAINING_SOURCE_DIR}/core/framework/checkpoint_common.h"
   )

onnxruntime/core/session/onnxruntime_c_api.cc

@@ -49,7 +49,7 @@ ProviderInfo_CUDA* TryGetProviderInfo_CUDA();
 
 #ifdef ENABLE_TRAINING_APIS
 #include "orttraining/training_api/include/onnxruntime_training_c_api.h"
-#include "orttraining/training_api/include/ort_training_apis.h"
+#include "orttraining/training_api/ort_training_apis.h"
 #endif
 
 #ifdef USE_CANN

orttraining/orttraining/python/orttraining_pybind_state.cc

@@ -35,8 +35,8 @@
 #endif
 
 #ifdef ENABLE_TRAINING_APIS
-#include "orttraining/training_api/include/checkpoint.h"
-#include "orttraining/training_api/include/lr_scheduler.h"
+#include "orttraining/training_api/checkpoint.h"
+#include "orttraining/training_api/lr_scheduler.h"
 
 #endif
 

orttraining/orttraining/test/training_api/core/checkpoint_test.cc

@@ -19,11 +19,11 @@
 #include "core/platform/path_lib.h"
 
 #include "orttraining/core/framework/checkpoint_common.h"
-#include "orttraining/training_api/include/module.h"
-#include "orttraining/training_api/include/optimizer.h"
-#include "orttraining/training_api/include/checkpoint_property.h"
-#include "orttraining/training_api/include/checkpoint.h"
-#include "orttraining/training_api/include/lr_scheduler.h"
+#include "orttraining/training_api/module.h"
+#include "orttraining/training_api/optimizer.h"
+#include "orttraining/training_api/checkpoint_property.h"
+#include "orttraining/training_api/checkpoint.h"
+#include "orttraining/training_api/lr_scheduler.h"
 
 #include "test/test_environment.h"
 #include "test/util/include/asserts.h"

orttraining/orttraining/test/training_api/core/training_api_tests.cc

@@ -10,12 +10,12 @@
 #include "test/framework/test_utils.h"
 #include "test/util/include/asserts.h"
 #include "core/framework/tensorprotoutils.h"
-#include "orttraining/training_api/include/utils.h"
-#include "orttraining/training_api/include/module.h"
-#include "orttraining/training_api/include/optimizer.h"
-#include "orttraining/training_api/include/checkpoint_property.h"
-#include "orttraining/training_api/include/checkpoint.h"
-#include "orttraining/training_api/include/lr_scheduler.h"
+#include "orttraining/training_api/utils.h"
+#include "orttraining/training_api/module.h"
+#include "orttraining/training_api/optimizer.h"
+#include "orttraining/training_api/checkpoint_property.h"
+#include "orttraining/training_api/checkpoint.h"
+#include "orttraining/training_api/lr_scheduler.h"
 #include "orttraining/test/training_api/core/data_utils.h"
 #include "test/util/include/temp_dir.h"
 #include "default_providers.h"

orttraining/orttraining/training_api/checkpoint.cc

@@ -14,8 +14,8 @@
 
 #include "orttraining/core/framework/checkpoint_common.h"
 #include "orttraining/core/framework/protobuf_message_sequence.h"
-#include "orttraining/training_api/include/checkpoint.h"
-#include "orttraining/training_api/include/utils.h"
+#include "orttraining/training_api/checkpoint.h"
+#include "orttraining/training_api/utils.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/checkpoint.h

@@ -7,9 +7,9 @@
 #include "core/platform/env.h"
 #include "onnx/defs/tensor_proto_util.h"
 
-#include "orttraining/training_api/include/module.h"
-#include "orttraining/training_api/include/optimizer.h"
-#include "orttraining/training_api/include/checkpoint_property.h"
+#include "orttraining/training_api/module.h"
+#include "orttraining/training_api/optimizer.h"
+#include "orttraining/training_api/checkpoint_property.h"
 
 /**
  * There are two representation for checkpoint respectively in memory and files:

orttraining/orttraining/training_api/checkpoint_property.cc

@@ -6,7 +6,7 @@
 #include "core/platform/path_lib.h"
 #include "core/platform/env.h"
 #include "core/framework/tensorprotoutils.h"
-#include "orttraining/training_api/include/checkpoint_property.h"
+#include "orttraining/training_api/checkpoint_property.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/lr_scheduler.cc

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-#include "orttraining/training_api/include/lr_scheduler.h"
+#include "orttraining/training_api/lr_scheduler.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/lr_scheduler.h

@@ -3,7 +3,7 @@
 
 #pragma once
 
-#include "orttraining/training_api/include/optimizer.h"
+#include "orttraining/training_api/optimizer.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/module.cc

@@ -8,8 +8,8 @@
 #include "core/session/environment.h"
 #include "core/session/onnxruntime_session_options_config_keys.h"
 
-#include "orttraining/training_api/include/module.h"
-#include "orttraining/training_api/include/utils.h"
+#include "orttraining/training_api/module.h"
+#include "orttraining/training_api/utils.h"
 
 using namespace onnxruntime;
 

orttraining/orttraining/training_api/onnxruntime_training_c_api.cc

@@ -8,9 +8,9 @@
 #include "core/session/abi_session_options_impl.h"
 #include "core/session/ort_apis.h"
 #include "core/session/ort_env.h"
-#include "orttraining/training_api/include/checkpoint.h"
-#include "orttraining/training_api/include/ort_training_apis.h"
-#include "orttraining/training_api/include/training_session.h"
+#include "orttraining/training_api/checkpoint.h"
+#include "orttraining/training_api/ort_training_apis.h"
+#include "orttraining/training_api/training_session.h"
 
 namespace {
 

orttraining/orttraining/training_api/optimizer.cc

@@ -6,8 +6,8 @@
 #include "core/session/inference_session.h"
 #include "core/session/environment.h"
 
-#include "orttraining/training_api/include/utils.h"
-#include "orttraining/training_api/include/optimizer.h"
+#include "orttraining/training_api/utils.h"
+#include "orttraining/training_api/optimizer.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/optimizer.h

@@ -6,7 +6,7 @@
 #include "core/session/inference_session.h"
 #include "core/session/environment.h"
 
-#include "orttraining/training_api/include/module.h"
+#include "orttraining/training_api/module.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/training_session.cc

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-#include "orttraining/training_api/include/training_session.h"
+#include "orttraining/training_api/training_session.h"
 
 namespace onnxruntime {
 namespace training {

orttraining/orttraining/training_api/utils.cc

@@ -8,7 +8,7 @@
 #include "core/framework/allocator.h"
 #include "core/framework/tensorprotoutils.h"
 
-#include "orttraining/training_api/include/utils.h"
+#include "orttraining/training_api/utils.h"
 
 namespace onnxruntime {
 namespace training {