Searched refs:source_tensor_ids (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 176 const gtl::ArraySlice<int64> source_tensor_ids,
685 const gtl::ArraySlice<int64> source_tensor_ids, in ComputeGradient() argument
689 std::unordered_set<int64> sources_set(source_tensor_ids.begin(), in ComputeGradient()
690 source_tensor_ids.end()); in ComputeGradient()
866 if (result.size() != source_tensor_ids.size()) { in ComputeGradient()
868 source_tensor_ids.size(), " found ", result.size(), in ComputeGradient()
871 std::unordered_set<int64> used_gradient_ids(source_tensor_ids.size()); in ComputeGradient()
872 for (int i = 0; i < source_tensor_ids.size(); i++) { in ComputeGradient()
873 int64 tensor_id = source_tensor_ids[i]; in ComputeGradient()
|
| D | gradients.cc | 278 std::vector<int64> source_tensor_ids = MakeTensorIDList(sources); in ComputeGradient() local
280 source_tensor_ids.begin(), source_tensor_ids.end()); in ComputeGradient()
292 vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets, in ComputeGradient()
|