Searched refs:sources_set (Results 1 – 2 of 2) sorted by relevance
326 const gtl::FlatSet<int64>& sources_set, bool persistent_tape) { in PrepareBackprop() argument482 gtl::FlatSet<int64> sources_set(source_tensor_ids.begin(), in ComputeGradient()485 target_tensor_ids, tensor_tape_, &op_tape_, sources_set, persistent_); in ComputeGradient()548 if (sources_set.find(grad_it->first) == sources_set.end()) { in ComputeGradient()
1698 tensorflow::gtl::FlatSet<tensorflow::int64> sources_set(sources_vec.begin(), local1708 if (sources_set.find(target_id) != sources_set.end()) {