Searched refs:hidden_new (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | rnn_declare.cc | 120 … {3, OUTPUT_DESC(reset)}, {4, OUTPUT_DESC(new)}, {5, OUTPUT_DESC(hidden_new)}};
128 {10, INPUT_DESC(reset)}, {11, INPUT_DESC(new)}, {12, INPUT_DESC(hidden_new)},
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 389 float cell_old, cell_new, hidden_old, hidden_new; in CheckBodyGraph() local
399 if (GetFloatScalarFromTensorInfo(hidden_zoneout_new_node, &hidden_new) != RET_OK) { in CheckBodyGraph()
406 if (hidden_old < 0.0f || hidden_old > 1.0f || hidden_new < 0.0f || hidden_new > 1.0f) { in CheckBodyGraph()
410 …if (std::abs(cell_old + cell_new - 1.0f) > EPSILON || std::abs(hidden_old + hidden_new - 1.0f) > E… in CheckBodyGraph()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_nn_ops.py | 1002 y, out_h, update, reset, new, hidden_new = out
1007 … reset, new, hidden_new, None, None)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 456 …alse, "param_type": "optional", "shape": "all"}, {"index": 5, "name": "hidden_new", "need_compile"…
457 …alse, "param_type": "optional", "shape": "all"}, {"index": 8, "name": "hidden_new", "need_compile"…
|