Searched refs:source_tensor_ids (Results 1 – 1 of 1) sorted by relevance
145 const gtl::ArraySlice<int64> source_tensor_ids,478 const gtl::ArraySlice<int64> source_tensor_ids, in ComputeGradient() argument482 gtl::FlatSet<int64> sources_set(source_tensor_ids.begin(), in ComputeGradient()483 source_tensor_ids.end()); in ComputeGradient()645 result->reserve(source_tensor_ids.size()); in ComputeGradient()646 gtl::FlatSet<int64> used_gradient_ids(source_tensor_ids.size()); in ComputeGradient()647 for (auto is : source_tensor_ids) { in ComputeGradient()