Searched refs:target_tensor_ids (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients.cc | 277 std::vector<int64> target_tensor_ids = MakeTensorIDList(targets); in ComputeGradient() local
282 for (int i = 0; i < target_tensor_ids.size(); ++i) { in ComputeGradient()
283 int64 target_id = target_tensor_ids[i]; in ComputeGradient()
292 vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets, in ComputeGradient()
|
| D | tape.h | 175 const gtl::ArraySlice<int64> target_tensor_ids,
599 gtl::ArraySlice<int64> target_tensor_ids, in InitialGradients() argument
604 for (int i = 0, end = target_tensor_ids.size(); i < end; ++i) { in InitialGradients()
605 const int64 id = target_tensor_ids[i]; in InitialGradients()
684 const gtl::ArraySlice<int64> target_tensor_ids, in ComputeGradient() argument
692 target_tensor_ids, tensor_tape_, &op_tape_, sources_set, persistent_); in ComputeGradient()
696 Status s = InitialGradients(vspace, target_tensor_ids, in ComputeGradient()
|