Searched refs:sources_set (Results 1 – 3 of 3) sorted by relevance
526 const std::unordered_set<int64_t>& sources_set, bool persistent_tape) { in PrepareBackprop() argument689 std::unordered_set<int64_t> 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_t> sources_set; in ForwardpropFromTape() local969 sources_set.insert(aid_id); in ForwardpropFromTape()1003 if (sources_set.find(tensor_id) != sources_set.end()) { in ForwardpropFromTape()
280 tensorflow::gtl::FlatSet<int64_t> sources_set(source_tensor_ids.begin(), in ComputeGradient() local285 if (sources_set.find(target_id) != sources_set.end()) { in ComputeGradient()
2824 tensorflow::gtl::FlatSet<int64_t> sources_set(sources_vec.begin(), local2834 if (sources_set.find(target_id) != sources_set.end()) {