Searched refs:tensor_tape (Results 1 – 1 of 1) sorted by relevance
324 gtl::ArraySlice<int64> target, const TensorTape& tensor_tape, in PrepareBackprop() argument336 auto op_id_it = tensor_tape.find(tensor_id); in PrepareBackprop()337 if (op_id_it == tensor_tape.end()) { in PrepareBackprop()354 if (tensor_tape.find(it) != tensor_tape.end()) { in PrepareBackprop()364 auto it = tensor_tape.find(pair.first); in PrepareBackprop()365 if (it != tensor_tape.end() && it->second != -1) { in PrepareBackprop()401 gtl::ArraySlice<Gradient*> output_gradients, const TensorTape& tensor_tape, in InitialGradients() argument407 auto tensor_it = tensor_tape.find(id); in InitialGradients()408 if (tensor_it != tensor_tape.end() && tensor_it->second != -1) { in InitialGradients()