From 679c78e12c6e21edc5fc46b6f1f3843da42df085 Mon Sep 17 00:00:00 2001 From: l00444296 Date: Tue, 15 Dec 2020 16:26:16 +0800 Subject: [PATCH] Feature: Get default op format from ge graph --- ge/generator/ge_generator.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 7a489bfd..2baa2c5b 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -269,14 +269,12 @@ static void ResetTensorVecShape(const vector &inputs, vector 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 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); }