MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SumGraph

Function SumGraph

tensorflow/compiler/tf2xla/tf2xla_test.cc:42–59  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

40}
41
42GraphDef SumGraph() {
43 GraphDef graph_def;
44 NodeDef* x = graph_def.add_node();
45 x->set_name("x");
46 x->set_op("Placeholder");
47 (*x->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32);
48 NodeDef* y = graph_def.add_node();
49 y->set_name("y");
50 y->set_op("Placeholder");
51 (*y->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32);
52 NodeDef* sum = graph_def.add_node();
53 sum->set_name("sum");
54 sum->set_op("Add");
55 sum->add_input("x");
56 sum->add_input("y");
57 (*sum->mutable_attr())["T"] = TypeAttrValue(DT_INT32);
58 return graph_def;
59}
60
61tf2xla::Config SumConfig() {
62 tf2xla::Config config;

Callers 1

TESTFunction · 0.70

Calls 4

set_opMethod · 0.80
TypeAttrValueFunction · 0.70
set_nameMethod · 0.45
add_inputMethod · 0.45

Tested by

no test coverage detected