| /third_party/mindspore/mindspore/core/ops/fusion/ |
| 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()
|
| D | mul_fusion.cc | 26 void MulFusion::set_activation_type(const ActivationType &activation_type) { in set_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()
|
| 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()
|
| 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()
|
| 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()
|
| 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()
|
| D | add_fusion.cc | 28 void AddFusion::set_activation_type(const ActivationType activation_type) { in set_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()
|
| D | conv2d_fusion.cc | 27 … const std::vector<int64_t> &pad_list, const ActivationType &activation_type) { in Init()
45 void Conv2DFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
|
| D | activation.cc | 32 void Activation::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
65 const ActivationType &activation_type, bool approximate) { in Init()
|
| D | conv2d_transpose_fusion.cc | 26 … const std::vector<int64_t> &output_paddings, ActivationType activation_type) { in Init()
69 void Conv2dTransposeFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
|
| D | full_connection.cc | 45 void FullConnection::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
55 const ActivationType &activation_type) { in Init()
|
| D | max_pool_fusion.cc | 23 … const RoundMode &round_mode, const bool global, const ActivationType activation_type) { in Init()
36 void MaxPoolFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
|
| D | avg_pool_fusion.cc | 23 … const RoundMode &round_mode, const bool global, const ActivationType activation_type) { in Init()
36 void AvgPoolFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | arithmetic_compare_fp16.cc | 46 ArithmeticCompareFuncFp16 GetArithmeticCompareFun(int primitive_type, int activation_type) { in GetArithmeticCompareFun()
57 …meticCompareOptFuncFp16 GetOptimizedArithmeticCompareFun(int primitive_type, int activation_type) { in GetOptimizedArithmeticCompareFun()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | de_conv2d_grad_filter.cc | 28 const ActivationType &activation_type, const bool has_bias) { in Init()
132 void DeConv2DGradFilter::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
|
| D | group_conv2d_grad_input.cc | 28 const ActivationType &activation_type, const bool has_bias) { in Init()
143 void GroupConv2DGradInput::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | activation_tensorrt.cc | 74 … schema::ActivationType activation_type, float alpha, in AddActivation()
|
| D | scale_tensorrt.cc | 56 schema::ActivationType activation_type = scale_op->activation_type(); in AddInnerOp() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_utils.h | 33 nvinfer1::ActivationType activation_type; member
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | affine.cc | 39 void Affine::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | scale_tests.cc | 25 OpParameter *CreateParameter(int axis, int activation_type = schema::ActivationType_NO_ACTIVATION) { in CreateParameter()
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/optimizer/fusion/ |
| D | conv_activation_fusion_test.cc | 73 MetaGraphTptr BuildGraph(schema::PrimitiveType conv_type, schema::ActivationType activation_type) { in BuildGraph()
|
| /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()
53 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_Activation_SetActivationType()
197 PrimitivePtr MindIR_AddFusion_CreatePrimitive(ActivationType activation_type) { in MindIR_AddFusion_CreatePrimitive()
223 void MindIR_AddFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_AddFusion_SetActivationType()
380 ActivationType activation_type) { in MindIR_AvgPoolFusion_CreatePrimitive()
673 void MindIR_AvgPoolFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type… in MindIR_AvgPoolFusion_SetActivationType()
853 ActivationType activation_type) { in MindIR_Conv2DFusion_CreatePrimitive()
1194 void MindIR_Conv2DFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type)… in MindIR_Conv2DFusion_SetActivationType()
1222 ActivationType activation_type, const std::vector<int64_t> &output_paddings) { in MindIR_Conv2dTransposeFusion_CreatePrimitive()
1572 …_Conv2dTransposeFusion_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_Conv2dTransposeFusion_SetActivationType()
[all …]
|