Searched refs:AvgPoolGradOp (Results 1 – 2 of 2) sorted by relevance
397 class AvgPoolGradOp : public XlaOpKernel { class399 AvgPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolGradOp() function in tensorflow::__anon9297e9860111::AvgPoolGradOp475 class AvgPool2DGradOp : public AvgPoolGradOp {478 : AvgPoolGradOp(ctx, /*num_spatial_dims=*/2) {} in AvgPool2DGradOp()484 class AvgPool3DGradOp : public AvgPoolGradOp {487 : AvgPoolGradOp(ctx, /*num_spatial_dims=*/3) {} in AvgPool3DGradOp()
2495 ConvertAvgPoolGradOp<TF::AvgPoolGradOp, /*num_dims=*/4>;