diff options
Diffstat (limited to 'reference_model/src/tensor.h')
| -rw-r--r-- | reference_model/src/tensor.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h index ede42a9..6b7d5f1 100644 --- a/reference_model/src/tensor.h +++ b/reference_model/src/tensor.h @@ -228,6 +228,16 @@ public: virtual int writeToNpyFile(const char* filename) const; virtual int copyValueFrom(Tensor* tensor) = 0; + virtual int readfromVector(const std::vector<float>& vals); + virtual int readfromVector(const std::vector<int32_t>& vals); + virtual int readfromVector(const std::vector<int64_t>& vals); + virtual int readfromVector(const std::vector<unsigned char>& vals); + + virtual int writeToVector(std::vector<float>& vals); + virtual int writeToVector(std::vector<int32_t>& vals); + virtual int writeToVector(std::vector<int64_t>& vals); + virtual int writeToVector(std::vector<unsigned char>& vals); + const char* bool_to_str(bool in) const { static const char* true_str = "true"; |