| /third_party/mindspore/mindspore/core/ops/ |
| D | scale.h | 38 explicit Scale(const std::string k_name) : PrimitiveC(k_name) {} in Scale() argument
|
| D | pow.h | 36 …explicit Pow(const std::string &k_name = kNamePow) : PrimitiveC(k_name) { InitIOName({"x", "y"}, {… in PrimitiveC() argument
|
| D | div.h | 35 …explicit Div(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "y"}, {"output"}); } in Div() argument
|
| D | topk.h | 34 explicit TopK(const std::string &k_name = kNameTopK) : PrimitiveC(k_name) { in PrimitiveC() argument
|
| D | prelu.h | 35 explicit PReLU(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"y"}); } in PReLU() argument
|
| D | sub.h | 36 …explicit Sub(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "y"}, {"output"}); } in Sub() argument
|
| D | tile.h | 36 …explicit Tile(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "multiples"}, {"ou… in Tile() argument
|
| D | add.h | 35 …explicit Add(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "y"}, {"output"}); } in Add() argument
|
| D | mul.h | 35 …explicit Mul(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "y"}, {"output"}); } in Mul() argument
|
| D | exp.h | 35 explicit Exp(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"y"}); } in Exp() argument
|
| D | reduce.h | 37 …explicit Reduce(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"input_x", "axis"}, {… in Reduce() argument
|
| D | reluv2.h | 37 …explicit ReLUV2(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output", "mas… in ReLUV2() argument
|
| D | arg_min.h | 37 explicit ArgMin(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in ArgMin() argument
|
| D | arg_max.h | 37 explicit ArgMax(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in ArgMax() argument
|
| D | layer_norm.h | 36 explicit LayerNorm(const std::string k_name) : PrimitiveC(k_name) {} in LayerNorm() argument
|
| D | pad.h | 35 explicit Pad(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"y"}); } in Pad() argument
|
| D | max_pool.h | 36 explicit MaxPool(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in MaxPool() argument
|
| D | avg_pool.h | 36 explicit AvgPool(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in AvgPool() argument
|
| D | adder.h | 35 explicit Adder(const std::string &k_name = kNameAdder) : PrimitiveC(k_name) {} in PrimitiveC() argument
|
| D | conv2d.h | 36 …explicit Conv2D(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "w"}, {"output"}… in Conv2D() argument
|
| D | conv2d_transpose.h | 38 explicit Conv2DTranspose(const std::string k_name) : PrimitiveC(k_name) { in Conv2DTranspose() argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | layer_norm_grad.h | 33 explicit LayerNormGrad(const std::string k_name) : PrimitiveC(k_name) {} in LayerNormGrad() argument
|
| D | pool_grad.h | 33 explicit PoolGrad(const std::string k_name) : PrimitiveC(k_name) { in PoolGrad() argument
|
| D | conv2d_backprop_filter.h | 35 explicit Conv2DBackpropFilter(const std::string k_name) : PrimitiveC(k_name) { in Conv2DBackpropFilter() argument
|
| D | conv2d_backprop_input.h | 33 …explicit Conv2DBackpropInput(const std::string &k_name = kNameConv2DBackpropInput) : PrimitiveC(k_… in PrimitiveC() argument
|