| 126 | } |
| 127 | |
| 128 | Status FileSliceSendOp::SendUInt64MetaMsg(OpKernelContext* ctx, |
| 129 | const FrameAndIter& frame_iter, |
| 130 | const string& name, |
| 131 | const uint64 val) { |
| 132 | Rendezvous::Args args; |
| 133 | args.device_context = ctx->op_device_context(); |
| 134 | args.alloc_attrs = AllocatorAttributes(); |
| 135 | |
| 136 | Rendezvous::ParsedKey parsed_key; |
| 137 | Tensor val_t; |
| 138 | TF_RETURN_IF_ERROR(ctx->allocate_temp(DT_UINT64, TensorShape({}), &val_t)); |
| 139 | val_t.scalar<uint64>()() = val; |
| 140 | slice_sendrecv::GetSliceRendezvousKey(key_prefix_, name, frame_iter, |
| 141 | &parsed_key.buf_); |
| 142 | VLOG(2) << "FileSliceSend " << parsed_key.buf_; |
| 143 | TF_RETURN_IF_ERROR(Rendezvous::ParseKey(parsed_key.buf_, &parsed_key)); |
| 144 | return ctx->rendezvous()->Send(parsed_key, args, val_t, ctx->is_input_dead()); |
| 145 | } |
| 146 | |
| 147 | Status FileSliceSendOp::SendTotalBytes(OpKernelContext* ctx, |
| 148 | const FrameAndIter& frame_iter, |
nothing calls this directly
no test coverage detected