Searched refs:dimensions_to_reduce_set (Results 1 – 1 of 1) sorted by relevance
2261 absl::flat_hash_set<int64_t> dimensions_to_reduce_set; in InferReduceShape() local2263 if (!dimensions_to_reduce_set.insert(dim_to_reduce).second) { in InferReduceShape()2272 if (dimensions_to_reduce_set.find(i) == dimensions_to_reduce_set.end()) { in InferReduceShape()