Searched refs:OpInputPortIdToArgId (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 52 EXPECT_EQ(OpInputPortIdToArgId(a_node_def, *a_op_def, 0), -1); in TEST_F()
57 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 0), 0); in TEST_F()
58 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 1), 0); in TEST_F()
59 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 2), 0); in TEST_F()
60 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 3), -1); in TEST_F()
83 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 0), 0); in TEST_F()
84 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 1), 1); in TEST_F()
85 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 2), 2); in TEST_F()
86 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 3), 3); in TEST_F()
87 EXPECT_EQ(OpInputPortIdToArgId(b_node_def, *b_op_def, 4), -1); in TEST_F()
|
| D | graph_view.cc | 62 int OpInputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id) { in OpInputPortIdToArgId() function
|
| D | graph_view.h | 44 int OpInputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 191 const int input_arg_id = OpInputPortIdToArgId(node, *op, port_id); in IsNodeInputPortHostFriendly()
|