1/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2 3Licensed under the Apache License, Version 2.0 (the "License"); 4you may not use this file except in compliance with the License. 5You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9Unless required by applicable law or agreed to in writing, software 10distributed under the License is distributed on an "AS IS" BASIS, 11WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12See the License for the specific language governing permissions and 13limitations under the License. 14==============================================================================*/ 15 16syntax = "proto3"; 17 18package tensorflow; 19 20option cc_enable_arenas = true; 21option java_outer_classname = "DevicePropertiesProtos"; 22option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto"; 23 24message DeviceProperties { 25 // Device type (CPU, GPU, ...) 26 string type = 1; 27 // Vendor (Intel, nvidia, ...) 28 string vendor = 2; 29 // Model (Haswell, K40, ...) 30 string model = 3; 31 // Core Frequency in Mhz 32 int64 frequency = 4; 33 // Number of cores 34 int64 num_cores = 5; 35 // Version of the tools and libraries used with this device (e.g. gcc 4.9, 36 // cudnn 5.1) 37 map<string, string> environment = 6; 38 // Number of registers per core. 39 int64 num_registers = 7; 40 // L1 cache size in bytes 41 int64 l1_cache_size = 8; 42 // L2 cache size in bytes 43 int64 l2_cache_size = 9; 44 // L3 cache size in bytes 45 int64 l3_cache_size = 10; 46 // Shared memory size per multiprocessor in bytes. This field is 47 // applicable to GPUs only. 48 int64 shared_memory_size_per_multiprocessor = 11; 49 // Memory size in bytes 50 int64 memory_size = 12; 51 // Memory bandwidth in KB/s 52 int64 bandwidth = 13; 53} 54 55message NamedDevice { 56 string name = 1; 57 DeviceProperties properties = 2; 58} 59