Function TEST

serving/processor/framework/util/utils_test.cc:25–37

namespace processor {

TEST(UtilsTest, HasDynamicShapeOutput0) {
  GraphDef graph_def;
  NodeDef* n0 = graph_def.add_node();
  n0->set_name("A");
  n0->set_op("A");
  AttrValue value0;
  tensorflow::TensorShapeProto tshape0;
  tshape0.add_dim()->set_size(-1);
  *value0.mutable_shape() = tshape0;
  (*n0->mutable_attr())["_output_shapes"] = value0;

  EXPECT_TRUE(HasDynamicShapeOutput(n0));
}

TEST(UtilsTest, HasDynamicShapeOutput1) {
  GraphDef graph_def;