Searched refs:dims_at_axis (Results 1 – 1 of 1) sorted by relevance
1708 const int dims_at_axis = input_shape.Dims(axis); in Reverse() local1710 for (int j = 0; j < dims_at_axis; ++j) { in Reverse()1711 const int start_pos = (i * dims_at_axis + j) * copy_size; in Reverse()1713 int loc = (i * dims_at_axis + dims_at_axis - j - 1) * copy_size; in Reverse()