Home
last modified time | relevance | path

Searched refs:largest_rank (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dbcast.h142 size_t largest_rank = 0; in BCastList() local
148 if (x[i].size() > largest_rank) { in BCastList()
149 largest_rank = x[i].size(); in BCastList()
185 if (copy[i].size() < largest_rank) { in BCastList()
186 copy[i].resize(largest_rank, 1); in BCastList()
205 for (int j = 0; j < largest_rank; ++j) { in BCastList()
238 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()
268 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()
277 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()