Home
last modified time | relevance | path

Searched defs:param_ptr (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/
Delewise_binary.cc31 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
49 void SetComputeType(internal::OpParamPtr param_ptr) override {} in SetComputeType()
52 auto param_ptr = std::make_shared<internal::AddParam>(); in CreateOpParam() local
71 void SetComputeType(internal::OpParamPtr param_ptr) override {} in SetComputeType()
74 auto param_ptr = std::make_shared<internal::SubParam>(); in CreateOpParam() local
92 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
106 void SetComputeType(internal::OpParamPtr param_ptr) override {} in SetComputeType()
109 auto param_ptr = std::make_shared<internal::CompareParam>(); in CreateOpParam() local
123 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
137 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
[all …]
Dactivation.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
39 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
53 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
Delewise_unary.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
38 void SetComputeType(internal::OpParamPtr param_ptr) override { in SetComputeType()
Dgelu.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dpaged_attention_mask.cc23 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dreshape_and_cache.cc28 auto param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dpaged_attention.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Drmsnorm.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dapply_rotary_pos_emb.cc26 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dadd_rms_norm.cc24 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dcast.cc29 auto param_ptr = std::make_shared<internal::CastParam>(); in CreateOpParam() local
Dreduce_sum.cc23 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dadd_layernorm.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dtranspose.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dquant_batch_matmul.cc28 auto param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dsoftmax.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dgather.cc28 internal::OpParamPtr param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
Dmatmul.cc29 auto param_ptr = std::make_shared<internal::MatMulExtParam>(); in CreateOpParam() local
Dflash_attention_score.cc31 auto param_ptr = std::make_shared<internal::FlashAttentionScoreParam>(); in CreateOpParam() local
Dmatmul_elemwise.cc35 auto param_ptr = std::make_shared<internal::MatMulExtParam>(); in CreateOpParam() local
Dinference_multi_matmul.cc30 auto param_ptr = std::make_shared<internal::OpParam>(); in CreateOpParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/common/
Dutils.cc155 std::vector<int> GetIntParameterData(const ParameterPtr &param_ptr) { in GetIntParameterData()
186 std::vector<int64_t> GetInt64ParameterData(const ParameterPtr &param_ptr) { in GetInt64ParameterData()
217 std::vector<float> GetFloatParameterData(const ParameterPtr &param_ptr) { in GetFloatParameterData()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/silent_check/
Dsilent_check.cc27 auto param_ptr = param->cast<ParameterPtr>(); in GetLossScale() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dreshape_checker.cc82 auto param_ptr = op->input(kInputIndex2)->cast<api::ParameterPtr>(); in Check() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/parallel_optimizer/
Dopt_param_mgr.cc101 auto param_ptr = parameter->cast<ParameterPtr>(); in SplitParam() local

12