def test_from_json_full(self):
config = Config.from_json({
'input_signatures': ['abc', 'def'],
'output_signatures': ['ghi', 'jkl', 'mno']
})
self.assertEqual(
config,
Config(input_signature=['abc', 'def'],
output_signature=['ghi', 'jkl', 'mno']))
def test_compile_with_explicit_signature(self):
with tf.Graph().as_default(), tf.compat.v1.Session().as_default(
) as session:
input_x = tf.compat.v1.placeholder(dtype=tf.float32,
shape=[3, 4],
name='x')
input_y = tf.compat.v1.placeholder(dtype=tf.float32,
shape=[3, 4],
name='y')
input_z = tf.add(input_x, input_y, name='z')
compiled = compiler.compile_source(source=TensorFlowModel(
inputs=[Input(tensor=input_x),
Input(tensor=input_y)],
outputs=[input_z],
session=session),
config=Config(
input_signature=['foo', 'bar'],
output_signature=['baz']))
self.assertEqual(len(compiled.inputs), 2)
self.assertEqual(compiled.inputs[0].name, 'foo')
self.assertEqual(compiled.inputs[0].tensor, input_x)
self.assertIsNone(compiled.inputs[0].data_format)
self.assertEqual(compiled.inputs[1].name, 'bar')
self.assertEqual(compiled.inputs[1].tensor, input_y)
self.assertIsNone(compiled.inputs[1].data_format)
self.assertEqual(len(compiled.outputs), 1)
self.assertEqual(compiled.outputs[0].name, 'baz')
self.assertIs(compiled.outputs[0].tensor, input_z)
self.assertIs(compiled.session, session)
def test_compile_simple(self):
with eager_context.graph_mode(), tf.Graph().as_default(), tf.compat.v1.Session().as_default() as session:
input_x = tf.compat.v1.placeholder(dtype=tf.float32, shape=[3, 4], name='x')
input_y = tf.compat.v1.placeholder(dtype=tf.float32, shape=[3, 4], name='y')
output_z = tf.add(input_x, input_y, name='z')
compiled = compiler.compile_source(source=TensorFlowModel(inputs=[Input(tensor=input_x), Input(tensor=input_y)],
outputs=[output_z],
session=session),
config=Config())
self.assertEqual(len(compiled.inputs), 2)
self.assertEqual(compiled.inputs[0].name, 'x:0')
self.assertEqual(compiled.inputs[0].tensor, input_x)
self.assertIsNone(compiled.inputs[0].data_format)
self.assertEqual(compiled.inputs[1].name, 'y:0')
self.assertEqual(compiled.inputs[1].tensor, input_y)
self.assertIsNone(compiled.inputs[1].data_format)
self.assertEqual(len(compiled.outputs), 1)
self.assertEqual(compiled.outputs[0].name, 'z:0')
self.assertIs(compiled.outputs[0].tensor, output_z)
self.assertIs(compiled.session, session)
def test_from_json_output_names_only(self):
config = Config.from_json({'output_signatures': ['abc', 'def']})
self.assertEqual(
config,
Config(input_signature=None, output_signature=['abc', 'def']))
def test_from_json_minimal(self):
config = Config.from_json({})
self.assertEqual(config,
Config(input_signature=None, output_signature=None))