Searched defs:backend_memory_format (Results 1 – 2 of 2) sorted by relevance
1416 at::MemoryFormat backend_memory_format = at::MemoryFormat::Contiguous; in determine_backend_memory_format() local1516 at::MemoryFormat backend_memory_format = determine_backend_memory_format(input, weight, backend); in _convolution() local2046 at::MemoryFormat backend_memory_format = determine_backend_memory_format(input, weight, backend); in convolution_backward() local
578 at::MemoryFormat backend_memory_format = at::MemoryFormat::Contiguous; in convolution_overrideable() local