def dummy_serving_receiver_fn():
feature_spec = {
'x':
array_ops.placeholder(dtype=dtypes.int64,
shape=(2, 1),
name='feature_x'),
}
return export.build_raw_serving_input_receiver_fn(feature_spec)
def test_build_raw_serving_input_receiver_fn_name(self):
"""Test case for issue #12755."""
f = {
"feature":
tf.compat.v1.placeholder(
name="feature", shape=[32], dtype=tf.dtypes.float32)
}
serving_input_receiver_fn = export.build_raw_serving_input_receiver_fn(f)
v = serving_input_receiver_fn()
self.assertIsInstance(v, export.ServingInputReceiver)
def test_build_raw_serving_input_receiver_fn_name(self):
"""Test case for issue #12755."""
f = {
"feature":
array_ops.placeholder(
name="feature", shape=[32], dtype=dtypes.float32)
}
serving_input_receiver_fn = export.build_raw_serving_input_receiver_fn(f)
v = serving_input_receiver_fn()
self.assertTrue(isinstance(v, export.ServingInputReceiver))
def test_build_raw_serving_input_receiver_fn_without_shape(self):
"""Test case for issue #21178."""
f = {
"feature_1": tf.compat.v1.placeholder(tf.dtypes.float32),
"feature_2": tf.compat.v1.placeholder(tf.dtypes.int32)
}
serving_input_receiver_fn = export.build_raw_serving_input_receiver_fn(f)
v = serving_input_receiver_fn()
self.assertIsInstance(v, export.ServingInputReceiver)
self.assertEqual(tensor_shape.unknown_shape(),
v.receiver_tensors["feature_1"].shape)
self.assertEqual(tensor_shape.unknown_shape(),
v.receiver_tensors["feature_2"].shape)
def test_build_raw_serving_input_receiver_fn(self):
features = {"feature_1": constant_op.constant(["hello"]),
"feature_2": constant_op.constant([42])}
serving_input_receiver_fn = export.build_raw_serving_input_receiver_fn(
features)
with ops.Graph().as_default():
serving_input_receiver = serving_input_receiver_fn()
self.assertEqual(set(["feature_1", "feature_2"]),
set(serving_input_receiver.features.keys()))
self.assertEqual(set(["feature_1", "feature_2"]),
set(serving_input_receiver.receiver_tensors.keys()))
self.assertEqual(
dtypes.string,
serving_input_receiver.receiver_tensors["feature_1"].dtype)
self.assertEqual(
dtypes.int32,
serving_input_receiver.receiver_tensors["feature_2"].dtype)
def dummy_serving_receiver_fn(): return export.build_raw_serving_input_receiver_fn(_serving_feature_dict())
