org.bytedeco.caffe

Class FloatCuDNNSoftmaxLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.caffe.FloatLayer

org.bytedeco.caffe.FloatSoftmaxLayer

org.bytedeco.caffe.FloatCuDNNSoftmaxLayer

All Implemented Interfaces:

AutoCloseable

@Platform(value={"linux-x86_64","macosx-x86_64"},
          extension="-gpu")
 @Name(value="caffe::CuDNNSoftmaxLayer<float>")
 @NoOffset
 @Properties(inherit=caffe.class)
public class FloatCuDNNSoftmaxLayer
extends FloatSoftmaxLayer

\brief cuDNN implementation of SoftmaxLayer. Fallback to SoftmaxLayer for CPU mode.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
FloatCuDNNSoftmaxLayer(LayerParameter param)
FloatCuDNNSoftmaxLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
protected void	Backward_gpu(FloatBlobVector top, BoolVector propagate_down, FloatBlobVector bottom)
protected void	Forward_gpu(FloatBlobVector bottom, FloatBlobVector top)
void	LayerSetUp(FloatBlobVector bottom, FloatBlobVector top) \brief Does layer-specific setup: your layer should implement this function as well as Reshape.
void	Reshape(FloatBlobVector bottom, FloatBlobVector top) \brief Adjust the shapes of top blobs and internal buffers to accommodate the shapes of the bottom blobs.

Methods inherited from class org.bytedeco.caffe.FloatSoftmaxLayer

Backward_cpu, ExactNumBottomBlobs, ExactNumTopBlobs, Forward_cpu, type

Methods inherited from class org.bytedeco.caffe.FloatLayer

AllowForceBackward, AutoTopBlobs, Backward, blobs, CheckBlobCounts, EqualNumBottomTopBlobs, Forward, layer_param, loss, MaxBottomBlobs, MaxTopBlobs, MinBottomBlobs, MinTopBlobs, param_propagate_down, set_loss, set_param_propagate_down, SetUp, ToProto

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, hashCode, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetof, parseBytes, physicalBytes, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, toString, totalBytes, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

FloatCuDNNSoftmaxLayer

public FloatCuDNNSoftmaxLayer(Pointer p)

Pointer cast constructor. Invokes Pointer.Pointer(Pointer).

FloatCuDNNSoftmaxLayer

public FloatCuDNNSoftmaxLayer(@Const @ByRef
                              LayerParameter param)

Method Detail

LayerSetUp

@Virtual
public void LayerSetUp(@Const @ByRef
                                FloatBlobVector bottom,
                                @Const @ByRef
                                FloatBlobVector top)

Description copied from class: FloatLayer

\brief Does layer-specific setup: your layer should implement this function as well as Reshape.

Overrides:

LayerSetUp in class FloatLayer

Parameters:

bottom - the preshaped input blobs, whose data fields store the input data for this layer

top - the allocated but unshaped output blobs This method should do one-time layer specific setup. This includes reading and processing relevent parameters from the layer_param_. Setting up the shapes of top blobs and internal buffers should be done in Reshape, which will be called before the forward pass to adjust the top blob sizes.

Reshape

@Virtual
public void Reshape(@Const @ByRef
                             FloatBlobVector bottom,
                             @Const @ByRef
                             FloatBlobVector top)

Description copied from class: FloatLayer

\brief Adjust the shapes of top blobs and internal buffers to accommodate the shapes of the bottom blobs.

Overrides:

Reshape in class FloatSoftmaxLayer

Parameters:

bottom - the input blobs, with the requested input shapes

top - the top blobs, which should be reshaped as needed This method should reshape top blobs as needed according to the shapes of the bottom (input) blobs, as well as reshaping any internal buffers and making any other necessary adjustments so that the layer can accommodate the bottom blobs.

Forward_gpu

@Virtual
protected void Forward_gpu(@Const @ByRef
                                    FloatBlobVector bottom,
                                    @Const @ByRef
                                    FloatBlobVector top)

Overrides:

Forward_gpu in class FloatSoftmaxLayer

Backward_gpu

@Virtual
protected void Backward_gpu(@Const @ByRef
                                     FloatBlobVector top,
                                     @Const @ByRef
                                     BoolVector propagate_down,
                                     @Const @ByRef
                                     FloatBlobVector bottom)

Overrides:

Backward_gpu in class FloatSoftmaxLayer