Searched refs:set_switch_input (Results 1 – 3 of 3) sorted by relevance
232 (*target_func_graph)->set_switch_input(func_graph->switch_input()); in SetFuncGraphInfo()756 new_func_graph->set_switch_input(func_graph->switch_input()); in TransformableClone()
395 void set_switch_input(std::shared_ptr<bool> switch_input) { switch_input_ = switch_input; } in set_switch_input() function
51 k_graph_->set_switch_input(primal_graph->switch_input()); in DFunctor()