Searched refs:weight_type (Results 1 – 5 of 5) sorted by relevance
53 const auto weight_type = in BidirectionalLSTMOpModel() local59 fw_input_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()62 fw_input_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()63 fw_input_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()64 fw_input_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()69 fw_recurrent_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()72 fw_recurrent_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()73 fw_recurrent_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()74 fw_recurrent_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()80 fw_cell_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()[all …]
41 const TensorType& weight_type = TensorType_FLOAT32, in LSTMOpModel() argument52 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()55 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()56 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()57 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()62 recurrent_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()65 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()66 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()67 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()73 cell_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()[all …]
39 TensorType weight_type = TensorType_FLOAT32) { in BaseEmbeddingLookupOpModel() argument41 weight_ = AddInput(weight_type); in BaseEmbeddingLookupOpModel()
2263 const TensorType weight_type) in LSTMOpModel() argument2268 weight_type_(weight_type) { in LSTMOpModel()2274 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()2277 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()2278 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()2279 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()2284 recurrent_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()2287 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()2288 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()2289 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()[all …]
351 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument377 raise ValueError('Provided `' + weight_type + '` was a list of ' +380 'You should provide one `' + weight_type + '`'389 raise TypeError('The model has multiple outputs, so `' + weight_type + '` '391 'Provided `' + weight_type + '` type not understood: ' +