/frameworks/ml/nn/runtime/test/specs/V1_3/ |
D | unidirectional_sequence_rnn.mod.py | 20 activation, time_major, output, output_state, input_data, weights_data, argument 27 time_major).To(output, output_state) 36 output_state: output_state_data, 190 output_state=Output("output_state", "TENSOR_FLOAT32", 216 output_state=Output("output_state", "TENSOR_FLOAT32",
|
/frameworks/ml/nn/runtime/test/specs/V1_2/ |
D | layer_norm_lstm.mod.py | 177 for test_input, output_state, cell_state, golden_state, golden_output in tests: 180 cur_input[output_state_in] = output_state 346 for test_input, output_state, cell_state, golden_state, golden_output in tests: 349 cur_input[output_state_in] = output_state
|
D | lstm_state2_float16.mod.py | 136 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable 146 input0[output_state_in] = output_state
|
D | lstm_state_float16.mod.py | 136 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable 146 input0[output_state_in] = output_state
|
D | lstm_float16.mod.py | 136 output_state = [0, 0, 0, 0] variable 146 input0[output_state_in] = output_state
|
/frameworks/ml/nn/runtime/test/specs/V1_1/ |
D | lstm_state2_relaxed.mod.py | 137 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable 147 input0[output_state_in] = output_state
|
D | lstm_state_relaxed.mod.py | 137 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable 147 input0[output_state_in] = output_state
|
D | lstm_relaxed.mod.py | 137 output_state = [0, 0, 0, 0] variable 147 input0[output_state_in] = output_state
|
/frameworks/ml/nn/tools/test_generator/tests/P_vts_backward_compatibility_float/ |
D | lstm_float.mod.py | 133 output_state = [0, 0, 0, 0] variable 143 input0[output_state_in] = output_state
|
/frameworks/ml/nn/tools/test_generator/tests/P_backward_compatibility_float/ |
D | lstm_float.mod.py | 133 output_state = [0, 0, 0, 0] variable 143 input0[output_state_in] = output_state
|
/frameworks/ml/nn/runtime/test/specs/V1_0/ |
D | lstm_state.mod.py | 136 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable 146 input0[output_state_in] = output_state
|
D | lstm_state2.mod.py | 136 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable 146 input0[output_state_in] = output_state
|
D | lstm.mod.py | 136 output_state = [0, 0, 0, 0] variable 146 input0[output_state_in] = output_state
|
/frameworks/ml/nn/common/operations/ |
D | BidirectionalSequenceLSTM.cpp | 197 params_.output_state = (operation.outputs.size() == 5 || operation.outputs.size() == 6); in BidirectionalSequenceLSTM() 198 if (params_.output_state) { in BidirectionalSequenceLSTM() 419 if (params_.output_state) { in Prepare() 472 if (params_.output_state) { in Eval() 524 if (params_.output_state) { in Eval() 595 if (params_.output_state) { in Eval() 648 if (params_.output_state) { in Eval()
|
D | LSTM.h | 43 bool output_state; member
|
/frameworks/ml/nn/common/ |
D | CpuExecutor.cpp | 1056 const bool output_state = (outs.size() == 5 || outs.size() == 6); in executeOperation() local 1072 if (output_state) { in executeOperation()
|