Home
last modified time | relevance | path

Searched defs:input_scale (Results 1 – 25 of 30) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dswish_int8.cc32 void CalculateSwishTableList(int8_t *table, const float input_scale, const int32_t input_zp, const … in CalculateSwishTableList()
60 const float input_scale = input->quant_params().front().scale; in Prepare() local
Dsigmoid_int8.cc33 void CalculateTableList(int8_t *table, const float input_scale, const int32_t input_zp, const float… in CalculateTableList()
62 const float input_scale = input->quant_params().front().scale; in Prepare() local
Dscale_int8.cc155 auto input_scale = input->quant_params().front().scale; in InitQuantArgs() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/
Dtest_batchnorm_grad.py28 …def construct(self, input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, inpu… argument
33 def get_output(input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, input_reve… argument
Dtest_batchnorm.py25 def __init__(self, input_scale, input_bias, input_mean, input_variance, is_training): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dsigmoid_int8_coder.cc28 void CalculateTableList(int8_t *table, const float input_scale, const int32_t input_zp, const float… in CalculateTableList()
47 const float input_scale = input_tensor_->quant_params().at(0).scale; in Prepare() local
Dtanh_int8_coder.cc37 const float input_scale = input_tensor_->quant_params().at(0).scale; in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dunsqueeze_int8.c26 float input_scale = para_->quant_arg.in_quant_args_.scale_; in Int8Unsqueeze() local
Dhswish_int8.h25 double input_scale; member
Dscale_int8.c23 int32x4_t input_scale = vmulq_s32(scaled_input0, scaled_input1); in ClacSumHalfWordMul2() local
39 int32x4_t input_scale = vmulq_s32(scaled_input0, scaled_input1); in ClacSumHalfWordMul3() local
Dpower_int8.c21 double input_scale = args->in_args_.scale_; in PowerInt8() local
Dslice_int8.c24 double input_scale = quant_arg->in_args_.scale_; in SliceInt8() local
Dpooling_int8.c36 float input_scale = quant_args[0][0].scale_; in AvgPoolingInt8() local
295 float input_scale = quant_args[0][0].scale_; in MaxPoolingInt8() local
341 float input_scale = quant_args[0][0].scale_; in MaxPoolingWithQuantInt8() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Delu_ext.cc45 static const ScalarPtr input_scale = std::make_shared<FP32Imm>(1.f); in EluExtAscendCustomize() local
Delu_grad_ext.cc46 static const ScalarPtr input_scale = std::make_shared<FP32Imm>(1.f); in EluGradExtAscendCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/cmsis-nn/int8/
Dconv2d_base_coder.cc30 auto input_scale = static_cast<double>(conv_quant_arg_->input_quant_args_[0].scale_); in SetQuantArgs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/
Dscale_fusion.cc46 const auto &input_scale = inputs[kScaleIdx]; in Expand() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dbatchnorm.cc58 const auto &input_scale = inputs[1]; in Train() local
146 const auto &input_scale = inputs[1]; in Expand() local
Dbatchnorm_grad.cc54 const auto &input_scale = inputs[2]; in Expand() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dbn_grad.cc63 auto *input_scale = in_tensors_.at(2); in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/
Dbn_fp16_grad.cc78 auto *input_scale = in_tensors_.at(kNumInputDim_2); in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_matmul_split_fusion.cc300 auto input_scale = qbmm_cnode->input(kIndex3); in CreateQuantbatchmatmulSplitNode() local
494 auto input_scale = qbmm_cnode->input(kIndex3); in CreateQuantbatchmatmulSplitSiluNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Df_f_n_info.cc296 size_t input_scale = GetStrategyRealIndex(kInputScale); in InferTensorMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dscale_and_translate_cpu_kernel.cc369 auto input_scale = reinterpret_cast<float *>(inputs[2]->device_ptr()); in LaunchKernel() local
430 auto input_scale = reinterpret_cast<float *>(inputs[2]->device_ptr()); in LaunchKernel() local
/third_party/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml.h90 float input_scale; member

12