| 32 | } |
| 33 | |
| 34 | void GetColocationConstraints(const Output& tensor, |
| 35 | std::vector<string>* constraints) { |
| 36 | constraints->clear(); |
| 37 | TF_EXPECT_OK(GetNodeAttr(tensor.op().node()->attrs(), kColocationAttrName, |
| 38 | constraints)); |
| 39 | } |
| 40 | |
| 41 | TEST(CCOpTest, Basic) { |
| 42 | Scope root = Scope::NewRootScope(); |