From c1721a26aa5084eb943510681c5b9332cd0df467 Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Tue, 1 Jun 2021 14:12:20 +0800 Subject: [PATCH] add inner offset for var addr judgement and update --- ge/graph/build/task_generator.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/build/task_generator.cc b/ge/graph/build/task_generator.cc index d2b1962f..2db0b6b7 100755 --- a/ge/graph/build/task_generator.cc +++ b/ge/graph/build/task_generator.cc @@ -193,7 +193,7 @@ Status TaskGenerator::UpdateOpIsVarAttr(const OpDescPtr &op_desc, uint64_t sessi vector input_offsets = op_desc->GetInputOffset(); if (!(input_offsets.empty())) { vector input_var; - int64_t valid_input_index = 0; + size_t valid_input_index = 0; for (uint32_t i = 0; i < op_desc->GetAllInputsSize(); i++) { vector output_list; auto input_tensor_desc = op_desc->MutableInputDesc(i); @@ -220,7 +220,7 @@ Status TaskGenerator::UpdateOpIsVarAttr(const OpDescPtr &op_desc, uint64_t sessi vector output_offsets = op_desc->GetOutputOffset(); if (!(output_offsets.empty())) { vector output_var; - int64_t valid_output_index = 0; + size_t valid_output_index = 0; for (uint32_t i = 0; i < op_desc->GetAllOutputsDescSize(); i++) { vector output_list; auto output_tensor_desc = op_desc->MutableOutputDesc(i);