Searched refs:sources_that_are_targets (Results 1 – 1 of 1) sorted by relevance
146 const gtl::FlatMap<int64, TapeTensor> sources_that_are_targets,400 gtl::FlatMap<int64, TapeTensor> sources_that_are_targets, in InitialGradients() argument433 auto source_tensor = sources_that_are_targets.find(id); in InitialGradients()434 if (source_tensor != sources_that_are_targets.end()) { in InitialGradients()479 const gtl::FlatMap<int64, TapeTensor> sources_that_are_targets, in ComputeGradient() argument490 sources_that_are_targets, output_gradients, in ComputeGradient()