Searched refs:reduced_indices (Results 1 – 1 of 1) sorted by relevance
65 int64 GetReductionIterSize(const gtl::InlinedVector<int32, 8>& reduced_indices, in GetReductionIterSize() argument68 for (int32 reduce_dim : reduced_indices) { in GetReductionIterSize()81 gtl::InlinedVector<int32, 8> reduced_indices(reduction_dims); in GetReducedIndices() local83 reduced_indices[i] = reduction_indices_flat(reduction_dims - i - 1); in GetReducedIndices()84 reduced_indices[i] += reduced_indices[i] < 0 ? input_dims : 0; in GetReducedIndices()87 return reduced_indices; in GetReducedIndices()148 gtl::InlinedVector<int32, 8> reduced_indices = in Compute() local162 GetReductionIterSize(reduced_indices, input_shape); in Compute()171 reduction_index, reduced_indices, input_shape, strides); in Compute()