Searched refs:sources_set (Results 1 – 3 of 3) sorted by relevance
526 const std::unordered_set<int64>& sources_set, bool persistent_tape) { in PrepareBackprop() argument689 std::unordered_set<int64> sources_set(source_tensor_ids.begin(), in ComputeGradient()692 target_tensor_ids, tensor_tape_, &op_tape_, sources_set, persistent_); in ComputeGradient()738 sources_set.find(in_tensor_id) == sources_set.end()) { in ComputeGradient()766 if (sources_set.find(grad_it->first) == sources_set.end()) { in ComputeGradient()946 std::unordered_set<int64> sources_set; in ForwardpropFromTape() local969 sources_set.insert(aid_id); in ForwardpropFromTape()1003 if (sources_set.find(tensor_id) != sources_set.end()) { in ForwardpropFromTape()
279 tensorflow::gtl::FlatSet<tensorflow::int64> sources_set( in ComputeGradient() local284 if (sources_set.find(target_id) != sources_set.end()) { in ComputeGradient()
2746 tensorflow::gtl::FlatSet<tensorflow::int64> sources_set(sources_vec.begin(), local2757 if (sources_set.find(target_id) != sources_set.end()) {