| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_crop_parser.cc | 29 prim->set_axis(2); in Parse()
38 prim->set_axis(cropParam.axis()); in Parse()
40 prim->set_axis(2); in Parse()
|
| D | caffe_softmax_parser.cc | 32 prim->set_axis({proto.softmax_param().axis()}); in Parse()
34 prim->set_axis({1}); in Parse()
|
| D | caffe_slice_parser.cc | 45 prim->set_axis(slice_param.axis()); in Parse()
47 prim->set_axis(slice_param.slice_dim()); in Parse()
|
| D | caffe_scale_parser.cc | 51 prim->set_axis(1); in Parse()
58 prim->set_axis(axis_index); in Parse()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | glu.cc | 22 void GLU::Init(int64_t axis) { set_axis(axis); } in Init()
24 void GLU::set_axis(int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::GLU
|
| D | scale.cc | 21 void Scale::Init(const int64_t axis) { set_axis(axis); } in Init()
23 void Scale::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Scale
|
| D | lp_normalization.cc | 24 this->set_axis(axis); in Init()
28 void LpNormalization::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::LpNormalization
|
| D | split.cc | 25 this->set_axis(axis); in Init()
32 void Split::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Split
|
| D | crop.cc | 26 this->set_axis(axis); in Init()
30 void Crop::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Crop
|
| D | l2_normalize.cc | 24 this->set_axis(axis); in Init()
28 void L2Normalize::set_axis(const std::vector<int64_t> &axis) { (void)AddAttr(kAxis, MakeValue(axis)… in set_axis() function in mindspore::ops::L2Normalize
|
| D | reverse_v2.cc | 42 void ReverseV2::Init(const std::vector<int64_t> &axis) { this->set_axis(axis); } in Init()
43 void ReverseV2::set_axis(const std::vector<int64_t> &axis) { (void)this->AddAttr(kAxis, MakeValue(a… in set_axis() function in mindspore::ops::ReverseV2
|
| D | arg_min.cc | 23 set_axis(axis); in Init()
27 void ArgMin::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::ArgMin
|
| D | unstack.cc | 21 void Unstack::Init(const int64_t axis) { this->set_axis(axis); } in Init()
22 void Unstack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Unstack
|
| D | unpack.cc | 21 void Unpack::Init(const int64_t axis) { this->set_axis(axis); } in Init()
22 void Unpack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Unpack
|
| D | unsqueeze.cc | 24 void Unsqueeze::Init(const std::vector<int64_t> axis) { this->set_axis(axis); } in Init()
26 void Unsqueeze::set_axis(const std::vector<int64_t> axis) { (void)this->AddAttr(kAxis, MakeValue(ax… in set_axis() function in mindspore::ops::Unsqueeze
|
| D | log_softmax.cc | 29 void LogSoftmax::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::LogSoftmax
33 void LogSoftmax::Init(const int64_t axis) { this->set_axis(axis); } in Init()
|
| D | arg_max.cc | 50 set_axis(axis); in Init()
54 void ArgMax::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::ArgMax
|
| D | stack.cc | 63 void Stack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Stack
67 void Stack::Init(const int64_t axis) { this->set_axis(axis); } in Init()
|
| D | pack.cc | 50 void Pack::set_axis(const int64_t &axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Pack
54 void Pack::Init(const int64_t &axis) { this->set_axis(axis); } in Init()
|
| D | squeeze.cc | 21 void Squeeze::Init(const std::vector<int64_t> &axis) { set_axis(axis); } in Init()
22 void Squeeze::set_axis(const std::vector<int64_t> &axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Squeeze
|
| D | softmax.cc | 29 void Softmax::set_axis(const std::vector<int64_t> &axis) { (void)this->AddAttr(kAxis, MakeValue(axi… in set_axis() function in mindspore::ops::Softmax
44 this->set_axis(axis_vec); in Init()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | topk_fusion.cc | 24 this->set_axis(axis); in Init()
29 void TopKFusion::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::TopKFusion
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_split_parser.cc | 29 prim->set_axis(0); in Parse()
35 prim->set_axis(onnx_node_attr.i()); in Parse()
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_softmax.cc | 39 softmax->set_axis(init_data); in TEST_F()
69 softmax->set_axis(init_data); in TEST_F()
99 softmax->set_axis(init_data); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/parser/ |
| D | node_parser_test.cc | 43 primc->set_axis(0); in Parse()
55 primc->set_axis(0); in Parse()
|