Searched refs:remain_flat_size (Results 1 – 1 of 1) sorted by relevance
2992 int remain_flat_size = params_shape.FlatSize(); in GatherNd() local2995 dims_to_count[i] = remain_flat_size / params_shape.Dims(i); in GatherNd()2996 remain_flat_size = dims_to_count[i]; in GatherNd()