Searched defs:accum_flat (Results 1 – 2 of 2) sorted by relevance
244 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction() local299 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter() local
3229 auto accum_flat = accum.flat_outer_dims<T>(); in Compute() local