Halide
NetworkSize.h
Go to the documentation of this file.
1 #ifndef HALIDE_NETWORK_SIZE_H
2 #define HALIDE_NETWORK_SIZE_H
3 
4 namespace Halide {
5 // The size of the best cost model network found. Needed by the cost
6 // model and also the cost model training script.
7 const int head1_channels = 8, head1_w = 40, head1_h = 7;
8 const int head2_channels = 24, head2_w = 39;
9 const int conv1_channels = 32;
10 } // namespace Halide
11 
12 #endif // HALIDE_NETWORK_SIZE_H
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AddAtomicMutex.h:21
Halide::head2_channels
const int head2_channels
Definition: NetworkSize.h:8
Halide::conv1_channels
const int conv1_channels
Definition: NetworkSize.h:9
Halide::head1_channels
const int head1_channels
Definition: NetworkSize.h:7
Halide::head1_h
const int head1_h
Definition: NetworkSize.h:7
Halide::head1_w
const int head1_w
Definition: NetworkSize.h:7
Halide::head2_w
const int head2_w
Definition: NetworkSize.h:8