We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 88a9bd4 commit d6ae9acCopy full SHA for d6ae9ac
aten/src/ATen/native/ConvUtils.h
@@ -392,6 +392,11 @@ inline at::MemoryFormat miopen_conv_suggest_memory_format(const at::Tensor& inpu
392
return at::MemoryFormat::Contiguous;
393
}
394
395
+// deprecated, but to remove would be BC-breaking
396
+inline bool miopen_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) {
397
+ return miopen_conv_suggest_memory_format(input, weight) != at::MemoryFormat::Contiguous;
398
+}
399
+
400
inline bool mkldnn_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) {
401
402
// disable NHWC for float64 input.
0 commit comments