Searched defs:batch_key (Results 1 – 1 of 1) sorted by relevance
578 const int64 batch_key = context->input(2).scalar<int64>()(); in Compute() local811 const int64 batch_key = context->input(3).scalar<int64>()(); in Compute() local828 const int64 batch_key = batch_index(i, 0); in Compute() local