Searched refs:MaxPoolGradOp (Results 1 – 1 of 1) sorted by relevance
266 class MaxPoolGradOp : public XlaOpKernel { class268 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp() function in tensorflow::__anon05e5a9280111::MaxPoolGradOp352 class MaxPool2DGradOp : public MaxPoolGradOp {355 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/2) { in MaxPool2DGradOp()368 class MaxPool3DGradOp : public MaxPoolGradOp {371 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/3) {} in MaxPool3DGradOp()