Searched refs:spatial_dimensions_ (Results 1 – 1 of 1) sorted by relevance
31 spatial_dimensions_(spatial_dimensions.begin(), in TensorFormat()38 int spatial_dimension(int dim) const { return spatial_dimensions_[dim]; } in spatial_dimension()40 int num_spatial_dims() const { return spatial_dimensions_.size(); } in num_spatial_dims()48 absl::InlinedVector<int, 4> spatial_dimensions_; variable