| 1122 | } |
| 1123 | |
| 1124 | tensorflow::Status ConvertSplitOperator( |
| 1125 | const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, |
| 1126 | const ModelFlags& model_flags, Model* model) { |
| 1127 | CHECK_EQ(node.op(), "Split"); |
| 1128 | TF_QCHECK_OK(CheckInputsCount(node, tf_import_flags, 2)); |
| 1129 | auto* op = new TensorFlowSplitOperator; |
| 1130 | op->inputs.push_back(node.input(0)); |
| 1131 | op->inputs.push_back(node.input(1)); |
| 1132 | const int num_split = GetIntAttr(node, "num_split"); |
| 1133 | op->outputs.push_back(node.name()); |
| 1134 | for (int i = 1; i < num_split; i++) { |
| 1135 | op->outputs.push_back(absl::StrCat(node.name(), ":", i)); |
| 1136 | } |
| 1137 | op->num_split = num_split; |
| 1138 | model->operators.emplace_back(op); |
| 1139 | return tensorflow::Status::OK(); |
| 1140 | } |
| 1141 | |
| 1142 | tensorflow::Status ConvertSplitVOperator( |
| 1143 | const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, |
nothing calls this directly
no test coverage detected