Searched defs:axis_list (Results 1 – 8 of 8) sorted by relevance
34 std::vector<int> axis_list; in InitKernel() local
50 std::vector<int> axis_list; in MatchFastPattern() local
355 std::vector<int64_t> axis_list; in PrepareAxisRelatedStrategy() local757 Dimensions axis_list; in GetAxisList() local786 auto axis_list = GetAxisList(ops, SizeToLong(incoming_op_index)); in ModifyStrategyIfSqueezeIncoming() local849 Dimensions axis_list; in ModifyStrategyIfReduceIncoming() local905 Dimensions axis_list; in ModifyStrategyIfArgIncoming() local1154 auto axis_list = GetAxisList(ops, SizeToLong(iter_ops)); in ModifyStrategyIfSqueezeOutgoing() local
100 std::vector<int> axis_list; in MatchFastPattern() local
78 std::vector<int> axis_list; in MatchParallelPattern() local
495 auto axis_list = value_tuple->value(); in InferAxis() local
818 std::vector<int64_t> axis_list; in GetReduceAttrAxis() local
368 AbstractBasePtrList axis_list = {dyn_cast<AbstractScalar>(args_spec_list[1])}; in InferImplReduceShape() local