1// This is used for convolution logging. Also see 2// tensorflow/core/protobuf/autotuing.h 3syntax = "proto3"; 4 5package tensorflow; 6 7import "tensorflow/stream_executor/dnn.proto"; 8 9option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto"; 10 11// A convolution. Currently it's only used for logging. In the future, we may 12// want to use it in the API as well. 13message ConvolutionProto { 14 stream_executor.dnn.ConvolutionKind kind = 1; 15 stream_executor.dnn.TensorDescriptorProto input = 2; 16 stream_executor.dnn.TensorDescriptorProto filter = 3; 17 stream_executor.dnn.TensorDescriptorProto output = 4; 18 stream_executor.dnn.ConvolutionDescriptorProto conv_desc = 5; 19 20 // result = conv_scale * conv(...) + side_value_scale * side_value. 21 // side_value is an arbitrary buffer if activation is not none. Otherwise, it 22 // has to be the result buffer (using its old values). 23 double conv_scale = 6; 24 double side_value_scale = 7; 25 26 stream_executor.dnn.ActivationMode activation = 8; 27 28 int64 input_address = 9; 29 int64 filter_address = 10; 30 int64 output_address = 11; 31 int64 bias_address = 12; 32 int64 side_input_address = 13; 33} 34