/// <summary>
/// Conv2s the forward frame.
/// </summary>
/// <param name="input">The input.</param>
/// <param name="output">The output.</param>
/// <param name="weight">The weight.</param>
/// <param name="bias">The bias.</param>
/// <param name="finput">The finput.</param>
/// <param name="kW">The k w.</param>
/// <param name="kH">The k h.</param>
/// <param name="dW">The d w.</param>
/// <param name="dH">The d h.</param>
/// <param name="padW">The pad w.</param>
/// <param name="padH">The pad h.</param>
/// <param name="nInputPlane">The n input plane.</param>
/// <param name="inputWidth">Width of the input.</param>
/// <param name="inputHeight">Height of the input.</param>
/// <param name="nOutputPlane">The n output plane.</param>
/// <param name="outputWidth">Width of the output.</param>
/// <param name="outputHeight">Height of the output.</param>
private static void Conv2ForwardFrame(NDArray input, NDArray output, NDArray weight, NDArray bias, NDArray finput,
int kW,
int kH,
int dW,
int dH,
int padW,
int padH,
long nInputPlane,
long inputWidth,
long inputHeight,
long nOutputPlane,
long outputWidth,
long outputHeight)
{
var inputRef = NativeWrapper.AllocTensorRef(input);
var finputRef = NativeWrapper.AllocTensorRef(finput);
var inputPtr = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(TensorRef64)));
Marshal.StructureToPtr(inputRef, inputPtr, false);
var finputPtr = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(TensorRef64)));
Marshal.StructureToPtr(finputRef, finputPtr, false);
try
{
CpuOpsNative.TS_Unfolded_Copy(finputPtr, inputPtr, kW, kH, dW, dH, padW, padH, (int)nInputPlane, (int)inputWidth, (int)inputHeight, (int)outputWidth, (int)outputHeight);
using (var output2d = output.View(nOutputPlane, outputHeight * outputWidth))
{
if (bias != null)
{
using (var biasExp = bias.Expand(nOutputPlane, output2d.Shape[1]))
{
Ops.Copy(output2d, biasExp);
}
}
else
{
Ops.Fill(output, 0);
}
Ops.Addmm(output2d, 1, output2d, 1, weight, finput);
}
}
finally
{
Marshal.FreeHGlobal(inputPtr);
Marshal.FreeHGlobal(finputPtr);
NativeWrapper.FreeTensorRef(inputRef);
NativeWrapper.FreeTensorRef(finputRef);
}
}
