auto* output = Output(0);
output->ResizeLike(input0);
- output->CopyFrom(input0, &context_);
+ output->CopyFrom(input0, /* async */ true);
if (InputSize() == 1) {
return true;
const auto& in = Input(0);
auto* out = Output(0);
if (out != &in) {
- out->CopyFrom(in, &context_);
+ out->CopyFrom(in, /* async */ true);
}
return true;
}
if (num_entries == 0) {
if (!output_initialized) {
// Get both shape and meta
- output->CopyFrom(input, &context_);
+ output->CopyFrom(input, /* async */ true);
}
return true;
}
for (int i = 0; i < OutputSize(); ++i) {
OperatorBase::template Output<Tensor>(i, Context::GetDeviceType())
->CopyFrom(
- prefetched_blobs_[i].template Get<TensorCPU>(), &this->context_);
+ prefetched_blobs_[i].template Get<TensorCPU>(), /* async */ true);
}
return true;
}
&context_);
} else {
if (input_tensor_data != clipped_tensor_data) {
- clipped->CopyFrom(input_tensor, &context_);
+ clipped->CopyFrom(input_tensor, /*async*/ true);
}
}