Home | History | Annotate | Download | only in framework 
      1 syntax = "proto3";
      2 
      3 package tensorflow;
      4 option cc_enable_arenas = true;
      5 option java_outer_classname = "DeviceAttributesProtos";
      6 option java_multiple_files = true;
      7 option java_package = "org.tensorflow.framework";
      8 
      9 message InterconnectLink {
     10   int32 device_id = 1;
     11   string type = 2;
     12   int32 strength = 3;
     13 };
     14 
     15 message LocalLinks {
     16   repeated InterconnectLink link = 1;
     17 };
     18 
     19 message DeviceLocality {
     20   // Optional bus locality of device.  Default value of 0 means
     21   // no specific locality.  Specific localities are indexed from 1.
     22   int32 bus_id = 1;
     23 
     24   // Optional NUMA locality of device.
     25   int32 numa_node = 2;
     26 
     27   // Optional local interconnect links to other devices.
     28   LocalLinks links = 3;
     29 };
     30 
     31 message DeviceAttributes {
     32   // Fully specified name of the device within a cluster.
     33   string name = 1;
     34 
     35   // String representation of device_type.
     36   string device_type = 2;
     37 
     38   // Memory capacity of device in bytes.
     39   int64 memory_limit = 4;
     40 
     41   // Platform-specific data about device that may be useful
     42   // for supporting efficient data transfers.
     43   DeviceLocality locality = 5;
     44 
     45   // A device is assigned a global unique number each time it is
     46   // initialized. "incarnation" should never be 0.
     47   fixed64 incarnation = 6;
     48 
     49   // String representation of the physical device that this device maps to.
     50   string physical_device_desc = 7;
     51 }
     52