Searched refs:AvgPoolOp (Results 1 – 1 of 1) sorted by relevance
202 class AvgPoolOp : public PoolingOp { class204 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp() function in tensorflow::__anon05e5a9280111::AvgPoolOp245 class AvgPool2DOp : public AvgPoolOp {248 : AvgPoolOp(ctx, /*num_spatial_dims=*/2) { in AvgPool2DOp()253 class AvgPool3DOp : public AvgPoolOp {256 : AvgPoolOp(ctx, /*num_spatial_dims=*/3) {} in AvgPool3DOp()