onnxruntime/orttraining/orttraining/training_api
Scott McKay 2102b8f67c
Avoid duplicate symbol error between ONNX and ORT for ostream operator<< with TensorShapeProto (#12651)
* Remove ostream operator<< definitions for TensorShapeProto and TensorProto as they clash with ONNX definitions in onnx/defs/printer.h/cc.

Currently printer.h (unnecessarily) pulls in a number of other ONNX headers which causes naming clashes with parts of ORT. It is also excluded in a minimal build.

Instead convert the onnx::TensorShapeProto to onnxruntime::TensorShape so we use the existing ostream operator<< for TensorShape.

Make GetTensorShapeFromTensorProto consistent with GetTensorShapeFromTensorShapeProto so both return a TensorShape (as the name implies).
2022-08-22 17:20:52 +10:00
..
include Refactor with std::variant (on device training) (#12383) 2022-08-17 08:31:23 +08:00
checkpoint.cc Refactor with std::variant (on device training) (#12383) 2022-08-17 08:31:23 +08:00
checkpoint_property.cc Avoid duplicate symbol error between ONNX and ORT for ostream operator<< with TensorShapeProto (#12651) 2022-08-22 17:20:52 +10:00
lr_scheduler.cc lr scheduler implementation (on device training) (#11714) 2022-06-09 08:04:30 +08:00
module.cc Add win-ci pipeline for on-device training (#12513) 2022-08-10 14:45:39 -07:00
onnxruntime_training_c_api.cc Add Learning Rate Scheduler C API (#11957) 2022-08-15 09:10:25 -07:00
optimizer.cc Make GetTrainingApi as a part of the OrtApis, add Training API documentation and address other pull request review comments 2022-07-21 18:11:48 +00:00
training_session.cc Add Learning Rate Scheduler C API (#11957) 2022-08-15 09:10:25 -07:00
utils.cc Add win-ci pipeline for on-device training (#12513) 2022-08-10 14:45:39 -07:00