Searched refs:AddControlInputs (Results 1 – 2 of 2) sorted by relevance
88 void AddControlInputs(const Node& node, absl::flat_hash_set<Node*>* deps) { in AddControlInputs() function279 AddControlInputs(*launch, &control_inputs); in BuildXlaLaunchOps()
1503 AddControlInputs(ctrl_inputs, root_node); in HoistUnaryOpChain()1522 void AddControlInputs(std::set<string>* new_ctrl_inputs, in AddControlInputs() function in tensorflow::grappler::__anonbb66c6470111::HoistCWiseUnaryChainsStage1689 AddControlInputs(ctrl_inputs, cur_copy); in HoistChainForSplit()