Searched refs:src_attr (Results 1 – 4 of 4) sorted by relevance
970 auto& src_attr = conv2d.attr(); in IsCpuCompatibleDataType() local972 (*attr)["T"] = src_attr.at("T"); in IsCpuCompatibleDataType()973 (*attr)["strides"] = src_attr.at("strides"); in IsCpuCompatibleDataType()974 (*attr)["padding"] = src_attr.at("padding"); in IsCpuCompatibleDataType()975 (*attr)["explicit_paddings"] = src_attr.at("explicit_paddings"); in IsCpuCompatibleDataType()976 (*attr)["dilations"] = src_attr.at("dilations"); in IsCpuCompatibleDataType()977 (*attr)["data_format"] = src_attr.at("data_format"); in IsCpuCompatibleDataType()978 (*attr)["use_cudnn_on_gpu"] = src_attr.at("use_cudnn_on_gpu"); in IsCpuCompatibleDataType()993 auto& src_attr = dw_conv2d.attr(); in IsCpuCompatibleDataType() local995 (*attr)["T"] = src_attr.at("T"); in IsCpuCompatibleDataType()[all …]
120 Device* dst_dev, const AllocatorAttributes& src_attr, in MemCpyAsync() argument130 src_attr.on_host() ? DEVICE_CPU : src_dev->attributes().device_type()); in MemCpyAsync()154 src_dev_ctx, dst_dev_ctx, src_dev, dst_dev, src_attr, in MemCpyAsync()
68 Device* dst_dev, const AllocatorAttributes& src_attr,
87 List<AttributeInfo> src_attr = src.getAttributes(); in CodeAttribute() local88 int num = src_attr.size(); in CodeAttribute()90 AttributeInfo ai = src_attr.get(i); in CodeAttribute()