Home | History | Annotate | Download | only in example 
      1 // Protocol messages for describing the configuration of the ExampleParserOp.
      2 
      3 syntax = "proto3";
      4 
      5 option cc_enable_arenas = true;
      6 option java_outer_classname = "ExampleParserConfigurationProtos";
      7 option java_multiple_files = true;
      8 option java_package = "org.tensorflow.example";
      9 package tensorflow;
     10 
     11 import "tensorflow/core/framework/tensor_shape.proto";
     12 import "tensorflow/core/framework/tensor.proto";
     13 import "tensorflow/core/framework/types.proto";
     14 
     15 message VarLenFeatureProto {
     16   tensorflow.DataType dtype = 1;
     17   string values_output_tensor_name = 2;
     18   string indices_output_tensor_name = 3;
     19   string shapes_output_tensor_name = 4;
     20 };
     21 
     22 message FixedLenFeatureProto {
     23   tensorflow.DataType dtype = 1;
     24   tensorflow.TensorShapeProto shape = 2;
     25   tensorflow.TensorProto default_value = 3;
     26   string values_output_tensor_name = 4;
     27 };
     28 
     29 message FeatureConfiguration {
     30   oneof config {
     31     FixedLenFeatureProto fixed_len_feature = 1;
     32     VarLenFeatureProto var_len_feature = 2;
     33   }
     34 };
     35 
     36 message ExampleParserConfiguration {
     37   map<string, FeatureConfiguration> feature_map = 1;
     38 };
     39