Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dtopk_op.cc137 auto SortIndices = [&, context](int start_batch, int limit_batch) { in Compute() argument
138 for (int32 b = start_batch; b < limit_batch; ++b) { in Compute()
Dparameterized_truncated_normal_op.cc73 int limit_batch) { in operator ()() argument
93 for (int64 b = start_batch; b < limit_batch; ++b) { in operator ()()