Searched refs:origin_grad_index (Results 1 – 1 of 1) sorted by relevance
287 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in gradient() local288 EXC_IF_VEC_IDX_OOB(inputs_, origin_grad_index); in gradient()289 MS_EXCEPTION_IF_NULL(inputs_[origin_grad_index]); in gradient()290 return inputs_[origin_grad_index]; in gradient()294 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in indices() local295 EXC_IF_VEC_IDX_OOB(inputs_, origin_grad_index); in indices()296 MS_EXCEPTION_IF_NULL(inputs_[origin_grad_index]); in indices()297 return inputs_[origin_grad_index]; in indices()347 size_t origin_grad_index = kSparseAdamOriginIdx.at("grad"); in gradient() local348 EXC_IF_VEC_IDX_OOB(inputs_, origin_grad_index); in gradient()[all …]