Searched defs:weight_type (Results 1 – 5 of 5) sorted by relevance
39 TensorType weight_type = TensorType_FLOAT32) { in BaseEmbeddingLookupOpModel()
41 const TensorType& weight_type = TensorType_FLOAT32, in LSTMOpModel()1638 const TensorType& weight_type = TensorType_FLOAT32) in LayerNormLSTMOpModel()
53 const auto weight_type = in BidirectionalLSTMOpModel() local
351 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
2263 const TensorType weight_type) in LSTMOpModel()3463 TensorType weight_type = TensorType_FLOAT32) { in BaseEmbeddingLookupOpModel()