Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "46c3722d3df3790784b2438954f4f7be07839eab"
Commit de319a78 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 351655727
parent fff890ec
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/serving/export_saved_model_lib.py
View file @ de319a78
...@@ -86,7 +86,7 @@ def export_inference_graph(input_type, batch_size, input_image_size, params, ...@@ -86,7 +86,7 @@ def export_inference_graph(input_type, batch_size, input_image_size, params,
if input_type == 'image_tensor': if input_type == 'image_tensor':
input_signature = tf.TensorSpec( input_signature = tf.TensorSpec(
shape=[batch_size, input_image_size[0], input_image_size[1], 3], shape=[batch_size, None, None, 3],
dtype=tf.uint8) dtype=tf.uint8)
signatures = { signatures = {
'serving_default': 'serving_default':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment