Searched refs:dimensions_to_reduce_set (Results 1 – 1 of 1) sorted by relevance
2157 absl::flat_hash_set<int64> dimensions_to_reduce_set; in InferReduceShape() local2159 if (!dimensions_to_reduce_set.insert(dim_to_reduce).second) { in InferReduceShape()2168 if (dimensions_to_reduce_set.find(i) == dimensions_to_reduce_set.end()) { in InferReduceShape()