Searched refs:MaxPoolGradOp (Results 1 – 3 of 3) sorted by relevance
273 class MaxPoolGradOp : public XlaOpKernel { class275 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp() function in tensorflow::__anon9297e9860111::MaxPoolGradOp373 class MaxPool2DGradOp : public MaxPoolGradOp {376 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/2) { in MaxPool2DGradOp()389 class MaxPool3DGradOp : public MaxPoolGradOp {392 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/3) {} in MaxPool3DGradOp()
2758 static LogicalResult Verify(MaxPoolGradOp op) { in Verify()
4090 ConvertMaxPoolGradOp<TF::MaxPoolGradOp, /*num_dims=*/4>;