Searched defs:axis_set (Results 1 – 7 of 7) sorted by relevance
64 std::set<int64_t> axis_set(axis_vec.begin(), axis_vec.end()); in GetUniqReduceAxes() local178 std::set<int64_t> axis_set = GetUniqReduceAxes(node); in SuitableForAtomicAdd() local
111 std::set<size_t> axis_set(axis.begin(), axis.end()); in Resize() local
100 HashSet<int64_t> axis_set; in Eval() local
136 std::unordered_set<int64_t> axis_set(axis.begin(), axis.end()); in GetReducedOriShape() local
117 std::set<size_t> axis_set(axis.begin(), axis.end()); in Resize() local
397 std::set<int64_t> axis_set; in UpdateParameters() local
482 std::set<int64_t> axis_set; in DoInferReduceShape() local