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