Searched refs:axis_vec (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | l2_norm_populate.cc | 39 auto axis_vec = value->axis(); in PopulateL2NormParameter() local
40 if (axis_vec == nullptr) { in PopulateL2NormParameter()
45 param->axis_num_ = axis_vec->size(); in PopulateL2NormParameter()
47 if (axis_vec->size() > MAX_SHAPE_SIZE) { in PopulateL2NormParameter()
48 MS_LOG(ERROR) << "axis_vec size " << axis_vec->size() << " is invalid!"; in PopulateL2NormParameter()
52 for (size_t i = 0; i < axis_vec->size(); i++) { in PopulateL2NormParameter()
53 param->axis_[i] = static_cast<int>(axis_vec->Get(i)); in PopulateL2NormParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | l2_norm_populate_v0.cc | 40 auto axis_vec = l2_norm_prim->axis(); in PopulateL2NormParameter() local
41 if (axis_vec == nullptr) { in PopulateL2NormParameter()
46 l2_norm_parameter->axis_num_ = axis_vec->size(); in PopulateL2NormParameter()
47 if (((size_t)axis_vec->size()) > MAX_SHAPE_SIZE) { in PopulateL2NormParameter()
53 for (size_t i = 0; i < axis_vec->size(); i++) { in PopulateL2NormParameter()
54 l2_norm_parameter->axis_[i] = *(axis_vec->begin() + i); in PopulateL2NormParameter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | axis_normalizer.cc | 48 ShapeVector axis_vec; in Process() local
52 axis_vec.push_back(v2); in Process()
59 axis_vec.push_back(i); in Process()
65 axis_vec.push_back(v2); in Process()
72 std::sort(axis_vec.begin(), axis_vec.end()); in Process()
73 SetNodeAttrSafely(kAttrAxis, MakeValue(axis_vec), node); in Process()
|
| D | add_atomic_clean.cc | 52 auto axis_vec = GetReduceAxis(node); in GetUniqReduceAxes() local
53 if (axis_vec.empty()) { in GetUniqReduceAxes()
55 (void)axis_vec.emplace_back(i); in GetUniqReduceAxes()
58 …(void)std::transform(axis_vec.begin(), axis_vec.end(), axis_vec.begin(), [&src_shape_vec](int64_t … in GetUniqReduceAxes()
63 std::set<int64_t> axis_set(axis_vec.begin(), axis_vec.end()); in GetUniqReduceAxes()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | softmax.cc | 38 std::vector<int64_t> axis_vec = {axis}; in Init() local
39 …(void)CheckAndConvertUtils::CheckInteger("axis_len", SizeToLong(axis_vec.size()), kEqual, 1, op_na… in Init()
40 auto rank = SizeToLong(axis_vec.size()); in Init()
41 for (auto &item : axis_vec) { in Init()
44 this->set_axis(axis_vec); in Init()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 717 std::vector<int64_t> axis_vec; in CreateSqueezeNode() local
718 std::transform(axis.begin(), axis.end(), std::back_inserter(axis_vec), in CreateSqueezeNode()
720 squeeze_prim->set_axis(axis_vec); in CreateSqueezeNode()
|