Searched defs:num_step (Results 1 – 5 of 5) sorted by relevance
25 int num_step; member
26 int num_step = gru_param->num_step; in CustomGru() local
25 int num_step = gru_param->num_step; in CustomGruFp16() local
71 int64_t num_step = x_shape[0]; in DynamicGRUV2GradCheckShapeValue() local153 int64_t num_step = -1; in DynamicGRUV2GradInferShape() local
146 int64_t num_step = x_shape[kDynRnnIdx0]; in DynamicRNNInferShape() local