Searched refs:PoolingOp (Results 1 – 1 of 1) sorted by relevance
39 class PoolingOp : public XlaOpKernel { class41 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp() function in tensorflow::__anon05e5a9280111::PoolingOp150 class MaxPoolOp : public PoolingOp {153 : PoolingOp(ctx, /*num_spatial_dims=*/num_spatial_dims, in MaxPoolOp()202 class AvgPoolOp : public PoolingOp {205 : PoolingOp(ctx, /*num_spatial_dims=*/num_spatial_dims, in AvgPoolOp()