Searched refs:input_node_and_ports (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 76 std::vector<std::pair<const NodeDef*, int>> input_node_and_ports; in IsTensorRTCandidate() local
77 input_node_and_ports.reserve(input_edges.size()); in IsTensorRTCandidate()
79 input_node_and_ports.emplace_back(&input_edge->src()->def(), in IsTensorRTCandidate()
82 return validator_.ValidateNode(node->def(), input_node_and_ports, in IsTensorRTCandidate()
|
| D | convert_nodes.h | 384 const std::vector<std::pair<const NodeDef*, int>>& input_node_and_ports,
|
| D | convert_nodes_test.cc | 1243 std::vector<std::pair<const NodeDef*, int>> input_node_and_ports; in RunValidation() local
1245 input_node_and_ports.emplace_back(&validator_inputs_[input], 0); in RunValidation()
1253 validator_->ValidateNode(node_def, input_node_and_ports, in RunValidation()
|
| D | convert_nodes.cc | 937 const std::vector<std::pair<const NodeDef*, int>>& input_node_and_ports, in ValidateNode() argument
960 for (int i = 0; i < input_node_and_ports.size(); ++i) { in ValidateNode()
961 const auto& pair = input_node_and_ports[i]; in ValidateNode()
|