Searched refs:LayerNormType (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | qlstm_noprojection.mod.py | 63 LayerNormType = ("TENSOR_QUANT16_SYMM", [num_units], 3.05182e-05, 0) variable
64 input_layer_norm_weights = Input("input_layer_norm_weights", LayerNormType)
65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType)
66 cell_layer_norm_weights = Input("cell_layer_norm_weights", LayerNormType)
67 output_layer_norm_weights = Input("output_layer_norm_weights", LayerNormType)
|
| D | qlstm_projection.mod.py | 63 LayerNormType = ("TENSOR_QUANT16_SYMM", [num_units], 3.05182e-05, 0) variable
64 input_layer_norm_weights = Input("input_layer_norm_weights", LayerNormType)
65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType)
66 cell_layer_norm_weights = Input("cell_layer_norm_weights", LayerNormType)
67 output_layer_norm_weights = Input("output_layer_norm_weights", LayerNormType)
|