| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 79 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
102 input_layer_norm_weights, forget_layer_norm_weights,
148 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
251 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
274 input_layer_norm_weights, forget_layer_norm_weights,
315 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
140 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 82 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", "{0}") variable
96 input_layer_norm_weights, forget_layer_norm_weights,
648 forget_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 84 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT16", variable
101 input_layer_norm_weights, forget_layer_norm_weights,
156 forget_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
145 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
147 forget_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
145 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | unidirectional_sequence_lstm_1step.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
145 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT16", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
145 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | qlstm_projection.mod.py | 65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType) variable
90 cell_state_in, input_layer_norm_weights, forget_layer_norm_weights,
131 forget_layer_norm_weights: [6553, 6553, 13107, 9830],
188 forget_layer_norm_weights: [6553, 6553, 13107, 9830],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 82 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
105 input_layer_norm_weights, forget_layer_norm_weights,
146 forget_layer_norm_weights: [0.2, 0.2, 0.4, 0.3],
|
| D | qlstm_noprojection.mod.py | 65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType) variable
90 cell_state_in, input_layer_norm_weights, forget_layer_norm_weights,
125 forget_layer_norm_weights: [6553, 6553, 13107, 9830],
|
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 148 const RunTimeOperandInfo* forget_layer_norm_weights, in CheckInputTensorDimensions() argument
268 if (!IsNullInput(forget_layer_norm_weights)) { in CheckInputTensorDimensions()
269 NN_CHECK_EQ(NumDimensions(forget_layer_norm_weights), 1); in CheckInputTensorDimensions()
270 NN_CHECK_EQ(SizeOfDimension(forget_layer_norm_weights, 0), n_cell); in CheckInputTensorDimensions()
285 (IsNullInput(forget_layer_norm_weights) && IsNullInput(cell_layer_norm_weights) && in CheckInputTensorDimensions()
287 (!IsNullInput(forget_layer_norm_weights) && !IsNullInput(cell_layer_norm_weights) && in CheckInputTensorDimensions()
292 (IsNullInput(input_layer_norm_weights) && IsNullInput(forget_layer_norm_weights) && in CheckInputTensorDimensions()
295 !IsNullInput(forget_layer_norm_weights) && !IsNullInput(cell_layer_norm_weights) && in CheckInputTensorDimensions()
|
| D | LSTM.h | 200 const RunTimeOperandInfo* forget_layer_norm_weights,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 4363 ANeuralNetworksOperandType forget_layer_norm_weights = quant16SymTensor1D; in TEST() local
4401 forget_layer_norm_weights, in TEST()
|