Home
last modified time | relevance | path

Searched refs:GetGradients (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/lite/src/cxx_api/model/
Dmodel.cc221 std::vector<MSTensor> Model::GetGradients() const { in GetGradients() function in mindspore::Model
227 return impl_->GetGradients(); in GetGradients()
Dmodel_impl.h76 std::vector<MSTensor> GetGradients() const;
Dmodel_impl.cc347 std::vector<MSTensor> ModelImpl::GetGradients() const { in GetGradients() function in mindspore::ModelImpl
353 auto params = session_->GetGradients(); in GetGradients()
/third_party/mindspore/mindspore/lite/include/
Dlite_session.h209 virtual std::vector<tensor::MSTensor *> GetGradients() const { in GetGradients() function
/third_party/mindspore/include/api/
Dmodel.h122 std::vector<MSTensor> GetGradients() const;
/third_party/mindspore/mindspore/lite/src/train/
Doptimizer_kernel.h103 lite::Tensor *GetGradients() { in GetGradients() function
Dtrain_session.h65 std::vector<tensor::MSTensor *> GetGradients() const override;
Dtrain_session.cc747 std::vector<tensor::MSTensor *> TrainSession::GetGradients() const { in GetGradients() function in mindspore::lite::TrainSession
752 auto kernelGradint = optimizer->GetGradients(); in GetGradients()
762 auto current_gradients = GetGradients(); in ApplyGradients()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/cxx_api/
Dmodel_test.cc167 auto graients = model.GetGradients(); in TEST_F()