Home
last modified time | relevance | path

Searched defs:num_step (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dcustom_gru_parameter.h25 int num_step; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dcustom_gru_fp32.c26 int num_step = gru_param->num_step; in CustomGru() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dcustom_gru_fp16.c25 int num_step = gru_param->num_step; in CustomGruFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Ddynamic_gru_v2_grad.cc71 int64_t num_step = x_shape[0]; in DynamicGRUV2GradCheckShapeValue() local
153 int64_t num_step = -1; in DynamicGRUV2GradInferShape() local
Ddynamic_rnn.cc146 int64_t num_step = x_shape[kDynRnnIdx0]; in DynamicRNNInferShape() local