Searched refs:accum_flat (Results 1 – 2 of 2) sorted by relevance
265 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction() local281 T* accum_slice_ptr = &accum_flat(idx_a, 0); in AddToAccumGradFunction()291 T* accum_slice_ptr = &accum_flat(idx_a, 0); in AddToAccumGradFunction()324 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter() local339 Eigen::DSizes<Eigen::DenseIndex, 1> slice_shape(accum_flat.dimension(1)); in DivideAccumGradByCounter()341 T* accum_slice_ptr = &accum_flat(i, 0); in DivideAccumGradByCounter()
1389 auto accum_flat = accum.flat_outer_dims<T>(); in Compute() local1401 auto a = accum_flat.template chip<0>(index); in Compute()1412 auto accum_flat = accum.flat<T>(); in Compute() local1415 const Tindex first_dim_size = accum_flat.size(); in Compute()1423 T& a = accum_flat(index); in Compute()1547 auto accum_flat = accum.flat_outer_dims<T>(); in Compute() local1559 auto a = accum_flat.template chip<0>(index); in Compute()1582 auto accum_flat = accum.flat<T>(); in Compute() local1587 const Tindex first_dim_size = accum_flat.size(); in Compute()1595 T& a = accum_flat(index); in Compute()[all …]