diff --git a/mindspore/ccsrc/plugin/device/ascend/kernel/host/host_kernel_mod.cc b/mindspore/ccsrc/plugin/device/ascend/kernel/host/host_kernel_mod.cc index a4c34fbedc1..ec3a618d1a2 100644 --- a/mindspore/ccsrc/plugin/device/ascend/kernel/host/host_kernel_mod.cc +++ b/mindspore/ccsrc/plugin/device/ascend/kernel/host/host_kernel_mod.cc @@ -81,6 +81,10 @@ void HostKernelMod::InferOp() { MS_LOG(EXCEPTION) << "The node is not dynamic shape."; } KernelMod::InferShape(); + + input_size_list_.clear(); + output_size_list_.clear(); + HostKernelMod::Init(node); } std::vector HostKernelMod::GenTask(const std::vector &, const std::vector &,