| 152 | } |
| 153 | |
| 154 | Status FileSliceSendOp::SendScalarShape(OpKernelContext* ctx, |
| 155 | const FrameAndIter& frame_iter) { |
| 156 | Rendezvous::Args args; |
| 157 | args.device_context = ctx->op_device_context(); |
| 158 | args.alloc_attrs = AllocatorAttributes(); |
| 159 | Rendezvous::ParsedKey parsed_key; |
| 160 | |
| 161 | Tensor shape_t; |
| 162 | TF_RETURN_IF_ERROR(ctx->allocate_temp(DT_INT64, TensorShape({0}), &shape_t)); |
| 163 | slice_sendrecv::GetSliceRendezvousKey(key_prefix_, |
| 164 | "_slice_transfer_shape", frame_iter, &parsed_key.buf_); |
| 165 | VLOG(2) << "FileSliceSend " << parsed_key.buf_; |
| 166 | TF_RETURN_IF_ERROR(Rendezvous::ParseKey(parsed_key.buf_, &parsed_key)); |
| 167 | |
| 168 | return ctx->rendezvous()->Send(parsed_key, args, shape_t, |
| 169 | ctx->is_input_dead()); |
| 170 | } |
| 171 | |
| 172 | Status FileSliceSendOp::SendElementBytes(OpKernelContext* ctx, |
| 173 | const FrameAndIter& frame_iter, |
nothing calls this directly
no test coverage detected