Home
last modified time | relevance | path

Searched refs:output_state (Results 1 – 16 of 16) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_3/
Dunidirectional_sequence_rnn.mod.py20 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/
Dlayer_norm_lstm.mod.py177 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
Dlstm_state2_float16.mod.py136 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable
146 input0[output_state_in] = output_state
Dlstm_state_float16.mod.py136 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable
146 input0[output_state_in] = output_state
Dlstm_float16.mod.py136 output_state = [0, 0, 0, 0] variable
146 input0[output_state_in] = output_state
/frameworks/ml/nn/runtime/test/specs/V1_1/
Dlstm_state2_relaxed.mod.py137 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable
147 input0[output_state_in] = output_state
Dlstm_state_relaxed.mod.py137 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable
147 input0[output_state_in] = output_state
Dlstm_relaxed.mod.py137 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/
Dlstm_float.mod.py133 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/
Dlstm_float.mod.py133 output_state = [0, 0, 0, 0] variable
143 input0[output_state_in] = output_state
/frameworks/ml/nn/runtime/test/specs/V1_0/
Dlstm_state.mod.py136 output_state = [-0.0297319, 0.122947, 0.208851, -0.153588] variable
146 input0[output_state_in] = output_state
Dlstm_state2.mod.py136 output_state = [-0.0371611, 0.125073, 0.411934, -0.208605] variable
146 input0[output_state_in] = output_state
Dlstm.mod.py136 output_state = [0, 0, 0, 0] variable
146 input0[output_state_in] = output_state
/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp197 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()
DLSTM.h43 bool output_state; member
/frameworks/ml/nn/common/
DCpuExecutor.cpp1056 const bool output_state = (outs.size() == 5 || outs.size() == 6); in executeOperation() local
1072 if (output_state) { in executeOperation()