Searched refs:RecurrentWeightsType (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/runtime/test/specs/V1_2/ |
D | quantized_lstm.mod.py | 38 RecurrentWeightsType = ("TENSOR_QUANT8_ASYMM", variable 40 recurrent_to_input_weights = Input("recurrentToInputWeights", RecurrentWeightsType) 41 recurrent_to_forget_weights = Input("recurrentToForgetWeights", RecurrentWeightsType) 42 recurrent_to_cell_weights = Input("recurrentToCellWeights", RecurrentWeightsType) 43 recurrent_to_output_weights = Input("recurrentToOutputWeights", RecurrentWeightsType) 129 RecurrentWeightsType = ("TENSOR_QUANT8_ASYMM", variable 131 recurrent_to_input_weights = Parameter("recurrentToInputWeights", RecurrentWeightsType, 133 recurrent_to_forget_weights = Parameter("recurrentToForgetWeights", RecurrentWeightsType, 135 recurrent_to_cell_weights = Parameter("recurrentToCellWeights", RecurrentWeightsType, 137 recurrent_to_output_weights = Parameter("recurrentToOutputWeights", RecurrentWeightsType,
|
/frameworks/ml/nn/runtime/test/specs/V1_3/ |
D | qlstm_noprojection.mod.py | 36 RecurrentWeightsType = ("TENSOR_QUANT8_SYMM", [num_units, output_size], 0.00784314, 0) variable 37 recurrent_to_input_weights = Input("recurrent_to_input_weights", RecurrentWeightsType) 38 recurrent_to_forget_weights = Input("recurrent_to_forget_weights", RecurrentWeightsType) 39 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", RecurrentWeightsType) 40 recurrent_to_output_weights = Input("recurrent_to_output_weights", RecurrentWeightsType)
|
D | qlstm_projection.mod.py | 36 RecurrentWeightsType = ("TENSOR_QUANT8_SYMM", [num_units, output_size], 0.00784314, 0) variable 37 recurrent_to_input_weights = Input("recurrent_to_input_weights", RecurrentWeightsType) 38 recurrent_to_forget_weights = Input("recurrent_to_forget_weights", RecurrentWeightsType) 39 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", RecurrentWeightsType) 40 recurrent_to_output_weights = Input("recurrent_to_output_weights", RecurrentWeightsType)
|