Exemplos de DepthwiseConvolution2dDescriptor em Python

Exemplo n.º 1

Arquivo: test_network.py Projeto: ARM-software/armnn

def test_DepthwiseConvolution2d_layer_optional_none():
    net = ann.INetwork()
    layer = net.AddDepthwiseConvolution2dLayer(
        convolution2dDescriptor=ann.DepthwiseConvolution2dDescriptor(),
        weights=ann.ConstTensor())

    assert layer

Exemplo n.º 2

Arquivo: test_network.py Projeto: zhangzhe1103/armnn

def test_DepthwiseConvolution2d_layer_all_args():
    net = ann.INetwork()
    layer = net.AddDepthwiseConvolution2dLayer(convolution2dDescriptor=ann.DepthwiseConvolution2dDescriptor(),
                                       weights=ann.ConstTensor(),
                                       biases=ann.ConstTensor(),
                                       name='NAME1')

    assert layer
    assert 'NAME1' == layer.GetName()

Exemplo n.º 3

Arquivo: test_descriptors.py Projeto: vnn0804/armnn

def test_transpose_convolution2d_descriptor_default_values():
    desc = ann.DepthwiseConvolution2dDescriptor()
    assert desc.m_PadLeft == 0
    assert desc.m_PadTop == 0
    assert desc.m_PadRight == 0
    assert desc.m_PadBottom == 0
    assert desc.m_StrideX == 0
    assert desc.m_StrideY == 0
    assert desc.m_BiasEnabled == False
    assert desc.m_DataLayout == ann.DataLayout_NCHW