| 54 | } |
| 55 | |
| 56 | void add_dag_node_int_vector_params(DagNodeDef* node, |
| 57 | const std::string& name, |
| 58 | std::vector<int32_t> values) { |
| 59 | TensorValue* param = node->add_params(); |
| 60 | param->set_name(name); |
| 61 | param->set_dtype(kInt32); |
| 62 | param->set_length(values.size()); |
| 63 | for (auto& v: values) { |
| 64 | param->add_int32_values(v); |
| 65 | } |
| 66 | } |
| 67 | |
| 68 | void add_dag_node_float_vector_params(DagNodeDef* node, |
| 69 | const std::string& name, |
no test coverage detected