Searched refs:batch_key (Results 1 – 1 of 1) sorted by relevance
578 const int64 batch_key = context->input(2).scalar<int64>()(); in Compute() local604 auto tensor_it = waiting_tensors_.find(batch_key); in Compute()617 .emplace(batch_key, in Compute()811 const int64 batch_key = context->input(3).scalar<int64>()(); in Compute() local813 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()828 const int64 batch_key = batch_index(i, 0); in Compute() local829 if (available_tensors_.find(batch_key) == available_tensors_.end()) { in Compute()830 missing_tensors.emplace(batch_key); in Compute()837 .emplace(batch_key, Batch{missing_tensors, context, done}) in Compute()843 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()[all …]