Searched defs:weight_h (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_dynamic_gru_v2.py | 31 def construct(self, x, weight_i, weight_h, bias_i, bias_h, init_h): argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | lstm_fp32_tests.cc | 65 auto *weight_h = new lite::Tensor; in InitLstmForwardCreator() local
223 auto *weight_h = new lite::Tensor; in InitLstmBackwardCreator() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | lstm_fp32.cc | 117 auto weight_h = in_tensors_.at(weight_h_index); in InitStateWeightBias() local
349 int LstmCPUKernel::LstmUnidirectional(float *output, const float *weight_i, const float *weight_h, in LstmUnidirectional()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | lstm_fp16.cc | 144 auto weight_h = in_tensors_.at(2); in InitStateWeightBias() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | lstm_fp32_coder.cc | 58 Tensor *weight_h = input_tensors().at(kInputSize1); in InitStateWeightBias() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | lstm_fp32.c | 232 …directional(float *output, const float *packed_input, const float *weight_i, const float *weight_h, in LstmUnidirectional()
261 void Lstm(float *output, const float *input, const float *weight_i, const float *weight_h, const fl… in Lstm()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | lstm_fp16.c | 246 … const float16_t *weight_h, const float16_t *input_bias, const float16_t *state_bias, in LstmUnidirectionalFp16()
275 …16(float16_t *output, const float16_t *input, const float16_t *weight_i, const float16_t *weight_h, in LstmFp16()
|