| 195 | } |
| 196 | |
| 197 | static std::vector<size_t> parse_dims(const tensorflow::TensorShapeProto& s) |
| 198 | { |
| 199 | std::vector<size_t> dims; |
| 200 | auto input_dims = s.dim(); |
| 201 | std::transform(input_dims.begin(), |
| 202 | input_dims.end(), |
| 203 | std::back_inserter(dims), |
| 204 | [](const tensorflow::TensorShapeProto_Dim& dim) { return dim.size(); }); |
| 205 | return dims; |
| 206 | } |
| 207 | |
| 208 | template <class T> |
| 209 | static std::vector<T> get_data_vals(const google::protobuf::RepeatedField<T>& data, |
no test coverage detected