| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 77 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
102 input_layer_norm_weights, forget_layer_norm_weights,
147 input_layer_norm_weights: [0.1, 0.2, 0.3, 0.5],
249 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
274 input_layer_norm_weights, forget_layer_norm_weights,
314 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
139 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", "{0}") variable
96 input_layer_norm_weights, forget_layer_norm_weights,
647 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_1step.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
144 input_layer_norm_weights: [0.1, 0.2, 0.3, 0.5],
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT16", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
144 input_layer_norm_weights: [0.1, 0.2, 0.3, 0.5],
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
146 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 82 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT16", variable
101 input_layer_norm_weights, forget_layer_norm_weights,
155 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
144 input_layer_norm_weights: [0.1, 0.2, 0.3, 0.5],
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 81 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
100 input_layer_norm_weights, forget_layer_norm_weights,
144 input_layer_norm_weights: [0.1, 0.2, 0.3, 0.5],
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | qlstm_projection.mod.py | 64 input_layer_norm_weights = Input("input_layer_norm_weights", LayerNormType) variable
90 cell_state_in, input_layer_norm_weights, forget_layer_norm_weights,
130 input_layer_norm_weights: [3277, 6553, 9830, 16384],
187 input_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 80 input_layer_norm_weights = Input("input_layer_norm_weights", "TENSOR_FLOAT32", variable
105 input_layer_norm_weights, forget_layer_norm_weights,
145 input_layer_norm_weights: [],
|
| D | qlstm_noprojection.mod.py | 64 input_layer_norm_weights = Input("input_layer_norm_weights", LayerNormType) variable
90 cell_state_in, input_layer_norm_weights, forget_layer_norm_weights,
124 input_layer_norm_weights: [],
|
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 147 const RunTimeOperandInfo* input_layer_norm_weights, in CheckInputTensorDimensions() argument
264 if (!IsNullInput(input_layer_norm_weights)) { in CheckInputTensorDimensions()
265 NN_CHECK_EQ(NumDimensions(input_layer_norm_weights), 1); in CheckInputTensorDimensions()
266 NN_CHECK_EQ(SizeOfDimension(input_layer_norm_weights, 0), n_cell); in CheckInputTensorDimensions()
282 NN_RET_CHECK(IsNullInput(input_layer_norm_weights)) in CheckInputTensorDimensions()
292 (IsNullInput(input_layer_norm_weights) && IsNullInput(forget_layer_norm_weights) && in CheckInputTensorDimensions()
294 (!IsNullInput(input_layer_norm_weights) && in CheckInputTensorDimensions()
|
| D | LSTM.h | 199 const RunTimeOperandInfo* input_layer_norm_weights,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 4361 ANeuralNetworksOperandType input_layer_norm_weights = quant16SymTensor1D; in TEST() local
4399 input_layer_norm_weights, in TEST()
|