Searched defs:indices_row (Results 1 – 2 of 2) sorted by relevance
42 Tindex indices_row = grad_index / col_size; in SparseApplyAdagradKernel() local81 Tindex indices_row = grad_index / col_size; in SparseApplyProximalAdagradKernel() local124 const Tindex indices_row = grad_index / col_size; in SparseApplyFtrlKernel() local228 Tindex indices_row = grad_index / col_size; in SparseApplyKerasMomentumKernel() local808 Tindex indices_row = grad_index / col_size; in SparseApplyAdadeltaKernel() local
94 auto indices_row = indices_accessor[j]; in get_offsets() local143 const auto indices_row = indices_accessor[j]; in get_pools() local