include/edm4eic/TensorData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4EIC_TensorDATA_H
0004 #define EDM4EIC_TensorDATA_H
0005
0006 #include <cstdint>
0007
0008 namespace edm4eic {
0009
0010
0011 /** @class TensorData
0012  *  Tensor type for use in training in inference of ML models
0013  *  @author: D. Kalinkin
0014  */
0015 class TensorData {
0016 public:
0017   std::int32_t elementType{}; ///< Data type in the same encoding as "ONNXTensorElementDataType", 1 - float, 7 - int64
0018
0019   unsigned int shape_begin{};
0020   unsigned int shape_end{};
0021   unsigned int floatData_begin{};
0022   unsigned int floatData_end{};
0023   unsigned int int64Data_begin{};
0024   unsigned int int64Data_end{};
0025 };
0026
0027 } // namespace edm4eic
0028
0029
0030 #endif