Searched refs:spatial_dim_size (Results 1 – 1 of 1) sorted by relevance
804 const int64_t spatial_dim_size = in ChangeSpatialSizeOnSpaceToBatchedShape() local806 const int64_t reshaped_space_size = spatial_dim_size * ctrl_.number_of_splits; in ChangeSpatialSizeOnSpaceToBatchedShape()2392 const int64_t spatial_dim_size = new_instr->shape().dimensions(space_dim); in BatchToSpace() local2411 spatial_dim_size * ctrl_.number_of_splits; in BatchToSpace()