Searched refs:AvgPoolOp (Results 1 – 4 of 4) sorted by relevance
210 class AvgPoolOp : public PoolingOp { class212 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp() function in tensorflow::__anon9297e9860111::AvgPoolOp253 class AvgPool2DOp : public AvgPoolOp {256 : AvgPoolOp(ctx, /*num_spatial_dims=*/2) {} in AvgPool2DOp()260 class AvgPool3DOp : public AvgPoolOp {263 : AvgPoolOp(ctx, /*num_spatial_dims=*/3) {} in AvgPool3DOp()
930 rewriter.replaceOpWithNewOp<AvgPoolOp>( in replaceWithAvgPool()
495 auto tf_avgpool_op = cast<TF::AvgPoolOp>(op); in matchAndRewrite()
2249 Value GetAvgPoolInput(TF::AvgPoolOp op) { return op.value(); } in GetAvgPoolInput()2319 using ConvertAvgPool2DOp = ConvertAvgPoolOp<TF::AvgPoolOp, /*num_dims=*/4>;