| 66 | } |
| 67 | |
| 68 | void add_dag_node_float_vector_params(DagNodeDef* node, |
| 69 | const std::string& name, |
| 70 | std::vector<float> values) { |
| 71 | TensorValue* param = node->add_params(); |
| 72 | param->set_name(name); |
| 73 | param->set_dtype(kFloat); |
| 74 | param->set_length(values.size()); |
| 75 | for (auto& v: values) { |
| 76 | param->add_float_values(v); |
| 77 | } |
| 78 | } |
| 79 | |
| 80 | void add_dag_node_in_edge(DagNodeDef* node, const DagEdgeDef* edge) { |
| 81 | DagEdgeDef* e = node->add_in_edges(); |
no test coverage detected