Home
last modified time | relevance | path

Searched refs:input_to_forget_weights (Results 1 – 23 of 23) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/
Dlstm2_state.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [-0.55291498, -0.42866567, 0.13056988, -0.3633365, -0.22755712, 0.2825369…
Dlstm2.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [-0.55291498, -0.42866567, 0.13056988, -0.3633365, -0.22755712, 0.2825369…
Dlstm.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, 0.4478132…
Dlstm3_state2.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
68 input_to_forget_weights,
119 input_to_forget_weights: [
Dlstm3_state.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
68 input_to_forget_weights,
119 input_to_forget_weights: [
Dlstm2_state2.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [-0.55291498, -0.42866567, 0.13056988, -0.3633365, -0.22755712, 0.2825369…
Dlstm_state.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, 0.4478132…
Dlstm_state2.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, 0.4478132…
Dlstm3_state3.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
68 input_to_forget_weights,
119 input_to_forget_weights: [
Dlstm3.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
68 input_to_forget_weights,
119 input_to_forget_weights: [
/frameworks/ml/nn/tools/test_generator/tests/P_lstm/
Dlstm.mod.py30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, … variable
67 input_to_forget_weights,
98 …input_to_forget_weights: [0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, 0.4478132…
Dstdout.txt.expect16 auto input_to_forget_weights = model->addOperand(&type1);
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,…
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we…
/frameworks/ml/nn/runtime/test/generated/models/
Dlstm2.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm3_state2.model.cpp17 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
43 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
46 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm_state2.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm3_state.model.cpp17 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
43 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
46 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm3_state3.model.cpp17 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
43 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
46 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm3.model.cpp17 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
43 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
46 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm2_state.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm_state.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
Dlstm2_state2.model.cpp16 auto input_to_forget_weights = model->addOperand(&type1); in CreateModel() local
42 …model->addOperation(ANEURALNETWORKS_LSTM, {input, input_to_input_weights, input_to_forget_weights,… in CreateModel()
45 …{input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_we… in CreateModel()
/frameworks/ml/nn/common/operations/
DLSTM.cpp105 const RunTimeOperandInfo *input_to_forget_weights = in CheckInputTensorDimensions() local
107 NN_CHECK_EQ(NumDimensions(input_to_forget_weights), 2); in CheckInputTensorDimensions()
108 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 0), n_cell); in CheckInputTensorDimensions()
109 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 1), n_input); in CheckInputTensorDimensions()