Home
last modified time | relevance | path

Searched defs:EngineConnection (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h45 struct EngineConnection { struct
47 EngineConnection(const string& outside, int out_id, int out_port, in EngineConnection() argument
60 EngineConnection(const string& outside, int out_id, const string& inside, in EngineConnection() argument
71 bool is_control_edge() const { return port_number == Graph::kControlSlot; } in is_control_edge()
73 const string outside_node_name;
74 const int outside_id;
75 const int outside_port;
76 PartialTensorShape outside_shape; // Only set for input edge.
78 const string inside_node_name;
79 const int inside_id;
[all …]