Searched refs:tf_dwconv2d_op (Results 1 – 1 of 1) sorted by relevance
846 auto tf_dwconv2d_op = cast<TF::DepthwiseConv2dNativeOp>(op); in matchAndRewrite() local849 tf_dwconv2d_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()851 tf_dwconv2d_op.filter().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()853 tf_dwconv2d_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()863 auto tmpAttr = tf_dwconv2d_op.data_formatAttr(); in matchAndRewrite()870 auto tmpAttr = tf_dwconv2d_op.strides(); in matchAndRewrite()881 auto tmpAttr = tf_dwconv2d_op.dilations(); in matchAndRewrite()893 if (!GetPaddingFromString(tf_dwconv2d_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()897 if (!FormatFromString(tf_dwconv2d_op.data_format().str(), &data_format_tf)) in matchAndRewrite()902 tf_dwconv2d_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()[all …]