Halide 22.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
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 = 73;
9 const int conv1_channels = 32; // Only 30 are used (needs to be a multiple of 8 for vectorization in cost_model_generator.cpp)
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 AbstractGenerator.h:19
Halide::head2_w
const int head2_w
Definition NetworkSize.h:8
Halide::head2_channels
const int head2_channels
Definition NetworkSize.h:8
Halide::head1_w
const int head1_w
Definition NetworkSize.h:7
Halide::head1_channels
const int head1_channels
Definition NetworkSize.h:7
Halide::conv1_channels
const int conv1_channels
Definition NetworkSize.h:9
Halide::head1_h
const int head1_h
Definition NetworkSize.h:7