Home
last modified time | relevance | path

Searched refs:GetInputPort (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view_test.cc134 GraphView::InputPort input = graph.GetInputPort("AddN", 0); in TEST_F()
141 input = graph.GetInputPort("AddN", 1); in TEST_F()
208 GraphView::InputPort input = graph.GetInputPort("d", -1); in TEST_F()
216 input = graph.GetInputPort("d", 0); in TEST_F()
224 input = graph.GetInputPort("d", 1); in TEST_F()
Dgraph_view.h127 InputPort GetInputPort(absl::string_view node_name, int port_id) const { in GetInputPort() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcompute_batch_size_op.cc78 graph.GetInputPort(node.name(), input_index); in GetInputNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils.cc273 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), 0); in GetInputNode()
280 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), i); in GetInputNode()