Searched refs:dims_to_count (Results 1 – 1 of 1) sorted by relevance
1064 std::vector<int> dims_to_count; member1084 ret.dims_to_count = std::vector<int>(ret.indices_nd, 0); in GatherNdHelper()1086 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()1105 from_pos += indices_data[i * res.indices_nd + j] * res.dims_to_count[j]; in GatherNd()1127 from_pos += indices_data[i * res.indices_nd + j] * res.dims_to_count[j]; in GatherNdString()1159 std::vector<int> dims_to_count(indices_nd, 0); in ScatterNd()1161 dims_to_count[i] = remain_flat_size / output_shape.Dims(i); in ScatterNd()1162 remain_flat_size = dims_to_count[i]; in ScatterNd()1171 to_pos += idx * dims_to_count[j]; in ScatterNd()[all …]