Searched refs:reduction_index (Results 1 – 3 of 3) sorted by relevance
168 for (int64 reduction_index = 0; reduction_index < reduction_iter_size; in Compute() local169 ++reduction_index) { in Compute()171 reduction_index, reduced_indices, input_shape, strides); in Compute()172 curr_strings[reduction_index] = in Compute()
1137 const T reduction_index = reduction_indices(i); in ReductionShapeHelper() local1138 if (reduction_index < -input_rank || reduction_index >= input_rank) { in ReductionShapeHelper()1140 reduction_index, " for input with ", in ReductionShapeHelper()1144 auto wrapped_index = reduction_index; in ReductionShapeHelper()
577 const int32 reduction_index = reduction_indices[i]; in ProcessTensorFlowReductionOperator() local578 if (reduction_index < -input_rank || reduction_index >= input_rank) { in ProcessTensorFlowReductionOperator()579 CHECK(false) << "Invalid reduction dimension " << reduction_index in ProcessTensorFlowReductionOperator()582 int32 wrapped_index = reduction_index; in ProcessTensorFlowReductionOperator()