Searched refs:IsConv2D (Results 1 – 6 of 6) sorted by relevance
166 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataFormat()172 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsGpuCompatibleDataFormat()178 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleConv2D()184 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsGpuCompatibleConv2D()243 if (!conv2d.node || !IsConv2D(*conv2d.node) || in FindConv2DWithBias()324 if (!conv2d.node || !IsConv2D(*conv2d.node) || in FindConv2DWithSqueezeAndBias()371 if (!conv2d.node || !IsConv2D(*conv2d.node) || // in FindConv2DWithBatchNorm()
2017 } else if (IsConv2D(*node)) { in Expand()
2917 if (!IsConv2D(*conv_node) && !IsConv3D(*conv_node)) { in MulConvPushDown()
2070 return IsConv2D(*node) || IsConv3D(*node); in IsSupported()
56 bool IsConv2D(const NodeDef& node);
147 bool IsConv2D(const NodeDef& node) { return node.op() == "Conv2D"; } in IsConv2D() function