Program Listing for File tensor.h
↰ Return to documentation for file (include/converter/include/mindapi/ir/tensor.h
)
#ifndef MINDSPORE_CORE_MINDAPI_IR_TENSOR_H_
#define MINDSPORE_CORE_MINDAPI_IR_TENSOR_H_
#include <cstdint>
#include "mindapi/base/base.h"
#include "mindapi/base/shape_vector.h"
#include "mindapi/base/type_id.h"
#include "mindapi/ir/common.h"
#include "mindapi/ir/value.h"
namespace mindspore::api {
class MIND_API Tensor : public Value {
public:
MIND_API_BASE_MEMBER(Tensor);
Tensor(TypeId data_type, const ShapeVector &shape);
Tensor(TypeId data_type, const ShapeVector &shape, void *data, size_t data_len);
const ShapeVector &shape() const;
void set_shape(const ShapeVector &shape);
TypeId data_type() const;
void set_data_type(const TypeId data_type);
const void *data() const;
void *data();
size_t DataSize() const;
std::size_t Size() const;
};
using TensorPtr = SharedPtr<Tensor>;
} // namespace mindspore::api
#endif // MINDSPORE_CORE_MINDAPI_IR_TENSOR_H_