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

Method GetCollectiveKey

tensorflow/core/kernels/nccl_ops.cc:50–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

48 }
49
50 string GetCollectiveKey(OpKernelContext* c) {
51 return strings::StrCat(collective_prefix_, ";", c->step_id(), ";",
52 c->frame_iter().frame_id, ":",
53 c->frame_iter().iter_id);
54 }
55
56 int num_devices() const { return num_devices_; }
57

Callers

nothing calls this directly

Calls 3

frame_iterMethod · 0.80
StrCatFunction · 0.50
step_idMethod · 0.45

Tested by

no test coverage detected