Searched refs:convert_conv2d_weight_memory_format (Results 1 – 4 of 4) sorted by relevance
5 def convert_conv2d_weight_memory_format(module, memory_format): function76 convert_conv2d_weight_memory_format(child, memory_format)
12 convert_conv2d_weight_memory_format,
403 convert_conv2d_weight_memory_format
3936 model = nn.utils.convert_conv2d_weight_memory_format(model, memory_format)3946 model = nn.utils.convert_conv2d_weight_memory_format(model, memory_format)