Searched refs:middle_size (Results 1 – 2 of 2) sorted by relevance
50 const int64 middle_size = input.dim_size(1); in ReverseRows() local51 const int64 row_size = inner_size * middle_size; in ReverseRows()62 int remaining = middle_size; in ReverseRows()
480 for middle_size in list(range(50)) + [100000]:484 middle_size=middle_size):486 np.arange(outer_size * middle_size * 3, dtype=np.float32),487 newshape=(outer_size, middle_size, 3))497 for middle_size in list(range(50)) + [100000]:501 middle_size=middle_size):503 np.arange(outer_size * middle_size * 4, dtype=np.float32),504 newshape=(outer_size, middle_size, 4))514 for middle_size in (1, 2):518 middle_size=middle_size):[all …]