Searched defs:input_bias (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | gru_fp16.c | 87 … const float16_t *weight_r, const float16_t *input_bias, const float16_t *state_bias, in GruUnidirectionalFp16()
114 …const float16_t *input_bias, const float16_t *state_bias, float16_t *hidden_state, float16_t *buff… in GruFp16()
|
| D | lstm_fp16.c | 289 …te(float16_t *gate, const float16_t *input, const float16_t *weight_i, const float16_t *input_bias, in LstmGateCompute()
307 … const float16_t *weight_h, const float16_t *input_bias, const float16_t *state_bias, in LstmUnidirectionalFp16()
331 … const float16_t *input_bias, const float16_t *state_bias, const float16_t *weight_project, in LstmFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | inference_matmul_split_fusion.cc | 256 auto input_bias = biasAdd_cnode->input(kIndex2); in CreateMatmulBiasAddSplitNode() local
298 auto input_bias = qbmm_cnode->input(kIndex5); in CreateQuantbatchmatmulSplitNode() local
438 auto input_bias = biasAdd_cnode->input(kIndex2); in CreateMatmulBiasAddSplitSiluNode() local
492 auto input_bias = qbmm_cnode->input(kIndex5); in CreateQuantbatchmatmulSplitSiluNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | gru_fp32.c | 106 … const float *input_bias, const float *state_bias, float *hidden_state, float *buffer[4], in GruUnidirectional()
131 …*output, const float *input, const float *weight_g, const float *weight_r, const float *input_bias, in Gru()
|
| D | lstm_fp32.c | 277 … const float *input_bias, const float *state_bias, float *hidden_state, float *cell_state, in LstmUnidirectional()
305 …*output, const float *input, const float *weight_i, const float *weight_h, const float *input_bias, in Lstm()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/ |
| D | test_batchnorm.py | 25 def __init__(self, input_scale, input_bias, input_mean, input_variance, is_training): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | bias_add.cc | 120 auto input_bias = reinterpret_cast<T *>(calc_info.input_1->GetData()); in BiasAddCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | lstm_fp32_base.cc | 322 int LstmFp32BaseCPUKernel::LstmPreProcessWithInput(const float *weight_i, const float *input_bias, … in LstmPreProcessWithInput()
|