| /third_party/mindspore/mindspore/core/ops/ |
| D | scale.cc | 23 void Scale::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Scale
|
| D | glu.cc | 24 void GLU::set_axis(int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::GLU
|
| D | lp_normalization.cc | 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 | 32 void Split::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Split
|
| D | unpack.cc | 22 void Unpack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Unpack
|
| D | crop.cc | 30 void Crop::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Crop
|
| D | unstack.cc | 22 void Unstack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Unstack
|
| D | l2_normalize.cc | 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 | arg_min.cc | 27 void ArgMin::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::ArgMin
|
| D | reverse_v2.cc | 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 | unsqueeze.cc | 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 | stack.cc | 63 void Stack::set_axis(const int64_t axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Stack
|
| 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
|
| 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
|
| D | concat.cc | 30 void Concat::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::Concat
|
| D | arg_max.cc | 54 void ArgMax::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::ArgMax
|
| D | squeeze.cc | 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
|
| D | one_hot.cc | 26 void OneHot::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::OneHot
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | topk_fusion.cc | 29 void TopKFusion::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::TopKFusion
|
| D | full_connection.cc | 31 void FullConnection::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } in set_axis() function in mindspore::ops::FullConnection
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.h | 1084 void set_axis(int64_t axis) { axis_ = axis; } in set_axis() function
|