def test_GetIENetworkFromNGraph(): element_type = Type.f32 param = Parameter(element_type, Shape([1, 3, 22, 22])) relu = Relu(param) func = Function([relu], [param], 'test') caps = Function.to_capsule(func) cnnNetwork = IENetwork(caps) assert cnnNetwork != None assert cnnNetwork.get_function() != None caps2 = cnnNetwork.get_function() func2 = Function.from_capsule(caps2) assert func2 != None
def test_CreateIENetworkFromNGraph(): element_type = Type.f32 param = Parameter(element_type, Shape([1, 3, 22, 22])) relu = Relu(param) func = Function([relu], [param], 'test') caps = Function.to_capsule(func) cnnNetwork = IENetwork(caps) assert cnnNetwork != None assert cnnNetwork.get_function() != None assert len(cnnNetwork.layers) == 2