Home
last modified time | relevance | path

Searched defs:scale_param (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dscale_int8.c22 … int32x4_t output_multiplier_vec, const ScaleQuantParameter *scale_param) { in ClacSumHalfWordMul2()
34 const ScaleQuantParameter *scale_param) { in ClacSumHalfWordMul3()
54 …nst int8_t *in_data, int8_t *out_data, const int8_t *scale, const ScaleQuantParameter *scale_param, in DoScaleInt8()
108 const ScaleQuantParameter *scale_param, int real_dst_count) { in DoScaleWithBiasInt8()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dscale.cc49 auto *scale_param = reinterpret_cast<const ScaleParameter *>(op_parameter_); in CheckSpecs() local
166 auto *scale_param = reinterpret_cast<const ScaleParameter *>(op_parameter_); in Prepare() local
Dfusion_eltwise.cc60 auto *scale_param = reinterpret_cast<ScaleParameter *>(op_parameter); in CheckSupport() local
97 auto *scale_param = reinterpret_cast<ScaleParameter *>(op_parameter); local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dupsample_mapper.cc80 ParameterPtr scale_param = scale_input->cast<ParameterPtr>(); in AttrAdjust() local
Drandom_normal_mapper.cc101 …auto scale_param = opt::BuildFloatValueParameterNode(func_graph, scale, cnode->fullname_with_scope… in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dscale_fp16.c68 int task_id, const ScaleStruct *scale_param) { in DoScaleFp16()
140 int task_id, const ScaleStruct *scale_param) { in Fp16DoScaleRelu()
214 int task_id, const ScaleStruct *scale_param) { in DoScaleRelu6Fp16()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dscale_fp32.c88 const ScaleStruct *scale_param) { in DoScale()
188 const ScaleStruct *scale_param) { in DoScaleRelu()
292 const ScaleStruct *scale_param) { in DoScaleRelu6()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_megatron_op_adjust.cc116 …auto scale_param = opt::BuildFloatValueParameterNode(func_graph, scale, node_name + "MaskedSoftmax… in AdjustMegatronScaledMaskedSoftmax() local
Donnx_inputs_adjust.cc406 auto scale_param = scale_node->cast<ParameterPtr>(); in AdjustResize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/
Dopencl_fusion.cc434 auto *scale_param = in TryFusionConvScaleWeight() local
506 auto *scale_param = in TryMergeDeconvScale() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dgroupnormsilu_fusion.cc61 auto scale_param = instance_norm_input2->cast<ParameterPtr>(); in GetInstanceNormGroups() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc1167 auto scale_param = node->cast<CNodePtr>()->input(kIndexTwo); in GetScaleZpFromAntiQuantModeNodes() local
/third_party/mindspore/mindspore-src/source/third_party/proto/caffe/
Dcaffe.proto403 optional ScaleParameter scale_param = 142; field