Home
last modified time | relevance | path

Searched defs:activation_type (Results 1 – 25 of 39) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Ddiv_fusion.cc30 void DivFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
32 void DivFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dmul_fusion.cc30 void MulFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
41 void MulFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
Dsub_fusion.cc30 void SubFusion::Init(const ActivationType &activation_type) { this->set_activation_type(activation_… in Init()
32 void SubFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dadd_fusion.cc28 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()
Dscale_fusion.cc30 void ScaleFusion::Init(const int64_t axis, const ActivationType &activation_type) { in Init()
35 void ScaleFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dl2_normalize_fusion.cc31 const ActivationType &activation_type) { in Init()
37 void L2NormalizeFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dadder_fusion.cc33 const Format &format, const ActivationType activation_type) { in Init()
46 void AdderFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
Dmat_mul_fusion.cc29 void MatMulFusion::Init(bool transpose_a, bool transpose_b, const ActivationType &activation_type) { in Init()
35 void MatMulFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
Dmax_pool_fusion.cc29 … const RoundMode &round_mode, const bool global, const ActivationType activation_type) { in Init()
42 void MaxPoolFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
Dconv2d_backprop_filter_fusion.cc32 … const int64_t group, const Format &format, const ActivationType activation_type) { in Init()
45 void Conv2DBackpropFilterFusion::set_activation_type(const ActivationType activation_type) { in set_activation_type()
Dfull_connection.cc50 void FullConnection::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
60 const ActivationType &activation_type) { in Init()
Dconv2d_backprop_input_fusion.cc35 const ActivationType &activation_type) { in Init()
54 void Conv2DBackpropInputFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dconv2d_fusion.cc33 … const std::vector<int64_t> &pad_list, const ActivationType &activation_type) { in Init()
53 void Conv2DFusion::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dconv2d_transpose_fusion.cc36 … const std::vector<int64_t> &output_paddings, ActivationType activation_type) { in Init()
79 void Conv2dTransposeFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
Dactivation.cc41 void Activation::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
74 const ActivationType &activation_type, bool approximate) { in Init()
Davg_pool_fusion.cc29 … const RoundMode &round_mode, const bool global, const ActivationType activation_type) { in Init()
70 void AvgPoolFusion::set_activation_type(ActivationType activation_type) { in set_activation_type()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/
Dactivation.cc27 …vationExpander(const inner::GraphBuilder &gb, const NodePtrList &inputs, int64_t activation_type) { in GetActivationExpander()
55 auto activation_type = GetValue<int64_t>(attrs_["activation_type"]); in Expand() local
Dactivation.h35 auto activation_type = GetValue<int64_t>(iter->second); in Check() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dactivation_tensorrt.cc53 ActivationType activation_type = ActivationType::NO_ACTIVATION; in IsSupport() local
84 ActivationType activation_type = ActivationType::NO_ACTIVATION; in AddInnerOp() local
204 …yer *ActivationTensorRT::AddActivation(TensorRTContext *ctx, ActivationType activation_type, float… in AddActivation()
Dactivation_opt_plugin.h30 ActivationOptPlugin(const std::string name, ActivationType activation_type, uint32_t device_id) in ActivationOptPlugin()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dactivation_checker.cc45 auto activation_type = static_cast<mindspore::ActivationType>(api::GetValue<int64_t>(value_ptr)); in Check() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dactivation_nnapi.cc25 std::unordered_map<schema::ActivationType, OperationCode> activation_type = { variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/operator_populate/
Dactivation_populate.cc51 auto activation_type = ActivationType(GetValue<int64_t>(attr)); in PopulateActivationOpParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dde_conv2d_grad_filter.cc34 const ActivationType &activation_type, const bool has_bias) { in Init()
138 void DeConv2DGradFilter::set_activation_type(const ActivationType &activation_type) { in set_activation_type()
Dgroup_conv2d_grad_input.cc33 const ActivationType &activation_type, const bool has_bias) { in Init()
148 void GroupConv2DGradInput::set_activation_type(const ActivationType &activation_type) { in set_activation_type()

12