Searched refs:sources_that_are_targets (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 177 const std::unordered_map<int64, TapeTensor>& sources_that_are_targets,
600 const std::unordered_map<int64, TapeTensor>& sources_that_are_targets, in InitialGradients() argument
635 auto source_tensor = sources_that_are_targets.find(id); in InitialGradients()
636 if (source_tensor != sources_that_are_targets.end()) { in InitialGradients()
686 const std::unordered_map<int64, TapeTensor>& sources_that_are_targets, in ComputeGradient() argument
697 sources_that_are_targets, output_gradients, in ComputeGradient()
997 std::unordered_map<int64, TapeTensor> sources_that_are_targets; in ForwardpropFromTape() local
1004 sources_that_are_targets.emplace( in ForwardpropFromTape()
1017 sources_that_are_targets, used_in_grads, in ForwardpropFromTape()
|
| D | gradients.cc | 281 std::unordered_map<int64, TapeTensor> sources_that_are_targets; in ComputeGradient() local
286 sources_that_are_targets.insert( in ComputeGradient()
292 vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets, in ComputeGradient()
|