Browse Source

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

modified:   ge/graph/preprocess/graph_preprocess.cc
pull/931/head
zhaoxinxin 5 years ago
parent
commit
8ca55d3cb9
2 changed files with 1 additions and 15 deletions
  1. +1
    -0
      ge/graph/load/new_model_manager/model_manager.cc
  2. +0
    -15
      ge/graph/preprocess/graph_preprocess.cc

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

@@ -527,6 +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.blobs.push_back(data);
}
if (!GetLocalOmgContext().user_input_dims.empty() && GetLocalOmgContext().need_multi_batch) {


+ 0
- 15
ge/graph/preprocess/graph_preprocess.cc View File

@@ -1032,21 +1032,6 @@ Status UpdateDynamicInputShapeRange(const ge::GeAttrValue::INT index,
desc.SetShape(origin_shape);
desc.SetShapeRange(current_shape_range_vec);

int64_t dynamic_shape_size = 1;
for (const auto range_pair : range_vec.at(index)) {
FMK_INT64_MULCHECK(dynamic_shape_size, range_pair.second);
dynamic_shape_size *= range_pair.second;
}
auto data_type_size = GetSizeByDataType(desc.GetDataType());
if (data_type_size < 0) {
GELOGE(PARAM_INVALID, "Input data type is %s, is not supported.",
TypeUtils::DataTypeToSerialString(desc.GetDataType()).c_str());
return PARAM_INVALID;
}
FMK_INT64_MULCHECK(dynamic_shape_size, data_type_size);
dynamic_shape_size *= data_type_size;
GELOGI("In dynamic_execute mode ,set input %s shape range size %ld", op->GetName().c_str(), dynamic_shape_size);
ge::TensorUtils::SetSize(desc, dynamic_shape_size);
graphStatus graph_ret = op->UpdateInputDesc(0, desc);
GE_CHK_STATUS_RET(graph_ret, "UpdateInputDesc fail, graph ret: %u", graph_ret);
graph_ret = op->UpdateOutputDesc(0, desc);


Loading…
Cancel
Save