Browse Source

modified: ge/graph/load/new_model_manager/model_manager.cc

pull/931/head
zhaoxinxin 5 years ago
parent
commit
51d703ba15
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/new_model_manager/model_manager.cc

+ 1
- 1
ge/graph/load/new_model_manager/model_manager.cc View File

@@ -527,7 +527,7 @@ Status ModelManager::DataInputTensor(uint32_t model_id, const std::vector<InputT
DataBuffer data;
data.data = inputs[i].data;
data.length = inputs[i].length;
input_data.shapes.emplace_back(inputs[i].data);
input_data.shapes.emplace_back(inputs[i].dims);
input_data.blobs.push_back(data);
}
if (!GetLocalOmgContext().user_input_dims.empty() && GetLocalOmgContext().need_multi_batch) {


Loading…
Cancel
Save