Searched refs:switch_layer_input (Results 1 – 6 of 6) sorted by relevance
60 *(abstract->func_graph()->switch_layer_input()) = true; in operator()
48 *(fg->switch_layer_input())) { in operator()
396 std::shared_ptr<bool> switch_layer_input() const { return switch_layer_input_; } in switch_layer_input() function397 …set_switch_layer_input(std::shared_ptr<bool> switch_layer_input) { switch_layer_input_ = switch_la… in set_switch_layer_input() argument
233 (*target_func_graph)->set_switch_layer_input(func_graph->switch_layer_input()); in SetFuncGraphInfo()757 new_func_graph->set_switch_layer_input(func_graph->switch_layer_input()); in TransformableClone()
584 if (*(func_graph->switch_layer_input())) { in ExportOneFuncGraph()585 ofs << "switch_layer_input: " << *(func_graph->switch_layer_input()) << "\n"; in ExportOneFuncGraph()
52 k_graph_->set_switch_layer_input(primal_graph->switch_layer_input()); in DFunctor()