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())