Searched refs:CopyToDeviceAsync (Results 1 – 2 of 2) sorted by relevance
361 TEST(PjRtClientTest, CopyToDeviceAsync) { in TEST() argument
561 Status CopyToDeviceAsync(OpKernelContext* ctx, const Tensor& native_tensor, in CopyToDeviceAsync() function636 ctx, CopyToDeviceAsync(ctx, native_outputs->at(t), t, *stream), in ExecuteNativeSegment()