Searched refs:DEVICE_TO_HOST (Results 1 – 11 of 11) sorted by relevance
38 type != ChannelHandle::DEVICE_TO_HOST) { in NewChannel()101 if (channel.type == ChannelHandle::DEVICE_TO_HOST) { in RegisterRecvInternal()
165 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier); in CopyDeviceToHost()383 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
58 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
146 T, VariantDeviceCopyDirection::DEVICE_TO_HOST, \
96 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
180 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
55 DEVICE_TO_HOST = 2, enumerator
513 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_HOST); in CreateDeviceToHostChannelHandle()
2426 if (handle.type() != ChannelHandle::DEVICE_TO_HOST) { in SendToHost()
95 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
330 DEVICE_TO_HOST = 2; enumerator