Searched defs:spatial_dimension (Results 1 – 3 of 3) sorted by relevance
38 int spatial_dimension(int dim) const { return spatial_dimensions_[dim]; } in spatial_dimension() function
597 for (int64 spatial_dimension = 0; in HandleConvolution() local622 for (int64 spatial_dimension = 0; in HandleConvolution() local
571 int64 spatial_dimension, int64 new_spatial_dim_size) { in IncreaseSpatialSizeOnSpaceToBatchedShape()