Searched refs:VerifyNodeHasColocation (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | colocation_test.cc | 28 bool VerifyNodeHasColocation(const NodeDef& ndef, const string& coloc) { in VerifyNodeHasColocation() function
94 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(0), "loc:@X")); // A in TEST()
95 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(1), "loc:@X")); // B in TEST()
96 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(2), "loc:@X")); // C in TEST()
97 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(3), "loc:@C")); // D in TEST()
98 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(4), "loc:@D")); // E in TEST()
103 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(0), "loc:@E")); // A in TEST()
104 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(1), "loc:@E")); // B in TEST()
105 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(2), "loc:@E")); // C in TEST()
106 EXPECT_TRUE(VerifyNodeHasColocation(gdef.node(3), "loc:@E")); // D in TEST()
[all …]
|