Searched refs:dimensions_to_reduce_set (Results 1 – 1 of 1) sorted by relevance
2088 std::set<int64> dimensions_to_reduce_set(dimensions_to_reduce.begin(), in InferReduceShape() local2093 if (dimensions_to_reduce_set.find(i) == dimensions_to_reduce_set.end()) { in InferReduceShape()