Searched defs:input_gamma (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | layer_norm_grad_grad.cc | 74 … const size_t &mean_num, T *input_x, T *input_dy, T *input_mean, T *input_gamma, in SwitchParallelCompute0()
100 … const size_t &mean_num, T *input_x, T *input_dy, T *input_mean, T *input_gamma, in SwitchParallelCompute1()
126 … const size_t &mean_num, T *input_gamma, T *input_d_dx, T *input_d_dg, T *input_d_db, in SwitchParallelCompute2()
172 auto input_gamma = reinterpret_cast<T *>(ctx.Input(4)->GetData()); in LayerNormGradGradCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | layer_norm_grad_grad_cpu_kernel.cc | 76 … DATA_T *sum4, DATA_T *inv_std, DATA_T *input_d_dx, DATA_T *input_dy, DATA_T *input_gamma, in shard_inner_mean()
122 … DATA_T *input_gamma, DATA_T *input_d_dg, DATA_T *input_d_db, DATA_T *x_hat, DATA_T *output_sopd_x, in shard_input_prop()
186 auto input_gamma = static_cast<DATA_T *>(inputs[4]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/ |
| D | layernorm.cc | 37 const auto &input_gamma = inputs[1]; in Expand() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | layernorm_grad.cc | 82 auto input_gamma = in_tensors_.at(4); in DoExecute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/ |
| D | layernorm_fp16_grad.cc | 91 auto input_gamma = in_tensors_.at(kNumInputDim_4); in DoExecute() local
|
| /third_party/skia/third_party/externals/libjpeg-turbo/ |
| D | jpeglib.h | 319 double input_gamma; /* image gamma of input image */ member
|