Searched refs:bias_input (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | matmul_biasadd_fusion.cc | 70 auto bias_input = utils::cast<AnfNodePtr>((*equiv)[bias_]); in Process() local
74 MS_EXCEPTION_IF_NULL(bias_input); in Process()
88 std::vector<AnfNodePtr> inputs = {prim_value, x_input, w_input, bias_input}; in Process()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | rnn_declare.cc | 107 … {4, INPUT_DESC(bias_input)}, {5, INPUT_DESC(bias_hidden)}, {6, INPUT_DESC(seq_length)},
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 84 VectorRef bias_input = VectorRef({is_var10, matmul_input, placeholders[16]}); in GenerateBodyGraphCellPattern() local
100 VectorRef input_gate = VectorRef({is_var_sigmoid1, bias_input}); in GenerateBodyGraphCellPattern()
|
| /third_party/mindspore/config/ |
| D | op_info.config | 456 …equired", "shape": "all", "reshape_type": "CN"}, {"index": 3, "name": "bias_input", "need_compile"…
|