Searched refs:axes_set (Results 1 – 2 of 2) sorted by relevance
69 std::unordered_set<int32_t> axes_set(Axes().cbegin(), Axes().cend()); in OutputShape()71 if (axes_set.count(i) != 0) { in OutputShape()84 std::unordered_set<int32_t> axes_set(Axes().cbegin(), Axes().cend()); in OutputSize()86 if (axes_set.count(i) == 0) { in OutputSize()
276 llvm::SmallSet<uint64_t, 4> axes_set; in ShapeMatchesReduceWithKeepAxes() local278 axes_set.insert(a.getZExtValue()); in ShapeMatchesReduceWithKeepAxes()283 if (axes_set.contains(i)) { in ShapeMatchesReduceWithKeepAxes()