Searched refs:axis_list (Results 1 – 8 of 8) sorted by relevance
34 std::vector<int> axis_list; in InitKernel() local36 (void)std::transform(axis_list_me.begin(), axis_list_me.end(), std::back_inserter(axis_list), in InitKernel()38 if (axis_list.size() != 1) { in InitKernel()41 int axis = axis_list[0]; in InitKernel()
355 std::vector<int64_t> axis_list; in PrepareAxisRelatedStrategy() local367 axis_list.push_back(iter->second->cast<Int64ImmPtr>()->value()); in PrepareAxisRelatedStrategy()374 (void)std::transform(value_vector.begin(), value_vector.end(), std::back_inserter(axis_list), in PrepareAxisRelatedStrategy()380 axis_list.push_back(default_axis); in PrepareAxisRelatedStrategy()383 for (auto &axis : axis_list) { in PrepareAxisRelatedStrategy()757 Dimensions axis_list; in GetAxisList() local773 axis_list.push_back(axis); in GetAxisList()775 return axis_list; in GetAxisList()786 auto axis_list = GetAxisList(ops, SizeToLong(incoming_op_index)); in ModifyStrategyIfSqueezeIncoming() local787 for (auto axis : axis_list) { in ModifyStrategyIfSqueezeIncoming()[all …]
50 std::vector<int> axis_list; in MatchFastPattern() local53 axis_list.emplace_back(i); in MatchFastPattern()56 if (axis_list.size() == 1) { in MatchFastPattern()57 split_axis_ = axis_list.front(); in MatchFastPattern()
100 std::vector<int> axis_list; in MatchFastPattern() local103 axis_list.emplace_back(i); in MatchFastPattern()106 if (axis_list.size() == 1) { in MatchFastPattern()107 split_axis_ = axis_list.front(); in MatchFastPattern()
78 std::vector<int> axis_list; in MatchParallelPattern() local81 (void)axis_list.emplace_back(i); in MatchParallelPattern()84 if (axis_list.size() == 1) { in MatchParallelPattern()85 split_axis_ = axis_list.front(); in MatchParallelPattern()
495 auto axis_list = value_tuple->value(); in InferAxis() local503 if (axis_list.empty()) { in InferAxis()514 for (auto &dim : axis_list) { in InferAxis()
818 std::vector<int64_t> axis_list; in GetReduceAttrAxis() local820 (void)axis_list.emplace_back(GetValue<int64_t>(axis_attr)); in GetReduceAttrAxis()822 axis_list = GetValue<std::vector<int64_t>>(axis_attr); in GetReduceAttrAxis()824 for (const auto &elem : axis_list) { in GetReduceAttrAxis()
368 AbstractBasePtrList axis_list = {dyn_cast<AbstractScalar>(args_spec_list[1])}; in InferImplReduceShape() local369 axis = std::make_shared<AbstractTuple>(axis_list); in InferImplReduceShape()