Searched refs:IsConv2D (Results 1 – 8 of 8) sorted by relevance
234 if (IsConv2D(*contraction) || IsDepthwiseConv2dNative(*contraction) || in IsCpuCompatibleDataType()238 if (IsConv2D(*contraction) || IsDepthwiseConv2dNative(*contraction) || in IsCpuCompatibleDataType()243 if (IsConv2D(*contraction)) { in IsCpuCompatibleDataType()256 if (IsConv2D(*contraction)) { in IsCpuCompatibleDataType()264 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()274 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()280 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()286 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()306 if (IsConv2D(node)) { in IsCpuCompatibleDataType()333 if (!IsConv2D(contraction_node)) return false; in IsCpuCompatibleDataType()[all …]
80 if (!IsConv2D(*node_def) && !IsConv3D(*node_def)) { in NumConvOnDeviceWithDataTypeOverThreshold()
2075 } else if (IsConv2D(*node)) { in Expand()
91 if (IsConv2D(*node.node()) || IsConv2DBackpropInput(*node.node())) { in IsNonFloatingConv2D()
2173 return IsConv2D(*node) || IsConv3D(*node); in IsSupported()
3465 if (!IsConv2D(*conv_node) && !IsConv3D(*conv_node)) { in MulConvPushDown()
63 bool IsConv2D(const NodeDef& node);
179 bool IsConv2D(const NodeDef& node) { return node.op() == "Conv2D"; } in IsConv2D() function