Browse Source

Feature: Get default op format from ge graph

pull/618/head
l00444296 4 years ago
parent
commit
679c78e12c
1 changed files with 1 additions and 3 deletions
  1. +1
    -3
      ge/generator/ge_generator.cc

+ 1
- 3
ge/generator/ge_generator.cc View File

@@ -269,14 +269,12 @@ static void ResetTensorVecShape(const vector<GeTensor> &inputs, vector<GeTensor>
for (auto input : inputs) {
auto input_desc = input.GetTensorDesc();
GeShape shape_ori = input_desc.GetShape();
Format format_ori = input_desc.GetFormat();
DataType type_ori = input_desc.GetDataType();

std::vector<int64_t> dynamic_shape_dims = {kDynamicDimValue};
GeShape dynamic_shape(dynamic_shape_dims);

ge::GeTensor inputTensor;
ge::GeTensorDesc desc(shape_ori, format_ori, type_ori);
ge::GeTensorDesc desc(input_desc);
if (shape_ori.GetDims().size() > 0) {
desc.SetShape(dynamic_shape);
}


Loading…
Cancel
Save