Searched refs:dst_attr (Results 1 – 2 of 2) sorted by relevance
86 const AllocatorAttributes& dst_attr, const Tensor* src, Tensor* dst, in MemCpyAsync() argument97 dst_attr.on_host() ? DEVICE_CPU : dst_dev->attributes().device_type()); in MemCpyAsync()120 dst_attr, src, dst, dev_to_dev_stream_index, done); in MemCpyAsync()
79 const AllocatorAttributes& dst_attr,