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