| 59 | } |
| 60 | |
| 61 | tf2xla::Config SumConfig() { |
| 62 | tf2xla::Config config; |
| 63 | config.add_feed()->mutable_id()->set_node_name("x"); |
| 64 | config.add_feed()->mutable_id()->set_node_name("y"); |
| 65 | config.add_fetch()->mutable_id()->set_node_name("sum"); |
| 66 | return config; |
| 67 | } |
| 68 | |
| 69 | TEST(ConvertGraphDefToXla, Sum) { |
| 70 | GraphDef graph_def = SumGraph(); |