| 1159 | } |
| 1160 | |
| 1161 | tensorflow::Status ConvertSwitchOperator( |
| 1162 | const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, |
| 1163 | const ModelFlags& model_flags, Model* model) { |
| 1164 | CHECK_EQ(node.op(), "Switch"); |
| 1165 | TF_QCHECK_OK(CheckInputsCount(node, tf_import_flags, 2)); |
| 1166 | auto* op = new TensorFlowSwitchOperator; |
| 1167 | op->inputs.push_back(node.input(0)); |
| 1168 | op->inputs.push_back(node.input(1)); |
| 1169 | op->outputs.push_back(node.name()); |
| 1170 | // Switch operators have two outputs: "name" and "name:1". |
| 1171 | op->outputs.push_back(node.name() + ":1"); |
| 1172 | model->operators.emplace_back(op); |
| 1173 | return tensorflow::Status::OK(); |
| 1174 | } |
| 1175 | |
| 1176 | tensorflow::Status ConvertSoftmaxOperator( |
| 1177 | const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, |
nothing calls this directly
no test coverage detected