protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
return new SigmoidCrossEntropyLossLayerTest<double>(strName, nDeviceID, engine);
else
return new SigmoidCrossEntropyLossLayerTest<float>(strName, nDeviceID, engine);
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (strName.Contains("SIAMESE"))
{
if (dt == common.DataType.DOUBLE)
{
return(new ResNetSiameseModelBuilderTest <double>(strName, nDeviceID, engine));
}
else
{
return(new ResNetSiameseModelBuilderTest <float>(strName, nDeviceID, engine));
}
}
else
{
if (dt == common.DataType.DOUBLE)
{
return(new ResNetModelBuilderTest <double>(strName, nDeviceID, engine));
}
else
{
return(new ResNetModelBuilderTest <float>(strName, nDeviceID, engine));
}
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new ContrastiveLossLayerTest <double>(strName, nDeviceID, engine));
}
else
{
return(new ContrastiveLossLayerTest <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new MyCaffePreProcessorTest <double>(strName, nDeviceID, engine));
}
else
{
return(new MyCaffePreProcessorTest <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new NeuralStyleTransferTest <double>(strName, nDeviceID, engine));
}
else
{
return(new NeuralStyleTransferTest <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
if (dt == common.DataType.DOUBLE)
{
return(new Im2ColLayerTest <double>(strName, nDeviceID));
}
else
{
return(new Im2ColLayerTest <float>(strName, nDeviceID));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new BeamSearchTest2 <double>(strName, nDeviceID, engine));
}
else
{
return(new BeamSearchTest2 <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new RandomNumberGeneratorTest <double>(strName, nDeviceID, engine));
}
else
{
return(new RandomNumberGeneratorTest <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
string strPath = TestBase.CudaPath;
if (dt == common.DataType.DOUBLE)
{
CudaDnn <double> .SetDefaultCudaPath(strPath);
return(new AnnotatedDataLayerTest <double>(strName, nDeviceID, engine, this));
}
else
{
CudaDnn <float> .SetDefaultCudaPath(strPath);
return(new AnnotatedDataLayerTest <float>(strName, nDeviceID, engine, this));
}
}
