Searched refs:batch_key (Results 1 – 1 of 1) sorted by relevance
899 const int64 batch_key = context->input(2).scalar<int64>()(); in Compute() local935 auto tensor_it = waiting_tensors_.find(batch_key); in Compute()948 .emplace(batch_key, in Compute()1141 const int64 batch_key = context->input(3).scalar<int64>()(); in Compute() local1143 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()1158 const int64 batch_key = batch_index(i, 0); in Compute() local1159 if (available_tensors_.find(batch_key) == available_tensors_.end()) { in Compute()1160 missing_tensors.emplace(batch_key); in Compute()1167 .emplace(batch_key, Batch{missing_tensors, context, done}) in Compute()1173 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()[all …]