Searched defs:switch_input (Results 1 – 4 of 4) sorted by relevance
121 auto switch_input = utils::cast<CNodePtr>(switch_node->input(1)); in IdentifySubgraphInput() local
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()
716 auto switch_input = NodeInputIsSwitch(node); in InferCallShape() local
1008 auto switch_input = CreateSwitchInput(cnode, node_input, graph); in CreateCNodeInputs() local