-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsilicon_valley_2.cc
More file actions
31 lines (25 loc) · 1.01 KB
/
silicon_valley_2.cc
File metadata and controls
31 lines (25 loc) · 1.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
/* Silicon Valley s03e06 12:44
Sources:
- (fictional) https://github.com/Stitchpunk/atari-ai/blob/master/src/lib/common.cc
- https://github.com/tensorflow/tensorflow/blob/master/tensorflow/stream_executor/stream.cc
*/
namespace {
// Code to turn parameters to functions on stream into strings that
// will be VLOG'ed. We need overloads, instead of
// e.g. BatchDescriptorToVlogString(), as the code that calls these
// functions does not know what the type of the parameter is.
string ToVlogString(const dnn::BatchDescriptor &descriptor) {
return descriptor.ToShortString();
}
string ToVlogString(const dnn::FilterDescriptor &descriptor) {
return descriptor.ToShortString();
}
string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) {
return descriptor.ToShortString();
}
string ToVlogString(const dnn::PoolingDescriptor &descriptor) {
return descriptor.ToShortString();
}
string ToVlogString(const dnn::NormalizeDescriptor &descriptor) {
return descriptor.ToShortString();
}