Searched refs:padding_memory_format_3d (Results 1 – 1 of 1) sorted by relevance
477 at::MemoryFormat padding_memory_format_3d(const Tensor& input) { in padding_memory_format_3d() function560 switch (padding_memory_format_3d(input)) { in reflection_pad3d_kernel_impl()583 switch (padding_memory_format_3d(grad_output)) { in reflection_pad3d_backward_kernel_impl()668 switch (padding_memory_format_3d(input)) { in replication_pad3d_kernel_impl()691 switch (padding_memory_format_3d(grad_output)) { in replication_pad3d_backward_kernel_impl()