/* expected_dtype = */ DT_FLOAT))
      << "Expected " << flattened[0]->DebugString();
}

TEST(SignatureFlatteningTest, SingleReturnOutputSignature) {
  std::vector<const TensorSpecProto*> flattened;
  StructuredValue value = testing::SingleReturnOutputSignature();
  TF_EXPECT_OK(internal::FlattenSignature(value, &flattened));
  EXPECT_EQ(flattened.size(), 1);
  EXPECT_TRUE(TensorSpecsAreEqual(*flattened[0],

StructuredValue ZeroReturnOutputSignature();

// Returns a StructuredValue corresponding to the serialized OutputSignature of
// a tf.function with a single tensor output
StructuredValue SingleReturnOutputSignature();

// Returns a StructuredValue corresponding to the serialized OutputSignature of
// a tf.function with three tensor outputs
StructuredValue ThreeReturnOutputSignature();

}  // namespace testing

  return ParseStructuredValue(kThreeArgInputSignatureTextProto);
}

StructuredValue ZeroReturnOutputSignature() {
  return ParseStructuredValue(kZeroReturnOutputSignatureTextProto);
}

StructuredValue SingleReturnOutputSignature() {
  return ParseStructuredValue(kSingleReturnOutputSignatureTextProto);
}

StructuredValue ThreeReturnOutputSignature() {
  return ParseStructuredValue(kThreeReturnOutputSignatureTextProto);
}