Browse Source

fix adam op accu bug

tags/v1.3.0
lizhenyu 4 years ago
parent
commit
2d2bcef88d
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/session_basic.cc

+ 1
- 0
mindspore/ccsrc/backend/session/session_basic.cc View File

@@ -306,6 +306,7 @@ ParameterPtr ConstructRunOpParameter(const std::shared_ptr<KernelGraph> &graph,
kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{device_address->format()});
kernel_build_info_builder->SetOutputsDeviceType(std::vector<TypeId>{device_address->type_id()});
kernel_build_info_builder->SetOutputsReshapeType({input_tensor->padding_type()});
AnfAlgo::SetOutputAddr(device_address, 0, param.get());
}
AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), param.get());
// construct abstract of parameter


Loading…
Cancel
Save