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