Searched defs:MaxPoolGradOp (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
H A D | pooling_ops.cc | 275 class MaxPoolGradOp : public XlaOpKernel { class in namespace:tensorflow::__anon25440 277 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) function in class:tensorflow::__anon25440::MaxPoolGradOp 362 class MaxPool2DGradOp : public MaxPoolGradOp { 365 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/2) { 378 class MaxPool3DGradOp : public MaxPoolGradOp { 381 : MaxPoolGradOp(ctx, /*num_spatial_dims=*/3) {} |
Completed in 79 milliseconds