Searched defs:axis_reduce (Results 1 – 1 of 1) sorted by relevance
29 void reduce_one_axis(const int64_t one_axis, const int64_t dim, std::set<int64_t> axis_reduce) { in reduce_one_axis()39 std::set<int64_t> axis_reduce; in infer_shape_reduce() local