1// This proto describes the types of hardware profiled by the TensorFlow
2// profiler.
3syntax = "proto3";
4
5package tensorflow.profiler;
6
7// Types of hardware profiled.
8enum HardwareType {
9  // Unknown hardware.
10  UNKNOWN_HARDWARE = 0;
11  // CPU only without any hardware accelerator.
12  CPU_ONLY = 1;
13  // GPU.
14  GPU = 2;
15  // TPU.
16  TPU = 3;
17}
18
19message CudaComputeCapability {
20  uint32 major = 1;
21  uint32 minor = 2;
22}
23
24message DeviceCapabilities {
25  double clock_rate_in_ghz = 1;
26  uint32 num_cores = 2;
27  uint64 memory_size_in_bytes = 3;
28  uint64 memory_bandwidth = 4;  // Bytes/s.
29  CudaComputeCapability compute_capability = 5;
30}
31