repository

Contents

struct dnnl::ukernel::attr_params

Overview

Ukernel attributes memory storage. More…

#include <dnnl_ukernel.hpp>

struct attr_params: public dnnl::handle
{
    // methods

    void set_post_ops_args(const void** post_ops_args);
    void set_A_scales(const void* a_scales);
    void set_B_scales(const void* b_scales);
    void set_D_scales(const void* d_scales);
};

Inherited Members

public:
    // methods

    handle<T, traits>& operator = (const handle<T, traits>&);
    handle<T, traits>& operator = (handle<T, traits>&&);
    void reset(T t, bool weak = false);
    T get(bool allow_empty = false) const;
    operator T () const;
    operator bool () const;
    bool operator == (const handle<T, traits>& other) const;
    bool operator != (const handle& other) const;

Detailed Documentation

Ukernel attributes memory storage.

Methods

void set_post_ops_args(const void** post_ops_args)

Sets post-operations arguments to a storage.

Parameters:

post_ops_args

Pointer to pointers of post_ops storages. Expected to be packed together.

void set_A_scales(const void* a_scales)

Sets tensor A scales arguments to a storage.

Parameters:

a_scales

Pointer to scales storage.

void set_B_scales(const void* b_scales)

Sets tensor B scales arguments to a storage.

If attr_params::set_B_scales used mask of 2, then at least N values of selected data type are expected.

Parameters:

b_scales

Pointer to scales storage.

void set_D_scales(const void* d_scales)

Sets tensor D scales arguments to a storage.

Parameters:

d_scales

Pointer to scales storage.

enum dnnl::ukernel::pack_type enum dnnl_pack_type_t