Browse Source

handle empty tensor in ge adapter

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
tags/v1.2.0-rc1
zhoufeng 5 years ago
parent
commit
f107fbd51e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc

+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc View File

@@ -132,7 +132,7 @@ GeTensor VectorToTensorUtil(const ValuePtr &value) {
auto vec = value->isa<ValueTuple>() ? value->cast<ValueTuplePtr>()->value() : value->cast<ValueListPtr>()->value();
if (vec.empty()) {
MS_LOG(WARNING) << "Convert a none tuple to an empty ge tensor";
return GeTensor();
return GeTensor(GeTensorDesc(ge::Shape({0})));
}
MS_EXCEPTION_IF_NULL(vec[0]);
if (vec[0]->isa<Int32Imm>()) {


Loading…
Cancel
Save