Searched refs:num_spatial_dims_ (Results 1 – 1 of 1) sorted by relevance
46 num_spatial_dims_(num_spatial_dims), in PoolingOp()74 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()126 const int num_spatial_dims_; member in tensorflow::__anon9297e9860111::PoolingOp276 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in MaxPoolGradOp()284 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()366 const int num_spatial_dims_; member in tensorflow::__anon9297e9860111::MaxPoolGradOp400 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in AvgPoolGradOp()422 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims()468 const int num_spatial_dims_; member in tensorflow::__anon9297e9860111::AvgPoolGradOp496 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) { in MaxPoolGradGradOp()[all …]