Halide
Halide::Internal::Weights Struct Reference

#include <Weights.h>

Public Member Functions

template<typename F >
void for_each_buffer (F f)
 
void randomize (uint32_t seed)
 
bool load (std::istream &i)
 
bool save (std::ostream &o) const
 
bool load_from_file (const std::string &filename)
 
bool save_to_file (const std::string &filename) const
 
bool load_from_dir (const std::string &dir)
 
bool save_to_dir (const std::string &dir) const
 

Public Attributes

uint32_t pipeline_features_version = PipelineFeatures::version()
 
uint32_t schedule_features_version = ScheduleFeatures::version()
 
Halide::Runtime::Buffer< float > head1_filter {head1_channels, head1_w, head1_h}
 
Halide::Runtime::Buffer< float > head1_bias {head1_channels}
 
Halide::Runtime::Buffer< float > head2_filter {head2_channels, head2_w}
 
Halide::Runtime::Buffer< float > head2_bias {head2_channels}
 
Halide::Runtime::Buffer< float > conv1_filter {conv1_channels, head1_channels + head2_channels}
 
Halide::Runtime::Buffer< float > conv1_bias {conv1_channels}
 

Detailed Description

Definition at line 15 of file Weights.h.

Member Function Documentation

◆ for_each_buffer()

template<typename F >
void Halide::Internal::Weights::for_each_buffer ( f)
inline

Definition at line 29 of file Weights.h.

References conv1_bias, conv1_filter, head1_bias, head1_filter, head2_bias, and head2_filter.

◆ randomize()

void Halide::Internal::Weights::randomize ( uint32_t  seed)

◆ load()

bool Halide::Internal::Weights::load ( std::istream &  i)

◆ save()

bool Halide::Internal::Weights::save ( std::ostream &  o) const

◆ load_from_file()

bool Halide::Internal::Weights::load_from_file ( const std::string &  filename)

◆ save_to_file()

bool Halide::Internal::Weights::save_to_file ( const std::string &  filename) const

◆ load_from_dir()

bool Halide::Internal::Weights::load_from_dir ( const std::string &  dir)

◆ save_to_dir()

bool Halide::Internal::Weights::save_to_dir ( const std::string &  dir) const

Member Data Documentation

◆ pipeline_features_version

uint32_t Halide::Internal::Weights::pipeline_features_version = PipelineFeatures::version()

Definition at line 16 of file Weights.h.

◆ schedule_features_version

uint32_t Halide::Internal::Weights::schedule_features_version = ScheduleFeatures::version()

Definition at line 17 of file Weights.h.

◆ head1_filter

Halide::Runtime::Buffer<float> Halide::Internal::Weights::head1_filter {head1_channels, head1_w, head1_h}

Definition at line 19 of file Weights.h.

Referenced by for_each_buffer().

◆ head1_bias

Halide::Runtime::Buffer<float> Halide::Internal::Weights::head1_bias {head1_channels}

Definition at line 20 of file Weights.h.

Referenced by for_each_buffer().

◆ head2_filter

Halide::Runtime::Buffer<float> Halide::Internal::Weights::head2_filter {head2_channels, head2_w}

Definition at line 22 of file Weights.h.

Referenced by for_each_buffer().

◆ head2_bias

Halide::Runtime::Buffer<float> Halide::Internal::Weights::head2_bias {head2_channels}

Definition at line 23 of file Weights.h.

Referenced by for_each_buffer().

◆ conv1_filter

Halide::Runtime::Buffer<float> Halide::Internal::Weights::conv1_filter {conv1_channels, head1_channels + head2_channels}

Definition at line 25 of file Weights.h.

Referenced by for_each_buffer().

◆ conv1_bias

Halide::Runtime::Buffer<float> Halide::Internal::Weights::conv1_bias {conv1_channels}

Definition at line 26 of file Weights.h.

Referenced by for_each_buffer().


The documentation for this struct was generated from the following file: