def test_sequence_repeat_converter(self): input_dim = (3, 1, 1) output_dim = (9, 1, 1) inputs = [('input', datatypes.Array(*input_dim))] outputs = [('output', datatypes.Array(*output_dim))] builder = NeuralNetworkBuilder(inputs, outputs) builder.add_sequence_repeat(name='Repeat', input_name='input', output_name='output', nrep=3) model_onnx = convert_coreml(builder.spec) self.assertTrue(model_onnx is not None)
def test_sequence_repeat_converter(self): input_dim = (3, 1, 1) output_dim = (9, 1, 1) inputs = [('input', datatypes.Array(*input_dim))] outputs = [('output', datatypes.Array(*output_dim))] builder = NeuralNetworkBuilder(inputs, outputs) builder.add_sequence_repeat(name='Repeat', input_name='input', output_name='output', nrep=3) context = ConvertContext() node = SequenceRepeatLayerConverter.convert( context, builder.spec.neuralNetwork.layers[0], ['input'], ['output']) self.assertTrue(node is not None)