Searched refs:OpOutputPortIdToArgId (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 54 EXPECT_EQ(OpOutputPortIdToArgId(a_node_def, *a_op_def, 0), 0); in TEST_F()
55 EXPECT_EQ(OpOutputPortIdToArgId(a_node_def, *a_op_def, 1), -1); in TEST_F()
62 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, 0), 0); in TEST_F()
63 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, 1), 0); in TEST_F()
64 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, 2), 0); in TEST_F()
65 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, 3), -1); in TEST_F()
66 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, 4), -1); in TEST_F()
95 EXPECT_EQ(OpOutputPortIdToArgId(b_node_def, *b_op_def, port_id), arg_id); in TEST_F()
116 EXPECT_EQ(OpOutputPortIdToArgId(c_node_def, *c_op_def, 0), 0); in TEST_F()
117 EXPECT_EQ(OpOutputPortIdToArgId(c_node_def, *c_op_def, 1), 0); in TEST_F()
[all …]
|
| D | graph_view.cc | 58 int OpOutputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id) { in OpOutputPortIdToArgId() function
|
| D | graph_view.h | 43 int OpOutputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 152 const int output_arg_id = OpOutputPortIdToArgId(node, *op, port_id); in IsNodeOutputPortHostFriendly()
|