diff options
Diffstat (limited to 'tensorflow/core/example/example_parser_configuration_test.cc')
-rw-r--r-- | tensorflow/core/example/example_parser_configuration_test.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/example/example_parser_configuration_test.cc b/tensorflow/core/example/example_parser_configuration_test.cc index 6206d2bd51..647f727d83 100644 --- a/tensorflow/core/example/example_parser_configuration_test.cc +++ b/tensorflow/core/example/example_parser_configuration_test.cc @@ -14,7 +14,8 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/core/example/example_parser_configuration.h" -#include "tensorflow/core/example/example.pb.h" +#include "tensorflow/core/framework/attr_value.pb.h" +#include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/tensor_testutil.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/lib/io/path.h" |