Searched refs:dim_shift_array (Results 1 – 1 of 1) sorted by relevance
70 int64_t dim_shift_array[kTensorDimensionLimit]; in roll_out() local72 dim_shift_array[i] = 0; in roll_out()76 dim_shift_array[d] += shifts[i]; in roll_out()80 IntArrayRef dim_shifts(dim_shift_array, dim_shift_array_length); in roll_out()