Searched refs:tfl_avgpool_op (Results 1 – 1 of 1) sorted by relevance
869 auto tfl_avgpool_op = cast<TFL::AveragePool2DOp>(op); in matchAndRewrite() local872 tfl_avgpool_op.input().getType().dyn_cast<ShapedType>(); in matchAndRewrite()874 tfl_avgpool_op.getResult().getType().dyn_cast<ShapedType>(); in matchAndRewrite()884 int64_t kernel_h = tfl_avgpool_op.filter_height(); in matchAndRewrite()885 int64_t kernel_w = tfl_avgpool_op.filter_width(); in matchAndRewrite()892 int64_t stride_h = tfl_avgpool_op.stride_h(); in matchAndRewrite()893 int64_t stride_w = tfl_avgpool_op.stride_w(); in matchAndRewrite()898 if (!GetPaddingFromString(tfl_avgpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()928 rewriter, op->getLoc(), average_type, tfl_avgpool_op.input(), in matchAndRewrite()932 rewriter, op->getLoc(), average_type, tfl_avgpool_op.input(), in matchAndRewrite()