Searched refs:bcast_ (Results 1 – 1 of 1) sorted by relevance
95 const Vec& bcast(int i) const { return bcast_[i]; } in bcast()115 Vec bcast_[N]; variable169 bcast_[i].push_back(1); in BCastList()232 bcast_[i].push_back(1); in BCastList()265 bcast_[i].back() = in BCastList()266 mul_dims(bcast_[i].back(), current_is_one[i] ? output_dim : 1); in BCastList()275 bcast_[i].push_back(current_is_one[i] ? output_dim : 1); in BCastList()290 bcast_[i].push_back(1); in BCastList()296 Reverse(&bcast_[i]); in BCastList()307 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()[all …]