Lines Matching full:at
11 namespace at::native {
13 using conv_depthwise2d_backward_fn = std::tuple<at::Tensor,at::Tensor>(*)(
14 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
15 at::IntArrayRef, at::IntArrayRef, std::array<bool, 2>);
17 using conv_depthwise3d_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
18 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
19 at::IntArrayRef, at::IntArrayRef, std::array<bool, 3>);
21 using cudnn_convolution_backward_fn = std::tuple<at::Tensor,at::Tensor>(*)(
22 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
23 at::IntArrayRef, int64_t, bool, bool, bool, std::array<bool,2>);
25 using mps_convolution_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
26 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
27 at::IntArrayRef, int64_t, std::array<bool,3>);
29 using cudnn_convolution_transpose_backward_fn = std::tuple<at::Tensor,at::Tensor>(*)(
30 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
31 at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool, std::array<bool,2>);
33 using miopen_convolution_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
34 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
35 at::IntArrayRef, int64_t, bool, bool, std::array<bool,3>);
37 using miopen_convolution_transpose_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
38 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
39 at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, std::array<bool,3>);
41 using miopen_depthwise_convolution_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
42 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
43 at::IntArrayRef, int64_t, bool, bool, std::array<bool,3>);
45 using mkldnn_convolution_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
46 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
47 at::IntArrayRef, int64_t, std::array<bool,3>);
52 using mkldnn_convolution_transpose_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
53 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
54 at::IntArrayRef, at::IntArrayRef, int64_t, std::array<bool,3>);
56 using slow_conv_dilated2d_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
57 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
58 at::IntArrayRef, at::IntArrayRef, std::array<bool, 3>);
60 using slow_conv_dilated3d_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
61 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
62 at::IntArrayRef, at::IntArrayRef, std::array<bool, 3>);
64 using slow_conv_transpose2d_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
65 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
66 at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, std::array<bool,3>);
68 using slow_conv_transpose3d_backward_fn = std::tuple<at::Tensor,at::Tensor,at::Tensor>(*)(
69 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
70 at::IntArrayRef, at::IntArrayRef, at::IntArrayRef, std::array<bool,3>);
127 …bool transposed, SymIntArrayRef output_padding, c10::SymInt groups, const at::OptionalSymIntArrayR…
129 TORCH_API at::MemoryFormat _determine_backend_memory_format(const Tensor& input,
320 inline at::MemoryFormat cudnn_conv_suggest_memory_format(const at::Tensor& input, const at::Tensor&… in cudnn_conv_suggest_memory_format()
322 if (!at::detail::getCUDAHooks().compiledWithCuDNN() || in cudnn_conv_suggest_memory_format()
323 input.scalar_type() == at::kDouble || in cudnn_conv_suggest_memory_format()
324 weight.scalar_type() == at::kDouble) { in cudnn_conv_suggest_memory_format()
325 return at::MemoryFormat::Contiguous; in cudnn_conv_suggest_memory_format()
327 long cudnn_version = at::detail::getCUDAHooks().versionCuDNN(); in cudnn_conv_suggest_memory_format()
333 (input_memory_format == at::MemoryFormat::ChannelsLast) || in cudnn_conv_suggest_memory_format()
334 (weight_memory_format == at::MemoryFormat::ChannelsLast) in cudnn_conv_suggest_memory_format()
337 return at::MemoryFormat::ChannelsLast; in cudnn_conv_suggest_memory_format()
341 (input_memory_format == at::MemoryFormat::ChannelsLast3d) || in cudnn_conv_suggest_memory_format()
342 (weight_memory_format == at::MemoryFormat::ChannelsLast3d) in cudnn_conv_suggest_memory_format()
345 return at::MemoryFormat::ChannelsLast3d; in cudnn_conv_suggest_memory_format()
348 return at::MemoryFormat::Contiguous; in cudnn_conv_suggest_memory_format()
356 inline bool miopen_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in miopen_conv_use_channels_last()
359 if (!at::detail::getCUDAHooks().compiledWithMIOpen() || in miopen_conv_use_channels_last()
360 input.scalar_type() == at::kDouble || in miopen_conv_use_channels_last()
361 weight.scalar_type() == at::kDouble) { in miopen_conv_use_channels_last()
374 ( (input_memory_format == at::MemoryFormat::ChannelsLast) || in miopen_conv_use_channels_last()
375 (weight_memory_format == at::MemoryFormat::ChannelsLast) ) in miopen_conv_use_channels_last()
383 inline bool mkldnn_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in mkldnn_conv_use_channels_last()
386 if (input.scalar_type() == at::kDouble || in mkldnn_conv_use_channels_last()
387 weight.scalar_type() == at::kDouble) { in mkldnn_conv_use_channels_last()
400 (input_memory_format == at::MemoryFormat::ChannelsLast) || in mkldnn_conv_use_channels_last()
401 (weight_memory_format == at::MemoryFormat::ChannelsLast); in mkldnn_conv_use_channels_last()
404 (input_memory_format == at::MemoryFormat::ChannelsLast3d) || in mkldnn_conv_use_channels_last()
405 (weight_memory_format == at::MemoryFormat::ChannelsLast3d); in mkldnn_conv_use_channels_last()
410 inline bool thnn_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in thnn_conv_use_channels_last()
416 (input_memory_format == at::MemoryFormat::ChannelsLast) || ( in thnn_conv_use_channels_last()
417 weight_memory_format == at::MemoryFormat::ChannelsLast)); in thnn_conv_use_channels_last()
422 inline bool xpu_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in xpu_conv_use_channels_last()
430 if (input.scalar_type() == at::kDouble || in xpu_conv_use_channels_last()
431 weight.scalar_type() == at::kDouble) { in xpu_conv_use_channels_last()
439 (input_memory_format == at::MemoryFormat::ChannelsLast) || in xpu_conv_use_channels_last()
440 (weight_memory_format == at::MemoryFormat::ChannelsLast); in xpu_conv_use_channels_last()
443 (input_memory_format == at::MemoryFormat::ChannelsLast3d) || in xpu_conv_use_channels_last()
444 (weight_memory_format == at::MemoryFormat::ChannelsLast3d); in xpu_conv_use_channels_last()
449 } // namespace at::native