Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.h62 struct EngineConnection { struct
64 EngineConnection(const string& outside, int out_id, int out_port, in EngineConnection() argument
77 EngineConnection(const string& outside, int out_id, const string& inside, in EngineConnection() argument
88 bool is_control_edge() const { return port_number == Graph::kControlSlot; } in is_control_edge()
90 const string outside_node_name;
91 const int outside_id;
92 const int outside_port;
93 PartialTensorShape outside_shape; // Only set for input edge.
95 const string inside_node_name;
96 const int inside_id;
[all …]