Caffe2 - C++ API: GLSoftmaxScale Class Reference

Inheritance diagram for GLSoftmaxScale:

Public Member Functions
const std::vector< binding * >	input_bindings ()

	GLSoftmaxScale (bool _compute_exp=false, bool tiled=false)

template<typename T >
void	scale (const GLImage< T > input_image, const GLImage< T > max_image, const GLImage< T > sum_image, const GLImage< T > output_image)

Public Member Functions inherited from GLFilter
	GLFilter (const std::string kernel_name, const std::string vertex_shader, const std::string fragment_shader, const std::vector< binding * > uniforms, const std::vector< binding * > uniform_blocks={}, const std::vector< binding * > attributes={}, const replacements_t &replacements={})

void	throwRuntimeError (std::function< void(std::stringstream &errmsg)> error_formatter) const

void	checkGLError (std::function< void(std::stringstream &errmsg)> error_formatter) const

template<typename T >
void	attach_uniform_buffer (const binding block, GLuint bindingPoint, std::function< void(T , size_t)> loader)

void	run (const std::vector< texture_attachment > &input, const std::vector< const GLTexture * > &output, std::function< void(void)> uniforms_initializer, int width, int height)

void	releaseBuffers ()

void	deleteProgram ()

void	deleteBindings ()

Static Public Attributes
static const char *	fragment_shader

Static Public Attributes inherited from GLFilter
static const char *	vertex_shader

Additional Inherited Members
Public Types inherited from GLFilter
typedef std::vector< std::pair< std::string, std::string > >	replacements_t

Protected Attributes inherited from GLFilter
const std::string	kernel_name

GLuint	program = 0

GLuint	frameBuffer = 0

GLuint	uniformBlock [kMaxUniformBlocks] = {0}

GLint	blockSize [kMaxUniformBlocks] = {0}

bool	frame_buffer_initialized = false

Static Protected Attributes inherited from GLFilter
static constexpr int	kMaxUniformBlocks = 12

static constexpr bool	check_opengl_errors = true

Detailed Description

Definition at line 172 of file GLSoftmax.cc.

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