| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 83 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
103 cell_layer_norm_weights, output_layer_norm_weights).To(
150 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
255 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
275 cell_layer_norm_weights, output_layer_norm_weights).To(
317 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
142 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 84 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", "{0}") variable
97 cell_layer_norm_weights, output_layer_norm_weights).To([output])
650 output_layer_norm_weights: []
|
| D | unidirectional_sequence_lstm_1step.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
147 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT16", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
147 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
149 output_layer_norm_weights: []
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 88 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT16", variable
102 cell_layer_norm_weights, output_layer_norm_weights).To([output])
158 output_layer_norm_weights: []
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
147 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 87 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
147 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | qlstm_projection.mod.py | 67 output_layer_norm_weights = Input("output_layer_norm_weights", LayerNormType) variable
91 cell_layer_norm_weights, output_layer_norm_weights, cell_clip, projection_clip,
133 output_layer_norm_weights: [19660, 6553, 6553, 16384],
190 output_layer_norm_weights: [19660, 6553, 6553, 16384],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 86 output_layer_norm_weights = Input("output_layer_norm_weights", "TENSOR_FLOAT32", variable
106 cell_layer_norm_weights, output_layer_norm_weights).To(
148 output_layer_norm_weights: [0.6, 0.2, 0.2, 0.5]
|
| D | qlstm_noprojection.mod.py | 67 output_layer_norm_weights = Input("output_layer_norm_weights", LayerNormType) variable
91 cell_layer_norm_weights, output_layer_norm_weights, cell_clip, projection_clip,
127 output_layer_norm_weights: [19660, 6553, 6553, 16384],
|
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 150 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input, uint32_t n_output, in CheckInputTensorDimensions() argument
222 params->use_layer_norm = !IsNullInput(output_layer_norm_weights); in CheckInputTensorDimensions()
276 if (!IsNullInput(output_layer_norm_weights)) { in CheckInputTensorDimensions()
277 NN_CHECK_EQ(NumDimensions(output_layer_norm_weights), 1); in CheckInputTensorDimensions()
278 NN_CHECK_EQ(SizeOfDimension(output_layer_norm_weights, 0), n_cell); in CheckInputTensorDimensions()
286 IsNullInput(output_layer_norm_weights)) || in CheckInputTensorDimensions()
288 !IsNullInput(output_layer_norm_weights)); in CheckInputTensorDimensions()
293 IsNullInput(cell_layer_norm_weights) && IsNullInput(output_layer_norm_weights)) || in CheckInputTensorDimensions()
296 !IsNullInput(output_layer_norm_weights)); in CheckInputTensorDimensions()
|
| D | LSTM.h | 202 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 4364 ANeuralNetworksOperandType output_layer_norm_weights = quant16SymTensor1D; in TEST() local
4402 output_layer_norm_weights, in TEST()
|