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.h46 struct EngineConnection { struct
48 EngineConnection(const string& outside, int out_id, int out_port, in EngineConnection() function
61 EngineConnection(const string& outside, int out_id, const string& inside, in EngineConnection() function
72 bool is_control_edge() const { return port_number == Graph::kControlSlot; } in is_control_edge()
74 const string outside_node_name;
75 const int outside_id;
76 const int outside_port;
77 PartialTensorShape outside_shape; // Only set for input edge.
79 const string inside_node_name;
80 const int inside_id;
[all …]