Searched refs:reduction_indices_flat (Results 1 – 1 of 1) sorted by relevance
78 const auto reduction_indices_flat = reduction_indices.flat<int32>(); in GetReducedIndices() local79 const int32 reduction_dims = reduction_indices_flat.size(); in GetReducedIndices()83 reduced_indices[i] = reduction_indices_flat(reduction_dims - i - 1); in GetReducedIndices()130 const auto reduction_indices_flat = reduction_indices.flat<int32>(); in Compute() local131 const int32 reduction_dims = reduction_indices_flat.size(); in Compute()135 int32 reduce_index = reduction_indices_flat(i); in Compute()