Searched refs:tensor_tape (Results 1 – 1 of 1) sorted by relevance
524 gtl::ArraySlice<int64> target, const TensorTape& tensor_tape, in PrepareBackprop() argument536 auto op_id_it = tensor_tape.find(tensor_id); in PrepareBackprop()537 if (op_id_it == tensor_tape.end()) { in PrepareBackprop()554 if (tensor_tape.find(it) != tensor_tape.end()) { in PrepareBackprop()564 auto it = tensor_tape.find(pair.first); in PrepareBackprop()565 if (it != tensor_tape.end() && it->second != -1) { in PrepareBackprop()601 gtl::ArraySlice<Gradient*> output_gradients, const TensorTape& tensor_tape, in InitialGradients() argument607 auto tensor_it = tensor_tape.find(id); in InitialGradients()608 if (tensor_it != tensor_tape.end() && tensor_it->second != -1) { in InitialGradients()