예제 #1
0
def unittest():
    for dtype, atol in dtypesSupported():
        calcTest(dtype, atol)
        batchCalcTest(dtype, atol)

        speedTest(dtype)
        batchSpeedTest(dtype)
예제 #2
0
def unittest():
	for dtype, atol in dtypesSupported():
		batchNorm2dTest(dtype, atol)
		batchNorm3dTest(dtype, atol)
		instanceNorm2dTest(dtype, atol)

		mapLRN2dTest(dtype, atol)
		crossMapLRN2dTest(dtype, atol)
예제 #3
0
def unittest():
    vectorTest()

    for dtype, atol in dtypesSupported():
        matrixTest(dtype, atol)
        gbpGbpTest(dtype, atol)
        gbpBgpTest(dtype, atol)
        bgpGbpTest(dtype, atol)
        bgpBgpTest(dtype, atol)
예제 #4
0
파일: CuDnn.py 프로젝트: rsarbaev/PuzzleLib 
def unittest():
    for dtype, atol in dtypesSupported():
        conv2dTest(dtype, atol)
        conv3dTest(dtype, atol)
        convGroupTest(dtype, atol)

        deconv2dTest(dtype, atol)
        deconv3dTest(dtype, atol)
        deconvGroupTest(dtype, atol)

        maxpool2dTest(dtype, atol)
        maxpool3dTest(dtype, atol)

        softmax2dTest(dtype, atol)
예제 #5
0
def unittest():
    for dtype, atol in dtypesSupported():
        reluTest(dtype, atol)
        tanhTest(dtype, atol)
        lstmTest(dtype, atol)
        gruTest(dtype, atol)
예제 #6
0
파일: Pad.py 프로젝트: rsarbaev/PuzzleLib 
def unittest():
    for dtype, _ in dtypesSupported():
        reflectpad1dTest(dtype)
        reflectpad2dTest(dtype)
예제 #7
0
def unittest():
	for dtype, atol in dtypesSupported():
		spatialTfTest(dtype, atol)
예제 #8
0
def unittest():
    for dtype, _ in dtypesSupported():
        transposeTest(dtype)
        moveAxisTest(dtype)
        swapAxesTest(dtype)
        depthConcatTest(dtype)