Searched refs:remain_flat_size (Results 1 – 1 of 1) sorted by relevance
1083 int remain_flat_size = params_shape.FlatSize(); in GatherNdHelper() local1086 ret.dims_to_count[i] = remain_flat_size / params_shape.Dims(i); in GatherNdHelper()1087 remain_flat_size = ret.dims_to_count[i]; in GatherNdHelper()1158 int remain_flat_size = output_flat_size; in ScatterNd() local1161 dims_to_count[i] = remain_flat_size / output_shape.Dims(i); in ScatterNd()1162 remain_flat_size = dims_to_count[i]; in ScatterNd()