Searched refs:switch_input (Results 1 – 9 of 9) sorted by relevance
38 *(true_abstract->func_graph()->switch_input()) = true; in operator()42 *(false_abstract->func_graph()->switch_input()) = true; in operator()
47 if (fg->has_flag(FUNC_GRAPH_FLAG_DEFER_INLINE) || fg->stub() || *(fg->switch_input()) || in operator()
121 auto switch_input = utils::cast<CNodePtr>(switch_node->input(1)); in IdentifySubgraphInput() local122 auto pos = PosInInputNodes(switch_input); in IdentifySubgraphInput()
394 std::shared_ptr<bool> switch_input() const { return switch_input_; } in switch_input() function395 void set_switch_input(std::shared_ptr<bool> switch_input) { switch_input_ = switch_input; } in set_switch_input() argument
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()
581 if (*(func_graph->switch_input())) { in ExportOneFuncGraph()582 ofs << "switch_input: " << *(func_graph->switch_input()) << "\n"; in ExportOneFuncGraph()
716 auto switch_input = NodeInputIsSwitch(node); in InferCallShape() local717 if (switch_input) { in InferCallShape()718 return InferSwitchShape(switch_input); in InferCallShape()
51 k_graph_->set_switch_input(primal_graph->switch_input()); in DFunctor()
1008 auto switch_input = CreateSwitchInput(cnode, node_input, graph); in CreateCNodeInputs() local1009 (void)cnode_inputs->emplace_back(switch_input); in CreateCNodeInputs()