Ph/ BUddlmZmZmZmZddlZddlmZddlmZddl m Z dee fdZ dee fdZ eeeeZe ed <eeZe ed <e dd ed edeeej&ej(feeefffd Zdeedej&defdZy))ListDictTupleOptionalN)Tensor)no_grad) TypeAliasreturncDddtjjgS)z:Return the device type list that supports foreach kernels.cudaxputorch_C_get_privateuse1_backend_nameeC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/utils/_foreach_utils.py&_get_foreach_kernels_supported_devicesr E588AAC DDrcDddtjjgS)zEReturn the device type list that supports fused kernels in optimizer.r r rrrr$_get_fused_kernels_supported_devicesr rrTensorListListIndicestensorlistlist with_indicesc tjj||jDcic]\\}}}|t t|f|c}}}Scc}}}wN)rr"_group_tensors_by_device_and_dtypeitemsgetattr)rrdevice str_dtypevalues rrrsc 33NLQWWY  Z ' VY  *+U2Y  s Atensorsr"c|jttdgzvstjj ryt d|DS)NcpuFc3`K|]&}|duxst|tjk((ywr)typerr).0ts r z'_has_foreach_support..-s*EWqDy3DGu||33Ws,.)r)setrrjit is_scriptingall)r%r"s r_has_foreach_supportr1*sD {{#DF%PQQUZU^U^UkUkUm EWE EEr)F)typingrrrrrrtorch.autograd.grad_modertyping_extensionsr strrrr__annotations__intrboolr"dtyperr1rrrr:s.. ,'ES EEd3iE!hv&6!78 8#Y " % ekk) *E.'2I,J JK F$v,F FFr