Searched refs:wrapped_index (Results 1 – 2 of 2) sorted by relevance
1144 auto wrapped_index = reduction_index; in ReductionShapeHelper() local1145 if (wrapped_index < 0) { in ReductionShapeHelper()1146 wrapped_index += input_rank; in ReductionShapeHelper()1149 true_indices->insert(wrapped_index); in ReductionShapeHelper()
582 int32 wrapped_index = reduction_index; in ProcessTensorFlowReductionOperator() local583 if (wrapped_index < 0) { in ProcessTensorFlowReductionOperator()584 wrapped_index += input_rank; in ProcessTensorFlowReductionOperator()586 true_indices.insert(wrapped_index); in ProcessTensorFlowReductionOperator()