MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetColocationConstraints

Function GetColocationConstraints

tensorflow/core/graph/stream_subgraph.cc:28–32  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

26namespace {
27
28bool GetColocationConstraints(const Node* node,
29 std::vector<string>* constraints) {
30 return TryGetNodeAttr(node->attrs(),
31 kColocationAttrName, constraints);
32}
33
34std::string GetDeviceNamePrefix(const std::string& device_name) {
35 std::string prefix_token("device:GPU:");

Callers 1

MarkEmbeddingGraphFunction · 0.70

Calls 1

attrsMethod · 0.45

Tested by

no test coverage detected