MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsConv1x1StrideNot1

Function IsConv1x1StrideNot1

tensorflow/core/util/mkl_util.h:2127–2133  ·  view source on GitHub ↗

utility function to determine if it is conv 1x1 and stride != 1 for purpose of temporarily disabling primitive reuse

Source from the content-addressed store, hash-verified

2125// utility function to determine if it is conv 1x1 and stride != 1
2126// for purpose of temporarily disabling primitive reuse
2127inline bool IsConv1x1StrideNot1(memory::dims filter_dims,
2128 memory::dims strides) {
2129 if (filter_dims.size() != 4 || strides.size() != 2) return false;
2130
2131 return ((filter_dims[2] == 1) && (filter_dims[3] == 1) &&
2132 ((strides[0] != 1) || (strides[1] != 1)));
2133}
2134
2135inline bool IsPrimitiveRefPath(dnnl::primitive_desc_base& base_desc) {
2136 string impl_info = string(base_desc.impl_info_str());

Callers 2

ComputeMethod · 0.85
ComputeMethod · 0.85

Calls 1

sizeMethod · 0.45

Tested by

no test coverage detected