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

Function GetFrameAndIter

tensorflow/core/kernels/sendrecv_ops.cc:42–53  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

40}
41
42static FrameAndIter GetFrameAndIter(OpKernelContext* ctx,
43 bool hostmem_sendrecv) {
44 if (hostmem_sendrecv && ctx->call_frame() != nullptr) {
45 // Host memory send/recv pairs are added by
46 // common_runtime/memory_types.cc. When the pair of nodes are
47 // added inside a function, we need to use the function call frame
48 // to formulate the unique rendezvous key.
49 return FrameAndIter(reinterpret_cast<uint64>(ctx->call_frame()), 0);
50 } else {
51 return ctx->frame_iter();
52 }
53}
54
55SendOp::SendOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
56 string send_device;

Callers 2

ComputeMethod · 0.70
ComputeAsyncMethod · 0.70

Calls 3

FrameAndIterClass · 0.85
frame_iterMethod · 0.80
call_frameMethod · 0.45

Tested by

no test coverage detected