Searched refs:num_spatial_dims_ (Results 1 – 1 of 1) sorted by relevance
44 num_spatial_dims_(num_spatial_dims), in PoolingOp()72 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()124 const int num_spatial_dims_; member in tensorflow::__anon05e5a9280111::PoolingOp269 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in MaxPoolGradOp()277 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()345 const int num_spatial_dims_; member in tensorflow::__anon05e5a9280111::MaxPoolGradOp379 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in AvgPoolGradOp()401 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()447 const int num_spatial_dims_; member in tensorflow::__anon05e5a9280111::AvgPoolGradOp476 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in MaxPoolGradGradOp()[all …]