mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!46917 Tensor tostring support uint64_t 

Merge pull request !46917 from caifubi/r2.0.0-alpha-input-to-attr
This commit is contained in:
i-robot 2022-12-17 02:38:47 +00:00 committed by Gitee
parent 6f6ebcdd7c c57b67050d
commit 53ce6bf714
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/utils/anfalgo.cc
View File

@ -1731,6 +1731,8 @@ std::string AnfAlgo::GetTensorValueString(const tensor::TensorPtr &tensor) {
fn(reinterpret_cast<uint32_t *>(tensor->data_c()));
} else if (dtype->type_id() == kNumberTypeInt64) {
fn(reinterpret_cast<int64_t *>(tensor->data_c()));
} else if (dtype->type_id() == kNumberTypeUInt64) {
fn(reinterpret_cast<uint64_t *>(tensor->data_c()));
} else if (dtype->type_id() == kNumberTypeFloat16) {
fn(reinterpret_cast<float16 *>(tensor->data_c()));
} else if (dtype->type_id() == kNumberTypeFloat64) {