Searched refs:tf_avgpool_op (Results 1 – 1 of 1) sorted by relevance
495 auto tf_avgpool_op = cast<TF::AvgPoolOp>(op); in matchAndRewrite() local498 tf_avgpool_op.value().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()500 tf_avgpool_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()504 auto tmpAttr = tf_avgpool_op.data_formatAttr(); in matchAndRewrite()511 auto tmpAttr = tf_avgpool_op.strides(); in matchAndRewrite()522 auto tmpAttr = tf_avgpool_op.ksize(); in matchAndRewrite()534 if (!GetPaddingFromString(tf_avgpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()542 for (auto& elem : tf_avgpool_op.ksize()) { in matchAndRewrite()559 op, output_type, tf_avgpool_op.value(), kernel, stride, pad); in matchAndRewrite()