Home
last modified time | relevance | path

Searched refs:determine_backend_memory_format (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp1412 static inline at::MemoryFormat determine_backend_memory_format( in determine_backend_memory_format() function
1465 return determine_backend_memory_format(input, weight, backend); in _determine_backend_memory_format()
1516 at::MemoryFormat backend_memory_format = determine_backend_memory_format(input, weight, backend); in _convolution()
2046 at::MemoryFormat backend_memory_format = determine_backend_memory_format(input, weight, backend); in convolution_backward()