Searched defs:param_num (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | layernorm_grad.c | 21 … const float16_t *gamma, int param_num, int param_size, int block_num, int block_size, in LayerNormFp16Grad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | layernorm_grad.c | 22 … int param_num, int param_size, int block_num, int block_size, float *dx, float *dg, float *db) { in LayerNormGrad()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | dfunctor.h | 237 auto param_num = param_size - 1; in BpropToK() local
|
| /third_party/curl/lib/ |
| D | mprintf.c | 232 int param_num = 0; in dprintf_Pass1() local
581 long param_num = 0; /* parameter counter */ in dprintf_formatf() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 234 size_t param_num = parameters_indexs.size(); in IsAtomicNode() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_lstm_cell_fusion.cc | 193 const auto param_num = shape[0] * shape[1] * shape[kInputIndexTwo]; in SetWeightAbstractAndDefault() local
|
| /third_party/iowow/src/kv/benchmark/ |
| D | bmbase.c | 25 int param_num; member
|