Searched refs:flat_index (Results 1 – 3 of 3) sorted by relevance
118 int flat_index = 0; in GetFlatIndex() local137 flat_index += index_in_input * (*eval_data->input_dims_num_elements)[i]; in GetFlatIndex()140 return flat_index; in GetFlatIndex()
156 int flat_index = index[0];158 flat_index = flat_index * dims[i] + index[i];160 return flat_index;
1595 int flat_index = i; in SelectTrueCoords() local1597 int coord_j = flat_index / dims_to_count[j]; in SelectTrueCoords()1599 flat_index %= dims_to_count[j]; in SelectTrueCoords()