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
824 for (auto spatial_dimension : spatial_dimensions) { in ChangeSpatialSizeOnSpaceToBatchedShape() local841 for (auto spatial_dimension : spatial_dimensions) { in ChangeSpatialSizeOnSpaceToBatchedShape() local861 for (auto spatial_dimension : spatial_dimensions) { in ChangeSpatialSizeOnSpaceToBatchedShape() local2409 for (auto spatial_dimension : split_spatial_dimensions) { in BatchToSpace() local2424 for (auto spatial_dimension : split_spatial_dimensions) { in BatchToSpace() local
615 for (int64_t spatial_dimension = 0; in GetConvolutionFlops() local640 for (int64_t spatial_dimension = 0; in GetConvolutionFlops() local