more covering (wip)

Author: JulioJerez

newton-4.00/sdk/dBrain/ndBrainLayerActivationCategoricalSoftmax.cpp

@@ -98,4 +98,4 @@ void ndBrainLayerActivationCategoricalSoftmax::BackPropagate(const ndBrainLayerB
 	inputDerivative.Set(output);
 	inputDerivative.Sub(outputDerivative);
 	inputDerivative.FlushToZero();
-}
+}

newton-4.00/sdk/dBrain/ndBrainLayerActivationSoftmax.cpp

@@ -25,6 +25,7 @@
 #include "gpu/ndBrainGpuContext.h"
 #include "ndBrainTrainerCpuInference.h"
 #include "ndBrainLayerActivationSoftmax.h"
+#include "gpu/ndBrainTrainerGpuInference.h"
 
 ndBrainLayerActivationSoftmax::ndBrainLayerActivationSoftmax(ndInt32 neurons)
 	:ndBrainLayerActivation(neurons)
@@ -142,16 +143,6 @@ ndBrainLayerBackPropagateCpuCommand* ndBrainLayerActivationSoftmax::GetLayerCpuB
 	return command;
 }
 
-//ndBrainLayer::ndLayerUniformDataGpu ndBrainLayerActivationSoftmax::GetLayerUniformDataGpu(const ndBrainGpuContext* const context) const
-//{
-//	ndLayerUniformDataGpu data;
-//	data.m_shader = context->m_ndBrainLayerSoftmaxActivation;
-//	data.m_inputSize = GetInputSize();
-//	data.m_outputSize = GetOutputSize();
-//
-//	return data;
-//}
-
 void ndBrainLayerActivationSoftmax::FeedForward(const ndBrainLayerFeedFowardCpuCommand* const command, ndInt32 miniBatchIndex) const
 {
 	const ndCommandShareInfo* const info = &command->m_info;
@@ -189,4 +180,27 @@ void ndBrainLayerActivationSoftmax::FeedForward(const ndBrainLayerFeedFowardCpuC
 void ndBrainLayerActivationSoftmax::BackPropagate(const ndBrainLayerBackPropagateCpuCommand* const, ndInt32) const
 {
 	ndAssert(0);
-}
+}
+
+//ndBrainLayer::ndLayerUniformDataGpu ndBrainLayerActivationSoftmax::GetLayerUniformDataGpu(const ndBrainGpuContext* const context) const
+//{
+//	ndLayerUniformDataGpu data;
+//	data.m_shader = context->m_ndBrainLayerSoftmaxActivation;
+//	data.m_inputSize = GetInputSize();
+//	data.m_outputSize = GetOutputSize();
+//
+//	return data;
+//}
+
+ndBrainTrainerGpuCommand* ndBrainLayerActivationSoftmax::CreateGpuFeedForwardCommand(
+	ndBrainTrainerGpuInference* const owner,
+	const ndBrainLayer::ndCommandShareInfo& info,
+	ndBrainGpuContext* const context, ndInt32 miniBatchSize,
+	const ndSharedPtr<ndBrainGpuBuffer>& uniformBuffer,
+	ndBrainGpuBuffer* const buffer1,
+	ndBrainGpuBuffer* const buffer2) const
+{
+	ndBrainTrainerGpuCommand* const command = new ndBrainTrainerGpuCommand(owner,
+		info, size_t(this), context, context->m_ndBrainLayerSoftmaxActivation, miniBatchSize, uniformBuffer, buffer1, buffer2);
+	return command;
+}

newton-4.00/sdk/dBrain/ndBrainLayerActivationSoftmax.h

@@ -46,6 +46,12 @@ class ndBrainLayerActivationSoftmax : public ndBrainLayerActivation
 
 	virtual bool HasGpuSupport() const override;
 	//virtual ndLayerUniformDataGpu GetLayerUniformDataGpu(const ndBrainGpuContext* const context) const override;
+	virtual ndBrainTrainerGpuCommand* CreateGpuFeedForwardCommand(ndBrainTrainerGpuInference* const owner,
+		const ndBrainLayer::ndCommandShareInfo& info,
+		ndBrainGpuContext* const context, ndInt32 miniBatchSize,
+		const ndSharedPtr<ndBrainGpuBuffer>& uniformBuffer,
+		ndBrainGpuBuffer* const buffer1,
+		ndBrainGpuBuffer* const buffer2) const override;
 };
 
 #endif