Searched defs:origin_grad_index (Results 1 – 1 of 1) sorted by relevance
287 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in gradient() local294 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in indices() local347 size_t origin_grad_index = kSparseAdamOriginIdx.at("grad"); in gradient() local390 size_t origin_grad_index = kSparseFtrlOriginIdx.at("grad"); in gradient() local