public static class softmax_backward.primitive_desc extends primitive_desc
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
primitive_desc()
Default constructor.
|
primitive_desc(dnnl_primitive_desc pd)
Constructs a primitive descriptor for a softmax backward
propagation primitive from a C API primitive descriptor that must
have a matching kind.
|
primitive_desc(engine aengine,
dnnl.algorithm aalgorithm,
memory.desc diff_src_desc,
memory.desc diff_dst_desc,
memory.desc dst_desc,
int axis,
softmax_forward.primitive_desc hint_fwd_pd) |
primitive_desc(engine aengine,
dnnl.algorithm aalgorithm,
memory.desc diff_src_desc,
memory.desc diff_dst_desc,
memory.desc dst_desc,
int axis,
softmax_forward.primitive_desc hint_fwd_pd,
primitive_attr attr,
boolean allow_empty)
Constructs a primitive descriptor for a softmax backward propagation
primitive.
|
primitive_desc(engine aengine,
int aalgorithm,
memory.desc diff_src_desc,
memory.desc diff_dst_desc,
memory.desc dst_desc,
int axis,
softmax_forward.primitive_desc hint_fwd_pd) |
primitive_desc(engine aengine,
int aalgorithm,
memory.desc diff_src_desc,
memory.desc diff_dst_desc,
memory.desc dst_desc,
int axis,
softmax_forward.primitive_desc hint_fwd_pd,
primitive_attr attr,
boolean allow_empty) |
primitive_desc(long size)
Native array allocator.
|
primitive_desc(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
memory.desc |
diff_dst_desc()
\copydoc dnnl::primitive_desc_base::dst_desc()const
|
memory.desc |
diff_src_desc()
\copydoc dnnl::primitive_desc_base::diff_src_desc()const
|
memory.desc |
dst_desc()
\copydoc dnnl::primitive_desc_base::dst_desc()const
|
dnnl.algorithm |
get_algorithm()
\copydoc dnnl::primitive_desc_base::get_algorithm()const
|
int |
get_axis()
\copydoc dnnl::primitive_desc_base::get_axis()const
|
dnnl.prop_kind |
get_prop_kind()
\copydoc dnnl::primitive_desc_base::get_prop_kind()const
|
softmax_backward.primitive_desc |
getPointer(long i) |
softmax_backward.primitive_desc |
position(long position) |
next_impl
diff_dst_desc, diff_src_desc, diff_weights_desc, diff_weights_desc, dst_desc, get_activation_kind, get_alpha, get_beta, get_cache_blob_id, get_cell_kind, get_dilations, get_direction, get_engine, get_epsilon, get_factors, get_group_size, get_k, get_kernel, get_kind, get_local_size, get_p, get_padding_l, get_padding_r, get_primitive_attr, get_strides, impl_info_str, query_md, query_md, query_md, query_md, query_s64, query_s64, scratchpad_desc, scratchpad_engine, src_desc, src_desc, weights_desc, weights_desc, workspace_desc
asBoolean, asDnnl_primitive_desc, equals, get, get, notEquals, put, reset, reset
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public primitive_desc(Pointer p)
Pointer(Pointer)
.public primitive_desc(long size)
Pointer.position(long)
.public primitive_desc()
public primitive_desc(@Const @ByRef engine aengine, dnnl.algorithm aalgorithm, @Const @ByRef memory.desc diff_src_desc, @Const @ByRef memory.desc diff_dst_desc, @Const @ByRef memory.desc dst_desc, int axis, @Const @ByRef softmax_forward.primitive_desc hint_fwd_pd, @Const @ByRef(nullValue="dnnl::primitive_attr()") primitive_attr attr, @Cast(value="bool") boolean allow_empty)
aengine
- Engine to use.aalgorithm
- Softmax algorithm kind: either
#dnnl::algorithm::softmax_accurate,
or #dnnl::algorithm::softmax_log.diff_src_desc
- Diff source memory descriptor.diff_dst_desc
- Diff destination memory descriptor.dst_desc
- Destination memory descriptor.axis
- Axis over which softmax is computed.hint_fwd_pd
- Primitive descriptor for a softmax
forward propagation primitive. It is used as a hint for
deciding which memory format to use.attr
- Primitive attributes to use. Attributes are optional
and default to empty attributes.allow_empty
- A flag signifying whether construction is
allowed to fail without throwing an exception. In this case an
empty object will be produced. This flag is optional and
defaults to false.public primitive_desc(@Const @ByRef engine aengine, dnnl.algorithm aalgorithm, @Const @ByRef memory.desc diff_src_desc, @Const @ByRef memory.desc diff_dst_desc, @Const @ByRef memory.desc dst_desc, int axis, @Const @ByRef softmax_forward.primitive_desc hint_fwd_pd)
public primitive_desc(@Const @ByRef engine aengine, @Cast(value="dnnl::algorithm") int aalgorithm, @Const @ByRef memory.desc diff_src_desc, @Const @ByRef memory.desc diff_dst_desc, @Const @ByRef memory.desc dst_desc, int axis, @Const @ByRef softmax_forward.primitive_desc hint_fwd_pd, @Const @ByRef(nullValue="dnnl::primitive_attr()") primitive_attr attr, @Cast(value="bool") boolean allow_empty)
public primitive_desc(@Const @ByRef engine aengine, @Cast(value="dnnl::algorithm") int aalgorithm, @Const @ByRef memory.desc diff_src_desc, @Const @ByRef memory.desc diff_dst_desc, @Const @ByRef memory.desc dst_desc, int axis, @Const @ByRef softmax_forward.primitive_desc hint_fwd_pd)
public primitive_desc(dnnl_primitive_desc pd)
pd
- C API primitive descriptor for a softmax backward
propagation primitive.public softmax_backward.primitive_desc position(long position)
position
in class primitive_desc
public softmax_backward.primitive_desc getPointer(long i)
getPointer
in class primitive_desc
@ByVal public memory.desc dst_desc()
dst_desc
in class primitive_desc_base
@ByVal public memory.desc diff_src_desc()
diff_src_desc
in class primitive_desc_base
@ByVal public memory.desc diff_dst_desc()
diff_dst_desc
in class primitive_desc_base
public dnnl.algorithm get_algorithm()
get_algorithm
in class primitive_desc_base
public dnnl.prop_kind get_prop_kind()
get_prop_kind
in class primitive_desc_base
public int get_axis()
get_axis
in class primitive_desc_base
Copyright © 2024. All rights reserved.