|
|
|
@@ -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); |
|
|
|
|