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