| 1160 | } |
| 1161 | |
| 1162 | void ConvertSplitVOperator(const Model& model, |
| 1163 | const TensorFlowSplitVOperator& src_op, |
| 1164 | GraphDef* tensorflow_graph) { |
| 1165 | tensorflow::NodeDef* split_v_op = tensorflow_graph->add_node(); |
| 1166 | split_v_op->set_op("SplitV"); |
| 1167 | split_v_op->set_name(src_op.outputs[0]); |
| 1168 | for (const auto& input : src_op.inputs) { |
| 1169 | *split_v_op->add_input() = input; |
| 1170 | } |
| 1171 | (*split_v_op->mutable_attr())["T"].set_type( |
| 1172 | GetTensorFlowDataType(model, src_op.outputs[0])); |
| 1173 | (*split_v_op->mutable_attr())["Tlen"].set_type( |
| 1174 | GetTensorFlowDataType(model, src_op.inputs[1])); |
| 1175 | (*split_v_op->mutable_attr())["num_split"].set_i(src_op.num_split); |
| 1176 | ConvertIntTensorConst(model, src_op.inputs[1], tensorflow_graph); |
| 1177 | } |
| 1178 | |
| 1179 | void ConvertCastOperator(const Model& model, const CastOperator& src_op, |
| 1180 | GraphDef* tensorflow_graph) { |
no test coverage detected