alanlxl | 30f15bd | 2020-08-11 21:26:12 +1000 | [diff] [blame^] | 1 | // Copyright 2020 The Chromium OS Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef FEDERATED_UTILS_H_ |
| 6 | #define FEDERATED_UTILS_H_ |
| 7 | |
| 8 | #include "chrome/knowledge/federated/example.pb.h" |
| 9 | #include "chrome/knowledge/federated/feature.pb.h" |
| 10 | #include "federated/mojom/example.mojom.h" |
| 11 | |
| 12 | namespace federated { |
| 13 | |
| 14 | // Converts the mojom Example struct to a TensorFlow Example proto. |
| 15 | tensorflow::Example ConvertToTensorFlowExampleProto( |
| 16 | const chromeos::federated::mojom::ExamplePtr& example); |
| 17 | |
| 18 | } // namespace federated |
| 19 | |
| 20 | #endif // FEDERATED_UTILS_H_ |