Searched refs:limit_group (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op.cc | 81 int64 start_group, int64 limit_group, Distribution dist) { in Run()
88 int64 limit_group_full = std::min(limit_group, size / kGroupSize); in Run()
96 if (limit_group_full < limit_group) { in Run()
112 int64 start_group, int64 limit_group, Distribution dist) { in Run()
122 int64 limit_group_full = std::min(limit_group, size / kGroupSize); in Run()
139 if (limit_group_full < limit_group) { in Run()
169 [&gen, data, size, dist](int64 start_group, int64 limit_group) { in operator ()() argument
174 limit_group, dist); in operator ()()
|