pull/1037/head
wjm 4 years ago
parent df17218f41
commit 9284635a22

@ -180,7 +180,7 @@ Status MultiBatchClonePass::CheckAndParseDynamicData(){
size_t unknown_shape_count = 0;
auto data_name_and_shape = GetLocalOmgContext().user_input_dims;
std::vector<std::string> data_name_order;
for (auto &item : GetLocalOmgContext().user_input_dims) {
for (auto &item : data_name_and_shape) {
data_name_order.push_back(item.first);
}
if (!getnext_sink_dynamic_dims_) {
@ -190,7 +190,6 @@ Status MultiBatchClonePass::CheckAndParseDynamicData(){
auto data_format = data_desc.GetFormat() == Format::FORMAT_NCHW ? "NCHW" :
data_desc.GetFormat() == Format::FORMAT_NHWC ? "NHWC" : "Others";
auto data_name = node->GetName();
GELOGI("CheckAndParseDynamicData shape_dims is %s.", formats::JoinToString(data_shape.GetDims()).c_str());
std::vector<int64_t> data_shape_dims = data_shape.GetDims();
if (std::all_of(data_shape_dims.begin(), data_shape_dims.end(), [](int64_t val) { return val >= 0; })) {

Loading…
Cancel
Save