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