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
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
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
src
autoschedulers
adams2019
NetworkSize.h
Generated by
1.12.0