1 #ifndef LITE_TENSOR_H 2 #define LITE_TENSOR_H 3 #include "mindir_lite_graph.h" 4 #include "mindir_types.h" 5 6 namespace mindspore { 7 namespace lite { 8 9 // ********** Tensor ********** 10 TensorPtr MindIR_Tensor_Create(); 11 TensorPtr MindIR_Tensor_Create(const char *name, DataType data_type, const int32_t *dims, uint32_t dims_size, 12 Format format, const uint8_t *data, uint32_t data_size, 13 const QuantParam *quant_params, uint32_t quant_params_size); 14 std::string MindIR_Tensor_GetName(ConstTensorPtr tensor); 15 void MindIR_Tensor_SetName(TensorPtr *tensor, const std::string &name); 16 DataType MindIR_Tensor_GetDataType(ConstTensorPtr tensor); 17 void MindIR_Tensor_SetDataType(TensorPtr *tensor, DataType data_type); 18 std::vector<int32_t> MindIR_Tensor_GetDims(ConstTensorPtr tensor); 19 void MindIR_Tensor_SetDims(TensorPtr *tensor, const std::vector<int32_t> &dims); 20 Format MindIR_Tensor_GetFormat(ConstTensorPtr tensor); 21 void MindIR_Tensor_SetFormat(TensorPtr *tensor, Format format); 22 void MindIR_Tensor_SetData(TensorPtr *tensor, const std::vector<uint8_t> &data); 23 std::vector<uint8_t> MindIR_Tensor_GetData(ConstTensorPtr tensor); 24 std::vector<QuantParam> MindIR_Tensor_GetQuantParams(ConstTensorPtr tensor); 25 void MindIR_Tensor_SetQuantParams(TensorPtr *tensor, const std::vector<QuantParam> &quant_params); 26 int32_t MindIR_Tensor_GetNodeType(ConstTensorPtr tensor); 27 28 void MindIR_Tensor_Destroy(TensorPtr *tensor); 29 30 } // namespace lite 31 } // namespace mindspore 32 33 #endif // LITE_TENSOR_H 34