Searched refs:InputWeightsType (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/runtime/test/specs/V1_2/ |
D | quantized_lstm.mod.py | 31 InputWeightsType = ("TENSOR_QUANT8_ASYMM", variable 33 input_to_input_weights = Input("inputToInputWeights", InputWeightsType) 34 input_to_forget_weights = Input("inputToForgetWeights", InputWeightsType) 35 input_to_cell_weights = Input("inputToCellWeights", InputWeightsType) 36 input_to_output_weights = Input("inputToOutputWeights", InputWeightsType) 118 InputWeightsType = ("TENSOR_QUANT8_ASYMM", variable 120 input_to_input_weights = Parameter("inputToInputWeights", InputWeightsType, 122 input_to_forget_weights = Parameter("inputToForgetWeights", InputWeightsType, 124 input_to_cell_weights = Parameter("inputToCellWeights", InputWeightsType, 126 input_to_output_weights = Parameter("inputToOutputWeights", InputWeightsType,
|
/frameworks/ml/nn/runtime/test/specs/V1_3/ |
D | qlstm_noprojection.mod.py | 30 InputWeightsType = ("TENSOR_QUANT8_SYMM", [num_units, input_size], 0.00784314, 0) variable 31 input_to_input_weights = Input("input_to_input_weights", InputWeightsType) 32 input_to_forget_weights = Input("input_to_forget_weights", InputWeightsType) 33 input_to_cell_weights = Input("input_to_cell_weights", InputWeightsType) 34 input_to_output_weights = Input("input_to_output_weights", InputWeightsType)
|
D | qlstm_projection.mod.py | 30 InputWeightsType = ("TENSOR_QUANT8_SYMM", [num_units, input_size], 0.00784314, 0) variable 31 input_to_input_weights = Input("input_to_input_weights", InputWeightsType) 32 input_to_forget_weights = Input("input_to_forget_weights", InputWeightsType) 33 input_to_cell_weights = Input("input_to_cell_weights", InputWeightsType) 34 input_to_output_weights = Input("input_to_output_weights", InputWeightsType)
|