Searched refs:src_attr (Results 1 – 3 of 3) sorted by relevance
85 Device* dst_dev, const AllocatorAttributes& src_attr, in MemCpyAsync() argument95 src_attr.on_host() ? DEVICE_CPU : src_dev->attributes().device_type()); in MemCpyAsync()119 src_dev_ctx, dst_dev_ctx, src_dev, dst_dev, src_attr, in MemCpyAsync()
78 Device* dst_dev, const AllocatorAttributes& src_attr,
468 auto src_attr = conv2d->attr(); in CopyConv2DAttributes() local470 (*attr)["T"] = src_attr.at("T"); in CopyConv2DAttributes()471 (*attr)["strides"] = src_attr.at("strides"); in CopyConv2DAttributes()472 (*attr)["padding"] = src_attr.at("padding"); in CopyConv2DAttributes()473 (*attr)["dilations"] = src_attr.at("dilations"); in CopyConv2DAttributes()474 (*attr)["data_format"] = src_attr.at("data_format"); in CopyConv2DAttributes()475 (*attr)["use_cudnn_on_gpu"] = src_attr.at("use_cudnn_on_gpu"); in CopyConv2DAttributes()