Inspect if model2function ops are placed on a device during function creation or when the function is called

[matiaslindgren]

lidbox/lidbox/util.py

Lines 101 to 106 in 49c27a0

	def model2function(model):
	model_input = model.inputs[0]
	model_fn = tf.function(
	lambda x: model(x, training=False),
	input_signature=[tf.TensorSpec(model_input.shape, model_input.dtype)])
	return model_fn.get_concrete_function()