Searched refs:true_indices (Results 1 – 2 of 2) sorted by relevance
1134 std::set<int64>* true_indices) { in ReductionShapeHelper() argument1149 true_indices->insert(wrapped_index); in ReductionShapeHelper()1184 std::set<int64> true_indices; in ReductionShape() local1187 input_rank, &true_indices)); in ReductionShape()1190 input_rank, &true_indices)); in ReductionShape()1198 if (true_indices.count(i) > 0) { in ReductionShape()
573 std::set<int32> true_indices; in ProcessTensorFlowReductionOperator() local586 true_indices.insert(wrapped_index); in ProcessTensorFlowReductionOperator()592 if (true_indices.count(i) > 0) { in ProcessTensorFlowReductionOperator()