Searched refs:reduce_index (Results 1 – 1 of 1) sorted by relevance
135 int32 reduce_index = reduction_indices_flat(i); in Compute() local137 reduce_index < 0 ? reduce_index + input_dims : reduce_index; in Compute()139 context, reduce_index >= -input_dims && reduce_index < input_dims, in Compute()140 errors::OutOfRange("Invalid reduction dimension ", reduce_index, in Compute()144 reduce_index)); in Compute()