#include <mkldnn.hpp>

Inheritance diagram for mkldnn::primitive_attr:

Collaboration diagram for mkldnn::primitive_attr:

Public Member Functions
	primitive_attr ()

round_mode	get_int_output_round_mode () const

void	set_int_output_round_mode (round_mode mode)

void	get_output_scales (int &mask, std::vector< float > &scales) const

void	set_output_scales (int mask, const std::vector< float > &scales)

const post_ops	get_post_ops () const

void	set_post_ops (post_ops ops)

void	set_rnn_data_qparams (const float scale, const float shift)

void	set_rnn_weights_qparams (int mask, const std::vector< float > &scales)

Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_attr_t >
	handle (mkldnn_primitive_attr_t t=0, bool weak=false)
	Constructs a C handle wrapper. More...

	handle (const handle &other)

handle &	operator= (const handle &other)

bool	operator== (const handle &other) const

bool	operator!= (const handle &other) const

void	reset (mkldnn_primitive_attr_t t, bool weak=false)
	Resets the value of a C handle. More...

mkldnn_primitive_attr_t	get () const
	Returns the value of the underlying C handle. More...

Additional Inherited Members
Protected Member Functions inherited from mkldnn::handle< mkldnn_primitive_attr_t >
bool	operator== (const mkldnn_primitive_attr_t other) const

bool	operator!= (const mkldnn_primitive_attr_t other) const

Constructor & Destructor Documentation

◆ primitive_attr()

mkldnn::primitive_attr::primitive_attr ( )

inline

round_mode mkldnn::primitive_attr::get_int_output_round_mode ( ) const

inline

void mkldnn::primitive_attr::set_int_output_round_mode ( round_mode mode )

inline

void mkldnn::primitive_attr::get_output_scales	(	int &	mask,
		std::vector< float > &	scales
	)		const

inline

void mkldnn::primitive_attr::set_output_scales	(	int	mask,
		const std::vector< float > &	scales
	)

inline

const post_ops mkldnn::primitive_attr::get_post_ops ( ) const

inline

void mkldnn::primitive_attr::set_post_ops ( post_ops ops )

inline

void mkldnn::primitive_attr::set_rnn_data_qparams	(	const float	scale,
		const float	shift
	)

inline

void mkldnn::primitive_attr::set_rnn_weights_qparams	(	int	mask,
		const std::vector< float > &	scales
	)

inline

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