Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!38721 revert fuzzy fix 

Merge pull request !38721 from liyan2022/master
This commit is contained in:
i-robot 2022-07-26 01:50:19 +00:00 committed by Gitee
parent eafd3e9e59 9cd4a07c97
commit 42389e62bf
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/tensor_category.cc
View File

@ -20,7 +20,7 @@
namespace mindspore {
namespace lite {
Category TensorCategory(const int node_type, const size_t shape_num, const TypeId data_type, const size_t data_size) {
return (node_type == NodeType_ValueNode && data_size != 0)
return (node_type == NodeType_ValueNode)
? (shape_num == 0 && data_size == DataTypeSize(data_type) ? Category::CONST_SCALAR : Category::CONST_TENSOR)
: Category::VAR;
}