| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 81 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
103 cell_layer_norm_weights, output_layer_norm_weights).To(
149 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
253 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
275 cell_layer_norm_weights, output_layer_norm_weights).To(
316 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
141 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 83 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", "{0}") variable
97 cell_layer_norm_weights, output_layer_norm_weights).To([output])
649 cell_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_1step.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
146 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT16", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
146 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
148 cell_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 86 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT16", variable
102 cell_layer_norm_weights, output_layer_norm_weights).To([output])
157 cell_layer_norm_weights: [],
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
146 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 85 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
101 cell_layer_norm_weights, output_layer_norm_weights).To([output])
146 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | qlstm_projection.mod.py | 66 cell_layer_norm_weights = Input("cell_layer_norm_weights", LayerNormType) variable
91 cell_layer_norm_weights, output_layer_norm_weights, cell_clip, projection_clip,
132 cell_layer_norm_weights: [22937, 6553, 9830, 26214],
189 cell_layer_norm_weights: [22937, 6553, 9830, 26214],
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 84 cell_layer_norm_weights = Input("cell_layer_norm_weights", "TENSOR_FLOAT32", variable
106 cell_layer_norm_weights, output_layer_norm_weights).To(
147 cell_layer_norm_weights: [0.7, 0.2, 0.3, 0.8],
|
| D | qlstm_noprojection.mod.py | 66 cell_layer_norm_weights = Input("cell_layer_norm_weights", LayerNormType) variable
91 cell_layer_norm_weights, output_layer_norm_weights, cell_clip, projection_clip,
126 cell_layer_norm_weights: [22937, 6553, 9830, 26214],
|
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 149 const RunTimeOperandInfo* cell_layer_norm_weights, in CheckInputTensorDimensions() argument
272 if (!IsNullInput(cell_layer_norm_weights)) { in CheckInputTensorDimensions()
273 NN_CHECK_EQ(NumDimensions(cell_layer_norm_weights), 1); in CheckInputTensorDimensions()
274 NN_CHECK_EQ(SizeOfDimension(cell_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()
293 IsNullInput(cell_layer_norm_weights) && IsNullInput(output_layer_norm_weights)) || in CheckInputTensorDimensions()
295 !IsNullInput(forget_layer_norm_weights) && !IsNullInput(cell_layer_norm_weights) && in CheckInputTensorDimensions()
|
| D | LSTM.h | 201 const RunTimeOperandInfo* cell_layer_norm_weights,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 4363 ANeuralNetworksOperandType cell_layer_norm_weights = quant16SymTensor1D; in TEST() local
4401 cell_layer_norm_weights, in TEST()
|