Searched refs:tensor_it (Results 1 – 2 of 2) sorted by relevance
407 auto tensor_it = tensor_tape.find(id); in InitialGradients() local408 if (tensor_it != tensor_tape.end() && tensor_it->second != -1) { in InitialGradients()409 auto op_it = op_tape.find(tensor_it->second); in InitialGradients()
935 auto tensor_it = waiting_tensors_.find(batch_key); in Compute() local936 if (tensor_it != waiting_tensors_.end()) { in Compute()937 context->set_output(0, tensor_it->second.tensor); in Compute()938 waiting_tensors_.erase(tensor_it); in Compute()