Searched refs:MaxPoolGradGradOp (Results 1 – 2 of 2) sorted by relevance
493 class MaxPoolGradGradOp : public XlaOpKernel { class495 MaxPoolGradGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradGradOp() function in tensorflow::__anon9297e9860111::MaxPoolGradGradOp648 class MaxPool2DGradGradOp : public MaxPoolGradGradOp {651 : MaxPoolGradGradOp(ctx, /*num_spatial_dims=*/2) { in MaxPool2DGradGradOp()666 class MaxPool3DGradGradOp : public MaxPoolGradGradOp {669 : MaxPoolGradGradOp(ctx, /*num_spatial_dims=*/3) { in MaxPool3DGradGradOp()
183 TypeID::get<TF::MaxPoolGradGradOp>(), in IsOpAllowedTf2XlaFallback()