Browse Source

!4746 Do not expand the value tuple for tensor inputs

Merge pull request !4746 from YuJianfeng/internal_output
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
9f9705baa1
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      mindspore/ccsrc/vm/backend.cc

+ 2
- 3
mindspore/ccsrc/vm/backend.cc View File

@@ -104,9 +104,8 @@ void PushInputTensor(const BaseRef &arg, std::vector<tensor::TensorPtr> *inputs)
auto value_tuple = value->cast<ValueTuplePtr>();
MS_EXCEPTION_IF_NULL(value_tuple);
auto tuple_value = value_tuple->value();
for (const auto &v : tuple_value) {
PushInputTensor(v, inputs);
}
(void)std::transform(tuple_value.begin(), tuple_value.end(), std::back_inserter(*inputs),
[](const ValuePtr &v) { return v->cast<tensor::TensorPtr>(); });
} else if (value->isa<Scalar>()) {
tensor::TensorPtr scalar_tensor = ScalarToTensor(value->cast<ScalarPtr>());
inputs->push_back(scalar_tensor);


Loading…
Cancel
Save