Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
Visible to Intel only — GUID: GUID-FEA0E010-ADBA-4ED5-A07B-0FFAC96C5A13
Visible to Intel only — GUID: GUID-FEA0E010-ADBA-4ED5-A07B-0FFAC96C5A13
struct dnnl::softmax_forward
Overview
Softmax forward propagation primitive. More…
#include <dnnl.hpp> struct softmax_forward: public dnnl::primitive { // structs struct primitive_desc; // construction softmax_forward(); softmax_forward(const primitive_desc& pd); softmax_forward( const primitive_desc& pd, const std::vector<uint8_t>& cache_blob ); };
Inherited Members
public:
// enums
enum kind;
// 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;
const_dnnl_primitive_desc_t get_primitive_desc() const;
kind get_kind() const;
std::vector<uint8_t> get_cache_blob() const;
void execute(const stream& astream, const std::unordered_map<int, memory>& args) const;
handle();
handle(const handle<T, traits>&);
handle(handle<T, traits>&&);
handle(T t, bool weak = false);
Detailed Documentation
Softmax forward propagation primitive.
Construction
softmax_forward()
Default constructor. Produces an empty object.
softmax_forward(const primitive_desc& pd)
Constructs a softmax forward propagation primitive.
Parameters:
pd |
Primitive descriptor for a softmax forward propagation primitive. |
softmax_forward(
const primitive_desc& pd,
const std::vector<uint8_t>& cache_blob
)
Constructs a softmax forward propagation primitive from a cache blob.
Parameters:
pd |
Primitive descriptor for a softmax forward propagation primitive. |
cache_blob |
Cache blob. |