Searched defs:is_backward (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | gru_fp16.c | 92 bool is_backward) { in GruUnidirectionalFp16()
|
| D | lstm_fp16.c | 248 const LstmParameter *lstm_param, bool is_backward) { in LstmUnidirectionalFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_fusion.cc | 269 … const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, bool is_backward, in SearchFromNodes()
310 … const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, bool is_backward, in SearchStreamFromMultiRelationNode()
333 … const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, bool is_backward, in SearchStreamFromUnidirectionalNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | gru_fp32.c | 96 const GruParameter *gru_param, bool is_backward) { in GruUnidirectional()
|
| D | lstm_fp32.c | 234 float *buffer[7], const LstmParameter *lstm_param, bool is_backward) { in LstmUnidirectional()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | lstm_fp32.cc | 351 float *cell_state, bool is_backward) { in LstmUnidirectional()
|