Searched refs:dw_conv2d (Results 1 – 1 of 1) sorted by relevance
296 bool IsCpuCompatibleDepthwiseConv2dNative(const NodeDef* dw_conv2d) { in IsCpuCompatibleDataType() argument297 DCHECK(IsDepthwiseConv2dNative(*dw_conv2d)) in IsCpuCompatibleDataType()299 return NodeIsOnCpu(dw_conv2d) && IsCpuCompatibleDataType(dw_conv2d); in IsCpuCompatibleDataType()986 void CopyDepthwiseConv2dNativeAttributes(const NodeDef& dw_conv2d, in IsCpuCompatibleDataType() argument989 DCHECK(IsDepthwiseConv2dNative(dw_conv2d)) in IsCpuCompatibleDataType()993 auto& src_attr = dw_conv2d.attr(); in IsCpuCompatibleDataType()