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