Skip to main content

pytorch_layer_mapping

axonml_serialize

Function pytorch_layer_mapping 

Source 
pub fn pytorch_layer_mapping() -> HashMap<&'static str, &'static str>
Expand description

Map of PyTorch layer names to Axonml equivalents.