pooling_ops.cc | 39 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { 63 int num_dims() const { return num_spatial_dims_ + 2; } 122 const int num_spatial_dims_; member in class:tensorflow::__anon38761::PoolingOp 245 ksize_, stride_, num_spatial_dims_, 278 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { 286 int num_dims() const { return num_spatial_dims_ + 2; } 355 const int num_spatial_dims_; member in class:tensorflow::__anon38761::MaxPoolGradOp 389 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { 406 int num_dims() const { return num_spatial_dims_ + 2; } 442 for (int i = 0; i < num_spatial_dims_; ++i) 500 const int num_spatial_dims_; member in class:tensorflow::__anon38761::AvgPoolGradOp [all...] |