Searched refs:current_is_one (Results 1 – 1 of 1) sorted by relevance
195 bool current_is_one[N]; in BCastList() local198 current_is_one[i] = false; in BCastList()213 current_is_one[i] = true; in BCastList()216 current_is_one[i] = false; in BCastList()257 std::equal(current_is_one, current_is_one + N, prev_is_one) && in BCastList()266 mul_dims(bcast_[i].back(), current_is_one[i] ? output_dim : 1); in BCastList()267 if (current_is_one[i] && !none_is_one) { in BCastList()275 bcast_[i].push_back(current_is_one[i] ? output_dim : 1); in BCastList()276 if (current_is_one[i] && !none_is_one) { in BCastList()283 prev_is_one[i] = current_is_one[i]; in BCastList()