Searched refs:grad_ptr (Results 1 – 2 of 2) sorted by relevance
34 const float *grad_ptr = gradient + start; in AdamFp32() local40 grad_r = _mm256_loadu_ps(grad_ptr); in AdamFp32()72 grad_ptr += C8NUM; in AdamFp32()
673 GradPtr grad_ptr = std::make_shared<std::vector<float>>(data_len, 0); in HandleInitWeights() local674 MS_EXCEPTION_IF_NULL(grad_ptr); in HandleInitWeights()675 ps_->InitGrad(key, grad_ptr); in HandleInitWeights()