Searched refs:MaxPoolOp (Results 1 – 1 of 1) sorted by relevance
150 class MaxPoolOp : public PoolingOp { class152 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp() function in tensorflow::__anon05e5a9280111::MaxPoolOp183 class MaxPool2DOp : public MaxPoolOp {186 : MaxPoolOp(ctx, /*num_spatial_dims=*/2) { in MaxPool2DOp()195 class MaxPool3DOp : public MaxPoolOp {198 : MaxPoolOp(ctx, /*num_spatial_dims=*/3) {} in MaxPool3DOp()