| /third_party/mindspore/mindspore/lite/mindir/include/ |
| D | mindir.h | 41 PrimitivePtr MindIR_Activation_CreatePrimitive(ActivationType activation_type, float alpha, float m…
43 ActivationType MindIR_Activation_GetActivationType(ConstPrimitivePtr primitive);
44 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type);
55 PrimitivePtr MindIR_AddFusion_CreatePrimitive(ActivationType activation_type);
56 ActivationType MindIR_AddFusion_GetActivationType(ConstPrimitivePtr primitive);
57 void MindIR_AddFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type);
74 ActivationType activation_type);
89 ActivationType MindIR_AvgPoolFusion_GetActivationType(ConstPrimitivePtr primitive);
90 void MindIR_AvgPoolFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type…
117 ActivationType activation_type);
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_activation_parser.cc | 33 prim->set_activation_type(mindspore::ActivationType::RELU); in Parse()
35 prim->set_activation_type(mindspore::ActivationType::RELU6); in Parse()
37 prim->set_activation_type(mindspore::ActivationType::SIGMOID); in Parse()
39 prim->set_activation_type(mindspore::ActivationType::TANH); in Parse()
41 prim->set_activation_type(mindspore::ActivationType::SELU); in Parse()
43 prim->set_activation_type(mindspore::ActivationType::SOFTPLUS); in Parse()
45 prim->set_activation_type(mindspore::ActivationType::LEAKY_RELU); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_activation_parser.cc | 27 prim->set_activation_type(mindspore::ActivationType::RELU); in Parse()
32 prim->set_activation_type(mindspore::ActivationType::LEAKY_RELU); in Parse()
43 prim->set_activation_type(mindspore::ActivationType::RELU6); in Parse()
51 prim->set_activation_type(mindspore::ActivationType::SIGMOID); in Parse()
59 prim->set_activation_type(mindspore::ActivationType::TANH); in Parse()
67 prim->set_activation_type(mindspore::ActivationType::ELU); in Parse()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | mul_fusion.cc | 26 void MulFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
30 ActivationType MulFusion::get_activation_type() const { in get_activation_type()
33 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
35 void MulFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
|
| D | sub_fusion.cc | 23 void SubFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
25 void SubFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
30 ActivationType SubFusion::get_activation_type() const { in get_activation_type()
33 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | div_fusion.cc | 23 void DivFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
25 void DivFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
30 ActivationType DivFusion::get_activation_type() const { in get_activation_type()
33 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | scale_fusion.cc | 24 void ScaleFusion::Init(const int64_t axis, const ActivationType &activation_type) { in Init()
29 void ScaleFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
34 ActivationType ScaleFusion::get_activation_type() const { in get_activation_type()
37 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | l2_normalize_fusion.cc | 24 const ActivationType &activation_type) { in Init()
30 void L2NormalizeFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
35 ActivationType L2NormalizeFusion::get_activation_type() const { in get_activation_type()
38 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | adder_fusion.cc | 25 const Format &format, const ActivationType activation_type) { in Init()
38 void AdderFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
43 ActivationType AdderFusion::get_activation_type() const { in get_activation_type()
46 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | conv2d_backprop_filter_fusion.cc | 27 … const int64_t group, const Format &format, const ActivationType activation_type) { in Init()
44 void Conv2DBackpropFilterFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
53 ActivationType Conv2DBackpropFilterFusion::get_activation_type() const { in get_activation_type()
56 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | activation.cc | 32 void Activation::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
52 ActivationType Activation::get_activation_type() const { in get_activation_type()
54 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
65 const ActivationType &activation_type, bool approximate) { in Init()
|
| D | add_fusion.cc | 28 void AddFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
32 ActivationType AddFusion::get_activation_type() const { in get_activation_type()
34 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
36 void AddFusion::Init(const ActivationType activation_type) { this->set_activation_type(activation_t… in Init()
|
| D | conv2d_backprop_input_fusion.cc | 29 const ActivationType &activation_type) { in Init()
48 void Conv2DBackpropInputFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
58 ActivationType Conv2DBackpropInputFusion::get_activation_type() const { in get_activation_type()
61 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | mul_fusion.h | 40 void Init(const ActivationType &activation_type);
45 void set_activation_type(const ActivationType &activation_type);
50 ActivationType get_activation_type() const;
|
| D | div_fusion.h | 40 void Init(const ActivationType &activation_type = NO_ACTIVATION);
45 void set_activation_type(const ActivationType &activation_type);
50 ActivationType get_activation_type() const;
|
| D | sub_fusion.h | 40 void Init(const ActivationType &activation_type = NO_ACTIVATION);
45 void set_activation_type(const ActivationType &activation_type);
50 ActivationType get_activation_type() const;
|
| D | scale_fusion.h | 41 void Init(const int64_t axis = -1, const ActivationType &activation_type = NO_ACTIVATION);
46 void set_activation_type(const ActivationType &activation_type);
51 ActivationType get_activation_type() const;
|
| D | l2_normalize_fusion.h | 45 const ActivationType &activation_type = NO_ACTIVATION);
50 void set_activation_type(const ActivationType &activation_type);
55 ActivationType get_activation_type() const;
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | activation_grad.cc | 29 void ActivationGrad::Init(const ActivationType &type, const float alpha) { in Init()
34 void ActivationGrad::set_activation_type(const ActivationType &type) { in set_activation_type()
39 ActivationType ActivationGrad::get_activation_type() const { in get_activation_type()
42 return ActivationType(GetValue<int64_t>(value_ptr)); in get_activation_type()
|
| D | activation_grad.h | 35 void Init(const ActivationType &type = NO_ACTIVATION, const float alpha = 0.2);
36 void set_activation_type(const ActivationType &type);
39 ActivationType get_activation_type() const;
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 25 PrimitivePtr MindIR_Activation_CreatePrimitive(ActivationType activation_type, float alpha, float m… in MindIR_Activation_CreatePrimitive()
28 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(activation_typ… in MindIR_Activation_CreatePrimitive()
37 ActivationType MindIR_Activation_GetActivationType(ConstPrimitivePtr primitive) { in MindIR_Activation_GetActivationType()
42 return static_cast<ActivationType>(value->activation_type()); in MindIR_Activation_GetActivationType()
44 ActivationType en = static_cast<ActivationType>(0); in MindIR_Activation_GetActivationType()
48 ActivationType en = static_cast<ActivationType>(0); in MindIR_Activation_GetActivationType()
53 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_Activation_SetActivationType()
60 …schema::CreateActivation(fbb, static_cast<schema::ActivationType>(activation_type), value->alpha(), in MindIR_Activation_SetActivationType()
91 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetAlpha()
122 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetMinVal()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_utils.cc | 146 ActivationParams ConvertActivationType(schema::ActivationType activation_type) { in ConvertActivationType()
147 std::map<schema::ActivationType, ActivationParams> action_map = { in ConvertActivationType()
148 …{schema::ActivationType_RELU, ActivationParams{nvinfer1::ActivationType::kRELU, false, 0, false, 0… in ConvertActivationType()
149 …{schema::ActivationType_SIGMOID, ActivationParams{nvinfer1::ActivationType::kSIGMOID, false, 0, fa… in ConvertActivationType()
150 …{schema::ActivationType_TANH, ActivationParams{nvinfer1::ActivationType::kTANH, false, 0, false, 0… in ConvertActivationType()
151 …{schema::ActivationType_LEAKY_RELU, ActivationParams{nvinfer1::ActivationType::kLEAKY_RELU, true, … in ConvertActivationType()
152 … {schema::ActivationType_ELU, ActivationParams{nvinfer1::ActivationType::kELU, true, 0, false, 0}}, in ConvertActivationType()
153 …{schema::ActivationType_SELU, ActivationParams{nvinfer1::ActivationType::kSELU, true, 0, true, 0}}, in ConvertActivationType()
154 …{schema::ActivationType_SOFTSIGN, ActivationParams{nvinfer1::ActivationType::kSOFTSIGN, false, 0, … in ConvertActivationType()
155 …{schema::ActivationType_SOFTPLUS, ActivationParams{nvinfer1::ActivationType::kSOFTPLUS, true, 0, t… in ConvertActivationType()
[all …]
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops.fbs | 227 activation_type: ActivationType = 0;
235 activation_type: ActivationType;
245 activation_type: ActivationType = 0;
258 activation_type: ActivationType = 0;
315 activation_type: ActivationType = 0;
393 activation_type: ActivationType = 0;
408 activation_type: ActivationType = 0;
422 activation_type: ActivationType = 0;
437 activation_type: ActivationType = 0;
475 activation_type: ActivationType;
[all …]
|
| D | ops_v0.fbs | 63 enum ActivationType : byte {
161 type: ActivationType = 0;
167 type: ActivationType = 0;
189 activationType: ActivationType = 0;
209 activationType: ActivationType = 0;
230 activationType: ActivationType = 0;
251 activationType: ActivationType = 0;
272 activationType: ActivationType = 0;
362 activationType: ActivationType = 0;
382 activationType: ActivationType = 0;
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_util.cc | 29 std::map<tflite::ActivationFunctionType, mindspore::ActivationType> tfMsActivationFunctionMap{
30 {tflite::ActivationFunctionType_NONE, mindspore::ActivationType::NO_ACTIVATION},
31 {tflite::ActivationFunctionType_RELU, mindspore::ActivationType::RELU},
32 {tflite::ActivationFunctionType_RELU6, mindspore::ActivationType::RELU6},
33 {tflite::ActivationFunctionType_TANH, mindspore::ActivationType::TANH},
44 mindspore::ActivationType GetActivationFunctionType(tflite::ActivationFunctionType tfliteAFType) { in GetActivationFunctionType()
|