pytorch-metadata.json 200 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304630563066307630863096310631163126313631463156316631763186319632063216322632363246325632663276328632963306331633263336334633563366337633863396340634163426343634463456346634763486349635063516352635363546355635663576358635963606361636263636364636563666367636863696370637163726373637463756376637763786379638063816382638363846385638663876388638963906391639263936394639563966397639863996400640164026403640464056406640764086409641064116412641364146415641664176418641964206421642264236424642564266427642864296430643164326433643464356436643764386439644064416442644364446445644664476448644964506451645264536454645564566457645864596460646164626463646464656466646764686469647064716472647364746475647664776478647964806481648264836484648564866487648864896490649164926493649464956496649764986499650065016502650365046505650665076508650965106511651265136514651565166517651865196520652165226523652465256526652765286529653065316532653365346535653665376538653965406541654265436544654565466547654865496550655165526553655465556556655765586559656065616562656365646565656665676568656965706571657265736574657565766577657865796580658165826583658465856586658765886589659065916592659365946595659665976598659966006601660266036604660566066607660866096610661166126613661466156616661766186619662066216622662366246625662666276628662966306631663266336634663566366637663866396640664166426643664466456646664766486649665066516652665366546655665666576658665966606661666266636664666566666667666866696670667166726673667466756676667766786679668066816682668366846685668666876688668966906691669266936694669566966697669866996700670167026703670467056706670767086709671067116712671367146715671667176718671967206721672267236724672567266727672867296730673167326733673467356736673767386739674067416742674367446745674667476748674967506751675267536754675567566757675867596760676167626763676467656766676767686769677067716772677367746775677667776778677967806781678267836784678567866787678867896790679167926793679467956796679767986799680068016802680368046805680668076808680968106811681268136814681568166817681868196820682168226823682468256826682768286829683068316832683368346835683668376838683968406841684268436844684568466847684868496850685168526853685468556856685768586859686068616862686368646865686668676868686968706871687268736874687568766877687868796880688168826883688468856886688768886889689068916892689368946895689668976898689969006901690269036904690569066907690869096910691169126913691469156916691769186919692069216922692369246925692669276928692969306931693269336934693569366937693869396940694169426943694469456946694769486949695069516952695369546955695669576958695969606961696269636964696569666967696869696970697169726973697469756976697769786979698069816982698369846985698669876988698969906991699269936994699569966997699869997000700170027003700470057006700770087009701070117012701370147015701670177018701970207021702270237024702570267027702870297030703170327033703470357036703770387039704070417042704370447045704670477048704970507051705270537054705570567057705870597060706170627063706470657066706770687069707070717072707370747075707670777078707970807081708270837084708570867087708870897090709170927093709470957096709770987099710071017102710371047105710671077108710971107111711271137114711571167117711871197120712171227123712471257126712771287129713071317132713371347135713671377138713971407141714271437144714571467147714871497150715171527153715471557156715771587159716071617162716371647165716671677168716971707171717271737174717571767177717871797180718171827183718471857186718771887189719071917192719371947195719671977198719972007201720272037204720572067207720872097210721172127213721472157216721772187219722072217222722372247225722672277228722972307231723272337234723572367237723872397240724172427243724472457246724772487249725072517252725372547255725672577258725972607261726272637264726572667267726872697270727172727273727472757276727772787279728072817282728372847285728672877288728972907291729272937294729572967297
  1. [
  2. {
  3. "name": "torch.nn.modules.conv.Conv1d",
  4. "category": "Layer",
  5. "attributes": [
  6. { "name": "output_padding", "visible": false },
  7. { "name": "in_channels", "visible": false },
  8. { "name": "out_channels", "visible": false },
  9. { "name": "groups", "default": 1 },
  10. { "name": "transposed", "default": false },
  11. { "name": "padding", "default": [ 0 ] },
  12. { "name": "dilation", "default": [ 1 ] },
  13. { "name": "stride", "default": [ 1 ] }
  14. ]
  15. },
  16. {
  17. "name": "torch.nn.modules.conv.ConvTranspose1d",
  18. "category": "Layer",
  19. "attributes": [
  20. { "name": "output_padding", "visible": false },
  21. { "name": "in_channels", "visible": false },
  22. { "name": "out_channels", "visible": false },
  23. { "name": "groups", "default": 1 },
  24. { "name": "transposed", "default": true },
  25. { "name": "padding", "default": [ 0 ] },
  26. { "name": "dilation", "default": [ 1 ] },
  27. { "name": "stride", "default": [ 1 ] }
  28. ]
  29. },
  30. {
  31. "name": "torch.nn.modules.conv.Conv2d",
  32. "category": "Layer",
  33. "attributes": [
  34. { "name": "output_padding", "visible": false },
  35. { "name": "in_channels", "visible": false },
  36. { "name": "out_channels", "visible": false },
  37. { "name": "groups", "default": 1 },
  38. { "name": "transposed", "default": false },
  39. { "name": "padding", "default": [ 0, 0 ] },
  40. { "name": "dilation", "default": [ 1, 1 ] },
  41. { "name": "stride", "default": [ 1, 1 ] }
  42. ]
  43. },
  44. {
  45. "name": "torch.nn.modules.conv.ConvTranspose2d",
  46. "category": "Layer",
  47. "attributes": [
  48. { "name": "output_padding", "visible": false },
  49. { "name": "in_channels", "visible": false },
  50. { "name": "out_channels", "visible": false },
  51. { "name": "groups", "default": 1 },
  52. { "name": "transposed", "default": true },
  53. { "name": "padding", "default": [ 0, 0 ] },
  54. { "name": "dilation", "default": [ 1, 1 ] },
  55. { "name": "stride", "default": [ 1, 1 ] }
  56. ]
  57. },
  58. {
  59. "name": "torch.nn.modules.conv.Conv3d",
  60. "category": "Layer"
  61. },
  62. {
  63. "name": "torch.nn.modules.conv.ConvTranspose3d",
  64. "category": "Layer"
  65. },
  66. {
  67. "name": "torch.nn.modules.linear.Linear",
  68. "category": "Layer",
  69. "attributes": []
  70. },
  71. {
  72. "name": "torch.nn.modules.activation.Sigmoid",
  73. "category": "Activation"
  74. },
  75. {
  76. "name": "torch.nn.modules.activation.Softmax",
  77. "category": "Activation"
  78. },
  79. {
  80. "name": "torch.nn.modules.activation.Softmax2d",
  81. "category": "Activation"
  82. },
  83. {
  84. "name": "torch.nn.modules.activation.Softplus",
  85. "category": "Activation"
  86. },
  87. {
  88. "name": "torch.nn.modules.activation.Tanh",
  89. "category": "Activation"
  90. },
  91. {
  92. "name": "torch.nn.modules.activation.LogSoftmax",
  93. "category": "Activation"
  94. },
  95. {
  96. "name": "torch.nn.modules.activation.ReLU",
  97. "category": "Activation",
  98. "attributes": [
  99. { "name": "inplace", "default": false, "visible": false },
  100. { "name": "threshold", "default": 0 },
  101. { "name": "value", "default": 0 }
  102. ]
  103. },
  104. {
  105. "name": "torch.nn.modules.activation.ReLU6",
  106. "category": "Activation"
  107. },
  108. {
  109. "name": "torch.nn.modules.activation.PReLU",
  110. "category": "Activation",
  111. "attributes": []
  112. },
  113. {
  114. "name": "torch.nn.modules.activation.ELU",
  115. "category": "Activation",
  116. "attributes": []
  117. },
  118. {
  119. "name": "torch.nn.modules.activation.GELU",
  120. "category": "Activation",
  121. "attributes": []
  122. },
  123. {
  124. "name": "torch.nn.modules.activation.GLU",
  125. "category": "Activation",
  126. "attributes": []
  127. },
  128. {
  129. "name": "torch.nn.modules.activation.Hardtanh",
  130. "category": "Activation",
  131. "attributes": []
  132. },
  133. {
  134. "name": "torch.nn.modules.activation.Hardswish",
  135. "category": "Activation",
  136. "attributes": []
  137. },
  138. {
  139. "name": "torch.nn.modules.activation.Hardsigmoid",
  140. "category": "Activation",
  141. "attributes": []
  142. },
  143. {
  144. "name": "torch.nn.modules.activation.LeakyReLU",
  145. "category": "Activation",
  146. "attributes": []
  147. },
  148. {
  149. "name": "torch.nn.modules.pooling.MaxPool1d",
  150. "category": "Pool"
  151. },
  152. {
  153. "name": "torch.nn.modules.pooling.MaxPool2d",
  154. "category": "Pool",
  155. "attributes": [
  156. { "name": "padding", "default": 0 },
  157. { "name": "dilation", "default": 1 },
  158. { "name": "return_indices", "default": false },
  159. { "name": "ceil_mode", "visible": false }
  160. ]
  161. },
  162. {
  163. "name": "torch.nn.modules.pooling.MaxPool3d",
  164. "category": "Pool"
  165. },
  166. {
  167. "name": "torch.nn.modules.pooling.MaxUnpool1d",
  168. "category": "Pool"
  169. },
  170. {
  171. "name": "torch.nn.modules.pooling.MaxUnpool2d",
  172. "category": "Pool"
  173. },
  174. {
  175. "name": "torch.nn.modules.pooling.MaxUnpool3d",
  176. "category": "Pool"
  177. },
  178. {
  179. "name": "torch.nn.modules.pooling.AvgPool2d",
  180. "category": "Pool",
  181. "attributes": [
  182. { "name": "padding", "default": 0 },
  183. { "name": "count_include_pad", "default": true },
  184. { "name": "ceil_mode", "visible": false }
  185. ]
  186. },
  187. {
  188. "name": "torch.nn.modules.pooling.AvgPool3d",
  189. "category": "Pool"
  190. },
  191. {
  192. "name": "torch.nn.modules.pooling.AdaptiveAvgPool1d",
  193. "category": "Pool"
  194. },
  195. {
  196. "name": "torch.nn.modules.pooling.AdaptiveAvgPool2d",
  197. "category": "Pool"
  198. },
  199. {
  200. "name": "torch.nn.modules.pooling.AdaptiveAvgPool3d",
  201. "category": "Pool"
  202. },
  203. {
  204. "name": "torch.nn.modules.pooling.AdaptiveMaxPool1d",
  205. "category": "Pool"
  206. },
  207. {
  208. "name": "torch.nn.modules.pooling.AdaptiveMaxPool2d",
  209. "category": "Pool"
  210. },
  211. {
  212. "name": "torch.nn.modules.pooling.AdaptiveMaxPool3d",
  213. "category": "Pool"
  214. },
  215. {
  216. "name": "torch.nn.modules.batchnorm.BatchNorm1d",
  217. "category": "Normalization",
  218. "attributes": []
  219. },
  220. {
  221. "name": "torch.nn.modules.batchnorm.BatchNorm2d",
  222. "category": "Normalization",
  223. "attributes": [
  224. { "name": "eps", "default": 0.00001 },
  225. { "name": "momentum", "default": 0.1 },
  226. { "name": "affine", "default": true },
  227. { "name": "track_running_stats", "default": true }
  228. ],
  229. "inputs": [
  230. { "name": "input" },
  231. { "name": "weight" },
  232. { "name": "bias" },
  233. { "name": "running_mean" },
  234. { "name": "running_var" },
  235. { "name": "num_batches_tracked", "visible": false }
  236. ]
  237. },
  238. {
  239. "name": "torch.nn.modules.normalization.GroupNorm",
  240. "category": "Normalization"
  241. },
  242. {
  243. "name": "torch.nn.modules.normalization.LayerNorm",
  244. "category": "Normalization"
  245. },
  246. {
  247. "name": "torch.nn.modules.dropout.Dropout2d",
  248. "category": "Dropout",
  249. "attributes": [
  250. { "name": "inplace", "default": false, "visible": false },
  251. { "name": "p", "default": 0.5 }
  252. ]
  253. },
  254. {
  255. "name": "torch.nn.modules.dropout.Dropout",
  256. "category": "Dropout",
  257. "attributes": [
  258. { "name": "inplace", "default": false, "visible": false },
  259. { "name": "p", "default": 0.5 }
  260. ]
  261. },
  262. {
  263. "name": "torch.nn.modules.upsampling.Upsample",
  264. "category": "Data"
  265. },
  266. {
  267. "name": "torch.nn.modules.rnn.GRU",
  268. "category": "Layer"
  269. },
  270. {
  271. "name": "torch.nn.modules.rnn.GRUCell",
  272. "category": "Layer"
  273. },
  274. {
  275. "name": "torch.nn.modules.rnn.LSTM",
  276. "category": "Layer",
  277. "attributes": [
  278. { "name": "dropout", "default": 0 },
  279. { "name": "dropout_state", "default": {} },
  280. { "name": "num_layers", "default": 1 },
  281. { "name": "batch_first", "visible": false },
  282. { "name": "bidirectional", "visible": false },
  283. { "name": "bias", "visible": false }
  284. ],
  285. "inputs": [
  286. { "name": "input" },
  287. { "name": "weight_ih_l0", "visible": false },
  288. { "name": "weight_hh_l0", "visible": false },
  289. { "name": "bias_ih_l0", "visible": false },
  290. { "name": "bias_hh_l0", "visible": false },
  291. { "name": "weight_ih_l1", "visible": false },
  292. { "name": "weight_hh_l1", "visible": false },
  293. { "name": "bias_ih_l1", "visible": false },
  294. { "name": "bias_hh_l1", "visible": false }
  295. ]
  296. },
  297. {
  298. "name": "torch.nn.modules.rnn.LSTMCell",
  299. "category": "Layer"
  300. },
  301. {
  302. "name": "torch.nn.modules.rnn.RNN",
  303. "category": "Layer"
  304. },
  305. {
  306. "name": "torch.nn.modules.sparse.Embedding",
  307. "category": "Transform",
  308. "attributes": [
  309. { "name": "norm_type", "default": 2 },
  310. { "name": "scale_grad_by_freq", "default": false },
  311. { "name": "sparse", "default": false },
  312. { "name": "max_norm", "default": null },
  313. { "name": "padding_idx", "default": null }
  314. ]
  315. },
  316. {
  317. "name": "torch.nn.modules.padding.ReflectionPad1d",
  318. "category": "Tensor"
  319. },
  320. {
  321. "name": "torch.nn.modules.padding.ReflectionPad2d",
  322. "category": "Tensor"
  323. },
  324. {
  325. "name": "torch.nn.modules.padding.ReplicationPad1d",
  326. "category": "Tensor"
  327. },
  328. {
  329. "name": "torch.nn.modules.padding.ReplicationPad2d",
  330. "category": "Tensor"
  331. },
  332. {
  333. "name": "torch.nn.modules.padding.ReplicationPad3d",
  334. "category": "Tensor"
  335. },
  336. {
  337. "name": "torch.nn.modules.padding.ZeroPad2d",
  338. "category": "Tensor"
  339. },
  340. {
  341. "name": "torch.nn.modules.padding.ConstantPad1d",
  342. "category": "Tensor"
  343. },
  344. {
  345. "name": "torch.nn.modules.padding.ConstantPad2d",
  346. "category": "Tensor"
  347. },
  348. {
  349. "name": "torch.nn.modules.padding.ConstantPad3d",
  350. "category": "Tensor"
  351. },
  352. {
  353. "name": "torch.nn.modules.pixelshuffle.PixelShuffle"
  354. },
  355. {
  356. "name": "torch.nn.modules.instancenorm.InstanceNorm1d"
  357. },
  358. {
  359. "name": "torch.nn.modules.instancenorm.InstanceNorm2d"
  360. },
  361. {
  362. "name": "torch.nn.modules.instancenorm.InstanceNorm3d"
  363. },
  364. {
  365. "name": "torch.nn.modules.normalization.CrossMapLRN2d",
  366. "category": "Normalization",
  367. "attributes": [
  368. { "name": "alpha", "default": 0.0001 },
  369. { "name": "beta", "default": 0.75 },
  370. { "name": "k", "default": 1 }
  371. ]
  372. },
  373. {
  374. "name": "torch._convolution:",
  375. "category": "Layer",
  376. "inputs": [
  377. { "name": "input", "type": "Tensor" },
  378. { "name": "weight", "type": "Tensor" },
  379. { "name": "bias", "type": "Tensor", "optional": true },
  380. { "name": "stride", "type": "int64[]", "default": 1 },
  381. { "name": "padding", "type": "int64[]", "default": 0 },
  382. { "name": "dilation", "type": "int64[]", "default": 1 },
  383. { "name": "transposed", "type": "boolean", "default": false },
  384. { "name": "output_padding", "type": "int64[]", "default": 0 },
  385. { "name": "groups", "type": "int64", "default": 1 },
  386. { "name": "benchmark", "type": "boolean", "visible": false },
  387. { "name": "deterministic", "type": "boolean", "visible": false },
  388. { "name": "cudnn_enabled", "type": "boolean", "visible": false },
  389. { "name": "allow_tf32", "type": "boolean", "visible": false }
  390. ],
  391. "outputs": [
  392. { "name": "output", "type": "Tensor" }
  393. ]
  394. },
  395. {
  396. "name": "torch._convolution_mode:",
  397. "inputs": [
  398. { "name": "input", "type": "Tensor" },
  399. { "name": "weight", "type": "Tensor" },
  400. { "name": "bias", "type": "Tensor", "optional": true },
  401. { "name": "stride", "type": "int64[]" },
  402. { "name": "padding", "type": "string" },
  403. { "name": "dilation", "type": "int64[]" },
  404. { "name": "groups", "type": "int64" }
  405. ],
  406. "outputs": [
  407. { "name": "output", "type": "Tensor" }
  408. ]
  409. },
  410. {
  411. "name": "torch._convolution_nogroup:",
  412. "inputs": [
  413. { "name": "input", "type": "Tensor" },
  414. { "name": "weight", "type": "Tensor" },
  415. { "name": "bias", "type": "Tensor", "optional": true },
  416. { "name": "stride", "type": "int64[]" },
  417. { "name": "padding", "type": "string" },
  418. { "name": "dilation", "type": "int64[]" },
  419. { "name": "transposed", "type": "boolean" },
  420. { "name": "output_padding", "type": "int64[]" }
  421. ],
  422. "outputs": [
  423. { "name": "output", "type": "Tensor" }
  424. ]
  425. },
  426. {
  427. "name": "torch._convolution:deprecated",
  428. "category": "Layer",
  429. "inputs": [
  430. { "name": "input", "type": "Tensor" },
  431. { "name": "weight", "type": "Tensor" },
  432. { "name": "bias", "type": "Tensor", "optional": true },
  433. { "name": "stride", "type": "int64[]", "default": 1 },
  434. { "name": "padding", "type": "int64[]", "default": 0 },
  435. { "name": "dilation", "type": "int64[]", "default": 1 },
  436. { "name": "transposed", "type": "boolean", "default": false },
  437. { "name": "output_padding", "type": "int64[]", "default": 0 },
  438. { "name": "groups", "type": "int64", "default": 1 },
  439. { "name": "benchmark", "type": "boolean", "visible": false },
  440. { "name": "deterministic", "type": "boolean", "visible": false },
  441. { "name": "cudnn_enabled", "type": "boolean", "visible": false }
  442. ],
  443. "outputs": [
  444. { "name": "output", "type": "Tensor" }
  445. ]
  446. },
  447. {
  448. "name": "torch._convolution_mode:",
  449. "category": "Layer",
  450. "inputs": [
  451. { "name": "input", "type": "Tensor" },
  452. { "name": "weight", "type": "Tensor" },
  453. { "name": "bias", "type": "Tensor", "optional": true },
  454. { "name": "stride", "type": "int64[]" },
  455. { "name": "padding", "type": "string" },
  456. { "name": "dilation", "type": "int64[]" },
  457. { "name": "groups", "type": "int64" }
  458. ],
  459. "outputs": [
  460. { "name": "output", "type": "Tensor" }
  461. ]
  462. },
  463. {
  464. "name": "torch.conv1d:",
  465. "category": "Layer",
  466. "inputs": [
  467. { "name": "input", "type": "Tensor" },
  468. { "name": "weight", "type": "Tensor" },
  469. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  470. { "name": "stride", "type": "int64[]", "default": 1 },
  471. { "name": "padding", "type": "int64[]", "default": 0 },
  472. { "name": "dilation", "type": "int64[]", "default": 1 },
  473. { "name": "groups", "type": "int64", "default": 1 }
  474. ],
  475. "outputs": [
  476. { "name": "output", "type": "Tensor" }
  477. ]
  478. },
  479. {
  480. "name": "torch.conv1d:padding",
  481. "category": "Layer",
  482. "inputs": [
  483. { "name": "input", "type": "Tensor" },
  484. { "name": "weight", "type": "Tensor" },
  485. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  486. { "name": "stride", "type": "int64[]", "default": 1 },
  487. { "name": "padding", "type": "string", "default": "valid" },
  488. { "name": "dilation", "type": "int64[]", "default": 1 },
  489. { "name": "groups", "type": "int64", "default": 1 }
  490. ],
  491. "outputs": [
  492. { "name": "output", "type": "Tensor" }
  493. ]
  494. },
  495. {
  496. "name": "torch.conv2d:",
  497. "category": "Layer",
  498. "inputs": [
  499. { "name": "input", "type": "Tensor" },
  500. { "name": "weight", "type": "Tensor" },
  501. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  502. { "name": "stride", "type": "int64[]", "default": 1 },
  503. { "name": "padding", "type": "int64[]", "default": 0 },
  504. { "name": "dilation", "type": "int64[]", "default": 1 },
  505. { "name": "groups", "type": "int64", "default": 1 }
  506. ],
  507. "outputs": [
  508. { "name": "output", "type": "Tensor" }
  509. ]
  510. },
  511. {
  512. "name": "torch.conv2d:padding",
  513. "category": "Layer",
  514. "inputs": [
  515. { "name": "input", "type": "Tensor" },
  516. { "name": "weight", "type": "Tensor" },
  517. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  518. { "name": "stride", "type": "int64[]", "default": 1 },
  519. { "name": "padding", "type": "string", "default": "valid" },
  520. { "name": "dilation", "type": "int64[]", "default": 1 },
  521. { "name": "groups", "type": "int64", "default": 1 }
  522. ],
  523. "outputs": [
  524. { "name": "output", "type": "Tensor" }
  525. ]
  526. },
  527. {
  528. "name": "torch.conv3d:",
  529. "category": "Layer",
  530. "inputs": [
  531. { "name": "input", "type": "Tensor" },
  532. { "name": "weight", "type": "Tensor" },
  533. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  534. { "name": "stride", "type": "int64[]", "default": 1 },
  535. { "name": "padding", "type": "int64[]", "default": 0 },
  536. { "name": "dilation", "type": "int64[]", "default": 1 },
  537. { "name": "groups", "type": "int64", "default": 1 }
  538. ],
  539. "outputs": [
  540. { "name": "output", "type": "Tensor" }
  541. ]
  542. },
  543. {
  544. "name": "torch.conv3d:padding",
  545. "category": "Layer",
  546. "inputs": [
  547. { "name": "input", "type": "Tensor" },
  548. { "name": "weight", "type": "Tensor" },
  549. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  550. { "name": "stride", "type": "int64[]", "default": 1 },
  551. { "name": "padding", "type": "string", "default": "valid" },
  552. { "name": "dilation", "type": "int64[]", "default": 1 },
  553. { "name": "groups", "type": "int64", "default": 1 }
  554. ],
  555. "outputs": [
  556. { "name": "output", "type": "Tensor" }
  557. ]
  558. },
  559. {
  560. "name": "torch.conv_transpose1d:input",
  561. "category": "Layer",
  562. "attributes": [
  563. { "name": "stride", "type": "int64[]", "default": 1 },
  564. { "name": "padding", "type": "int64[]", "default": 0 },
  565. { "name": "output_padding", "type": "int64[]", "default": 0 },
  566. { "name": "groups", "type": "int64", "default": 1 },
  567. { "name": "dilation", "type": "int64[]", "default": 1 }
  568. ],
  569. "inputs": [
  570. { "name": "input", "type": "Tensor" },
  571. { "name": "weight", "type": "Tensor" },
  572. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  573. ],
  574. "outputs": [
  575. { "name": "output", "type": "Tensor" }
  576. ]
  577. },
  578. {
  579. "name": "torch.conv_transpose2d:input",
  580. "category": "Layer",
  581. "attributes": [
  582. { "name": "stride", "type": "int64[]", "default": 1 },
  583. { "name": "padding", "type": "int64[]", "default": 0 },
  584. { "name": "output_padding", "type": "int64[]", "default": 0 },
  585. { "name": "groups", "type": "int64", "default": 1 },
  586. { "name": "dilation", "type": "int64[]", "default": 1 }
  587. ],
  588. "inputs": [
  589. { "name": "input", "type": "Tensor" },
  590. { "name": "weight", "type": "Tensor" },
  591. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  592. ],
  593. "outputs": [
  594. { "name": "output", "type": "Tensor" }
  595. ]
  596. },
  597. {
  598. "name": "torch.conv_transpose3d:input",
  599. "category": "Layer",
  600. "attributes": [
  601. { "name": "stride", "type": "int64[]", "default": 1 },
  602. { "name": "padding", "type": "int64[]", "default": 0 },
  603. { "name": "output_padding", "type": "int64[]", "default": 0 },
  604. { "name": "groups", "type": "int64", "default": 1 },
  605. { "name": "dilation", "type": "int64[]", "default": 1 }
  606. ],
  607. "inputs": [
  608. { "name": "input", "type": "Tensor" },
  609. { "name": "weight", "type": "Tensor" },
  610. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  611. ],
  612. "outputs": [
  613. { "name": "output", "type": "Tensor" }
  614. ]
  615. },
  616. {
  617. "name": "ops.torchaudio.sox_effects_apply_effects_tensor",
  618. "inputs": [
  619. { "name": "tensor", "type": "Tensor" },
  620. { "name": "sample_rate", "type": "int64" },
  621. { "name": "effects", "type": "string[][]" },
  622. { "name": "channels_first", "type": "boolean", "default": true }
  623. ],
  624. "outputs": [
  625. { "name": "output", "type": "Tensor" },
  626. { "name": "?", "type": "Tensor" }
  627. ]
  628. },
  629. {
  630. "name": "ops.prepacked.conv2d_clamp_run",
  631. "category": "Layer",
  632. "inputs": [
  633. { "name": "input", "type": "Tensor" },
  634. { "name": "weight", "type": "Tensor", "context": true },
  635. { "name": "bias", "type": "Tensor", "context": true },
  636. { "name": "stride", "type": "int64[]", "context": true },
  637. { "name": "padding", "type": "int64[]", "context": true },
  638. { "name": "dilation", "type": "int64[]", "context": true },
  639. { "name": "groups", "type": "int64", "context": true },
  640. { "name": "output_min", "type": "Scalar", "context": true },
  641. { "name": "output_max", "type": "Scalar", "context": true }
  642. ],
  643. "outputs": [
  644. { "name": "output", "type": "Tensor" }
  645. ]
  646. },
  647. {
  648. "name": "ops.prepacked.linear_clamp_run",
  649. "category": "Layer",
  650. "inputs": [
  651. { "name": "input", "type": "Tensor" },
  652. { "name": "weight", "type": "Tensor", "context": true },
  653. { "name": "bias", "type": "Tensor", "context": true },
  654. { "name": "output_min", "type": "Scalar", "context": true },
  655. { "name": "output_max", "type": "Scalar", "context": true }
  656. ],
  657. "outputs": [
  658. { "name": "output", "type": "Tensor" }
  659. ]
  660. },
  661. {
  662. "name": "ops.quantized.add",
  663. "inputs": [
  664. { "name": "A", "type": "Tensor" },
  665. { "name": "B", "type": "Tensor" },
  666. { "name": "scale", "type": "float32" },
  667. { "name": "zero_point", "type": "int64" }
  668. ],
  669. "outputs": [
  670. { "name": "C", "type": "Tensor" }
  671. ]
  672. },
  673. {
  674. "name": "ops.quantized.add_scalar:",
  675. "inputs": [
  676. { "name": "qa", "type": "Tensor" },
  677. { "name": "b", "type": "Scalar" }
  678. ],
  679. "outputs": [
  680. { "name": "qc", "type": "Tensor" }
  681. ]
  682. },
  683. {
  684. "name": "ops.quantized.add_scalar_relu:",
  685. "inputs": [
  686. { "name": "qa", "type": "Tensor" },
  687. { "name": "b", "type": "Scalar" }
  688. ],
  689. "outputs": [
  690. { "name": "qc", "type": "Tensor" }
  691. ]
  692. },
  693. {
  694. "name": "ops.quantized.add_scalar:Tensor",
  695. "inputs": [
  696. { "name": "qa", "type": "Tensor" },
  697. { "name": "b", "type": "Tensor" }
  698. ],
  699. "outputs": [
  700. { "name": "qc", "type": "Tensor" }
  701. ]
  702. },
  703. {
  704. "name": "ops.quantized.add_scalar_relu:Tensor",
  705. "inputs": [
  706. { "name": "qa", "type": "Tensor" },
  707. { "name": "b", "type": "Tensor" }
  708. ],
  709. "outputs": [
  710. { "name": "qc", "type": "Tensor" }
  711. ]
  712. },
  713. {
  714. "name": "ops.quantized.linear",
  715. "category": "Layer",
  716. "inputs": [
  717. { "name": "X", "type": "Tensor" },
  718. { "name": "weight", "type": "Tensor", "context": true },
  719. { "name": "bias", "type": "Tensor", "context": true },
  720. { "name": "Y_scale_i", "type": "float32" },
  721. { "name": "Y_zero_point_i", "type": "int64" }
  722. ],
  723. "outputs": [
  724. { "name": "Y", "type": "Tensor" }
  725. ]
  726. },
  727. {
  728. "name": "ops.quantized.linear_relu",
  729. "category": "Layer",
  730. "inputs": [
  731. { "name": "X", "type": "Tensor" },
  732. { "name": "weight", "type": "Tensor", "context": true },
  733. { "name": "bias", "type": "Tensor", "context": true },
  734. { "name": "Y_scale_i", "type": "float32" },
  735. { "name": "Y_zero_point_i", "type": "int64" }
  736. ],
  737. "outputs": [
  738. { "name": "Y", "type": "Tensor" }
  739. ]
  740. },
  741. {
  742. "name": "ops.quantized.linear_dynamic",
  743. "category": "Layer",
  744. "inputs": [
  745. { "name": "X", "type": "Tensor" },
  746. { "name": "weight", "type": "Tensor", "context": true },
  747. { "name": "bias", "type": "Tensor", "context": true },
  748. { "name": "reduce_range", "type": "boolean", "default": false }
  749. ],
  750. "outputs": [
  751. { "name": "Y", "type": "Tensor" }
  752. ]
  753. },
  754. {
  755. "name": "ops.quantized.linear_relu_dynamic",
  756. "category": "Layer",
  757. "inputs": [
  758. { "name": "X", "type": "Tensor" },
  759. { "name": "weight", "type": "Tensor", "context": true },
  760. { "name": "bias", "type": "Tensor", "context": true },
  761. { "name": "Y_scale_i", "type": "float32" },
  762. { "name": "Y_zero_point_i", "type": "int64" }
  763. ],
  764. "outputs": [
  765. { "name": "Y", "type": "Tensor" }
  766. ]
  767. },
  768. {
  769. "name": "torch.dequantize",
  770. "inputs": [
  771. { "name": "input", "type": "Tensor" }
  772. ],
  773. "outputs": [
  774. { "name": "output", "type": "Tensor" }
  775. ]
  776. },
  777. {
  778. "name": "ops.quantized.conv1d:",
  779. "category": "Layer",
  780. "inputs": [
  781. { "name": "input", "type": "Tensor" },
  782. { "name": "weight", "type": "Tensor", "context": true },
  783. { "name": "bias", "type": "Tensor", "context": true },
  784. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  785. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  786. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  787. { "name": "groups", "type": "int64", "default": 1, "context": true },
  788. { "name": "output_scale", "type": "float32" },
  789. { "name": "output_zero_point", "type": "int64" }
  790. ],
  791. "outputs": [
  792. { "name": "output", "type": "Tensor" }
  793. ]
  794. },
  795. {
  796. "name": "ops.quantized.conv2d:",
  797. "category": "Layer",
  798. "inputs": [
  799. { "name": "input", "type": "Tensor" },
  800. { "name": "weight", "type": "Tensor", "context": true },
  801. { "name": "bias", "type": "Tensor", "context": true },
  802. { "name": "stride", "type": "int64[]", "default": 1 },
  803. { "name": "padding", "type": "int64[]", "default": 0 },
  804. { "name": "dilation", "type": "int64[]", "default": 1 },
  805. { "name": "groups", "type": "int64", "default": 1 },
  806. { "name": "output_scale", "type": "float32" },
  807. { "name": "output_zero_point", "type": "int64" }
  808. ],
  809. "outputs": [
  810. { "name": "output", "type": "Tensor" }
  811. ]
  812. },
  813. {
  814. "name": "ops.quantized.conv2d:new",
  815. "category": "Layer",
  816. "inputs": [
  817. { "name": "input", "type": "Tensor" },
  818. { "name": "weight", "type": "Tensor", "context": true },
  819. { "name": "bias", "type": "Tensor", "context": true },
  820. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  821. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  822. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  823. { "name": "groups", "type": "int64", "default": 1, "context": true },
  824. { "name": "output_scale", "type": "float32" },
  825. { "name": "output_zero_point", "type": "int64" }
  826. ],
  827. "outputs": [
  828. { "name": "output", "type": "Tensor" }
  829. ]
  830. },
  831. {
  832. "name": "ops.quantized.conv3d:",
  833. "category": "Layer",
  834. "inputs": [
  835. { "name": "input", "type": "Tensor" },
  836. { "name": "weight", "type": "Tensor", "context": true },
  837. { "name": "bias", "type": "Tensor", "context": true },
  838. { "name": "stride", "type": "int64[]", "default": 1 },
  839. { "name": "padding", "type": "int64[]", "default": 0 },
  840. { "name": "dilation", "type": "int64[]", "default": 1 },
  841. { "name": "groups", "type": "int64", "default": 1 },
  842. { "name": "output_scale", "type": "float32" },
  843. { "name": "output_zero_point", "type": "int64" }
  844. ],
  845. "outputs": [
  846. { "name": "output", "type": "Tensor" }
  847. ]
  848. },
  849. {
  850. "name": "ops.quantized.conv3d:new",
  851. "category": "Layer",
  852. "inputs": [
  853. { "name": "input", "type": "Tensor" },
  854. { "name": "weight", "type": "Tensor", "context": true },
  855. { "name": "bias", "type": "Tensor", "context": true },
  856. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  857. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  858. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  859. { "name": "groups", "type": "int64", "default": 1, "context": true },
  860. { "name": "output_scale", "type": "float32" },
  861. { "name": "output_zero_point", "type": "int64" }
  862. ],
  863. "outputs": [
  864. { "name": "output", "type": "Tensor" }
  865. ]
  866. },
  867. {
  868. "name": "ops.quantized.conv1d_relu:",
  869. "category": "Layer",
  870. "inputs": [
  871. { "name": "input", "type": "Tensor" },
  872. { "name": "weight", "type": "Tensor", "context": true },
  873. { "name": "bias", "type": "Tensor", "context": true },
  874. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  875. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  876. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  877. { "name": "groups", "type": "int64", "default": 1, "context": true },
  878. { "name": "output_scale", "type": "float64" },
  879. { "name": "output_zero_point", "type": "int64" }
  880. ],
  881. "outputs": [
  882. { "name": "output", "type": "Tensor" }
  883. ]
  884. },
  885. {
  886. "name": "ops.quantized.conv2d_relu:",
  887. "category": "Layer",
  888. "inputs": [
  889. { "name": "input", "type": "Tensor" },
  890. { "name": "weight", "type": "Tensor", "context": true },
  891. { "name": "bias", "type": "Tensor", "context": true },
  892. { "name": "stride", "type": "int64[]", "default": 1 },
  893. { "name": "padding", "type": "int64[]", "default": 0 },
  894. { "name": "dilation", "type": "int64[]", "default": 1 },
  895. { "name": "groups", "type": "int64", "default": 1 },
  896. { "name": "output_scale", "type": "float64" },
  897. { "name": "output_zero_point", "type": "int64" }
  898. ],
  899. "outputs": [
  900. { "name": "output", "type": "Tensor" }
  901. ]
  902. },
  903. {
  904. "name": "ops.quantized.conv2d_relu:new",
  905. "category": "Layer",
  906. "inputs": [
  907. { "name": "input", "type": "Tensor" },
  908. { "name": "weight", "type": "Tensor", "context": true },
  909. { "name": "bias", "type": "Tensor", "context": true },
  910. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  911. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  912. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  913. { "name": "groups", "type": "int64", "default": 1, "context": true },
  914. { "name": "output_scale", "type": "float64" },
  915. { "name": "output_zero_point", "type": "int64" }
  916. ],
  917. "outputs": [
  918. { "name": "output", "type": "Tensor" }
  919. ]
  920. },
  921. {
  922. "name": "ops.quantized.conv3d_relu:",
  923. "category": "Layer",
  924. "inputs": [
  925. { "name": "input", "type": "Tensor" },
  926. { "name": "weight", "type": "Tensor", "context": true },
  927. { "name": "bias", "type": "Tensor", "context": true },
  928. { "name": "stride", "type": "int64[]", "default": 1 },
  929. { "name": "padding", "type": "int64[]", "default": 0 },
  930. { "name": "dilation", "type": "int64[]", "default": 1 },
  931. { "name": "groups", "type": "int64", "default": 1 },
  932. { "name": "output_scale", "type": "float64" },
  933. { "name": "output_zero_point", "type": "int64" }
  934. ],
  935. "outputs": [
  936. { "name": "output", "type": "Tensor" }
  937. ]
  938. },
  939. {
  940. "name": "ops.quantized.conv3d_relu:new",
  941. "category": "Layer",
  942. "inputs": [
  943. { "name": "input", "type": "Tensor" },
  944. { "name": "weight", "type": "Tensor", "context": true },
  945. { "name": "bias", "type": "Tensor", "context": true },
  946. { "name": "stride", "type": "int64[]", "default": 1, "context": true },
  947. { "name": "padding", "type": "int64[]", "default": 0, "context": true },
  948. { "name": "dilation", "type": "int64[]", "default": 1, "context": true },
  949. { "name": "groups", "type": "int64", "default": 1, "context": true },
  950. { "name": "output_scale", "type": "float64" },
  951. { "name": "output_zero_point", "type": "int64" }
  952. ],
  953. "outputs": [
  954. { "name": "output", "type": "Tensor" }
  955. ]
  956. },
  957. {
  958. "name": "ops.quantized.add_relu:",
  959. "inputs": [
  960. { "name": "qa", "type": "Tensor" },
  961. { "name": "qb", "type": "Tensor" },
  962. { "name": "scale", "type": "float32" },
  963. { "name": "zero_point", "type": "int64" }
  964. ],
  965. "outputs": [
  966. { "name": "qc", "type": "Tensor" }
  967. ]
  968. },
  969. {
  970. "name": "ops.quantized.add_relu:Scalar",
  971. "inputs": [
  972. { "name": "qa", "type": "Tensor" },
  973. { "name": "b", "type": "Scalar" }
  974. ],
  975. "outputs": [
  976. { "name": "qc", "type": "Tensor" }
  977. ]
  978. },
  979. {
  980. "name": "ops.quantized.cat",
  981. "category": "Tensor",
  982. "inputs": [
  983. { "name": "qx", "type": "Tensor[]" },
  984. { "name": "dim", "type": "int64" },
  985. { "name": "scale", "type": "float32", "optional": true },
  986. { "name": "zero_point", "type": "int64", "optional": true }
  987. ],
  988. "outputs": [
  989. { "name": "output", "type": "Tensor" }
  990. ]
  991. },
  992. {
  993. "name": "ops.quantized.cat_relu",
  994. "category": "Tensor",
  995. "inputs": [
  996. { "name": "qx", "type": "Tensor[]" },
  997. { "name": "dim", "type": "int64" },
  998. { "name": "scale", "type": "float32", "optional": true },
  999. { "name": "zero_point", "type": "int64", "optional": true }
  1000. ],
  1001. "outputs": [
  1002. { "name": "output", "type": "Tensor" }
  1003. ]
  1004. },
  1005. {
  1006. "name": "ops.quantized.batch_norm:",
  1007. "inputs": [
  1008. { "name": "qx", "type": "Tensor" },
  1009. { "name": "weight", "type": "Tensor", "optional": true },
  1010. { "name": "bias", "type": "Tensor", "optional": true },
  1011. { "name": "mean", "type": "Tensor" },
  1012. { "name": "var", "type": "Tensor" },
  1013. { "name": "eps", "type": "float32" },
  1014. { "name": "output_scale", "type": "float32", "optional": true },
  1015. { "name": "outputzero_point", "type": "int64", "optional": true }
  1016. ],
  1017. "outputs": [
  1018. { "name": "output", "type": "Tensor" }
  1019. ]
  1020. },
  1021. {
  1022. "name": "ops.quantized.batch_norm_relu",
  1023. "category": "Normalization",
  1024. "inputs": [
  1025. { "name": "qx", "type": "Tensor" },
  1026. { "name": "weight", "type": "Tensor", "optional": true },
  1027. { "name": "bias", "type": "Tensor", "optional": true },
  1028. { "name": "mean", "type": "Tensor" },
  1029. { "name": "var", "type": "Tensor" },
  1030. { "name": "eps", "type": "float32" },
  1031. { "name": "output_scale", "type": "float32", "optional": true },
  1032. { "name": "outputzero_point", "type": "int64", "optional": true }
  1033. ],
  1034. "outputs": [
  1035. { "name": "output", "type": "Tensor" }
  1036. ]
  1037. },
  1038. {
  1039. "name": "ops.quantized.batch_norm1d",
  1040. "category": "Normalization",
  1041. "inputs": [
  1042. { "name": "qx", "type": "Tensor" },
  1043. { "name": "weight", "type": "Tensor", "optional": true },
  1044. { "name": "bias", "type": "Tensor", "optional": true },
  1045. { "name": "mean", "type": "Tensor" },
  1046. { "name": "var", "type": "Tensor" },
  1047. { "name": "eps", "type": "float32" },
  1048. { "name": "output_scale", "type": "float32", "optional": true },
  1049. { "name": "outputzero_point", "type": "int64", "optional": true }
  1050. ],
  1051. "outputs": [
  1052. { "name": "output", "type": "Tensor" }
  1053. ]
  1054. },
  1055. {
  1056. "name": "ops.quantized.batch_norm1d_relu",
  1057. "category": "Normalization",
  1058. "inputs": [
  1059. { "name": "qx", "type": "Tensor" },
  1060. { "name": "weight", "type": "Tensor", "optional": true },
  1061. { "name": "bias", "type": "Tensor", "optional": true },
  1062. { "name": "mean", "type": "Tensor" },
  1063. { "name": "var", "type": "Tensor" },
  1064. { "name": "eps", "type": "float32" },
  1065. { "name": "output_scale", "type": "float32", "optional": true },
  1066. { "name": "outputzero_point", "type": "int64", "optional": true }
  1067. ],
  1068. "outputs": [
  1069. { "name": "output", "type": "Tensor" }
  1070. ]
  1071. },
  1072. {
  1073. "name": "ops.quantized.batch_norm2d",
  1074. "category": "Normalization",
  1075. "inputs": [
  1076. { "name": "qx", "type": "Tensor" },
  1077. { "name": "weight", "type": "Tensor", "optional": true },
  1078. { "name": "bias", "type": "Tensor", "optional": true },
  1079. { "name": "mean", "type": "Tensor" },
  1080. { "name": "var", "type": "Tensor" },
  1081. { "name": "eps", "type": "float32" },
  1082. { "name": "output_scale", "type": "float32", "optional": true },
  1083. { "name": "outputzero_point", "type": "int64", "optional": true }
  1084. ],
  1085. "outputs": [
  1086. { "name": "output", "type": "Tensor" }
  1087. ]
  1088. },
  1089. {
  1090. "name": "ops.quantized.batch_norm2d_relu",
  1091. "category": "Normalization",
  1092. "inputs": [
  1093. { "name": "qx", "type": "Tensor" },
  1094. { "name": "weight", "type": "Tensor", "optional": true },
  1095. { "name": "bias", "type": "Tensor", "optional": true },
  1096. { "name": "mean", "type": "Tensor" },
  1097. { "name": "var", "type": "Tensor" },
  1098. { "name": "eps", "type": "float32" },
  1099. { "name": "output_scale", "type": "float32", "optional": true },
  1100. { "name": "outputzero_point", "type": "int64", "optional": true }
  1101. ],
  1102. "outputs": [
  1103. { "name": "output", "type": "Tensor" }
  1104. ]
  1105. },
  1106. {
  1107. "name": "ops.quantized.batch_norm3d",
  1108. "category": "Normalization",
  1109. "inputs": [
  1110. { "name": "qx", "type": "Tensor" },
  1111. { "name": "weight", "type": "Tensor", "optional": true },
  1112. { "name": "bias", "type": "Tensor", "optional": true },
  1113. { "name": "mean", "type": "Tensor" },
  1114. { "name": "var", "type": "Tensor" },
  1115. { "name": "eps", "type": "float32" },
  1116. { "name": "output_scale", "type": "float32", "optional": true },
  1117. { "name": "outputzero_point", "type": "int64", "optional": true }
  1118. ],
  1119. "outputs": [
  1120. { "name": "output", "type": "Tensor" }
  1121. ]
  1122. },
  1123. {
  1124. "name": "ops.quantized.batch_norm3d_relu",
  1125. "category": "Normalization",
  1126. "inputs": [
  1127. { "name": "qx", "type": "Tensor" },
  1128. { "name": "weight", "type": "Tensor", "optional": true },
  1129. { "name": "bias", "type": "Tensor", "optional": true },
  1130. { "name": "mean", "type": "Tensor" },
  1131. { "name": "var", "type": "Tensor" },
  1132. { "name": "eps", "type": "float32" },
  1133. { "name": "output_scale", "type": "float32", "optional": true },
  1134. { "name": "outputzero_point", "type": "int64", "optional": true }
  1135. ],
  1136. "outputs": [
  1137. { "name": "output", "type": "Tensor" }
  1138. ]
  1139. },
  1140. {
  1141. "name": "torch.max_pool1d",
  1142. "category": "Pool",
  1143. "attributes": [
  1144. { "name": "kernel_size", "type": "int64[]" },
  1145. { "name": "stride", "type": "int64[]" },
  1146. { "name": "padding", "type": "int64[]", "default": 0 },
  1147. { "name": "dilation", "type": "int64[]", "default": 1 },
  1148. { "name": "ceil_mode", "type": "boolean", "default": false }
  1149. ],
  1150. "inputs": [
  1151. { "name": "self", "type": "Tensor" }
  1152. ],
  1153. "outputs": [
  1154. { "name": "output", "type": "Tensor" }
  1155. ]
  1156. },
  1157. {
  1158. "name": "torch.max_pool2d",
  1159. "category": "Pool",
  1160. "attributes": [
  1161. { "name": "kernel_size", "type": "int64[]" },
  1162. { "name": "stride", "type": "int64[]" },
  1163. { "name": "padding", "type": "int64[]", "default": 0 },
  1164. { "name": "dilation", "type": "int64[]", "default": 1 },
  1165. { "name": "ceil_mode", "type": "boolean", "default": false }
  1166. ],
  1167. "inputs": [
  1168. { "name": "self", "type": "Tensor" }
  1169. ],
  1170. "outputs": [
  1171. { "name": "output", "type": "Tensor" }
  1172. ]
  1173. },
  1174. {
  1175. "name": "torch.max_pool3d",
  1176. "category": "Pool",
  1177. "attributes": [
  1178. { "name": "kernel_size", "type": "int64[]" },
  1179. { "name": "stride", "type": "int64[]" },
  1180. { "name": "padding", "type": "int64[]", "default": 0 },
  1181. { "name": "dilation", "type": "int64[]", "default": 1 },
  1182. { "name": "ceil_mode", "type": "boolean", "default": false }
  1183. ],
  1184. "inputs": [
  1185. { "name": "self", "type": "Tensor" }
  1186. ],
  1187. "outputs": [
  1188. { "name": "output", "type": "Tensor" }
  1189. ]
  1190. },
  1191. {
  1192. "name": "torch.max_unpool2d",
  1193. "category": "Pool",
  1194. "inputs": [
  1195. { "name": "self", "type": "Tensor" },
  1196. { "name": "indices", "type": "Tensor" },
  1197. { "name": "output_size", "type": "int64[]" }
  1198. ],
  1199. "outputs": [
  1200. { "name": "output", "type": "Tensor" }
  1201. ]
  1202. },
  1203. {
  1204. "name": "torch.max_unpool3d",
  1205. "category": "Pool",
  1206. "inputs": [
  1207. { "name": "self", "type": "Tensor" },
  1208. { "name": "indices", "type": "Tensor" },
  1209. { "name": "output_size", "type": "int64[]" },
  1210. { "name": "stride", "type": "int64[]" },
  1211. { "name": "padding", "type": "int64[]" }
  1212. ],
  1213. "outputs": [
  1214. { "name": "output", "type": "Tensor" }
  1215. ]
  1216. },
  1217. {
  1218. "name": "torch.max_pool1d_with_indices",
  1219. "category": "Pool",
  1220. "attributes": [
  1221. { "name": "kernel_size", "type": "int64[]" },
  1222. { "name": "stride", "type": "int64[]" },
  1223. { "name": "padding", "type": "int64[]", "default": 0 },
  1224. { "name": "dilation", "type": "int64[]", "default": 1 },
  1225. { "name": "ceil_mode", "type": "boolean", "default": false },
  1226. { "name": "return_indices", "type": "boolean", "default": false }
  1227. ],
  1228. "inputs": [
  1229. { "name": "input", "type": "Tensor" }
  1230. ],
  1231. "outputs": [
  1232. { "name": "output", "type": "Tensor" },
  1233. { "name": "indices", "type": "Tensor" }
  1234. ]
  1235. },
  1236. {
  1237. "name": "torch.max_pool2d_with_indices",
  1238. "category": "Pool",
  1239. "attributes": [
  1240. { "name": "kernel_size", "type": "int64[]" },
  1241. { "name": "stride", "type": "int64[]" },
  1242. { "name": "padding", "type": "int64[]", "default": 0 },
  1243. { "name": "dilation", "type": "int64[]", "default": 1 },
  1244. { "name": "ceil_mode", "type": "boolean", "default": false },
  1245. { "name": "return_indices", "type": "boolean", "default": false }
  1246. ],
  1247. "inputs": [
  1248. { "name": "input", "type": "Tensor" }
  1249. ],
  1250. "outputs": [
  1251. { "name": "output", "type": "Tensor" },
  1252. { "name": "indices", "type": "Tensor" }
  1253. ]
  1254. },
  1255. {
  1256. "name": "torch.adaptive_avg_pool1d",
  1257. "category": "Pool",
  1258. "inputs": [
  1259. { "name": "input", "type": "Tensor" },
  1260. { "name": "output_size", "type": "int64[]", "visible": false }
  1261. ],
  1262. "outputs": [
  1263. { "name": "output", "type": "Tensor" }
  1264. ]
  1265. },
  1266. {
  1267. "name": "torch.adaptive_avg_pool2d",
  1268. "category": "Pool",
  1269. "inputs": [
  1270. { "name": "input", "type": "Tensor" },
  1271. { "name": "output_size", "type": "int64[]", "visible": false }
  1272. ],
  1273. "outputs": [
  1274. { "name": "output", "type": "Tensor" }
  1275. ]
  1276. },
  1277. {
  1278. "name": "torch.adaptive_avg_pool3d",
  1279. "category": "Pool",
  1280. "inputs": [
  1281. { "name": "input", "type": "Tensor" },
  1282. { "name": "output_size", "type": "int64[]", "visible": false }
  1283. ],
  1284. "outputs": [
  1285. { "name": "output", "type": "Tensor" }
  1286. ]
  1287. },
  1288. {
  1289. "name": "torch.adaptive_max_pool1d",
  1290. "category": "Pool",
  1291. "inputs": [
  1292. { "name": "input", "type": "Tensor" },
  1293. { "name": "output_size", "type": "int64[]", "visible": false }
  1294. ],
  1295. "outputs": [
  1296. { "name": "output", "type": "Tensor" },
  1297. { "name": "?", "type": "Tensor" }
  1298. ]
  1299. },
  1300. {
  1301. "name": "torch.adaptive_max_pool2d",
  1302. "category": "Pool",
  1303. "inputs": [
  1304. { "name": "input", "type": "Tensor" },
  1305. { "name": "output_size", "type": "int64[]", "visible": false }
  1306. ],
  1307. "outputs": [
  1308. { "name": "output", "type": "Tensor" },
  1309. { "name": "?", "type": "Tensor" }
  1310. ]
  1311. },
  1312. {
  1313. "name": "torch.adaptive_max_pool3d",
  1314. "category": "Pool",
  1315. "inputs": [
  1316. { "name": "input", "type": "Tensor" },
  1317. { "name": "output_size", "type": "int64[]", "visible": false }
  1318. ],
  1319. "outputs": [
  1320. { "name": "output", "type": "Tensor" },
  1321. { "name": "?", "type": "Tensor" }
  1322. ]
  1323. },
  1324. {
  1325. "name": "torch.atan",
  1326. "inputs": [
  1327. { "name": "self", "type": "Tensor" }
  1328. ],
  1329. "outputs": [
  1330. { "name": "output", "type": "Tensor" }
  1331. ]
  1332. },
  1333. {
  1334. "name": "torch.atan_",
  1335. "inputs": [
  1336. { "name": "self", "type": "Tensor" }
  1337. ],
  1338. "outputs": [
  1339. { "name": "output", "type": "Tensor" }
  1340. ]
  1341. },
  1342. {
  1343. "name": "torch.atanh",
  1344. "inputs": [
  1345. { "name": "self", "type": "Tensor" }
  1346. ],
  1347. "outputs": [
  1348. { "name": "output", "type": "Tensor" }
  1349. ]
  1350. },
  1351. {
  1352. "name": "torch.atanh_",
  1353. "inputs": [
  1354. { "name": "self", "type": "Tensor" }
  1355. ],
  1356. "outputs": [
  1357. { "name": "output", "type": "Tensor" }
  1358. ]
  1359. },
  1360. {
  1361. "name": "torch.arctan",
  1362. "inputs": [
  1363. { "name": "self", "type": "Tensor" }
  1364. ],
  1365. "outputs": [
  1366. { "name": "output", "type": "Tensor" }
  1367. ]
  1368. },
  1369. {
  1370. "name": "torch.arctan_",
  1371. "inputs": [
  1372. { "name": "self", "type": "Tensor" }
  1373. ],
  1374. "outputs": [
  1375. { "name": "output", "type": "Tensor" }
  1376. ]
  1377. },
  1378. {
  1379. "name": "torch.avg_pool1d",
  1380. "category": "Pool",
  1381. "inputs": [
  1382. { "name": "input", "type": "Tensor" },
  1383. { "name": "kernel_size", "type": "int64[]" },
  1384. { "name": "stride", "type": "int64[]" },
  1385. { "name": "padding", "type": "int64[]", "default": 0 },
  1386. { "name": "ceil_mode", "type": "boolean", "default": false },
  1387. { "name": "count_include_pad", "type": "boolean", "default": true },
  1388. { "name": "divisor_override" }
  1389. ],
  1390. "outputs": [
  1391. { "name": "output", "type": "Tensor" }
  1392. ]
  1393. },
  1394. {
  1395. "name": "torch.avg_pool2d",
  1396. "category": "Pool",
  1397. "attributes": [
  1398. { "name": "kernel_size", "type": "int64[]" },
  1399. { "name": "stride", "type": "int64[]" },
  1400. { "name": "padding", "type": "int64[]" },
  1401. { "name": "ceil_mode", "type": "boolean", "default": false },
  1402. { "name": "count_include_pad", "type": "boolean", "default": true },
  1403. { "name": "divisor_override", "type": "int64", "optional": true }
  1404. ],
  1405. "inputs": [
  1406. { "name": "input", "type": "Tensor" }
  1407. ],
  1408. "outputs": [
  1409. { "name": "output", "type": "Tensor" }
  1410. ]
  1411. },
  1412. {
  1413. "name": "torch.avg_pool3d",
  1414. "category": "Pool",
  1415. "attributes": [
  1416. { "name": "kernel_size", "type": "int64[]" },
  1417. { "name": "stride", "type": "int64[]" },
  1418. { "name": "padding", "type": "int64[]" },
  1419. { "name": "ceil_mode", "type": "boolean", "default": false },
  1420. { "name": "count_include_pad", "type": "boolean", "default": true },
  1421. { "name": "divisor_override", "type": "int64", "optional": true }
  1422. ],
  1423. "inputs": [
  1424. { "name": "input", "type": "Tensor" }
  1425. ],
  1426. "outputs": [
  1427. { "name": "output", "type": "Tensor" }
  1428. ]
  1429. },
  1430. {
  1431. "name": "torch.batch_norm",
  1432. "category": "Normalization",
  1433. "attributes": [
  1434. { "name": "training", "type": "boolean", "visible": false },
  1435. { "name": "momentum", "type": "float32", "default": 0.1 },
  1436. { "name": "eps", "type": "float32", "default": 0.00001 },
  1437. { "name": "cudnn_enabled", "type": "boolean", "visible": false }
  1438. ],
  1439. "inputs": [
  1440. { "name": "input", "type": "Tensor" },
  1441. { "name": "weight", "type": "Tensor", "optional": true },
  1442. { "name": "bias", "type": "Tensor", "optional": true },
  1443. { "name": "running_mean", "type": "Tensor", "optional": true },
  1444. { "name": "running_var", "type": "Tensor", "optional": true }
  1445. ],
  1446. "outputs": [
  1447. { "name": "output", "type": "Tensor" }
  1448. ]
  1449. },
  1450. {
  1451. "name": "torch.gather:",
  1452. "inputs": [
  1453. { "name": "selm", "type": "Tensor" },
  1454. { "name": "dim", "type": "int64" },
  1455. { "name": "index", "type": "Tensor" },
  1456. { "name": "sparse_grad", "type": "boolean", "default": false }
  1457. ],
  1458. "outputs": [
  1459. { "name": "output", "type": "Tensor" }
  1460. ]
  1461. },
  1462. {
  1463. "name": "torch.gather:dimname",
  1464. "inputs": [
  1465. { "name": "selm", "type": "Tensor" },
  1466. { "name": "dim", "type": "Dimname" },
  1467. { "name": "index", "type": "Tensor" },
  1468. { "name": "sparse_grad", "type": "boolean", "default": false }
  1469. ],
  1470. "outputs": [
  1471. { "name": "output", "type": "Tensor" }
  1472. ]
  1473. },
  1474. {
  1475. "name": "torch.group_norm",
  1476. "category": "Normalization",
  1477. "inputs": [
  1478. { "name": "input", "type": "Tensor" },
  1479. { "name": "num_groups", "type": "int64" },
  1480. { "name": "weight", "type": "Tensor", "optional": true },
  1481. { "name": "bias", "type": "Tensor", "optional": true },
  1482. { "name": "eps", "type": "float32", "default": 0.00001 },
  1483. { "name": "cudnn_enabled", "type": "boolean", "visible": true }
  1484. ],
  1485. "outputs": [
  1486. { "name": "output", "type": "Tensor" }
  1487. ]
  1488. },
  1489. {
  1490. "name": "torch.relu",
  1491. "category": "Activation",
  1492. "inputs": [
  1493. { "name": "self", "type": "Tensor" }
  1494. ],
  1495. "outputs": [
  1496. { "name": "output", "type": "Tensor" }
  1497. ]
  1498. },
  1499. {
  1500. "name": "torch.relu_",
  1501. "category": "Activation",
  1502. "inputs": [
  1503. { "name": "self", "type": "Tensor" }
  1504. ],
  1505. "outputs": [
  1506. { "name": "output", "type": "Tensor" }
  1507. ]
  1508. },
  1509. {
  1510. "name": "torch.relu6",
  1511. "category": "Activation",
  1512. "inputs": [
  1513. { "name": "self", "type": "Tensor" }
  1514. ],
  1515. "outputs": [
  1516. { "name": "output", "type": "Tensor" }
  1517. ]
  1518. },
  1519. {
  1520. "name": "torch.relu6_",
  1521. "category": "Activation",
  1522. "inputs": [
  1523. { "name": "self", "type": "Tensor" }
  1524. ],
  1525. "outputs": [
  1526. { "name": "output", "type": "Tensor" }
  1527. ]
  1528. },
  1529. {
  1530. "name": "torch.prelu",
  1531. "category": "Activation",
  1532. "inputs": [
  1533. { "name": "self", "type": "Tensor" },
  1534. { "name": "weight", "type": "Tensor" }
  1535. ],
  1536. "outputs": [
  1537. { "name": "output", "type": "Tensor" }
  1538. ]
  1539. },
  1540. {
  1541. "name": "torch.hardtanh",
  1542. "category": "Activation",
  1543. "attributes": [
  1544. { "name": "min_value", "type": "float64" },
  1545. { "name": "max_value", "type": "float64" }
  1546. ],
  1547. "inputs": [
  1548. { "name": "self", "type": "Tensor" }
  1549. ],
  1550. "outputs": [
  1551. { "name": "output", "type": "Tensor" }
  1552. ]
  1553. },
  1554. {
  1555. "name": "torch.hardtanh_",
  1556. "category": "Activation",
  1557. "attributes": [
  1558. { "name": "min_value", "type": "float64" },
  1559. { "name": "max_value", "type": "float64" }
  1560. ],
  1561. "inputs": [
  1562. { "name": "self", "type": "Tensor" }
  1563. ],
  1564. "outputs": [
  1565. { "name": "output", "type": "Tensor" }
  1566. ]
  1567. },
  1568. {
  1569. "name": "torch.hardswish",
  1570. "category": "Activation",
  1571. "inputs": [
  1572. { "name": "self", "type": "Tensor" }
  1573. ],
  1574. "outputs": [
  1575. { "name": "output", "type": "Tensor" }
  1576. ]
  1577. },
  1578. {
  1579. "name": "torch.hardswish_",
  1580. "category": "Activation",
  1581. "inputs": [
  1582. { "name": "self", "type": "Tensor" }
  1583. ],
  1584. "outputs": [
  1585. { "name": "output", "type": "Tensor" }
  1586. ]
  1587. },
  1588. {
  1589. "name": "torch.hardsigmoid",
  1590. "category": "Activation",
  1591. "inputs": [
  1592. { "name": "self", "type": "Tensor" }
  1593. ],
  1594. "outputs": [
  1595. { "name": "output", "type": "Tensor" }
  1596. ]
  1597. },
  1598. {
  1599. "name": "torch.hardsigmoid_",
  1600. "category": "Activation",
  1601. "inputs": [
  1602. { "name": "self", "type": "Tensor" }
  1603. ],
  1604. "outputs": [
  1605. { "name": "output", "type": "Tensor" }
  1606. ]
  1607. },
  1608. {
  1609. "name": "torch.sigmoid",
  1610. "category": "Activation",
  1611. "inputs": [
  1612. { "name": "input", "type": "Tensor" }
  1613. ],
  1614. "outputs": [
  1615. { "name": "output", "type": "Tensor" }
  1616. ]
  1617. },
  1618. {
  1619. "name": "torch.sigmoid_",
  1620. "category": "Activation",
  1621. "inputs": [
  1622. { "name": "input", "type": "Tensor" }
  1623. ],
  1624. "outputs": [
  1625. { "name": "output", "type": "Tensor" }
  1626. ]
  1627. },
  1628. {
  1629. "name": "torch.threshold",
  1630. "category": "Activation",
  1631. "inputs": [
  1632. { "name": "input", "type": "Tensor" },
  1633. { "name": "threshold", "type": "float64", "default": 0 },
  1634. { "name": "value", "type": "float64", "default": 0 },
  1635. { "name": "inplace", "type": "boolean", "default": false }
  1636. ],
  1637. "outputs": [
  1638. { "name": "output", "type": "Tensor" }
  1639. ]
  1640. },
  1641. {
  1642. "name": "torch.threshold_",
  1643. "category": "Activation",
  1644. "inputs": [
  1645. { "name": "input", "type": "Tensor" },
  1646. { "name": "threshold", "type": "float64" },
  1647. { "name": "value", "type": "float64" },
  1648. { "name": "inplace", "type": "boolean" }
  1649. ],
  1650. "outputs": [
  1651. { "name": "output", "type": "Tensor" }
  1652. ]
  1653. },
  1654. {
  1655. "name": "torch.tile",
  1656. "inputs": [
  1657. { "name": "self", "type": "Tensor" },
  1658. { "name": "dims", "type": "int64[]" }
  1659. ],
  1660. "outputs": [
  1661. { "name": "output", "type": "Tensor" }
  1662. ]
  1663. },
  1664. {
  1665. "name": "torch.softmax:int",
  1666. "category": "Activation",
  1667. "inputs": [
  1668. { "name": "input", "type": "Tensor" },
  1669. { "name": "dim", "type": "int64" },
  1670. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1671. ],
  1672. "outputs": [
  1673. { "name": "output", "type": "Tensor" }
  1674. ]
  1675. },
  1676. {
  1677. "name": "torch.softmax:Dimname",
  1678. "category": "Activation",
  1679. "inputs": [
  1680. { "name": "input", "type": "Tensor" },
  1681. { "name": "dim", "type": "Dimname" },
  1682. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1683. ],
  1684. "outputs": [
  1685. { "name": "output", "type": "Tensor" }
  1686. ]
  1687. },
  1688. {
  1689. "name": "torch.log_softmax:int",
  1690. "category": "Activation",
  1691. "attributes": [
  1692. { "name": "dim", "type": "int64" },
  1693. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1694. ],
  1695. "inputs": [
  1696. { "name": "input", "type": "Tensor" }
  1697. ],
  1698. "outputs": [
  1699. { "name": "output", "type": "Tensor" }
  1700. ]
  1701. },
  1702. {
  1703. "name": "torch.log_softmax:Dimname",
  1704. "category": "Activation",
  1705. "attributes": [
  1706. { "name": "dim", "type": "Dimname" },
  1707. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1708. ],
  1709. "inputs": [
  1710. { "name": "input", "type": "Tensor" }
  1711. ],
  1712. "outputs": [
  1713. { "name": "output", "type": "Tensor" }
  1714. ]
  1715. },
  1716. {
  1717. "name": "torch.tanh",
  1718. "category": "Activation",
  1719. "inputs": [
  1720. { "name": "input", "type": "Tensor" }
  1721. ],
  1722. "outputs": [
  1723. { "name": "output", "type": "Tensor" }
  1724. ]
  1725. },
  1726. {
  1727. "name": "torch.take",
  1728. "category": "Activation",
  1729. "inputs": [
  1730. { "name": "self", "type": "Tensor" },
  1731. { "name": "index", "type": "Tensor" }
  1732. ],
  1733. "outputs": [
  1734. { "name": "output", "type": "Tensor" }
  1735. ]
  1736. },
  1737. {
  1738. "name": "torch.leaky_relu",
  1739. "category": "Activation",
  1740. "attributes": [
  1741. { "name": "negative_slope", "type": "Scalar", "default": 0.01 }
  1742. ],
  1743. "inputs": [
  1744. { "name": "input", "type": "Tensor" }
  1745. ],
  1746. "outputs": [
  1747. { "name": "output", "type": "Tensor" }
  1748. ]
  1749. },
  1750. {
  1751. "name": "torch.slice:Tensor",
  1752. "category": "Tensor",
  1753. "inputs": [
  1754. { "name": "self", "type": "Tensor" },
  1755. { "name": "dim", "type": "int64", "default": 0 },
  1756. { "name": "start", "type": "int64", "optional": true, "default": null },
  1757. { "name": "end", "type": "int64", "optional": true, "default": null },
  1758. { "name": "step", "type": "int64", "default": 1 }
  1759. ],
  1760. "outputs": [
  1761. { "name": "output", "type": "Tensor" }
  1762. ]
  1763. },
  1764. {
  1765. "name": "torch.dropout",
  1766. "category": "Dropout",
  1767. "attributes": [
  1768. { "name": "p", "type": "float64", "default": 0.5 },
  1769. { "name": "train", "type": "boolean", "visible": false }
  1770. ],
  1771. "inputs": [
  1772. { "name": "input", "type": "Tensor" }
  1773. ],
  1774. "outputs": [
  1775. { "name": "output", "type": "Tensor" }
  1776. ]
  1777. },
  1778. {
  1779. "name": "torch.dropout_",
  1780. "category": "Dropout",
  1781. "attributes": [
  1782. { "name": "p", "type": "float64", "default": 0.5 },
  1783. { "name": "train", "type": "boolean", "visible": false }
  1784. ],
  1785. "inputs": [
  1786. { "name": "input", "type": "Tensor" }
  1787. ],
  1788. "outputs": [
  1789. { "name": "output", "type": "Tensor" }
  1790. ]
  1791. },
  1792. {
  1793. "name": "torch.feature_dropout",
  1794. "category": "Dropout",
  1795. "attributes": [
  1796. { "name": "p", "type": "float64" },
  1797. { "name": "train", "type": "boolean" }
  1798. ],
  1799. "inputs": [
  1800. { "name": "input", "type": "Tensor" }
  1801. ],
  1802. "outputs": [
  1803. { "name": "output", "type": "Tensor" }
  1804. ]
  1805. },
  1806. {
  1807. "name": "torch.feature_dropout_",
  1808. "category": "Dropout",
  1809. "attributes": [
  1810. { "name": "p", "type": "float64" },
  1811. { "name": "train", "type": "boolean" }
  1812. ],
  1813. "inputs": [
  1814. { "name": "input", "type": "Tensor" }
  1815. ],
  1816. "outputs": [
  1817. { "name": "output", "type": "Tensor" }
  1818. ]
  1819. },
  1820. {
  1821. "name": "torch.alpha_dropout",
  1822. "category": "Dropout",
  1823. "attributes": [
  1824. { "name": "p", "type": "float64" },
  1825. { "name": "train", "type": "boolean" }
  1826. ],
  1827. "inputs": [
  1828. { "name": "input", "type": "Tensor" }
  1829. ],
  1830. "outputs": [
  1831. { "name": "output", "type": "Tensor" }
  1832. ]
  1833. },
  1834. {
  1835. "name": "torch.alpha_dropout_",
  1836. "category": "Dropout",
  1837. "attributes": [
  1838. { "name": "p", "type": "float64" },
  1839. { "name": "train", "type": "boolean" }
  1840. ],
  1841. "inputs": [
  1842. { "name": "input", "type": "Tensor" }
  1843. ],
  1844. "outputs": [
  1845. { "name": "output", "type": "Tensor" }
  1846. ]
  1847. },
  1848. {
  1849. "name": "torch.feature_alpha_dropout",
  1850. "category": "Dropout",
  1851. "attributes": [
  1852. { "name": "p", "type": "float64" },
  1853. { "name": "train", "type": "boolean" }
  1854. ],
  1855. "inputs": [
  1856. { "name": "input", "type": "Tensor" }
  1857. ],
  1858. "outputs": [
  1859. { "name": "output", "type": "Tensor" }
  1860. ]
  1861. },
  1862. {
  1863. "name": "torch.feature_alpha_dropout_",
  1864. "category": "Dropout",
  1865. "attributes": [
  1866. { "name": "p", "type": "float64" },
  1867. { "name": "train", "type": "boolean" }
  1868. ],
  1869. "inputs": [
  1870. { "name": "input", "type": "Tensor" }
  1871. ],
  1872. "outputs": [
  1873. { "name": "output", "type": "Tensor" }
  1874. ]
  1875. },
  1876. {
  1877. "name": "torch.mm",
  1878. "inputs": [
  1879. { "name": "self", "type": "Tensor" },
  1880. { "name": "mat2", "type": "Tensor" }
  1881. ],
  1882. "outputs": [
  1883. { "name": "output", "type": "Tensor" }
  1884. ]
  1885. },
  1886. {
  1887. "name": "torch.addmm",
  1888. "category": "Layer",
  1889. "attributes": [
  1890. { "name": "alpha", "type": "Scalar", "default": 1 },
  1891. { "name": "beta", "type": "Scalar", "default": 1 }
  1892. ],
  1893. "inputs": [
  1894. { "name": "self", "type": "Tensor" },
  1895. { "name": "mat1", "type": "Tensor" },
  1896. { "name": "mat2", "type": "Tensor" }
  1897. ],
  1898. "outputs": [
  1899. { "name": "output", "type": "Tensor" }
  1900. ]
  1901. },
  1902. {
  1903. "name": "torch.add_:Tensor",
  1904. "inputs": [
  1905. { "name": "self", "type": "Tensor" },
  1906. { "name": "other", "type": "Tensor" },
  1907. { "name": "alpha", "type": "Scalar", "default": 1 }
  1908. ],
  1909. "outputs": [
  1910. { "name": "output", "type": "Tensor" }
  1911. ]
  1912. },
  1913. {
  1914. "name": "torch.add_:Scalar",
  1915. "inputs": [
  1916. { "name": "self", "type": "Tensor" },
  1917. { "name": "other", "type": "Scalar" },
  1918. { "name": "alpha", "type": "Scalar", "default": 1 }
  1919. ],
  1920. "outputs": [
  1921. { "name": "output", "type": "Tensor" }
  1922. ]
  1923. },
  1924. {
  1925. "name": "torch._add_relu:Tensor",
  1926. "attributes": [
  1927. { "name": "alpha", "type": "Scalar", "default": 1 }
  1928. ],
  1929. "inputs": [
  1930. { "name": "self", "type": "Tensor" },
  1931. { "name": "other", "type": "Tensor" }
  1932. ],
  1933. "outputs": [
  1934. { "name": "output", "type": "Tensor" }
  1935. ]
  1936. },
  1937. {
  1938. "name": "torch._add_relu_:Tensor",
  1939. "attributes": [
  1940. { "name": "alpha", "type": "Scalar", "default": 1 }
  1941. ],
  1942. "inputs": [
  1943. { "name": "self", "type": "Tensor" },
  1944. { "name": "other", "type": "Tensor" }
  1945. ],
  1946. "outputs": [
  1947. { "name": "output", "type": "Tensor" }
  1948. ]
  1949. },
  1950. {
  1951. "name": "torch.sub_:Tensor",
  1952. "inputs": [
  1953. { "name": "input", "type": "Tensor" },
  1954. { "name": "other", "type": "Tensor" },
  1955. { "name": "alpha", "type": "Scalar", "default": 1 }
  1956. ],
  1957. "outputs": [
  1958. { "name": "output", "type": "Tensor" }
  1959. ]
  1960. },
  1961. {
  1962. "name": "torch.sub_:Scalar",
  1963. "inputs": [
  1964. { "name": "input", "type": "Tensor" },
  1965. { "name": "other", "type": "Scalar" },
  1966. { "name": "alpha", "type": "Scalar", "default": 1 }
  1967. ],
  1968. "outputs": [
  1969. { "name": "output", "type": "Tensor" }
  1970. ]
  1971. },
  1972. {
  1973. "name": "torch.pow:Tensor_Tensor",
  1974. "inputs": [
  1975. { "name": "self", "type": "Tensor" },
  1976. { "name": "exponent", "type": "Tensor" }
  1977. ],
  1978. "outputs": [
  1979. { "name": "output", "type": "Tensor" }
  1980. ]
  1981. },
  1982. {
  1983. "name": "torch.pow:Scalar",
  1984. "inputs": [
  1985. { "name": "self", "type": "Scalar" },
  1986. { "name": "exponent", "type": "Tensor" }
  1987. ],
  1988. "outputs": [
  1989. { "name": "output", "type": "Tensor" }
  1990. ]
  1991. },
  1992. {
  1993. "name": "torch.pow:Tensor_Scalar",
  1994. "inputs": [
  1995. { "name": "self", "type": "Tensor" },
  1996. { "name": "exponent", "type": "Scalar" }
  1997. ],
  1998. "outputs": [
  1999. { "name": "output", "type": "Tensor" }
  2000. ]
  2001. },
  2002. {
  2003. "name": "torch.pow_:Tensor",
  2004. "inputs": [
  2005. { "name": "self", "type": "Tensor" },
  2006. { "name": "exponent", "type": "Tensor" }
  2007. ],
  2008. "outputs": [
  2009. { "name": "output", "type": "Tensor" }
  2010. ]
  2011. },
  2012. {
  2013. "name": "torch.pow_:Scalar",
  2014. "inputs": [
  2015. { "name": "self", "type": "Tensor" },
  2016. { "name": "exponent", "type": "Scalar" }
  2017. ],
  2018. "outputs": [
  2019. { "name": "output", "type": "Tensor" }
  2020. ]
  2021. },
  2022. {
  2023. "name": "torch.add:Tensor",
  2024. "inputs": [
  2025. { "name": "input", "type": "Tensor" },
  2026. { "name": "other", "type": "Tensor" },
  2027. { "name": "alpha", "type": "Scalar", "default": 1 }
  2028. ],
  2029. "outputs": [
  2030. { "name": "output", "type": "Tensor" }
  2031. ]
  2032. },
  2033. {
  2034. "name": "torch.add:Scalar",
  2035. "inputs": [
  2036. { "name": "input", "type": "Tensor" },
  2037. { "name": "other", "type": "Scalar" },
  2038. { "name": "alpha", "type": "Scalar", "default": 1 }
  2039. ],
  2040. "outputs": [
  2041. { "name": "output", "type": "Tensor" }
  2042. ]
  2043. },
  2044. {
  2045. "name": "torch.sub:Scalar",
  2046. "inputs": [
  2047. { "name": "input", "type": "Tensor" },
  2048. { "name": "other", "type": "Scalar" },
  2049. { "name": "alpha", "type": "Scalar", "default": 1 }
  2050. ],
  2051. "outputs": [
  2052. { "name": "output", "type": "Tensor" }
  2053. ]
  2054. },
  2055. {
  2056. "name": "torch.rsub:Scalar",
  2057. "attributes": [
  2058. { "name": "other", "type": "Scalar" },
  2059. { "name": "alpha", "type": "Scalar", "default": 1 }
  2060. ],
  2061. "inputs": [
  2062. { "name": "input", "type": "Tensor" }
  2063. ],
  2064. "outputs": [
  2065. { "name": "output", "type": "Tensor" }
  2066. ]
  2067. },
  2068. {
  2069. "name": "torch.sub:Tensor",
  2070. "inputs": [
  2071. { "name": "input", "type": "Tensor" },
  2072. { "name": "other", "type": "Tensor" },
  2073. { "name": "alpha", "type": "Scalar", "default": 1 }
  2074. ],
  2075. "outputs": [
  2076. { "name": "output", "type": "Tensor" }
  2077. ]
  2078. },
  2079. {
  2080. "name": "torch.rsub:Tensor",
  2081. "inputs": [
  2082. { "name": "input", "type": "Tensor" },
  2083. { "name": "other", "type": "Tensor" },
  2084. { "name": "alpha", "type": "Scalar", "default": 1 }
  2085. ],
  2086. "outputs": [
  2087. { "name": "output", "type": "Tensor" }
  2088. ]
  2089. },
  2090. {
  2091. "name": "torch.diag:Tensor",
  2092. "inputs": [
  2093. { "name": "self", "type": "Tensor" },
  2094. { "name": "diagonal", "type": "int64", "default": 0 }
  2095. ],
  2096. "outputs": [
  2097. { "name": "output", "type": "Tensor" }
  2098. ]
  2099. },
  2100. {
  2101. "name": "torch.block_diag:",
  2102. "inputs": [
  2103. { "name": "tensors", "type": "Tensor[]" }
  2104. ],
  2105. "outputs": [
  2106. { "name": "output", "type": "Tensor" }
  2107. ]
  2108. },
  2109. {
  2110. "name": "torch.div:Tensor",
  2111. "inputs": [
  2112. { "name": "input", "type": "Tensor" },
  2113. { "name": "other", "type": "Tensor" }
  2114. ],
  2115. "outputs": [
  2116. { "name": "output", "type": "Tensor" }
  2117. ]
  2118. },
  2119. {
  2120. "name": "torch.div:Scalar",
  2121. "inputs": [
  2122. { "name": "input", "type": "Tensor" },
  2123. { "name": "other", "type": "Scalar" }
  2124. ],
  2125. "outputs": [
  2126. { "name": "output", "type": "Tensor" }
  2127. ]
  2128. },
  2129. {
  2130. "name": "torch.floordiv",
  2131. "inputs": [
  2132. { "name": "input", "type": "Tensor" },
  2133. { "name": "other", "type": "Tensor" }
  2134. ],
  2135. "outputs": [
  2136. { "name": "output", "type": "Tensor" }
  2137. ]
  2138. },
  2139. {
  2140. "name": "torch.sum:",
  2141. "inputs": [
  2142. { "name": "self", "type": "Tensor" }
  2143. ],
  2144. "outputs": [
  2145. { "name": "output", "type": "Tensor" }
  2146. ]
  2147. },
  2148. {
  2149. "name": "torch.sum:dim_IntList",
  2150. "inputs": [
  2151. { "name": "self", "type": "Tensor" },
  2152. { "name": "dim", "type": "int64[]" },
  2153. { "name": "keepdim", "type": "boolean", "default": false },
  2154. { "name": "dtype", "type": "ScalarType", "default": null }
  2155. ],
  2156. "outputs": [
  2157. { "name": "output", "type": "Tensor" }
  2158. ]
  2159. },
  2160. {
  2161. "name": "torch.matmul",
  2162. "inputs": [
  2163. { "name": "self", "type": "Tensor" },
  2164. { "name": "other", "type": "Tensor" }
  2165. ],
  2166. "outputs": [
  2167. { "name": "output", "type": "Tensor" }
  2168. ]
  2169. },
  2170. {
  2171. "name": "torch.transpose",
  2172. "inputs": [
  2173. { "name": "input", "type": "Tensor" },
  2174. { "name": "dim0", "type": "int64" },
  2175. { "name": "dim1", "type": "int64" }
  2176. ],
  2177. "outputs": [
  2178. { "name": "output", "type": "Tensor" }
  2179. ]
  2180. },
  2181. {
  2182. "name": "torch.chunk",
  2183. "attributes": [
  2184. { "name": "chunks", "type": "int64" },
  2185. { "name": "dim", "type": "int64", "default": 0 }
  2186. ],
  2187. "inputs": [
  2188. { "name": "self", "type": "Tensor" }
  2189. ],
  2190. "outputs": [
  2191. { "name": "output", "type": "Tensor[]" }
  2192. ]
  2193. },
  2194. {
  2195. "name": "torch.unsafe_chunk",
  2196. "inputs": [
  2197. { "name": "self", "type": "Tensor" },
  2198. { "name": "chunks", "type": "int64" },
  2199. { "name": "dim", "type": "int64", "default": 0 }
  2200. ],
  2201. "outputs": [
  2202. { "name": "output", "type": "Tensor[]" }
  2203. ]
  2204. },
  2205. {
  2206. "name": "torch.view:",
  2207. "inputs": [
  2208. { "name": "self", "type": "Tensor" },
  2209. { "name": "size", "type": "int64[]", "visible": false }
  2210. ],
  2211. "outputs": [
  2212. { "name": "output", "type": "Tensor" }
  2213. ]
  2214. },
  2215. {
  2216. "name": "torch.view:SymInt",
  2217. "inputs": [
  2218. { "name": "self", "type": "Tensor" },
  2219. { "name": "size", "type": "SymInt[]", "visible": false }
  2220. ],
  2221. "outputs": [
  2222. { "name": "output", "type": "Tensor" }
  2223. ]
  2224. },
  2225. {
  2226. "name": "torch.to:dtype_layout",
  2227. "inputs": [
  2228. { "name": "self", "type": "Tensor" },
  2229. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  2230. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  2231. { "name": "device", "type": "Device", "optional": true, "default": null },
  2232. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  2233. { "name": "non_blocking", "type": "boolean", "default": false },
  2234. { "name": "copy", "type": "boolean", "default": false },
  2235. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  2236. ],
  2237. "outputs": [
  2238. { "name": "output", "type": "Tensor" }
  2239. ]
  2240. },
  2241. {
  2242. "name": "torch.to:device",
  2243. "inputs": [
  2244. { "name": "self", "type": "Tensor" },
  2245. { "name": "device", "type": "Device" },
  2246. { "name": "dtype", "type": "ScalarType" },
  2247. { "name": "non_blocking", "type": "boolean", "default": false },
  2248. { "name": "copy", "type": "boolean", "default": false },
  2249. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  2250. ],
  2251. "outputs": [
  2252. { "name": "output", "type": "Tensor" }
  2253. ]
  2254. },
  2255. {
  2256. "name": "torch.to:dtype",
  2257. "inputs": [
  2258. { "name": "self", "type": "Tensor" },
  2259. { "name": "dtype", "type": "ScalarType" },
  2260. { "name": "non_blocking", "type": "boolean", "default": false },
  2261. { "name": "copy", "type": "boolean", "default": false },
  2262. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  2263. ],
  2264. "outputs": [
  2265. { "name": "output", "type": "Tensor" }
  2266. ]
  2267. },
  2268. {
  2269. "name": "torch.to:other",
  2270. "inputs": [
  2271. { "name": "self", "type": "Tensor" },
  2272. { "name": "other", "type": "Tensor" },
  2273. { "name": "non_blocking", "type": "boolean", "default": false },
  2274. { "name": "copy", "type": "boolean", "default": false },
  2275. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  2276. ],
  2277. "outputs": [
  2278. { "name": "output", "type": "Tensor" }
  2279. ]
  2280. },
  2281. {
  2282. "name": "torch.clip:",
  2283. "inputs": [
  2284. { "name": "input", "type": "Tensor" },
  2285. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  2286. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  2287. ],
  2288. "outputs": [
  2289. { "name": "output", "type": "Tensor" }
  2290. ]
  2291. },
  2292. {
  2293. "name": "torch.clip:Tensor",
  2294. "inputs": [
  2295. { "name": "input", "type": "Tensor" },
  2296. { "name": "min", "type": "Tensor", "optional": true, "default": null },
  2297. { "name": "max", "type": "Tensor", "optional": true, "default": null }
  2298. ],
  2299. "outputs": [
  2300. { "name": "output", "type": "Tensor" }
  2301. ]
  2302. },
  2303. {
  2304. "name": "torch.clip_:",
  2305. "inputs": [
  2306. { "name": "input", "type": "Tensor" },
  2307. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  2308. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  2309. ],
  2310. "outputs": [
  2311. { "name": "output", "type": "Tensor" }
  2312. ]
  2313. },
  2314. {
  2315. "name": "torch.clip_:Tensor",
  2316. "inputs": [
  2317. { "name": "input", "type": "Tensor" },
  2318. { "name": "min", "type": "Tensor", "optional": true, "default": null },
  2319. { "name": "max", "type": "Tensor", "optional": true, "default": null }
  2320. ],
  2321. "outputs": [
  2322. { "name": "output", "type": "Tensor" }
  2323. ]
  2324. },
  2325. {
  2326. "name": "torch.clone",
  2327. "inputs": [
  2328. { "name": "input", "type": "Tensor" }
  2329. ],
  2330. "outputs": [
  2331. { "name": "output", "type": "Tensor" }
  2332. ]
  2333. },
  2334. {
  2335. "name": "torch.complex",
  2336. "inputs": [
  2337. { "name": "real", "type": "Tensor" },
  2338. { "name": "imag", "type": "Tensor" }
  2339. ],
  2340. "outputs": [
  2341. { "name": "output", "type": "Tensor" }
  2342. ]
  2343. },
  2344. {
  2345. "name": "torch.expand",
  2346. "inputs": [
  2347. { "name": "input", "type": "Tensor" },
  2348. { "name": "size", "type": "int64[]" },
  2349. { "name": "implicit", "type": "boolean", "default": false }
  2350. ],
  2351. "outputs": [
  2352. { "name": "output", "type": "Tensor" }
  2353. ]
  2354. },
  2355. {
  2356. "name": "torch.copy_",
  2357. "inputs": [
  2358. { "name": "self", "type": "Tensor" },
  2359. { "name": "src", "type": "Tensor" },
  2360. { "name": "non_blocking", "type": "boolean", "default": false }
  2361. ],
  2362. "outputs": [
  2363. { "name": "output", "type": "Tensor" }
  2364. ]
  2365. },
  2366. {
  2367. "name": "torch.reshape",
  2368. "category": "Shape",
  2369. "inputs": [
  2370. { "name": "self", "type": "Tensor" },
  2371. { "name": "shape", "type": "int64[]" }
  2372. ],
  2373. "outputs": [
  2374. { "name": "output", "type": "Tensor" }
  2375. ]
  2376. },
  2377. {
  2378. "name": "torch.rot90",
  2379. "inputs": [
  2380. { "name": "self", "type": "Tensor" },
  2381. { "name": "k", "type": "int64", "default": 1 },
  2382. { "name": "dims", "type": "int64[]", "default": [ 0, 1 ] }
  2383. ],
  2384. "outputs": [
  2385. { "name": "output", "type": "Tensor" }
  2386. ]
  2387. },
  2388. {
  2389. "name": "torch.round:",
  2390. "inputs": [
  2391. { "name": "self", "type": "Tensor" }
  2392. ],
  2393. "outputs": [
  2394. { "name": "output", "type": "Tensor" }
  2395. ]
  2396. },
  2397. {
  2398. "name": "torch.round_:",
  2399. "inputs": [
  2400. { "name": "self", "type": "Tensor" }
  2401. ],
  2402. "outputs": [
  2403. { "name": "output", "type": "Tensor" }
  2404. ]
  2405. },
  2406. {
  2407. "name": "torch.round:decimals",
  2408. "inputs": [
  2409. { "name": "self", "type": "Tensor" },
  2410. { "name": "decimals", "type": "int64" }
  2411. ],
  2412. "outputs": [
  2413. { "name": "output", "type": "Tensor" }
  2414. ]
  2415. },
  2416. {
  2417. "name": "torch.round_:decimals",
  2418. "inputs": [
  2419. { "name": "self", "type": "Tensor" },
  2420. { "name": "decimals", "type": "int64" }
  2421. ],
  2422. "outputs": [
  2423. { "name": "output", "type": "Tensor" }
  2424. ]
  2425. },
  2426. {
  2427. "name": "torch.select:int",
  2428. "inputs": [
  2429. { "name": "self", "type": "Tensor" },
  2430. { "name": "dim", "type": "int64" },
  2431. { "name": "index", "type": "int64" }
  2432. ],
  2433. "outputs": [
  2434. { "name": "output", "type": "Tensor" }
  2435. ]
  2436. },
  2437. {
  2438. "name": "torch.select:Dimname",
  2439. "inputs": [
  2440. { "name": "self", "type": "Tensor" },
  2441. { "name": "dim", "type": "Dimname" },
  2442. { "name": "index", "type": "int64" }
  2443. ],
  2444. "outputs": [
  2445. { "name": "output", "type": "Tensor" }
  2446. ]
  2447. },
  2448. {
  2449. "name": "torch.repeat",
  2450. "inputs": [
  2451. { "name": "input", "type": "Tensor" },
  2452. { "name": "repeats", "type": "int64[]" }
  2453. ],
  2454. "outputs": [
  2455. { "name": "output", "type": "Tensor" }
  2456. ]
  2457. },
  2458. {
  2459. "name": "torch.mean:",
  2460. "inputs": [
  2461. { "name": "self", "type": "Tensor" },
  2462. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  2463. ],
  2464. "outputs": [
  2465. { "name": "output", "type": "Tensor" }
  2466. ]
  2467. },
  2468. {
  2469. "name": "torch.mean:dim",
  2470. "inputs": [
  2471. { "name": "self", "type": "Tensor" },
  2472. { "name": "dim", "type": "int64[]" },
  2473. { "name": "keepdim", "type": "boolean", "default": false },
  2474. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  2475. ],
  2476. "outputs": [
  2477. { "name": "output", "type": "Tensor" }
  2478. ]
  2479. },
  2480. {
  2481. "name": "torch.median:",
  2482. "inputs": [
  2483. { "name": "self", "type": "Tensor" }
  2484. ],
  2485. "outputs": [
  2486. { "name": "output", "type": "Tensor" }
  2487. ]
  2488. },
  2489. {
  2490. "name": "torch.median:dim",
  2491. "inputs": [
  2492. { "name": "self", "type": "Tensor" },
  2493. { "name": "dim", "type": "int64" },
  2494. { "name": "keepdim", "type": "boolean", "default": false }
  2495. ],
  2496. "outputs": [
  2497. { "name": "values", "type": "Tensor" },
  2498. { "name": "indices", "type": "Tensor" }
  2499. ]
  2500. },
  2501. {
  2502. "name": "torch.moveaxis:int",
  2503. "inputs": [
  2504. { "name": "self", "type": "Tensor" },
  2505. { "name": "source", "type": "int64" },
  2506. { "name": "destination", "type": "int64" }
  2507. ],
  2508. "outputs": [
  2509. { "name": "output", "type": "Tensor" }
  2510. ]
  2511. },
  2512. {
  2513. "name": "torch.moveaxis:intlist",
  2514. "inputs": [
  2515. { "name": "self", "type": "Tensor" },
  2516. { "name": "source", "type": "int64[]" },
  2517. { "name": "destination", "type": "int64[]" }
  2518. ],
  2519. "outputs": [
  2520. { "name": "output", "type": "Tensor" }
  2521. ]
  2522. },
  2523. {
  2524. "name": "torch.lstm:input",
  2525. "category": "Layer",
  2526. "inputs": [
  2527. { "name": "input", "type": "Tensor" },
  2528. { "name": "hx", "type": "Tensor[]" },
  2529. { "name": "params", "type": "Tensor[]" },
  2530. { "name": "has_biases", "type": "boolean" },
  2531. { "name": "num_layers", "type": "int64" },
  2532. { "name": "dropout", "type": "float64" },
  2533. { "name": "train", "type": "boolean" },
  2534. { "name": "bidirectional", "type": "boolean" },
  2535. { "name": "batch_first", "type": "boolean" }
  2536. ],
  2537. "outputs": [
  2538. { "name": "output", "type": "Tensor" },
  2539. { "name": "?", "type": "Tensor" },
  2540. { "name": "?", "type": "Tensor" }
  2541. ]
  2542. },
  2543. {
  2544. "name": "torch.lstm:data",
  2545. "category": "Layer",
  2546. "inputs": [
  2547. { "name": "data", "type": "Tensor" },
  2548. { "name": "batch_sizes", "type": "Tensor" },
  2549. { "name": "hx", "type": "Tensor[]" },
  2550. { "name": "params", "type": "Tensor[]" },
  2551. { "name": "has_biases", "type": "boolean" },
  2552. { "name": "num_layers", "type": "int64" },
  2553. { "name": "dropout", "type": "float64" },
  2554. { "name": "train", "type": "boolean" },
  2555. { "name": "bidirectional", "type": "boolean" }
  2556. ],
  2557. "outputs": [
  2558. { "name": "output", "type": "Tensor" },
  2559. { "name": "?", "type": "Tensor" },
  2560. { "name": "?", "type": "Tensor" }
  2561. ]
  2562. },
  2563. {
  2564. "name": "torch.gru:input",
  2565. "category": "Layer",
  2566. "inputs": [
  2567. { "name": "input", "type": "Tensor" },
  2568. { "name": "hx", "type": "Tensor" },
  2569. { "name": "params", "type": "Tensor[]" },
  2570. { "name": "has_biases", "type": "boolean" },
  2571. { "name": "num_layers", "type": "int64" },
  2572. { "name": "dropout", "type": "float64" },
  2573. { "name": "train", "type": "boolean" },
  2574. { "name": "bidirectional", "type": "boolean" },
  2575. { "name": "first_batch", "type": "boolean" }
  2576. ],
  2577. "outputs": [
  2578. { "name": "output", "type": "Tensor" },
  2579. { "name": "?", "type": "Tensor" }
  2580. ]
  2581. },
  2582. {
  2583. "name": "torch.gru:data",
  2584. "category": "Layer",
  2585. "inputs": [
  2586. { "name": "data", "type": "Tensor" },
  2587. { "name": "batch_sizes", "type": "Tensor" },
  2588. { "name": "hx", "type": "Tensor" },
  2589. { "name": "params", "type": "Tensor[]" },
  2590. { "name": "has_biases", "type": "boolean" },
  2591. { "name": "num_layers", "type": "int64" },
  2592. { "name": "dropout", "type": "float64" },
  2593. { "name": "train", "type": "boolean" },
  2594. { "name": "bidirectional", "type": "boolean" }
  2595. ],
  2596. "outputs": [
  2597. { "name": "output", "type": "Tensor" },
  2598. { "name": "?", "type": "Tensor" }
  2599. ]
  2600. },
  2601. {
  2602. "name": "torch.coalesce",
  2603. "inputs": [
  2604. { "name": "self", "type": "Tensor" }
  2605. ],
  2606. "outputs": [
  2607. { "name": "output", "type": "Tensor" }
  2608. ]
  2609. },
  2610. {
  2611. "name": "torch._coalesce",
  2612. "inputs": [
  2613. { "name": "self", "type": "Tensor" }
  2614. ],
  2615. "outputs": [
  2616. { "name": "output", "type": "Tensor" }
  2617. ]
  2618. },
  2619. {
  2620. "name": "torch.contiguous",
  2621. "inputs": [
  2622. { "name": "input", "type": "Tensor" },
  2623. { "name": "memory_format", "type": "MemoryFormat", "default": null }
  2624. ],
  2625. "outputs": [
  2626. { "name": "output", "type": "Tensor" }
  2627. ]
  2628. },
  2629. {
  2630. "name": "torch.squeeze:",
  2631. "category": "Transform",
  2632. "inputs": [
  2633. { "name": "self", "type": "Tensor" }
  2634. ],
  2635. "outputs": [
  2636. { "name": "output", "type": "Tensor" }
  2637. ]
  2638. },
  2639. {
  2640. "name": "torch.squeeze:dim",
  2641. "category": "Transform",
  2642. "inputs": [
  2643. { "name": "self", "type": "Tensor" },
  2644. { "name": "dim", "type": "int64" }
  2645. ],
  2646. "outputs": [
  2647. { "name": "output", "type": "Tensor" }
  2648. ]
  2649. },
  2650. {
  2651. "name": "torch.squeeze:dimname",
  2652. "category": "Transform",
  2653. "inputs": [
  2654. { "name": "self", "type": "Tensor" },
  2655. { "name": "dim", "type": "Dimname" }
  2656. ],
  2657. "outputs": [
  2658. { "name": "output", "type": "Tensor" }
  2659. ]
  2660. },
  2661. {
  2662. "name": "torch.squeeze_:",
  2663. "category": "Transform",
  2664. "inputs": [
  2665. { "name": "self", "type": "Tensor" }
  2666. ],
  2667. "outputs": [
  2668. { "name": "output", "type": "Tensor" }
  2669. ]
  2670. },
  2671. {
  2672. "name": "torch.squeeze_:dim",
  2673. "category": "Transform",
  2674. "inputs": [
  2675. { "name": "self", "type": "Tensor" },
  2676. { "name": "dim", "type": "int64" }
  2677. ],
  2678. "outputs": [
  2679. { "name": "output", "type": "Tensor" }
  2680. ]
  2681. },
  2682. {
  2683. "name": "torch.squeeze_:dimname",
  2684. "category": "Transform",
  2685. "inputs": [
  2686. { "name": "self", "type": "Tensor" },
  2687. { "name": "dim", "type": "Dimname" }
  2688. ],
  2689. "outputs": [
  2690. { "name": "output", "type": "Tensor" }
  2691. ]
  2692. },
  2693. {
  2694. "name": "torch.unsqueeze",
  2695. "category": "Transform",
  2696. "inputs": [
  2697. { "name": "self", "type": "Tensor" },
  2698. { "name": "dim", "type": "int64" }
  2699. ],
  2700. "outputs": [
  2701. { "name": "output", "type": "Tensor" }
  2702. ]
  2703. },
  2704. {
  2705. "name": "torch.unsqueeze_",
  2706. "category": "Transform",
  2707. "inputs": [
  2708. { "name": "self", "type": "Tensor" },
  2709. { "name": "dim", "type": "int64" }
  2710. ],
  2711. "outputs": [
  2712. { "name": "output", "type": "Tensor" }
  2713. ]
  2714. },
  2715. {
  2716. "name": "torch.max:",
  2717. "inputs": [
  2718. { "name": "self", "type": "Tensor" }
  2719. ],
  2720. "outputs": [
  2721. { "name": "output", "type": "Tensor" }
  2722. ]
  2723. },
  2724. {
  2725. "name": "torch.max:dim",
  2726. "inputs": [
  2727. { "name": "self", "type": "Tensor" },
  2728. { "name": "dim", "type": "int64", "default": 0 },
  2729. { "name": "keepdim", "type": "boolean", "default": false }
  2730. ],
  2731. "outputs": [
  2732. { "name": "values", "type": "Tensor" },
  2733. { "name": "indices", "type": "Tensor" }
  2734. ]
  2735. },
  2736. {
  2737. "name": "torch.max:other",
  2738. "inputs": [
  2739. { "name": "self", "type": "Tensor" },
  2740. { "name": "other", "type": "Tensor" }
  2741. ],
  2742. "outputs": [
  2743. { "name": "output", "type": "Tensor" }
  2744. ]
  2745. },
  2746. {
  2747. "name": "torch.maximum",
  2748. "inputs": [
  2749. { "name": "self", "type": "Tensor" },
  2750. { "name": "other", "type": "Tensor" }
  2751. ],
  2752. "outputs": [
  2753. { "name": "output", "type": "Tensor" }
  2754. ]
  2755. },
  2756. {
  2757. "name": "torch.min:",
  2758. "inputs": [
  2759. { "name": "self", "type": "Tensor" }
  2760. ],
  2761. "outputs": [
  2762. { "name": "output", "type": "Tensor" }
  2763. ]
  2764. },
  2765. {
  2766. "name": "torch.min:dim",
  2767. "inputs": [
  2768. { "name": "self", "type": "Tensor" },
  2769. { "name": "dim", "type": "int64", "default": 0 },
  2770. { "name": "keepdim", "type": "boolean", "default": false }
  2771. ],
  2772. "outputs": [
  2773. { "name": "values", "type": "Tensor" },
  2774. { "name": "indices", "type": "Tensor" }
  2775. ]
  2776. },
  2777. {
  2778. "name": "torch.min:other",
  2779. "inputs": [
  2780. { "name": "self", "type": "Tensor" },
  2781. { "name": "other", "type": "Tensor" }
  2782. ],
  2783. "outputs": [
  2784. { "name": "output", "type": "Tensor" }
  2785. ]
  2786. },
  2787. {
  2788. "name": "torch.minimum",
  2789. "inputs": [
  2790. { "name": "self", "type": "Tensor" },
  2791. { "name": "other", "type": "Tensor" }
  2792. ],
  2793. "outputs": [
  2794. { "name": "output", "type": "Tensor" }
  2795. ]
  2796. },
  2797. {
  2798. "name": "torch.mish_:",
  2799. "category": "Activation",
  2800. "inputs": [
  2801. { "name": "self", "type": "Tensor" }
  2802. ],
  2803. "outputs": [
  2804. { "name": "output", "type": "Tensor" }
  2805. ]
  2806. },
  2807. {
  2808. "name": "torch.mish:",
  2809. "category": "Activation",
  2810. "inputs": [
  2811. { "name": "self", "type": "Tensor" }
  2812. ],
  2813. "outputs": [
  2814. { "name": "output", "type": "Tensor" }
  2815. ]
  2816. },
  2817. {
  2818. "name": "torch.cat:",
  2819. "category": "Tensor",
  2820. "inputs": [
  2821. { "name": "tensors", "type": "Tensor[]" },
  2822. { "name": "dim", "type": "int64", "default": 0 }
  2823. ],
  2824. "outputs": [
  2825. { "name": "output", "type": "Tensor" }
  2826. ]
  2827. },
  2828. {
  2829. "name": "torch.cat:names",
  2830. "category": "Tensor",
  2831. "inputs": [
  2832. { "name": "tensors", "type": "Tensor[]" },
  2833. { "name": "dim", "type": "Dimname" }
  2834. ],
  2835. "outputs": [
  2836. { "name": "output", "type": "Tensor" }
  2837. ]
  2838. },
  2839. {
  2840. "name": "torch.concat:",
  2841. "category": "Tensor",
  2842. "inputs": [
  2843. { "name": "tensors", "type": "Tensor[]" },
  2844. { "name": "dim", "type": "int64", "default": 0 }
  2845. ],
  2846. "outputs": [
  2847. { "name": "output", "type": "Tensor" }
  2848. ]
  2849. },
  2850. {
  2851. "name": "torch.concat:names",
  2852. "category": "Tensor",
  2853. "inputs": [
  2854. { "name": "tensors", "type": "Tensor[]" },
  2855. { "name": "dim", "type": "Dimname" }
  2856. ],
  2857. "outputs": [
  2858. { "name": "output", "type": "Tensor" }
  2859. ]
  2860. },
  2861. {
  2862. "name": "torch.broadcast_tensors",
  2863. "inputs": [
  2864. { "name": "tensors", "type": "Tensor[]" }
  2865. ],
  2866. "outputs": [
  2867. { "name": "output", "type": "Tensor[]" }
  2868. ]
  2869. },
  2870. {
  2871. "name": "torch.broadcast_to:",
  2872. "inputs": [
  2873. { "name": "self", "type": "Tensor" },
  2874. { "name": "size", "type": "int64[]" }
  2875. ],
  2876. "outputs": [
  2877. { "name": "output", "type": "Tensor" }
  2878. ]
  2879. },
  2880. {
  2881. "name": "torch.meshgrid:",
  2882. "category": "Tensor",
  2883. "inputs": [
  2884. { "name": "tensors", "type": "Tensor[]" }
  2885. ],
  2886. "outputs": [
  2887. { "name": "output", "type": "Tensor[]" }
  2888. ]
  2889. },
  2890. {
  2891. "name": "torch.meshgrid:indexing",
  2892. "category": "Tensor",
  2893. "inputs": [
  2894. { "name": "tensors", "type": "Tensor[]" },
  2895. { "name": "indexing", "type": "string" }
  2896. ],
  2897. "outputs": [
  2898. { "name": "output", "type": "Tensor[]" }
  2899. ]
  2900. },
  2901. {
  2902. "name": "torch.stack:",
  2903. "category": "Tensor",
  2904. "inputs": [
  2905. { "name": "inputs", "type": "Tensor[]" },
  2906. { "name": "dim", "type": "int64", "default": 0 }
  2907. ],
  2908. "outputs": [
  2909. { "name": "output", "type": "Tensor" }
  2910. ]
  2911. },
  2912. {
  2913. "name": "torch.expand_as",
  2914. "inputs": [
  2915. { "name": "inputs", "type": "Tensor" },
  2916. { "name": "other", "type": "Tensor" }
  2917. ],
  2918. "outputs": [
  2919. { "name": "output", "type": "Tensor" }
  2920. ]
  2921. },
  2922. {
  2923. "name": "torch.eye:",
  2924. "inputs": [
  2925. { "name": "n", "type": "int64" },
  2926. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  2927. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  2928. { "name": "device", "type": "Device", "optional": true, "default": null },
  2929. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  2930. ],
  2931. "outputs": [
  2932. { "name": "output", "type": "Tensor" }
  2933. ]
  2934. },
  2935. {
  2936. "name": "torch.eye:m",
  2937. "inputs": [
  2938. { "name": "n", "type": "int64" },
  2939. { "name": "m", "type": "int64" },
  2940. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  2941. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  2942. { "name": "device", "type": "Device", "optional": true, "default": null },
  2943. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  2944. ],
  2945. "outputs": [
  2946. { "name": "output", "type": "Tensor" }
  2947. ]
  2948. },
  2949. {
  2950. "name": "torch.pairwise_distance:",
  2951. "inputs": [
  2952. { "name": "x1", "type": "Tensor" },
  2953. { "name": "x2", "type": "Tensor" },
  2954. { "name": "p", "type": "float32", "default": 2 },
  2955. { "name": "eps", "type": "float32", "default": 1e-06 },
  2956. { "name": "keepdim", "type": "boolean", "default": false }
  2957. ],
  2958. "outputs": [
  2959. { "name": "output", "type": "Tensor" }
  2960. ]
  2961. },
  2962. {
  2963. "name": "torch.pad:",
  2964. "category": "Tensor",
  2965. "inputs": [
  2966. { "name": "self", "type": "Tensor" },
  2967. { "name": "pad", "type": "int64[]" },
  2968. { "name": "mode", "type": "string", "default": "constant" },
  2969. { "name": "value", "type": "float32", "optional": true, "default": null }
  2970. ],
  2971. "outputs": [
  2972. { "name": "output", "type": "Tensor" }
  2973. ]
  2974. },
  2975. {
  2976. "name": "torch.permute",
  2977. "category": "Shape",
  2978. "inputs": [
  2979. { "name": "inputs", "type": "Tensor" },
  2980. { "name": "dims", "type": "int64[]" }
  2981. ],
  2982. "outputs": [
  2983. { "name": "output", "type": "Tensor" }
  2984. ]
  2985. },
  2986. {
  2987. "name": "torch.embedding",
  2988. "category": "Transform",
  2989. "inputs": [
  2990. { "name": "inputs", "type": "Tensor" },
  2991. { "name": "weight", "type": "Tensor" },
  2992. { "name": "padding_idx", "type": "int64" },
  2993. { "name": "scale_grad_by_freq", "type": "boolean", "default": false },
  2994. { "name": "sparse", "type": "boolean", "default": false }
  2995. ],
  2996. "outputs": [
  2997. { "name": "output", "type": "Tensor" }
  2998. ]
  2999. },
  3000. {
  3001. "name": "torch.embedding_bag",
  3002. "category": "Transform",
  3003. "inputs": [
  3004. { "name": "weight", "type": "Tensor" },
  3005. { "name": "indices", "type": "Tensor" },
  3006. { "name": "offsets", "type": "Tensor" },
  3007. { "name": "scale_grad_by_freq", "type": "boolean", "default": false },
  3008. { "name": "mode", "type": "int64", "default": 0 },
  3009. { "name": "sparse", "type": "boolean", "default": false },
  3010. { "name": "per_sample_weights", "type": "Tensor", "optional": true, "default": null },
  3011. { "name": "include_last_offset", "type": "boolean", "default": false }
  3012. ],
  3013. "outputs": [
  3014. { "name": "output1", "type": "Tensor" },
  3015. { "name": "output2", "type": "Tensor" },
  3016. { "name": "output3", "type": "Tensor" },
  3017. { "name": "output4", "type": "Tensor" }
  3018. ]
  3019. },
  3020. {
  3021. "name": "torch.flatten:using_ints",
  3022. "category": "Shape",
  3023. "inputs": [
  3024. { "name": "self", "type": "Tensor" },
  3025. { "name": "start_dim", "type": "int64", "default": 0 },
  3026. { "name": "end_dim", "type": "int64", "default": -1 }
  3027. ],
  3028. "outputs": [
  3029. { "name": "output", "type": "Tensor" }
  3030. ]
  3031. },
  3032. {
  3033. "name": "torch.flatten:named_out_dim",
  3034. "category": "Shape",
  3035. "inputs": [
  3036. { "name": "self", "type": "Tensor" },
  3037. { "name": "start_dim", "type": "int64" },
  3038. { "name": "end_dim", "type": "int64" },
  3039. { "name": "out_dim", "type": "Dimname" }
  3040. ],
  3041. "outputs": [
  3042. { "name": "output", "type": "Tensor" }
  3043. ]
  3044. },
  3045. {
  3046. "name": "torch.flatten:using_names",
  3047. "category": "Shape",
  3048. "inputs": [
  3049. { "name": "self", "type": "Tensor" },
  3050. { "name": "start_dim", "type": "Dimname" },
  3051. { "name": "end_dim", "type": "Dimname" },
  3052. { "name": "out_dim", "type": "Dimname" }
  3053. ],
  3054. "outputs": [
  3055. { "name": "output", "type": "Tensor" }
  3056. ]
  3057. },
  3058. {
  3059. "name": "torch.flatten:DimnameList",
  3060. "category": "Shape",
  3061. "inputs": [
  3062. { "name": "self", "type": "Tensor" },
  3063. { "name": "dims", "type": "Dimname[]" },
  3064. { "name": "out_dim", "type": "Dimname" }
  3065. ],
  3066. "outputs": [
  3067. { "name": "output", "type": "Tensor" }
  3068. ]
  3069. },
  3070. {
  3071. "name": "torch.unflatten:int",
  3072. "category": "Shape",
  3073. "inputs": [
  3074. { "name": "self", "type": "Tensor" },
  3075. { "name": "dim", "type": "int64" },
  3076. { "name": "sizes", "type": "int64[]" },
  3077. { "name": "names", "type": "Dimname[]", "optional": true, "default": null }
  3078. ],
  3079. "outputs": [
  3080. { "name": "output", "type": "Tensor" }
  3081. ]
  3082. },
  3083. {
  3084. "name": "torch.unflatten:Dimname",
  3085. "category": "Shape",
  3086. "inputs": [
  3087. { "name": "self", "type": "Tensor" },
  3088. { "name": "dim", "type": "Dimname" },
  3089. { "name": "sizes", "type": "int64[]" },
  3090. { "name": "names", "type": "Dimname[]", "optional": true, "default": null }
  3091. ],
  3092. "outputs": [
  3093. { "name": "output", "type": "Tensor" }
  3094. ]
  3095. },
  3096. {
  3097. "name": "torch._pack_padded_sequence",
  3098. "inputs": [
  3099. { "name": "input", "type": "Tensor" },
  3100. { "name": "lengths", "type": "Tensor" },
  3101. { "name": "batch_first", "type": "boolean" }
  3102. ],
  3103. "outputs": [
  3104. { "name": "output", "type": "Tensor" },
  3105. { "name": "?", "type": "Tensor" }
  3106. ]
  3107. },
  3108. {
  3109. "name": "torch._pad_packed_sequence",
  3110. "inputs": [
  3111. { "name": "data", "type": "Tensor" },
  3112. { "name": "batch_sizes", "type": "Tensor" },
  3113. { "name": "batch_first", "type": "boolean" },
  3114. { "name": "padding_value", "type": "Scalar" },
  3115. { "name": "total_length", "type": "int64" }
  3116. ],
  3117. "outputs": [
  3118. { "name": "output", "type": "Tensor" },
  3119. { "name": "?", "type": "Tensor" }
  3120. ]
  3121. },
  3122. {
  3123. "name": "torch.quantize_per_tensor:",
  3124. "inputs": [
  3125. { "name": "self", "type": "Tensor" },
  3126. { "name": "scale", "type": "float32" },
  3127. { "name": "zero_point", "type": "int64" },
  3128. { "name": "dtype", "type": "ScalarType" }
  3129. ],
  3130. "outputs": [
  3131. { "name": "output", "type": "Tensor" }
  3132. ]
  3133. },
  3134. {
  3135. "name": "torch.quantize_per_tensor:qparams",
  3136. "inputs": [
  3137. { "name": "self", "type": "Tensor" },
  3138. { "name": "scale", "type": "Tensor" },
  3139. { "name": "zero_point", "type": "Tensor" },
  3140. { "name": "dtype", "type": "ScalarType" }
  3141. ],
  3142. "outputs": [
  3143. { "name": "output", "type": "Tensor" }
  3144. ]
  3145. },
  3146. {
  3147. "name": "torch.quantize_per_tensor:tensors",
  3148. "inputs": [
  3149. { "name": "tensors", "type": "Tensor[]" },
  3150. { "name": "scales", "type": "Tensor" },
  3151. { "name": "zero_points", "type": "Tensor" },
  3152. { "name": "dtype", "type": "ScalarType" }
  3153. ],
  3154. "outputs": [
  3155. { "name": "outputs", "type": "Tensor[]" }
  3156. ]
  3157. },
  3158. {
  3159. "name": "torch.floor_divide",
  3160. "inputs": [
  3161. { "name": "self", "type": "Tensor" },
  3162. { "name": "other", "type": "Tensor" }
  3163. ],
  3164. "outputs": [
  3165. { "name": "output", "type": "Tensor" }
  3166. ]
  3167. },
  3168. {
  3169. "name": "torch.index",
  3170. "inputs": [
  3171. { "name": "self", "type": "Tensor" },
  3172. { "name": "indices", "type": "Tensor[]", "optional": true }
  3173. ],
  3174. "outputs": [
  3175. { "name": "output", "type": "Tensor" }
  3176. ]
  3177. },
  3178. {
  3179. "name": "torch._cast_Byte",
  3180. "inputs": [
  3181. { "name": "input", "type": "Tensor" },
  3182. { "name": "non_blocking", "type": "boolean", "default": false }
  3183. ],
  3184. "outputs": [
  3185. { "name": "output", "type": "Tensor" }
  3186. ]
  3187. },
  3188. {
  3189. "name": "torch._cast_Char",
  3190. "inputs": [
  3191. { "name": "input", "type": "Tensor" },
  3192. { "name": "non_blocking", "type": "boolean", "default": false }
  3193. ],
  3194. "outputs": [
  3195. { "name": "output", "type": "Tensor" }
  3196. ]
  3197. },
  3198. {
  3199. "name": "torch._cast_Short",
  3200. "inputs": [
  3201. { "name": "input", "type": "Tensor" },
  3202. { "name": "non_blocking", "type": "boolean", "default": false }
  3203. ],
  3204. "outputs": [
  3205. { "name": "output", "type": "Tensor" }
  3206. ]
  3207. },
  3208. {
  3209. "name": "torch._cast_Int",
  3210. "inputs": [
  3211. { "name": "input", "type": "Tensor" },
  3212. { "name": "non_blocking", "type": "boolean", "default": false }
  3213. ],
  3214. "outputs": [
  3215. { "name": "output", "type": "Tensor" }
  3216. ]
  3217. },
  3218. {
  3219. "name": "torch._cast_Long",
  3220. "inputs": [
  3221. { "name": "input", "type": "Tensor" },
  3222. { "name": "non_blocking", "type": "boolean", "default": false }
  3223. ],
  3224. "outputs": [
  3225. { "name": "output", "type": "Tensor" }
  3226. ]
  3227. },
  3228. {
  3229. "name": "torch._cast_Half",
  3230. "inputs": [
  3231. { "name": "input", "type": "Tensor" },
  3232. { "name": "non_blocking", "type": "boolean", "default": false }
  3233. ],
  3234. "outputs": [
  3235. { "name": "output", "type": "Tensor" }
  3236. ]
  3237. },
  3238. {
  3239. "name": "torch._cast_Float",
  3240. "inputs": [
  3241. { "name": "input", "type": "Tensor" },
  3242. { "name": "non_blocking", "type": "boolean", "default": false }
  3243. ],
  3244. "outputs": [
  3245. { "name": "output", "type": "Tensor" }
  3246. ]
  3247. },
  3248. {
  3249. "name": "torch._cast_Double",
  3250. "inputs": [
  3251. { "name": "input", "type": "Tensor" },
  3252. { "name": "non_blocking", "type": "boolean", "default": false }
  3253. ],
  3254. "outputs": [
  3255. { "name": "output", "type": "Tensor" }
  3256. ]
  3257. },
  3258. {
  3259. "name": "torch.frobenius_norm:",
  3260. "category": "Normalization",
  3261. "inputs": [
  3262. { "name": "self", "type": "Tensor" }
  3263. ],
  3264. "outputs": [
  3265. { "name": "output", "type": "Tensor" }
  3266. ]
  3267. },
  3268. {
  3269. "name": "torch.frobenius_norm:dim",
  3270. "category": "Normalization",
  3271. "inputs": [
  3272. { "name": "self", "type": "Tensor" },
  3273. { "name": "dim", "type": "int64[]" },
  3274. { "name": "keepdim", "type": "boolean", "default": false }
  3275. ],
  3276. "outputs": [
  3277. { "name": "output", "type": "Tensor" }
  3278. ]
  3279. },
  3280. {
  3281. "name": "torch.norm:ScalarOpt_dtype",
  3282. "inputs": [
  3283. { "name": "self", "type": "Tensor" },
  3284. { "name": "p", "type": "Scalar", "optional": true },
  3285. { "name": "dtype", "type": "ScalarType" }
  3286. ],
  3287. "outputs": [
  3288. { "name": "output", "type": "Tensor" }
  3289. ]
  3290. },
  3291. {
  3292. "name": "torch.norm:ScalarOpt_dim_dtype",
  3293. "inputs": [
  3294. { "name": "self", "type": "Tensor" },
  3295. { "name": "p", "type": "Scalar", "optional": true },
  3296. { "name": "dim", "type": "int64[1]" },
  3297. { "name": "keepdim", "type": "boolean" },
  3298. { "name": "dtype", "type": "ScalarType" }
  3299. ],
  3300. "outputs": [
  3301. { "name": "output", "type": "Tensor" }
  3302. ]
  3303. },
  3304. {
  3305. "name": "torch.norm:ScalarOpt_dim",
  3306. "inputs": [
  3307. { "name": "self", "type": "Tensor" },
  3308. { "name": "p", "type": "Scalar", "optional": true },
  3309. { "name": "dim", "type": "int64[1]" },
  3310. { "name": "keepdim", "type": "boolean", "default": false }
  3311. ],
  3312. "outputs": [
  3313. { "name": "output", "type": "Tensor" }
  3314. ]
  3315. },
  3316. {
  3317. "name": "torch.norm:dtype_out",
  3318. "inputs": [
  3319. { "name": "self", "type": "Tensor" },
  3320. { "name": "p", "type": "Scalar", "optional": true },
  3321. { "name": "dim", "type": "int64[1]" },
  3322. { "name": "keepdim", "type": "boolean", "default": false },
  3323. { "name": "dtype", "type": "ScalarType" }
  3324. ],
  3325. "outputs": [
  3326. { "name": "output", "type": "Tensor" }
  3327. ]
  3328. },
  3329. {
  3330. "name": "torch.norm:Scalar",
  3331. "inputs": [
  3332. { "name": "self", "type": "Tensor" },
  3333. { "name": "p", "type": "Scalar", "default": 2 }
  3334. ],
  3335. "outputs": [
  3336. { "name": "output", "type": "Tensor" }
  3337. ]
  3338. },
  3339. {
  3340. "name": "torch.numpy_T",
  3341. "inputs": [
  3342. { "name": "self", "type": "Tensor" }
  3343. ],
  3344. "outputs": [
  3345. { "name": "output", "type": "Tensor" }
  3346. ]
  3347. },
  3348. {
  3349. "name": "torch.glu",
  3350. "category": "Activation",
  3351. "inputs": [
  3352. { "name": "self", "type": "Tensor" },
  3353. { "name": "dim", "type": "int64", "default": -1 }
  3354. ],
  3355. "outputs": [
  3356. { "name": "output", "type": "Tensor" }
  3357. ]
  3358. },
  3359. {
  3360. "name": "torch.gelu",
  3361. "category": "Activation",
  3362. "inputs": [
  3363. { "name": "self", "type": "Tensor" },
  3364. { "name": "approximate", "type": "string", "default": "none" }
  3365. ],
  3366. "outputs": [
  3367. { "name": "output", "type": "Tensor" }
  3368. ]
  3369. },
  3370. {
  3371. "name": "torch.gelu_",
  3372. "category": "Activation",
  3373. "inputs": [
  3374. { "name": "self", "type": "Tensor" },
  3375. { "name": "approximate", "type": "string", "default": "none" }
  3376. ],
  3377. "outputs": [
  3378. { "name": "output", "type": "Tensor" }
  3379. ]
  3380. },
  3381. {
  3382. "name": "torch.celu",
  3383. "category": "Activation",
  3384. "inputs": [
  3385. { "name": "self", "type": "Tensor" },
  3386. { "name": "alpha", "type": "Scalar", "default": 1 }
  3387. ],
  3388. "outputs": [
  3389. { "name": "output", "type": "Tensor" }
  3390. ]
  3391. },
  3392. {
  3393. "name": "torch.elu",
  3394. "category": "Activation",
  3395. "inputs": [
  3396. { "name": "self", "type": "Tensor" },
  3397. { "name": "alpha", "type": "Scalar", "default": 1 },
  3398. { "name": "scale", "type": "Scalar", "default": 1 },
  3399. { "name": "input_scale", "type": "Scalar", "default": 1 }
  3400. ],
  3401. "outputs": [
  3402. { "name": "output", "type": "Tensor" }
  3403. ]
  3404. },
  3405. {
  3406. "name": "torch.elu_",
  3407. "category": "Activation",
  3408. "inputs": [
  3409. { "name": "self", "type": "Tensor" },
  3410. { "name": "alpha", "type": "Scalar", "default": 1 },
  3411. { "name": "scale", "type": "Scalar", "default": 1 },
  3412. { "name": "input_scale", "type": "Scalar", "default": 1 }
  3413. ],
  3414. "outputs": [
  3415. { "name": "output", "type": "Tensor" }
  3416. ]
  3417. },
  3418. {
  3419. "name": "torch.selu",
  3420. "category": "Activation",
  3421. "inputs": [
  3422. { "name": "self", "type": "Tensor" }
  3423. ],
  3424. "outputs": [
  3425. { "name": "output", "type": "Tensor" }
  3426. ]
  3427. },
  3428. {
  3429. "name": "torch.selu_",
  3430. "category": "Activation",
  3431. "inputs": [
  3432. { "name": "self", "type": "Tensor" }
  3433. ],
  3434. "outputs": [
  3435. { "name": "output", "type": "Tensor" }
  3436. ]
  3437. },
  3438. {
  3439. "name": "torch.silu",
  3440. "category": "Activation",
  3441. "inputs": [
  3442. { "name": "self", "type": "Tensor" }
  3443. ],
  3444. "outputs": [
  3445. { "name": "output", "type": "Tensor" }
  3446. ]
  3447. },
  3448. {
  3449. "name": "torch.silu_",
  3450. "category": "Activation",
  3451. "inputs": [
  3452. { "name": "self", "type": "Tensor" }
  3453. ],
  3454. "outputs": [
  3455. { "name": "output", "type": "Tensor" }
  3456. ]
  3457. },
  3458. {
  3459. "name": "torch.hardswish",
  3460. "category": "Activation",
  3461. "inputs": [
  3462. { "name": "self", "type": "Tensor" }
  3463. ],
  3464. "outputs": [
  3465. { "name": "output", "type": "Tensor" }
  3466. ]
  3467. },
  3468. {
  3469. "name": "torch.hardswish_",
  3470. "category": "Activation",
  3471. "inputs": [
  3472. { "name": "self", "type": "Tensor" }
  3473. ],
  3474. "outputs": [
  3475. { "name": "output", "type": "Tensor" }
  3476. ]
  3477. },
  3478. {
  3479. "name": "torch.pixel_shuffle",
  3480. "inputs": [
  3481. { "name": "self", "type": "Tensor" },
  3482. { "name": "upscale_factor", "type": "int64" }
  3483. ],
  3484. "outputs": [
  3485. { "name": "output", "type": "Tensor" }
  3486. ]
  3487. },
  3488. {
  3489. "name": "torch.nonzero",
  3490. "inputs": [
  3491. { "name": "self", "type": "Tensor" }
  3492. ],
  3493. "outputs": [
  3494. { "name": "output", "type": "Tensor" }
  3495. ]
  3496. },
  3497. {
  3498. "name": "torch.neg",
  3499. "inputs": [
  3500. { "name": "self", "type": "Tensor" }
  3501. ],
  3502. "outputs": [
  3503. { "name": "output", "type": "Tensor" }
  3504. ]
  3505. },
  3506. {
  3507. "name": "torch.ne:Scalar",
  3508. "inputs": [
  3509. { "name": "input", "type": "Tensor" },
  3510. { "name": "other", "type": "Scalar" }
  3511. ],
  3512. "outputs": [
  3513. { "name": "output", "type": "Tensor" }
  3514. ]
  3515. },
  3516. {
  3517. "name": "torch.bitwise_not",
  3518. "inputs": [
  3519. { "name": "self", "type": "Tensor" }
  3520. ],
  3521. "outputs": [
  3522. { "name": "output", "type": "Tensor" }
  3523. ]
  3524. },
  3525. {
  3526. "name": "torch.bitwise_not_",
  3527. "inputs": [
  3528. { "name": "self", "type": "Tensor" }
  3529. ],
  3530. "outputs": [
  3531. { "name": "output", "type": "Tensor" }
  3532. ]
  3533. },
  3534. {
  3535. "name": "torch.bucketize:Tensor",
  3536. "inputs": [
  3537. { "name": "self", "type": "Tensor" },
  3538. { "name": "boundaries", "type": "Tensor" },
  3539. { "name": "out_int32", "type": "boolean", "default": false },
  3540. { "name": "right", "type": "boolean", "default": false }
  3541. ],
  3542. "outputs": [
  3543. { "name": "output", "type": "Tensor" }
  3544. ]
  3545. },
  3546. {
  3547. "name": "torch.bucketize:Scalar",
  3548. "inputs": [
  3549. { "name": "self", "type": "Scalar" },
  3550. { "name": "boundaries", "type": "Tensor" },
  3551. { "name": "out_int32", "type": "boolean", "default": false },
  3552. { "name": "right", "type": "boolean", "default": false }
  3553. ],
  3554. "outputs": [
  3555. { "name": "output", "type": "Tensor" }
  3556. ]
  3557. },
  3558. {
  3559. "name": "torch.pinverse",
  3560. "inputs": [
  3561. { "name": "self", "type": "Scalar" },
  3562. { "name": "rcond", "type": "float32", "default": 1e-15 }
  3563. ],
  3564. "outputs": [
  3565. { "name": "output", "type": "Tensor" }
  3566. ]
  3567. },
  3568. {
  3569. "name": "torch.real",
  3570. "inputs": [
  3571. { "name": "self", "type": "Tensor" }
  3572. ],
  3573. "outputs": [
  3574. { "name": "output", "type": "Tensor" }
  3575. ]
  3576. },
  3577. {
  3578. "name": "torch.imag",
  3579. "inputs": [
  3580. { "name": "self", "type": "Tensor" }
  3581. ],
  3582. "outputs": [
  3583. { "name": "output", "type": "Tensor" }
  3584. ]
  3585. },
  3586. {
  3587. "name": "torch.conj",
  3588. "inputs": [
  3589. { "name": "self", "type": "Tensor" }
  3590. ],
  3591. "outputs": [
  3592. { "name": "output", "type": "Tensor" }
  3593. ]
  3594. },
  3595. {
  3596. "name": "torch._conj",
  3597. "inputs": [
  3598. { "name": "self", "type": "Tensor" }
  3599. ],
  3600. "outputs": [
  3601. { "name": "output", "type": "Tensor" }
  3602. ]
  3603. },
  3604. {
  3605. "name": "torch.rsqrt",
  3606. "inputs": [
  3607. { "name": "self", "type": "Tensor" }
  3608. ],
  3609. "outputs": [
  3610. { "name": "output", "type": "Tensor" }
  3611. ]
  3612. },
  3613. {
  3614. "name": "torch.grid_sampler",
  3615. "inputs": [
  3616. { "name": "input", "type": "Tensor" },
  3617. { "name": "grid", "type": "Tensor" },
  3618. { "name": "interpolation_mode", "type": "int64" },
  3619. { "name": "padding_mode", "type": "int64" },
  3620. { "name": "align_corners", "type": "boolean" }
  3621. ],
  3622. "outputs": [
  3623. { "name": "output", "type": "Tensor" }
  3624. ]
  3625. },
  3626. {
  3627. "name": "torch.sort",
  3628. "inputs": [
  3629. { "name": "self", "type": "Tensor" },
  3630. { "name": "dim", "type": "int64", "default": -1 },
  3631. { "name": "descending", "type": "boolean", "default": false }
  3632. ],
  3633. "outputs": [
  3634. { "name": "values", "type": "Tensor" },
  3635. { "name": "indices", "type": "Tensor" }
  3636. ]
  3637. },
  3638. {
  3639. "name": "torch.gt:Tensor",
  3640. "inputs": [
  3641. { "name": "input", "type": "Tensor" },
  3642. { "name": "other", "type": "Tensor" }
  3643. ],
  3644. "outputs": [
  3645. { "name": "output", "type": "Tensor" }
  3646. ]
  3647. },
  3648. {
  3649. "name": "torch.gt:Scalar",
  3650. "inputs": [
  3651. { "name": "input", "type": "Tensor" },
  3652. { "name": "other", "type": "Scalar" }
  3653. ],
  3654. "outputs": [
  3655. { "name": "output", "type": "Tensor" }
  3656. ]
  3657. },
  3658. {
  3659. "name": "torch.mul:Tensor",
  3660. "inputs": [
  3661. { "name": "input", "type": "Tensor" },
  3662. { "name": "other", "type": "Tensor" }
  3663. ],
  3664. "outputs": [
  3665. { "name": "output", "type": "Tensor" }
  3666. ]
  3667. },
  3668. {
  3669. "name": "torch.mul:Scalar",
  3670. "inputs": [
  3671. { "name": "input", "type": "Tensor" },
  3672. { "name": "other", "type": "Scalar" }
  3673. ],
  3674. "outputs": [
  3675. { "name": "output", "type": "Tensor" }
  3676. ]
  3677. },
  3678. {
  3679. "name": "torch.mul:ScalarT",
  3680. "inputs": [
  3681. { "name": "input", "type": "Tensor[]" },
  3682. { "name": "other", "type": "Scalar" }
  3683. ],
  3684. "outputs": [
  3685. { "name": "output", "type": "Tensor[]" }
  3686. ]
  3687. },
  3688. {
  3689. "name": "torch.mul_:Tensor",
  3690. "inputs": [
  3691. { "name": "input", "type": "Tensor" },
  3692. { "name": "other", "type": "Tensor" }
  3693. ],
  3694. "outputs": [
  3695. { "name": "output", "type": "Tensor" }
  3696. ]
  3697. },
  3698. {
  3699. "name": "torch.mul_:Scalar",
  3700. "inputs": [
  3701. { "name": "input", "type": "Tensor" },
  3702. { "name": "other", "type": "Scalar" }
  3703. ],
  3704. "outputs": [
  3705. { "name": "output", "type": "Tensor" }
  3706. ]
  3707. },
  3708. {
  3709. "name": "torch.multinomial",
  3710. "inputs": [
  3711. { "name": "self", "type": "Tensor" },
  3712. { "name": "num_samples", "type": "int64" },
  3713. { "name": "replacement", "type": "boolean", "default": false },
  3714. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  3715. ],
  3716. "outputs": [
  3717. { "name": "output", "type": "Tensor" }
  3718. ]
  3719. },
  3720. {
  3721. "name": "torch.div_:Tensor",
  3722. "inputs": [
  3723. { "name": "input", "type": "Tensor" },
  3724. { "name": "other", "type": "Tensor" }
  3725. ],
  3726. "outputs": [
  3727. { "name": "output", "type": "Tensor" }
  3728. ]
  3729. },
  3730. {
  3731. "name": "torch.div_:Scalar",
  3732. "inputs": [
  3733. { "name": "input", "type": "Tensor" },
  3734. { "name": "other", "type": "Scalar" }
  3735. ],
  3736. "outputs": [
  3737. { "name": "output", "type": "Tensor" }
  3738. ]
  3739. },
  3740. {
  3741. "name": "torch.bmm",
  3742. "inputs": [
  3743. { "name": "input", "type": "Tensor" },
  3744. { "name": "mat2", "type": "Tensor" }
  3745. ],
  3746. "outputs": [
  3747. { "name": "output", "type": "Tensor" }
  3748. ]
  3749. },
  3750. {
  3751. "name": "torch.constant_pad_nd",
  3752. "category": "Tensor",
  3753. "inputs": [
  3754. { "name": "self", "type": "Tensor" },
  3755. { "name": "pad", "type": "int64[]" },
  3756. { "name": "value", "type": "Scalar", "default": 0 }
  3757. ],
  3758. "outputs": [
  3759. { "name": "output", "type": "Tensor" }
  3760. ]
  3761. },
  3762. {
  3763. "name": "torch.ones",
  3764. "inputs": [
  3765. { "name": "size", "type": "int64[]" },
  3766. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3767. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3768. { "name": "device", "type": "Device", "optional": true, "default": null },
  3769. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3770. ],
  3771. "outputs": [
  3772. { "name": "output", "type": "Tensor" }
  3773. ]
  3774. },
  3775. {
  3776. "name": "torch.empty:names",
  3777. "inputs": [
  3778. { "name": "size", "type": "int64[]" },
  3779. { "name": "names", "type": "Dimname[]", "optional": true, "default": null },
  3780. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3781. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3782. { "name": "device", "type": "Device", "optional": true, "default": null },
  3783. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3784. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3785. ],
  3786. "outputs": [
  3787. { "name": "output", "type": "Tensor" }
  3788. ]
  3789. },
  3790. {
  3791. "name": "torch.empty:memory_format",
  3792. "inputs": [
  3793. { "name": "size", "type": "int64[]" },
  3794. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3795. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3796. { "name": "device", "type": "Device", "optional": true, "default": null },
  3797. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3798. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3799. ],
  3800. "outputs": [
  3801. { "name": "output", "type": "Tensor" }
  3802. ]
  3803. },
  3804. {
  3805. "name": "torch.zeros",
  3806. "inputs": [
  3807. { "name": "size", "type": "int64[]" },
  3808. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3809. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3810. { "name": "device", "type": "Device", "optional": true, "default": null },
  3811. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3812. ],
  3813. "outputs": [
  3814. { "name": "output", "type": "Tensor" }
  3815. ]
  3816. },
  3817. {
  3818. "name": "torch.zeros_like",
  3819. "inputs": [
  3820. { "name": "self", "type": "Tensor" },
  3821. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3822. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3823. { "name": "device", "type": "Device", "optional": true, "default": null },
  3824. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3825. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3826. ],
  3827. "outputs": [
  3828. { "name": "output", "type": "Tensor" }
  3829. ]
  3830. },
  3831. {
  3832. "name": "torch.ones_like",
  3833. "inputs": [
  3834. { "name": "input", "type": "Tensor" },
  3835. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3836. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3837. { "name": "device", "type": "Device", "optional": true, "default": null },
  3838. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3839. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3840. ],
  3841. "outputs": [
  3842. { "name": "output", "type": "Tensor" }
  3843. ]
  3844. },
  3845. {
  3846. "name": "torch.amax:",
  3847. "inputs": [
  3848. { "name": "self", "type": "Tensor" },
  3849. { "name": "dim", "type": "int64[]", "default": [] },
  3850. { "name": "keepdim", "type": "boolean", "default": false }
  3851. ],
  3852. "outputs": [
  3853. { "name": "output", "type": "Tensor" }
  3854. ]
  3855. },
  3856. {
  3857. "name": "torch.amin:",
  3858. "inputs": [
  3859. { "name": "self", "type": "Tensor" },
  3860. { "name": "dim", "type": "int64[]", "default": [] },
  3861. { "name": "keepdim", "type": "boolean", "default": false }
  3862. ],
  3863. "outputs": [
  3864. { "name": "output", "type": "Tensor" }
  3865. ]
  3866. },
  3867. {
  3868. "name": "torch._aminmax:",
  3869. "inputs": [
  3870. { "name": "self", "type": "Tensor" }
  3871. ],
  3872. "outputs": [
  3873. { "name": "min", "type": "Tensor" },
  3874. { "name": "max", "type": "Tensor" }
  3875. ]
  3876. },
  3877. {
  3878. "name": "torch._aminmax:dim",
  3879. "inputs": [
  3880. { "name": "self", "type": "Tensor" },
  3881. { "name": "dim", "type": "int64" },
  3882. { "name": "keepdim", "type": "boolean", "default": false }
  3883. ],
  3884. "outputs": [
  3885. { "name": "min", "type": "Tensor" },
  3886. { "name": "max", "type": "Tensor" }
  3887. ]
  3888. },
  3889. {
  3890. "name": "torch.aminmax:",
  3891. "inputs": [
  3892. { "name": "self", "type": "Tensor" },
  3893. { "name": "dim", "type": "int64", "optional": true, "default": null },
  3894. { "name": "keepdim", "type": "boolean", "default": false }
  3895. ],
  3896. "outputs": [
  3897. { "name": "min", "type": "Tensor" },
  3898. { "name": "max", "type": "Tensor" }
  3899. ]
  3900. },
  3901. {
  3902. "name": "torch.arange:",
  3903. "inputs": [
  3904. { "name": "end", "type": "Scalar" },
  3905. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3906. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3907. { "name": "device", "type": "Device", "optional": true, "default": null },
  3908. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3909. ],
  3910. "outputs": [
  3911. { "name": "output", "type": "Tensor" }
  3912. ]
  3913. },
  3914. {
  3915. "name": "torch.arange:start",
  3916. "inputs": [
  3917. { "name": "start", "type": "Scalar" },
  3918. { "name": "end", "type": "Scalar" },
  3919. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3920. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3921. { "name": "device", "type": "Device", "optional": true, "default": null },
  3922. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3923. ],
  3924. "outputs": [
  3925. { "name": "output", "type": "Tensor" }
  3926. ]
  3927. },
  3928. {
  3929. "name": "torch.arange:start_step",
  3930. "inputs": [
  3931. { "name": "start", "type": "Scalar" },
  3932. { "name": "end", "type": "Scalar" },
  3933. { "name": "step", "type": "Scalar" },
  3934. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3935. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3936. { "name": "device", "type": "Device", "optional": true, "default": null },
  3937. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3938. ],
  3939. "outputs": [
  3940. { "name": "output", "type": "Tensor" }
  3941. ]
  3942. },
  3943. {
  3944. "name": "torch.arange:start_out",
  3945. "inputs": [
  3946. { "name": "start", "type": "Scalar" },
  3947. { "name": "end", "type": "Scalar" },
  3948. { "name": "step", "type": "Scalar", "default": 1 }
  3949. ],
  3950. "outputs": [
  3951. { "name": "output", "type": "Tensor" }
  3952. ]
  3953. },
  3954. {
  3955. "name": "torch.clamp",
  3956. "inputs": [
  3957. { "name": "self", "type": "Tensor" },
  3958. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  3959. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  3960. ],
  3961. "outputs": [
  3962. { "name": "output", "type": "Tensor" }
  3963. ]
  3964. },
  3965. {
  3966. "name": "torch.clamp_",
  3967. "inputs": [
  3968. { "name": "self", "type": "Tensor" },
  3969. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  3970. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  3971. ],
  3972. "outputs": [
  3973. { "name": "output", "type": "Tensor" }
  3974. ]
  3975. },
  3976. {
  3977. "name": "torch.clamp_min",
  3978. "inputs": [
  3979. { "name": "self", "type": "Tensor" },
  3980. { "name": "min", "type": "Scalar" }
  3981. ],
  3982. "outputs": [
  3983. { "name": "output", "type": "Tensor" }
  3984. ]
  3985. },
  3986. {
  3987. "name": "torch.clamp_min_",
  3988. "inputs": [
  3989. { "name": "self", "type": "Tensor" },
  3990. { "name": "min", "type": "Scalar" }
  3991. ],
  3992. "outputs": [
  3993. { "name": "output", "type": "Tensor" }
  3994. ]
  3995. },
  3996. {
  3997. "name": "torch.clamp_max",
  3998. "inputs": [
  3999. { "name": "self", "type": "Tensor" },
  4000. { "name": "max", "type": "Scalar" }
  4001. ],
  4002. "outputs": [
  4003. { "name": "output", "type": "Tensor" }
  4004. ]
  4005. },
  4006. {
  4007. "name": "torch.clamp_max_",
  4008. "inputs": [
  4009. { "name": "self", "type": "Tensor" },
  4010. { "name": "max", "type": "Scalar" }
  4011. ],
  4012. "outputs": [
  4013. { "name": "output", "type": "Tensor" }
  4014. ]
  4015. },
  4016. {
  4017. "name": "torch.fill_:Scalar",
  4018. "inputs": [
  4019. { "name": "self", "type": "Tensor" },
  4020. { "name": "value", "type": "Scalar" }
  4021. ],
  4022. "outputs": [
  4023. { "name": "output", "type": "Tensor" }
  4024. ]
  4025. },
  4026. {
  4027. "name": "torch.fill_:Tensor",
  4028. "inputs": [
  4029. { "name": "self", "type": "Tensor" },
  4030. { "name": "value", "type": "Tensor" }
  4031. ],
  4032. "outputs": [
  4033. { "name": "output", "type": "Tensor" }
  4034. ]
  4035. },
  4036. {
  4037. "name": "torch.floor",
  4038. "inputs": [
  4039. { "name": "self", "type": "Tensor" }
  4040. ],
  4041. "outputs": [
  4042. { "name": "output", "type": "Tensor" }
  4043. ]
  4044. },
  4045. {
  4046. "name": "torch.floor_",
  4047. "inputs": [
  4048. { "name": "self", "type": "Tensor" }
  4049. ],
  4050. "outputs": [
  4051. { "name": "output", "type": "Tensor" }
  4052. ]
  4053. },
  4054. {
  4055. "name": "torch.floor_divide",
  4056. "inputs": [
  4057. { "name": "self", "type": "Tensor" },
  4058. { "name": "other", "type": "Tensor" }
  4059. ],
  4060. "outputs": [
  4061. { "name": "output", "type": "Tensor" }
  4062. ]
  4063. },
  4064. {
  4065. "name": "torch.floor_divide_:Tensor",
  4066. "inputs": [
  4067. { "name": "self", "type": "Tensor" },
  4068. { "name": "other", "type": "Tensor" }
  4069. ],
  4070. "outputs": [
  4071. { "name": "output", "type": "Tensor" }
  4072. ]
  4073. },
  4074. {
  4075. "name": "torch.floor_divide_:Scalar",
  4076. "inputs": [
  4077. { "name": "self", "type": "Tensor" },
  4078. { "name": "other", "type": "Scalar" }
  4079. ],
  4080. "outputs": [
  4081. { "name": "output", "type": "Tensor" }
  4082. ]
  4083. },
  4084. {
  4085. "name": "torch.abs",
  4086. "inputs": [
  4087. { "name": "self", "type": "Tensor" }
  4088. ],
  4089. "outputs": [
  4090. { "name": "output", "type": "Tensor" }
  4091. ]
  4092. },
  4093. {
  4094. "name": "torch.abs_",
  4095. "inputs": [
  4096. { "name": "self", "type": "Tensor" }
  4097. ],
  4098. "outputs": [
  4099. { "name": "output", "type": "Tensor" }
  4100. ]
  4101. },
  4102. {
  4103. "name": "torch.leaky_relu_",
  4104. "category": "Activation",
  4105. "inputs": [
  4106. { "name": "input", "type": "Tensor" },
  4107. { "name": "negative_slope", "type": "Scalar", "default": 0.01 }
  4108. ],
  4109. "outputs": [
  4110. { "name": "output", "type": "Tensor" }
  4111. ]
  4112. },
  4113. {
  4114. "name": "torch.instance_norm",
  4115. "category": "Normalization",
  4116. "inputs": [
  4117. { "name": "input", "type": "Tensor" },
  4118. { "name": "weight", "type": "Tensor", "optional": true },
  4119. { "name": "bias", "type": "Tensor", "optional": true },
  4120. { "name": "running_mean", "type": "Tensor", "optional": true },
  4121. { "name": "running_var", "type": "Tensor", "optional": true },
  4122. { "name": "use_input_stats", "type": "boolean" },
  4123. { "name": "momentum", "type": "float32" },
  4124. { "name": "eps", "type": "float32" },
  4125. { "name": "cudnn_enabled", "type": "boolean" }
  4126. ],
  4127. "outputs": [
  4128. { "name": "output", "type": "Tensor" }
  4129. ]
  4130. },
  4131. {
  4132. "name": "torch.reflection_pad1d",
  4133. "category": "Tensor",
  4134. "inputs": [
  4135. { "name": "input", "type": "Tensor" },
  4136. { "name": "padding", "type": "int64[]" }
  4137. ],
  4138. "outputs": [
  4139. { "name": "output", "type": "Tensor" }
  4140. ]
  4141. },
  4142. {
  4143. "name": "torch.replication_pad1d",
  4144. "category": "Tensor",
  4145. "inputs": [
  4146. { "name": "input", "type": "Tensor" },
  4147. { "name": "padding", "type": "int64[]" }
  4148. ],
  4149. "outputs": [
  4150. { "name": "output", "type": "Tensor" }
  4151. ]
  4152. },
  4153. {
  4154. "name": "torch.replication_pad2d",
  4155. "category": "Tensor",
  4156. "inputs": [
  4157. { "name": "input", "type": "Tensor" },
  4158. { "name": "padding", "type": "int64[]" }
  4159. ],
  4160. "outputs": [
  4161. { "name": "output", "type": "Tensor" }
  4162. ]
  4163. },
  4164. {
  4165. "name": "torch.replication_pad3d",
  4166. "category": "Tensor",
  4167. "inputs": [
  4168. { "name": "input", "type": "Tensor" },
  4169. { "name": "padding", "type": "int64[]" }
  4170. ],
  4171. "outputs": [
  4172. { "name": "output", "type": "Tensor" }
  4173. ]
  4174. },
  4175. {
  4176. "name": "torch.reflection_pad2d",
  4177. "category": "Tensor",
  4178. "inputs": [
  4179. { "name": "input", "type": "Tensor" },
  4180. { "name": "padding", "type": "int64[]" }
  4181. ],
  4182. "outputs": [
  4183. { "name": "output", "type": "Tensor" }
  4184. ]
  4185. },
  4186. {
  4187. "name": "torch.reflection_pad3d",
  4188. "category": "Tensor",
  4189. "inputs": [
  4190. { "name": "input", "type": "Tensor" },
  4191. { "name": "padding", "type": "int64[]" }
  4192. ],
  4193. "outputs": [
  4194. { "name": "output", "type": "Tensor" }
  4195. ]
  4196. },
  4197. {
  4198. "name": "torch.rnn_relu:input",
  4199. "category": "Layer",
  4200. "inputs": [
  4201. { "name": "input", "type": "Tensor" },
  4202. { "name": "hx", "type": "Tensor" },
  4203. { "name": "params", "type": "Tensor[]" },
  4204. { "name": "has_biases", "type": "boolean" },
  4205. { "name": "num_layers", "type": "int64" },
  4206. { "name": "dropout", "type": "float32" },
  4207. { "name": "train", "type": "boolean" },
  4208. { "name": "bidirectional", "type": "boolean" },
  4209. { "name": "batch_first", "type": "boolean" }
  4210. ],
  4211. "outputs": [
  4212. { "name": "output", "type": "Tensor" },
  4213. { "name": "?", "type": "Tensor" }
  4214. ]
  4215. },
  4216. {
  4217. "name": "torch.rnn_tanh:input",
  4218. "category": "Layer",
  4219. "inputs": [
  4220. { "name": "input", "type": "Tensor" },
  4221. { "name": "hx", "type": "Tensor" },
  4222. { "name": "params", "type": "Tensor[]" },
  4223. { "name": "has_biases", "type": "boolean" },
  4224. { "name": "num_layers", "type": "int64" },
  4225. { "name": "dropout", "type": "float32" },
  4226. { "name": "train", "type": "boolean" },
  4227. { "name": "bidirectional", "type": "boolean" },
  4228. { "name": "batch_first", "type": "boolean" }
  4229. ],
  4230. "outputs": [
  4231. { "name": "output" },
  4232. { "name": "?" }
  4233. ]
  4234. },
  4235. {
  4236. "name": "torch.roll:",
  4237. "category": "Layer",
  4238. "inputs": [
  4239. { "name": "self", "type": "Tensor" },
  4240. { "name": "shifts", "type": "int64[]" },
  4241. { "name": "dims", "type": "int64[]", "default": [] }
  4242. ],
  4243. "outputs": [
  4244. { "name": "output" }
  4245. ]
  4246. },
  4247. {
  4248. "name": "torch.detach",
  4249. "inputs": [
  4250. { "name": "self", "type": "Tensor" }
  4251. ],
  4252. "outputs": [
  4253. { "name": "output", "type": "Tensor" }
  4254. ]
  4255. },
  4256. {
  4257. "name": "torch.unique_consecutive:",
  4258. "category": "Layer",
  4259. "inputs": [
  4260. { "name": "self", "type": "Tensor" },
  4261. { "name": "return_inverse", "type": "boolean", "default": false },
  4262. { "name": "return_counts", "type": "boolean", "default": false },
  4263. { "name": "dim", "type": "int64", "optional": true, "default": null }
  4264. ],
  4265. "outputs": [
  4266. { "name": "output1", "type": "Tensor" },
  4267. { "name": "output2", "type": "Tensor" },
  4268. { "name": "output3", "type": "Tensor" }
  4269. ]
  4270. },
  4271. {
  4272. "name": "torch.unique_dim_consecutive:",
  4273. "category": "Layer",
  4274. "inputs": [
  4275. { "name": "self", "type": "Tensor" },
  4276. { "name": "dim", "type": "int64" },
  4277. { "name": "return_inverse", "type": "boolean", "default": false },
  4278. { "name": "return_counts", "type": "boolean", "default": false }
  4279. ],
  4280. "outputs": [
  4281. { "name": "output1", "type": "Tensor" },
  4282. { "name": "output2", "type": "Tensor" },
  4283. { "name": "output3", "type": "Tensor" }
  4284. ]
  4285. },
  4286. {
  4287. "name": "torch.upsample_bicubic2d:",
  4288. "category": "Layer",
  4289. "inputs": [
  4290. { "name": "self", "type": "Tensor" },
  4291. { "name": "output_size", "type": "int64[]" },
  4292. { "name": "align_corners", "type": "boolean" },
  4293. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  4294. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  4295. ],
  4296. "outputs": [
  4297. { "name": "output", "type": "Tensor" }
  4298. ]
  4299. },
  4300. {
  4301. "name": "torch.upsample_bicubic2d:vec",
  4302. "category": "Layer",
  4303. "inputs": [
  4304. { "name": "input", "type": "Tensor" },
  4305. { "name": "output_size", "type": "int64[]" },
  4306. { "name": "align_corners", "type": "boolean" },
  4307. { "name": "scales_factors", "type": "float32[]", "optional": true }
  4308. ],
  4309. "outputs": [
  4310. { "name": "output", "type": "Tensor" }
  4311. ]
  4312. },
  4313. {
  4314. "name": "torch.upsample_linear1d:",
  4315. "inputs": [
  4316. { "name": "self", "type": "Tensor" },
  4317. { "name": "output_size", "type": "int64[]" },
  4318. { "name": "align_corners", "type": "boolean" },
  4319. { "name": "scales", "type": "float32", "optional": true, "default": null }
  4320. ],
  4321. "outputs": [
  4322. { "name": "output", "type": "Tensor" }
  4323. ]
  4324. },
  4325. {
  4326. "name": "torch.upsample_linear1d:vec",
  4327. "inputs": [
  4328. { "name": "self", "type": "Tensor" },
  4329. { "name": "output_size", "type": "int64[]", "optional": true },
  4330. { "name": "align_corners", "type": "boolean" },
  4331. { "name": "scale_factors", "type": "float32[]", "optional": true }
  4332. ],
  4333. "outputs": [
  4334. { "name": "output", "type": "Tensor" }
  4335. ]
  4336. },
  4337. {
  4338. "name": "torch.upsample_bilinear2d:",
  4339. "category": "Layer",
  4340. "inputs": [
  4341. { "name": "self", "type": "Tensor" },
  4342. { "name": "output_size", "type": "int64[]" },
  4343. { "name": "align_corners", "type": "boolean" },
  4344. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  4345. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  4346. ],
  4347. "outputs": [
  4348. { "name": "output", "type": "Tensor" }
  4349. ]
  4350. },
  4351. {
  4352. "name": "torch.upsample_bilinear2d:vec",
  4353. "category": "Layer",
  4354. "inputs": [
  4355. { "name": "input", "type": "Tensor" },
  4356. { "name": "output_size", "type": "int64[]", "optional": "true" },
  4357. { "name": "align_corners", "type": "boolean" },
  4358. { "name": "scales_factors", "type": "float32[]", "optional": true }
  4359. ],
  4360. "outputs": [
  4361. { "name": "output", "type": "Tensor" }
  4362. ]
  4363. },
  4364. {
  4365. "name": "torch.upsample_bilinear2d:vec",
  4366. "category": "Layer",
  4367. "inputs": [
  4368. { "name": "input", "type": "Tensor" },
  4369. { "name": "output_size", "type": "int64[]", "optional": true },
  4370. { "name": "align_corners", "type": "boolean" },
  4371. { "name": "scale_factors", "type": "float32[]", "optional": true }
  4372. ],
  4373. "outputs": [
  4374. { "name": "output", "type": "Tensor" }
  4375. ]
  4376. },
  4377. {
  4378. "name": "torch.upsample_nearest1d:",
  4379. "category": "Layer",
  4380. "inputs": [
  4381. { "name": "self", "type": "Tensor" },
  4382. { "name": "output_size", "type": "int64[]" },
  4383. { "name": "scales", "type": "float32", "optional": true, "default": null }
  4384. ],
  4385. "outputs": [
  4386. { "name": "output", "type": "Tensor" }
  4387. ]
  4388. },
  4389. {
  4390. "name": "torch.upsample_nearest1d:vec",
  4391. "category": "Layer",
  4392. "inputs": [
  4393. { "name": "input", "type": "Tensor" },
  4394. { "name": "output_size", "type": "int64[]", "optional": true },
  4395. { "name": "scales_factors", "type": "float32[]", "optional": true }
  4396. ],
  4397. "outputs": [
  4398. { "name": "output", "type": "Tensor" }
  4399. ]
  4400. },
  4401. {
  4402. "name": "torch.upsample_nearest2d:",
  4403. "category": "Layer",
  4404. "inputs": [
  4405. { "name": "self", "type": "Tensor" },
  4406. { "name": "output_size", "type": "int64[]" },
  4407. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  4408. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  4409. ],
  4410. "outputs": [
  4411. { "name": "output", "type": "Tensor" }
  4412. ]
  4413. },
  4414. {
  4415. "name": "torch.upsample_nearest2d:vec",
  4416. "category": "Layer",
  4417. "inputs": [
  4418. { "name": "input", "type": "Tensor" },
  4419. { "name": "output_size", "type": "int64[]", "optional": true },
  4420. { "name": "scales_factors", "type": "float32[]", "optional": true }
  4421. ],
  4422. "outputs": [
  4423. { "name": "output", "type": "Tensor" }
  4424. ]
  4425. },
  4426. {
  4427. "name": "torch.upsample_nearest3d:",
  4428. "category": "Layer",
  4429. "inputs": [
  4430. { "name": "self", "type": "Tensor" },
  4431. { "name": "output_size", "type": "int64[]" },
  4432. { "name": "scales_d", "type": "float32", "optional": true, "default": null },
  4433. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  4434. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  4435. ],
  4436. "outputs": [
  4437. { "name": "output", "type": "Tensor" }
  4438. ]
  4439. },
  4440. {
  4441. "name": "torch.upsample_nearest3d:vec",
  4442. "category": "Layer",
  4443. "inputs": [
  4444. { "name": "input", "type": "Tensor" },
  4445. { "name": "output_size", "type": "int64[]", "optional": true },
  4446. { "name": "scales_factors", "type": "float32[]", "optional": true }
  4447. ],
  4448. "outputs": [
  4449. { "name": "output", "type": "Tensor" }
  4450. ]
  4451. },
  4452. {
  4453. "name": "torch.upsample_trilinear3d:",
  4454. "inputs": [
  4455. { "name": "self", "type": "Tensor" },
  4456. { "name": "output_size", "type": "int64[]" },
  4457. { "name": "align_corners", "type": "boolean" },
  4458. { "name": "scales_d", "type": "float32", "optional": true, "default": null },
  4459. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  4460. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  4461. ],
  4462. "outputs": [
  4463. { "name": "output", "type": "Tensor" }
  4464. ]
  4465. },
  4466. {
  4467. "name": "torch.upsample_trilinear3d:vec",
  4468. "inputs": [
  4469. { "name": "input", "type": "Tensor" },
  4470. { "name": "output_size", "type": "int64[]", "optional": true },
  4471. { "name": "align_corners", "type": "boolean" },
  4472. { "name": "scale_factors", "type": "float32[]", "optional": true }
  4473. ],
  4474. "outputs": [
  4475. { "name": "output", "type": "int64" }
  4476. ]
  4477. },
  4478. {
  4479. "name": "torch.size",
  4480. "inputs": [
  4481. { "name": "self", "type": "Tensor" },
  4482. { "name": "dim", "type": "int64" }
  4483. ],
  4484. "outputs": [
  4485. { "name": "output", "type": "int64" }
  4486. ]
  4487. },
  4488. {
  4489. "name": "torch.exp",
  4490. "inputs": [
  4491. { "name": "self", "type": "Tensor" }
  4492. ],
  4493. "outputs": [
  4494. { "name": "output", "type": "Tensor" }
  4495. ]
  4496. },
  4497. {
  4498. "name": "torch.exp_",
  4499. "inputs": [
  4500. { "name": "self", "type": "Tensor" }
  4501. ],
  4502. "outputs": [
  4503. { "name": "output", "type": "Tensor" }
  4504. ]
  4505. },
  4506. {
  4507. "name": "torch.reciprocal",
  4508. "inputs": [
  4509. { "name": "self", "type": "Tensor" }
  4510. ],
  4511. "outputs": [
  4512. { "name": "output", "type": "Tensor" }
  4513. ]
  4514. },
  4515. {
  4516. "name": "torch.square",
  4517. "inputs": [
  4518. { "name": "self", "type": "Tensor" }
  4519. ],
  4520. "outputs": [
  4521. { "name": "output", "type": "Tensor" }
  4522. ]
  4523. },
  4524. {
  4525. "name": "torch.square_",
  4526. "inputs": [
  4527. { "name": "self", "type": "Tensor" }
  4528. ],
  4529. "outputs": [
  4530. { "name": "output", "type": "Tensor" }
  4531. ]
  4532. },
  4533. {
  4534. "name": "torch.sqrt",
  4535. "inputs": [
  4536. { "name": "self", "type": "Tensor" }
  4537. ],
  4538. "outputs": [
  4539. { "name": "output", "type": "Tensor" }
  4540. ]
  4541. },
  4542. {
  4543. "name": "torch.std:",
  4544. "inputs": [
  4545. { "name": "self", "type": "Tensor" },
  4546. { "name": "unbiased", "type": "boolean", "default": true }
  4547. ],
  4548. "outputs": [
  4549. { "name": "output", "type": "Tensor" }
  4550. ]
  4551. },
  4552. {
  4553. "name": "torch.std:dim",
  4554. "inputs": [
  4555. { "name": "self", "type": "Tensor" },
  4556. { "name": "dim", "type": "int64[]" },
  4557. { "name": "unbiased", "type": "boolean", "default": true },
  4558. { "name": "keepdim", "type": "boolean", "default": false }
  4559. ],
  4560. "outputs": [
  4561. { "name": "output", "type": "Tensor" }
  4562. ]
  4563. },
  4564. {
  4565. "name": "torch.std_mean:",
  4566. "inputs": [
  4567. { "name": "self", "type": "Tensor" },
  4568. { "name": "unbiased", "type": "boolean", "default": true }
  4569. ],
  4570. "outputs": [
  4571. { "name": "output", "type": "Tensor" }
  4572. ]
  4573. },
  4574. {
  4575. "name": "torch.std_mean:dim",
  4576. "inputs": [
  4577. { "name": "self", "type": "Tensor" },
  4578. { "name": "dim", "type": "int64[]" },
  4579. { "name": "unbiased", "type": "boolean", "default": true },
  4580. { "name": "keepdim", "type": "boolean", "default": false }
  4581. ],
  4582. "outputs": [
  4583. { "name": "output", "type": "Tensor" }
  4584. ]
  4585. },
  4586. {
  4587. "name": "torch.erf",
  4588. "inputs": [
  4589. { "name": "input", "type": "Tensor" }
  4590. ],
  4591. "outputs": [
  4592. { "name": "output", "type": "Tensor" }
  4593. ]
  4594. },
  4595. {
  4596. "name": "torch.argmin",
  4597. "inputs": [
  4598. { "name": "input", "type": "Tensor" },
  4599. { "name": "dim", "type": "int64" },
  4600. { "name": "keepdim", "type": "boolean", "default": false }
  4601. ],
  4602. "outputs": [
  4603. { "name": "output", "type": "Tensor" }
  4604. ]
  4605. },
  4606. {
  4607. "name": "torch.argmax",
  4608. "inputs": [
  4609. { "name": "input", "type": "Tensor" },
  4610. { "name": "dim", "type": "int64" },
  4611. { "name": "keepdim", "type": "boolean", "default": false }
  4612. ],
  4613. "outputs": [
  4614. { "name": "output", "type": "Tensor" }
  4615. ]
  4616. },
  4617. {
  4618. "name": "torch.log",
  4619. "inputs": [
  4620. { "name": "self", "type": "Tensor" }
  4621. ],
  4622. "outputs": [
  4623. { "name": "output", "type": "Tensor" }
  4624. ]
  4625. },
  4626. {
  4627. "name": "torch.log10",
  4628. "inputs": [
  4629. { "name": "self", "type": "Tensor" }
  4630. ],
  4631. "outputs": [
  4632. { "name": "output", "type": "Tensor" }
  4633. ]
  4634. },
  4635. {
  4636. "name": "torch.log1p",
  4637. "inputs": [
  4638. { "name": "self", "type": "Tensor" }
  4639. ],
  4640. "outputs": [
  4641. { "name": "output", "type": "Tensor" }
  4642. ]
  4643. },
  4644. {
  4645. "name": "torch.log2",
  4646. "inputs": [
  4647. { "name": "self", "type": "Tensor" }
  4648. ],
  4649. "outputs": [
  4650. { "name": "output", "type": "Tensor" }
  4651. ]
  4652. },
  4653. {
  4654. "name": "torch.logdet",
  4655. "inputs": [
  4656. { "name": "self", "type": "Tensor" }
  4657. ],
  4658. "outputs": [
  4659. { "name": "output", "type": "Tensor" }
  4660. ]
  4661. },
  4662. {
  4663. "name": "torch.affine_grid_generator",
  4664. "inputs": [
  4665. { "name": "theta", "type": "Tensor" },
  4666. { "name": "size", "type": "int64[]" },
  4667. { "name": "align_corners", "type": "boolean" }
  4668. ],
  4669. "outputs": [
  4670. { "name": "output", "type": "Tensor" }
  4671. ]
  4672. },
  4673. {
  4674. "name": "torch.__interpolate",
  4675. "inputs": [
  4676. { "name": "input", "type": "Tensor" },
  4677. { "name": "size", "type": "int64" },
  4678. { "name": "scale_factor", "type": "float32[]" },
  4679. { "name": "mode", "type": "string" },
  4680. { "name": "align_corners", "type": "boolean" },
  4681. { "name": "recompute_scale_factor", "type": "boolean" }
  4682. ],
  4683. "outputs": [
  4684. { "name": "output", "type": "Tensor" }
  4685. ]
  4686. },
  4687. {
  4688. "name": "torch.rand:",
  4689. "inputs": [
  4690. { "name": "size", "type": "int64[]" },
  4691. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4692. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4693. { "name": "device", "type": "Device", "optional": true, "default": null },
  4694. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4695. ],
  4696. "outputs": [
  4697. { "name": "output", "type": "Tensor" }
  4698. ]
  4699. },
  4700. {
  4701. "name": "torch.rand:generator",
  4702. "inputs": [
  4703. { "name": "high", "type": "int64", "optional": true, "default": null },
  4704. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4705. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4706. { "name": "device", "type": "Device", "optional": true, "default": null },
  4707. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  4708. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  4709. ],
  4710. "outputs": [
  4711. { "name": "output", "type": "Tensor" }
  4712. ]
  4713. },
  4714. {
  4715. "name": "torch.rand_link",
  4716. "inputs": [
  4717. { "name": "size", "type": "int64[]" },
  4718. { "name": "generator", "type": "Generator", "optional": true, "default": null },
  4719. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4720. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4721. { "name": "device", "type": "Device", "optional": true, "default": null },
  4722. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4723. ],
  4724. "outputs": [
  4725. { "name": "output", "type": "Tensor" }
  4726. ]
  4727. },
  4728. {
  4729. "name": "torch.randint:generator",
  4730. "inputs": [
  4731. { "name": "high", "type": "int64", "optional": true, "default": null },
  4732. { "name": "size", "type": "int64[]" },
  4733. { "name": "generator", "type": "Generator" },
  4734. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4735. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4736. { "name": "device", "type": "Device", "optional": true, "default": null },
  4737. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4738. ],
  4739. "outputs": [
  4740. { "name": "output", "type": "Tensor" }
  4741. ]
  4742. },
  4743. {
  4744. "name": "torch.randint:low",
  4745. "inputs": [
  4746. { "name": "low", "type": "int64", "optional": true, "default": null },
  4747. { "name": "high", "type": "int64", "optional": true, "default": null },
  4748. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4749. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4750. { "name": "device", "type": "Device", "optional": true, "default": null },
  4751. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4752. ],
  4753. "outputs": [
  4754. { "name": "output", "type": "Tensor" }
  4755. ]
  4756. },
  4757. {
  4758. "name": "torch.randn",
  4759. "inputs": [
  4760. { "name": "size", "type": "int64[]" },
  4761. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4762. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4763. { "name": "device", "type": "Device", "optional": true, "default": null },
  4764. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4765. ],
  4766. "outputs": [
  4767. { "name": "output", "type": "Tensor" }
  4768. ]
  4769. },
  4770. {
  4771. "name": "torch.unsafe_split:Tensor",
  4772. "category": "Tensor",
  4773. "inputs": [
  4774. { "name": "input", "type": "Tensor" },
  4775. { "name": "split_size", "type": "int64" },
  4776. { "name": "dim", "type": "int64" }
  4777. ],
  4778. "outputs": [
  4779. { "name": "outputs", "type": "Tensor[]" }
  4780. ]
  4781. },
  4782. {
  4783. "name": "torch._sparse_coo_tensor_unsafe:",
  4784. "category": "Tensor",
  4785. "inputs": [
  4786. { "name": "indices", "type": "Tensor" },
  4787. { "name": "values", "type": "Tensor" },
  4788. { "name": "size", "type": "int64[]" },
  4789. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4790. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4791. { "name": "device", "type": "Device", "optional": true, "default": null },
  4792. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4793. ],
  4794. "outputs": [
  4795. { "name": "outputs", "type": "Tensor[]" }
  4796. ]
  4797. },
  4798. {
  4799. "name": "torch.split:Tensor",
  4800. "category": "Tensor",
  4801. "inputs": [
  4802. { "name": "self", "type": "Tensor" },
  4803. { "name": "split_size", "type": "int64" },
  4804. { "name": "dim", "type": "int64" }
  4805. ],
  4806. "outputs": [
  4807. { "name": "outputs", "type": "Tensor[]" }
  4808. ]
  4809. },
  4810. {
  4811. "name": "torch.split:Tensor_sizes",
  4812. "category": "Tensor",
  4813. "inputs": [
  4814. { "name": "self", "type": "Tensor" },
  4815. { "name": "split_sizes", "type": "int64[]" },
  4816. { "name": "dim", "type": "int64" }
  4817. ],
  4818. "outputs": [
  4819. { "name": "outputs", "type": "Tensor[]" }
  4820. ]
  4821. },
  4822. {
  4823. "name": "torch.split_with_sizes",
  4824. "category": "Tensor",
  4825. "inputs": [
  4826. { "name": "self", "type": "Tensor" },
  4827. { "name": "split_sizes", "type": "int64[]" },
  4828. { "name": "dim", "type": "int64" }
  4829. ],
  4830. "outputs": [
  4831. { "name": "outputs", "type": "Tensor[]" }
  4832. ]
  4833. },
  4834. {
  4835. "name": "torch.remainder:Tensor",
  4836. "inputs": [
  4837. { "name": "self", "type": "Tensor" },
  4838. { "name": "other", "type": "Tensor" }
  4839. ],
  4840. "outputs": [
  4841. { "name": "output", "type": "Tensor" }
  4842. ]
  4843. },
  4844. {
  4845. "name": "torch.remainder:Scalar",
  4846. "inputs": [
  4847. { "name": "self", "type": "Tensor" },
  4848. { "name": "other", "type": "Scalar" }
  4849. ],
  4850. "outputs": [
  4851. { "name": "output", "type": "Tensor" }
  4852. ]
  4853. },
  4854. {
  4855. "name": "torch.remainder_:Tensor",
  4856. "inputs": [
  4857. { "name": "self", "type": "Tensor" },
  4858. { "name": "other", "type": "Tensor" }
  4859. ],
  4860. "outputs": [
  4861. { "name": "output", "type": "Tensor" }
  4862. ]
  4863. },
  4864. {
  4865. "name": "torch.remainder_:Scalar",
  4866. "inputs": [
  4867. { "name": "self", "type": "Tensor" },
  4868. { "name": "other", "type": "Scalar" }
  4869. ],
  4870. "outputs": [
  4871. { "name": "output", "type": "Tensor" }
  4872. ]
  4873. },
  4874. {
  4875. "name": "torch.layer_norm",
  4876. "category": "Normalization",
  4877. "inputs": [
  4878. { "name": "input", "type": "Tensor" },
  4879. { "name": "normalized_shape", "type": "int64[]" },
  4880. { "name": "weight", "type": "Tensor", "optional": true, "default": null },
  4881. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  4882. { "name": "eps", "type": "float32", "default": 0.00001 },
  4883. { "name": "cudnn_enable", "type": "boolean", "default": true }
  4884. ],
  4885. "outputs": [
  4886. { "name": "output", "type": "Tensor" }
  4887. ]
  4888. },
  4889. {
  4890. "name": "torch.scatter:src",
  4891. "inputs": [
  4892. { "name": "self", "type": "Tensor" },
  4893. { "name": "dim", "type": "int64" },
  4894. { "name": "index", "type": "Tensor" },
  4895. { "name": "src", "type": "Tensor" }
  4896. ],
  4897. "outputs": [
  4898. { "name": "output", "type": "Tensor" }
  4899. ]
  4900. },
  4901. {
  4902. "name": "torch.scatter_:src",
  4903. "inputs": [
  4904. { "name": "self", "type": "Tensor" },
  4905. { "name": "dim", "type": "int64" },
  4906. { "name": "index", "type": "Tensor" },
  4907. { "name": "src", "type": "Tensor" }
  4908. ],
  4909. "outputs": [
  4910. { "name": "output", "type": "Tensor" }
  4911. ]
  4912. },
  4913. {
  4914. "name": "torch.scatter:value",
  4915. "inputs": [
  4916. { "name": "self", "type": "Tensor" },
  4917. { "name": "dim", "type": "int64" },
  4918. { "name": "index", "type": "Tensor" },
  4919. { "name": "src", "type": "Scalar" }
  4920. ],
  4921. "outputs": [
  4922. { "name": "output", "type": "Tensor" }
  4923. ]
  4924. },
  4925. {
  4926. "name": "torch.scatter_:value",
  4927. "inputs": [
  4928. { "name": "self", "type": "Tensor" },
  4929. { "name": "dim", "type": "int64" },
  4930. { "name": "index", "type": "Tensor" },
  4931. { "name": "src", "type": "Scalar" }
  4932. ],
  4933. "outputs": [
  4934. { "name": "output", "type": "Tensor" }
  4935. ]
  4936. },
  4937. {
  4938. "name": "torch.scatter:reduce",
  4939. "inputs": [
  4940. { "name": "self", "type": "Tensor" },
  4941. { "name": "dim", "type": "int64" },
  4942. { "name": "index", "type": "Tensor" },
  4943. { "name": "src", "type": "Tensor" },
  4944. { "name": "reduce", "type": "string" }
  4945. ],
  4946. "outputs": [
  4947. { "name": "output", "type": "Tensor" }
  4948. ]
  4949. },
  4950. {
  4951. "name": "torch.scatter_:reduce",
  4952. "inputs": [
  4953. { "name": "self", "type": "Tensor" },
  4954. { "name": "dim", "type": "int64" },
  4955. { "name": "index", "type": "Tensor" },
  4956. { "name": "src", "type": "Tensor" },
  4957. { "name": "reduce", "type": "string" }
  4958. ],
  4959. "outputs": [
  4960. { "name": "output", "type": "Tensor" }
  4961. ]
  4962. },
  4963. {
  4964. "name": "torch.scatter:value_reduce",
  4965. "inputs": [
  4966. { "name": "self", "type": "Tensor" },
  4967. { "name": "dim", "type": "int64" },
  4968. { "name": "index", "type": "Tensor" },
  4969. { "name": "src", "type": "Scalar" },
  4970. { "name": "reduce", "type": "string" }
  4971. ],
  4972. "outputs": [
  4973. { "name": "output", "type": "Tensor" }
  4974. ]
  4975. },
  4976. {
  4977. "name": "torch.scatter_:value_reduce",
  4978. "inputs": [
  4979. { "name": "self", "type": "Tensor" },
  4980. { "name": "dim", "type": "int64" },
  4981. { "name": "index", "type": "Tensor" },
  4982. { "name": "src", "type": "Scalar" },
  4983. { "name": "reduce", "type": "string" }
  4984. ],
  4985. "outputs": [
  4986. { "name": "output", "type": "Tensor" }
  4987. ]
  4988. },
  4989. {
  4990. "name": "torch.scatter_add_",
  4991. "inputs": [
  4992. { "name": "self", "type": "Tensor" },
  4993. { "name": "dim", "type": "int64"},
  4994. { "name": "index", "type": "Tensor" },
  4995. { "name": "src", "type": "Tensor" }
  4996. ],
  4997. "outputs": [
  4998. { "name": "output", "type": "Tensor" }
  4999. ]
  5000. },
  5001. {
  5002. "name": "torch.sin",
  5003. "inputs": [
  5004. { "name": "self", "type": "Tensor" }
  5005. ],
  5006. "outputs": [
  5007. { "name": "output", "type": "Tensor" }
  5008. ]
  5009. },
  5010. {
  5011. "name": "torch.sinh",
  5012. "inputs": [
  5013. { "name": "self", "type": "Tensor" }
  5014. ],
  5015. "outputs": [
  5016. { "name": "output", "type": "Tensor" }
  5017. ]
  5018. },
  5019. {
  5020. "name": "torch.cos",
  5021. "inputs": [
  5022. { "name": "self", "type": "Tensor" }
  5023. ],
  5024. "outputs": [
  5025. { "name": "output", "type": "Tensor" }
  5026. ]
  5027. },
  5028. {
  5029. "name": "torch.cosh",
  5030. "inputs": [
  5031. { "name": "self", "type": "Tensor" }
  5032. ],
  5033. "outputs": [
  5034. { "name": "output", "type": "Tensor" }
  5035. ]
  5036. },
  5037. {
  5038. "name": "torch.ctc_loss:IntList",
  5039. "inputs": [
  5040. { "name": "log_probs", "type": "Tensor" },
  5041. { "name": "targets", "type": "Tensor" },
  5042. { "name": "input_lengths", "type": "int64[]" },
  5043. { "name": "target_lengths", "type": "int64[]" },
  5044. { "name": "blank", "type": "int64", "default": 0 },
  5045. { "name": "reduction", "type": "int64" },
  5046. { "name": "zero_infinity", "type": "boolean", "default": "false" }
  5047. ],
  5048. "outputs": [
  5049. { "name": "output", "type": "Tensor" }
  5050. ]
  5051. },
  5052. {
  5053. "name": "torch._ctc_loss:",
  5054. "inputs": [
  5055. { "name": "log_probs", "type": "Tensor" },
  5056. { "name": "targets", "type": "Tensor" },
  5057. { "name": "input_lengths", "type": "int64[]" },
  5058. { "name": "target_lengths", "type": "int64[]" },
  5059. { "name": "blank", "type": "int64", "default": 0 },
  5060. { "name": "zero_infinity", "type": "boolean", "default": "false" }
  5061. ],
  5062. "outputs": [
  5063. { "name": "output", "type": "Tensor" },
  5064. { "name": "output", "type": "Tensor" }
  5065. ]
  5066. },
  5067. {
  5068. "name": "torch.tan",
  5069. "inputs": [
  5070. { "name": "self", "type": "Tensor" }
  5071. ],
  5072. "outputs": [
  5073. { "name": "output", "type": "Tensor" }
  5074. ]
  5075. },
  5076. {
  5077. "name": "torch.atan2:Tensor",
  5078. "inputs": [
  5079. { "name": "input", "type": "Tensor" },
  5080. { "name": "other", "type": "Tensor" }
  5081. ],
  5082. "outputs": [
  5083. { "name": "output", "type": "Tensor" }
  5084. ]
  5085. },
  5086. {
  5087. "name": "torch.tan_",
  5088. "inputs": [
  5089. { "name": "self", "type": "Tensor" }
  5090. ],
  5091. "outputs": [
  5092. { "name": "output", "type": "Tensor" }
  5093. ]
  5094. },
  5095. {
  5096. "name": "torch.tanh",
  5097. "inputs": [
  5098. { "name": "self", "type": "Tensor" }
  5099. ],
  5100. "outputs": [
  5101. { "name": "output", "type": "Tensor" }
  5102. ]
  5103. },
  5104. {
  5105. "name": "torch.tanh_",
  5106. "inputs": [
  5107. { "name": "self", "type": "Tensor" }
  5108. ],
  5109. "outputs": [
  5110. { "name": "output", "type": "Tensor" }
  5111. ]
  5112. },
  5113. {
  5114. "name": "torch.cdist:",
  5115. "inputs": [
  5116. { "name": "x1", "type": "Tensor" },
  5117. { "name": "x2", "type": "Tensor" },
  5118. { "name": "p", "type": "float32", "default": 2 },
  5119. { "name": "compute_mode", "type": "int64", "optional": true, "default": null }
  5120. ],
  5121. "outputs": [
  5122. { "name": "output", "type": "Tensor" }
  5123. ]
  5124. },
  5125. {
  5126. "name": "torch.pdist:",
  5127. "inputs": [
  5128. { "name": "self", "type": "Tensor" },
  5129. { "name": "p", "type": "float32", "default": 2 }
  5130. ],
  5131. "outputs": [
  5132. { "name": "output", "type": "Tensor" }
  5133. ]
  5134. },
  5135. {
  5136. "name": "torch.narrow:",
  5137. "inputs": [
  5138. { "name": "self", "type": "Tensor" },
  5139. { "name": "dim", "type": "int64" },
  5140. { "name": "start", "type": "int64" },
  5141. { "name": "length", "type": "int64" }
  5142. ],
  5143. "outputs": [
  5144. { "name": "output", "type": "Tensor" }
  5145. ]
  5146. },
  5147. {
  5148. "name": "torch.narrow:Tensor",
  5149. "inputs": [
  5150. { "name": "self", "type": "Tensor" },
  5151. { "name": "dim", "type": "int64" },
  5152. { "name": "start", "type": "Tensor" },
  5153. { "name": "length", "type": "int64" }
  5154. ],
  5155. "outputs": [
  5156. { "name": "output", "type": "Tensor" }
  5157. ]
  5158. },
  5159. {
  5160. "name": "torch.narrow_copy",
  5161. "inputs": [
  5162. { "name": "self", "type": "Tensor" },
  5163. { "name": "dim", "type": "int64" },
  5164. { "name": "start", "type": "int64" },
  5165. { "name": "length", "type": "int64" }
  5166. ],
  5167. "outputs": [
  5168. { "name": "output", "type": "Tensor" }
  5169. ]
  5170. },
  5171. {
  5172. "name": "torch.index_select:",
  5173. "inputs": [
  5174. { "name": "self", "type": "Tensor" },
  5175. { "name": "dim", "type": "int64" },
  5176. { "name": "index", "type": "Tensor" }
  5177. ],
  5178. "outputs": [
  5179. { "name": "output", "type": "Tensor" }
  5180. ]
  5181. },
  5182. {
  5183. "name": "torch.index_put",
  5184. "inputs": [
  5185. { "name": "self", "type": "Tensor" },
  5186. { "name": "indices", "type": "Tensor[]", "optional": true },
  5187. { "name": "values", "type": "Tensor" },
  5188. { "name": "accumulate", "type": "boolean", "default": false }
  5189. ],
  5190. "outputs": [
  5191. { "name": "output", "type": "Tensor" }
  5192. ]
  5193. },
  5194. {
  5195. "name": "torch.index_put_",
  5196. "inputs": [
  5197. { "name": "self", "type": "Tensor" },
  5198. { "name": "indices", "type": "Tensor[]", "optional": true },
  5199. { "name": "values", "type": "Tensor" },
  5200. { "name": "accumulate", "type": "boolean", "default": false }
  5201. ],
  5202. "outputs": [
  5203. { "name": "output", "type": "Tensor" }
  5204. ]
  5205. },
  5206. {
  5207. "name": "torch.new_zeros",
  5208. "inputs": [
  5209. { "name": "self", "type": "Tensor" },
  5210. { "name": "size", "type": "int64[]" },
  5211. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5212. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5213. { "name": "device", "type": "Device", "optional": true, "default": null },
  5214. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5215. ],
  5216. "outputs": [
  5217. { "name": "output", "type": "Tensor" }
  5218. ]
  5219. },
  5220. {
  5221. "name": "torch.new_full",
  5222. "inputs": [
  5223. { "name": "self", "type": "Tensor" },
  5224. { "name": "size", "type": "int64[]" },
  5225. { "name": "fill_value", "type": "Scalar" },
  5226. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5227. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5228. { "name": "device", "type": "Device", "optional": true, "default": null },
  5229. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5230. ],
  5231. "outputs": [
  5232. { "name": "output", "type": "Tensor" }
  5233. ]
  5234. },
  5235. {
  5236. "name": "torch.softplus",
  5237. "category": "Activation",
  5238. "attributes": [
  5239. { "name": "beta", "type": "Scalar", "default": 1 },
  5240. { "name": "threshold", "type": "Scalar", "default": 20 }
  5241. ],
  5242. "inputs": [
  5243. { "name": "self", "type": "Tensor" }
  5244. ],
  5245. "outputs": [
  5246. { "name": "output", "type": "Tensor" }
  5247. ]
  5248. },
  5249. {
  5250. "name": "torch.zero_",
  5251. "inputs": [
  5252. { "name": "self", "type": "Tensor" }
  5253. ],
  5254. "outputs": [
  5255. { "name": "output", "type": "Tensor" }
  5256. ]
  5257. },
  5258. {
  5259. "name": "torch.cumsum:",
  5260. "inputs": [
  5261. { "name": "self", "type": "Tensor" },
  5262. { "name": "dim", "type": "int64" },
  5263. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  5264. ],
  5265. "outputs": [
  5266. { "name": "output", "type": "Tensor" }
  5267. ]
  5268. },
  5269. {
  5270. "name": "torch.cumsum:dimname",
  5271. "inputs": [
  5272. { "name": "self", "type": "Tensor" },
  5273. { "name": "dim", "type": "Dimname" },
  5274. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  5275. ],
  5276. "outputs": [
  5277. { "name": "output", "type": "Tensor" }
  5278. ]
  5279. },
  5280. {
  5281. "name": "torch.cumsum_",
  5282. "inputs": [
  5283. { "name": "self", "type": "Tensor" },
  5284. { "name": "dim", "type": "int64" },
  5285. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  5286. ],
  5287. "outputs": [
  5288. { "name": "output", "type": "Tensor" }
  5289. ]
  5290. },
  5291. {
  5292. "name": "torch.masked_fill:Scalar",
  5293. "inputs": [
  5294. { "name": "self", "type": "Tensor" },
  5295. { "name": "mask", "type": "Tensor" },
  5296. { "name": "value", "type": "Scalar" }
  5297. ],
  5298. "outputs": [
  5299. { "name": "output", "type": "Tensor" }
  5300. ]
  5301. },
  5302. {
  5303. "name": "torch.masked_fill_:Scalar",
  5304. "inputs": [
  5305. { "name": "self", "type": "Tensor" },
  5306. { "name": "mask", "type": "Tensor" },
  5307. { "name": "value", "type": "Scalar" }
  5308. ],
  5309. "outputs": [
  5310. { "name": "output", "type": "Tensor" }
  5311. ]
  5312. },
  5313. {
  5314. "name": "torch.masked_fill:Tensor",
  5315. "inputs": [
  5316. { "name": "self", "type": "Tensor" },
  5317. { "name": "mask", "type": "Tensor" },
  5318. { "name": "value", "type": "Tensor" }
  5319. ],
  5320. "outputs": [
  5321. { "name": "output", "type": "Tensor" }
  5322. ]
  5323. },
  5324. {
  5325. "name": "torch.masked_fill_:Tensor",
  5326. "inputs": [
  5327. { "name": "self", "type": "Tensor" },
  5328. { "name": "mask", "type": "Tensor" },
  5329. { "name": "value", "type": "Tensor" }
  5330. ],
  5331. "outputs": [
  5332. { "name": "output", "type": "Tensor" }
  5333. ]
  5334. },
  5335. {
  5336. "name": "torch.masked_select:",
  5337. "inputs": [
  5338. { "name": "self", "type": "Tensor" },
  5339. { "name": "mask", "type": "Tensor" }
  5340. ],
  5341. "outputs": [
  5342. { "name": "output", "type": "Tensor" }
  5343. ]
  5344. },
  5345. {
  5346. "name": "torch.argmax",
  5347. "attributes": [
  5348. { "name": "dim", "type": "int64", "optional": true, "default": null },
  5349. { "name": "keepdim", "type": "boolean", "default": false }
  5350. ],
  5351. "inputs": [
  5352. { "name": "self", "type": "Tensor" }
  5353. ],
  5354. "outputs": [
  5355. { "name": "output", "type": "Tensor" }
  5356. ]
  5357. },
  5358. {
  5359. "name": "torch.argmin",
  5360. "attributes": [
  5361. { "name": "dim", "type": "int64", "optional": true, "default": null },
  5362. { "name": "keepdim", "type": "boolean", "default": false }
  5363. ],
  5364. "inputs": [
  5365. { "name": "self", "type": "Tensor" }
  5366. ],
  5367. "outputs": [
  5368. { "name": "output", "type": "Tensor" }
  5369. ]
  5370. },
  5371. {
  5372. "name": "torch.fft",
  5373. "inputs": [
  5374. { "name": "self", "type": "Tensor" },
  5375. { "name": "signal_ndim", "type": "int64" },
  5376. { "name": "normalized", "type": "boolean", "default": false }
  5377. ],
  5378. "outputs": [
  5379. { "name": "output", "type": "Tensor" }
  5380. ]
  5381. },
  5382. {
  5383. "name": "torch.fft_fft2",
  5384. "inputs": [
  5385. { "name": "self", "type": "Tensor" },
  5386. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5387. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5388. { "name": "norm", "type": "string", "optional": true, "default": null }
  5389. ],
  5390. "outputs": [
  5391. { "name": "output", "type": "Tensor" }
  5392. ]
  5393. },
  5394. {
  5395. "name": "torch.fft_ifft2",
  5396. "inputs": [
  5397. { "name": "self", "type": "Tensor" },
  5398. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5399. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5400. { "name": "norm", "type": "string", "optional": true, "default": null }
  5401. ],
  5402. "outputs": [
  5403. { "name": "output", "type": "Tensor" }
  5404. ]
  5405. },
  5406. {
  5407. "name": "torch.fft_rfft2",
  5408. "inputs": [
  5409. { "name": "self", "type": "Tensor" },
  5410. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5411. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5412. { "name": "norm", "type": "string", "optional": true, "default": null }
  5413. ],
  5414. "outputs": [
  5415. { "name": "output", "type": "Tensor" }
  5416. ]
  5417. },
  5418. {
  5419. "name": "torch.fft_irfft2",
  5420. "inputs": [
  5421. { "name": "self", "type": "Tensor" },
  5422. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5423. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5424. { "name": "norm", "type": "string", "optional": true, "default": null }
  5425. ],
  5426. "outputs": [
  5427. { "name": "output", "type": "Tensor" }
  5428. ]
  5429. },
  5430. {
  5431. "name": "torch.fft_hfft2",
  5432. "inputs": [
  5433. { "name": "self", "type": "Tensor" },
  5434. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5435. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5436. { "name": "norm", "type": "string", "optional": true, "default": null }
  5437. ],
  5438. "outputs": [
  5439. { "name": "output", "type": "Tensor" }
  5440. ]
  5441. },
  5442. {
  5443. "name": "torch.fft_ihfft2",
  5444. "inputs": [
  5445. { "name": "self", "type": "Tensor" },
  5446. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5447. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5448. { "name": "norm", "type": "string", "optional": true, "default": null }
  5449. ],
  5450. "outputs": [
  5451. { "name": "output", "type": "Tensor" }
  5452. ]
  5453. },
  5454. {
  5455. "name": "torch.fft_fftn",
  5456. "inputs": [
  5457. { "name": "self", "type": "Tensor" },
  5458. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5459. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5460. { "name": "norm", "type": "string", "optional": true, "default": null }
  5461. ],
  5462. "outputs": [
  5463. { "name": "output", "type": "Tensor" }
  5464. ]
  5465. },
  5466. {
  5467. "name": "torch.fft_ifftn",
  5468. "inputs": [
  5469. { "name": "self", "type": "Tensor" },
  5470. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5471. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5472. { "name": "norm", "type": "string", "optional": true, "default": null }
  5473. ],
  5474. "outputs": [
  5475. { "name": "output", "type": "Tensor" }
  5476. ]
  5477. },
  5478. {
  5479. "name": "torch.fft_rfftn",
  5480. "inputs": [
  5481. { "name": "self", "type": "Tensor" },
  5482. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5483. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5484. { "name": "norm", "type": "string", "optional": true, "default": null }
  5485. ],
  5486. "outputs": [
  5487. { "name": "output", "type": "Tensor" }
  5488. ]
  5489. },
  5490. {
  5491. "name": "torch.fft_irfftn",
  5492. "inputs": [
  5493. { "name": "self", "type": "Tensor" },
  5494. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5495. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5496. { "name": "norm", "type": "string", "optional": true, "default": null }
  5497. ],
  5498. "outputs": [
  5499. { "name": "output", "type": "Tensor" }
  5500. ]
  5501. },
  5502. {
  5503. "name": "torch.fft_hfftn",
  5504. "inputs": [
  5505. { "name": "self", "type": "Tensor" },
  5506. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5507. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5508. { "name": "norm", "type": "string", "optional": true, "default": null }
  5509. ],
  5510. "outputs": [
  5511. { "name": "output", "type": "Tensor" }
  5512. ]
  5513. },
  5514. {
  5515. "name": "torch.fft_ihfftn",
  5516. "inputs": [
  5517. { "name": "self", "type": "Tensor" },
  5518. { "name": "s", "s": "int64[]", "optional": true, "default": null },
  5519. { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] },
  5520. { "name": "norm", "type": "string", "optional": true, "default": null }
  5521. ],
  5522. "outputs": [
  5523. { "name": "output", "type": "Tensor" }
  5524. ]
  5525. },
  5526. {
  5527. "name": "torch.to_mkldnn:",
  5528. "inputs": [
  5529. { "name": "self", "type": "Tensor" },
  5530. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  5531. ],
  5532. "outputs": [
  5533. { "name": "output", "type": "Tensor" }
  5534. ]
  5535. },
  5536. {
  5537. "name": "torch.topk:",
  5538. "inputs": [
  5539. { "name": "self", "type": "Tensor" },
  5540. { "name": "k", "type": "int64" },
  5541. { "name": "dim", "type": "int64", "default": -1 },
  5542. { "name": "largest", "type": "boolean", "default": true },
  5543. { "name": "sorted", "type": "boolean", "default": true }
  5544. ],
  5545. "outputs": [
  5546. { "name": "values", "type": "Tensor" },
  5547. { "name": "indices", "type": "Tensor" }
  5548. ]
  5549. },
  5550. {
  5551. "name": "torch.random_:",
  5552. "attributes": [
  5553. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5554. ],
  5555. "inputs": [
  5556. { "name": "self", "type": "Tensor" }
  5557. ],
  5558. "outputs": [
  5559. { "name": "output", "type": "Tensor" }
  5560. ]
  5561. },
  5562. {
  5563. "name": "torch.random_:to",
  5564. "attributes": [
  5565. { "name": "to", "type": "int64" },
  5566. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5567. ],
  5568. "inputs": [
  5569. { "name": "self", "type": "Tensor" }
  5570. ],
  5571. "outputs": [
  5572. { "name": "output", "type": "Tensor" }
  5573. ]
  5574. },
  5575. {
  5576. "name": "torch.random_:from",
  5577. "attributes": [
  5578. { "name": "from", "type": "int64" },
  5579. { "name": "to", "type": "int64", "optional": true },
  5580. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5581. ],
  5582. "inputs": [
  5583. { "name": "self", "type": "Tensor" }
  5584. ],
  5585. "outputs": [
  5586. { "name": "output", "type": "Tensor" }
  5587. ]
  5588. },
  5589. {
  5590. "name": "torch.uniform_",
  5591. "attributes": [
  5592. { "name": "from", "type": "float32", "default": 0 },
  5593. { "name": "to", "type": "float32", "default": 1 },
  5594. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5595. ],
  5596. "inputs": [
  5597. { "name": "self", "type": "Tensor" }
  5598. ],
  5599. "outputs": [
  5600. { "name": "output", "type": "Tensor" }
  5601. ]
  5602. },
  5603. {
  5604. "name": "torch.cauchy_",
  5605. "attributes": [
  5606. { "name": "median", "type": "float32", "default": 0 },
  5607. { "name": "sigma", "type": "float32", "default": 1 },
  5608. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5609. ],
  5610. "inputs": [
  5611. { "name": "self", "type": "Tensor" }
  5612. ],
  5613. "outputs": [
  5614. { "name": "output", "type": "Tensor" }
  5615. ]
  5616. },
  5617. {
  5618. "name": "torch.log_normal_",
  5619. "attributes": [
  5620. { "name": "mean", "type": "float32", "default": 1 },
  5621. { "name": "std", "type": "float32", "default": 2 },
  5622. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5623. ],
  5624. "inputs": [
  5625. { "name": "self", "type": "Tensor" }
  5626. ],
  5627. "outputs": [
  5628. { "name": "output", "type": "Tensor" }
  5629. ]
  5630. },
  5631. {
  5632. "name": "torch.exponential_",
  5633. "attributes": [
  5634. { "name": "lambd", "type": "float32", "default": 1 },
  5635. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5636. ],
  5637. "inputs": [
  5638. { "name": "self", "type": "Tensor" }
  5639. ],
  5640. "outputs": [
  5641. { "name": "output", "type": "Tensor" }
  5642. ]
  5643. },
  5644. {
  5645. "name": "torch.geometric_",
  5646. "attributes": [
  5647. { "name": "p", "type": "float32" },
  5648. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5649. ],
  5650. "inputs": [
  5651. { "name": "self", "type": "Tensor" }
  5652. ],
  5653. "outputs": [
  5654. { "name": "output", "type": "Tensor" }
  5655. ]
  5656. },
  5657. {
  5658. "name": "torch.mv",
  5659. "inputs": [
  5660. { "name": "self", "type": "Tensor" },
  5661. { "name": "vec", "type": "Tensor" }
  5662. ],
  5663. "outputs": [
  5664. { "name": "output", "type": "Tensor" }
  5665. ]
  5666. },
  5667. {
  5668. "name": "torch.mvlgamma",
  5669. "attributes": [
  5670. { "name": "p", "type": "int64" }
  5671. ],
  5672. "inputs": [
  5673. { "name": "self", "type": "Tensor" }
  5674. ],
  5675. "outputs": [
  5676. { "name": "output", "type": "Tensor" }
  5677. ]
  5678. },
  5679. {
  5680. "name": "torch.mvlgamma_",
  5681. "attributes": [
  5682. { "name": "p", "type": "int64" }
  5683. ],
  5684. "inputs": [
  5685. { "name": "self", "type": "Tensor" }
  5686. ],
  5687. "outputs": [
  5688. { "name": "output", "type": "Tensor" }
  5689. ]
  5690. },
  5691. {
  5692. "name": "torch.quantize_per_channel",
  5693. "attributes": [
  5694. { "name": "axis", "type": "int64" },
  5695. { "name": "dtype", "type": "ScalarType" }
  5696. ],
  5697. "inputs": [
  5698. { "name": "self", "type": "Tensor" },
  5699. { "name": "scales", "type": "Tensor" },
  5700. { "name": "zero_points", "type": "Tensor" }
  5701. ],
  5702. "outputs": [
  5703. { "name": "output", "type": "Tensor" }
  5704. ]
  5705. },
  5706. {
  5707. "name": "torch._shape_as_tensor",
  5708. "inputs": [
  5709. { "name": "self", "type": "Tensor" }
  5710. ],
  5711. "outputs": [
  5712. { "name": "output", "type": "Tensor" }
  5713. ]
  5714. },
  5715. {
  5716. "name": "torch.clone",
  5717. "attributes": [
  5718. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  5719. ],
  5720. "inputs": [
  5721. { "name": "self", "type": "Tensor" }
  5722. ],
  5723. "outputs": [
  5724. { "name": "output", "type": "Tensor" }
  5725. ]
  5726. },
  5727. {
  5728. "name": "torch.im2col",
  5729. "attributes": [
  5730. { "name": "kernel_size", "type": "int64[]" },
  5731. { "name": "dilation", "type": "int64[]" },
  5732. { "name": "padding", "type": "int64[]" },
  5733. { "name": "stride", "type": "int64[]" }
  5734. ],
  5735. "inputs": [
  5736. { "name": "self", "type": "Tensor" }
  5737. ],
  5738. "outputs": [
  5739. { "name": "output", "type": "Tensor" }
  5740. ]
  5741. },
  5742. {
  5743. "name": "torch.inverse:",
  5744. "inputs": [
  5745. { "name": "self", "type": "Tensor" }
  5746. ],
  5747. "outputs": [
  5748. { "name": "output", "type": "Tensor" }
  5749. ]
  5750. },
  5751. {
  5752. "name": "torch.col2im",
  5753. "attributes": [
  5754. { "name": "output_size", "type": "int64[]" },
  5755. { "name": "kernel_size", "type": "int64[]" },
  5756. { "name": "dilation", "type": "int64[]" },
  5757. { "name": "padding", "type": "int64[]" },
  5758. { "name": "stride", "type": "int64[]" }
  5759. ],
  5760. "inputs": [
  5761. { "name": "self", "type": "Tensor" }
  5762. ],
  5763. "outputs": [
  5764. { "name": "output", "type": "Tensor" }
  5765. ]
  5766. },
  5767. {
  5768. "name": "torch.eq:Tensor",
  5769. "inputs": [
  5770. { "name": "self", "type": "Tensor" },
  5771. { "name": "other", "type": "Tensor" }
  5772. ],
  5773. "outputs": [
  5774. { "name": "output", "type": "Tensor" }
  5775. ]
  5776. },
  5777. {
  5778. "name": "torch.eq:Scalar",
  5779. "inputs": [
  5780. { "name": "self", "type": "Tensor" },
  5781. { "name": "other", "type": "Scalar" }
  5782. ],
  5783. "outputs": [
  5784. { "name": "output", "type": "Tensor" }
  5785. ]
  5786. },
  5787. {
  5788. "name": "torch.where:self",
  5789. "inputs": [
  5790. { "name": "condition", "type": "Tensor" },
  5791. { "name": "self", "type": "Tensor" },
  5792. { "name": "other", "type": "Tensor" }
  5793. ],
  5794. "outputs": [
  5795. { "name": "output", "type": "Tensor" }
  5796. ]
  5797. },
  5798. {
  5799. "name": "torch.__and__:Tensor",
  5800. "inputs": [
  5801. { "name": "self", "type": "Tensor" },
  5802. { "name": "other", "type": "Tensor" }
  5803. ],
  5804. "outputs": [
  5805. { "name": "output", "type": "Tensor" }
  5806. ]
  5807. },
  5808. {
  5809. "name": "torch.__or__:Tensor",
  5810. "inputs": [
  5811. { "name": "self", "type": "Tensor" },
  5812. { "name": "other", "type": "Tensor" }
  5813. ],
  5814. "outputs": [
  5815. { "name": "output", "type": "Tensor" }
  5816. ]
  5817. },
  5818. {
  5819. "name": "torch.ge:Tensor",
  5820. "inputs": [
  5821. { "name": "self", "type": "Tensor" },
  5822. { "name": "other", "type": "Tensor" }
  5823. ],
  5824. "outputs": [
  5825. { "name": "output", "type": "Tensor" }
  5826. ]
  5827. },
  5828. {
  5829. "name": "torch.ge:Scalar",
  5830. "attributes": [
  5831. { "name": "other", "type": "Scalar" }
  5832. ],
  5833. "inputs": [
  5834. { "name": "self", "type": "Tensor" }
  5835. ],
  5836. "outputs": [
  5837. { "name": "output", "type": "Tensor" }
  5838. ]
  5839. },
  5840. {
  5841. "name": "torch.lt:Tensor",
  5842. "inputs": [
  5843. { "name": "self", "type": "Tensor" },
  5844. { "name": "other", "type": "Tensor" }
  5845. ],
  5846. "outputs": [
  5847. { "name": "output", "type": "Tensor" }
  5848. ]
  5849. },
  5850. {
  5851. "name": "torch.lt:Scalar",
  5852. "attributes": [
  5853. { "name": "other", "type": "Scalar" }
  5854. ],
  5855. "inputs": [
  5856. { "name": "self", "type": "Tensor" }
  5857. ],
  5858. "outputs": [
  5859. { "name": "output", "type": "Tensor" }
  5860. ]
  5861. },
  5862. {
  5863. "name": "torch.flip",
  5864. "attributes": [
  5865. { "name": "dims", "type": "int64[]" }
  5866. ],
  5867. "inputs": [
  5868. { "name": "self", "type": "Tensor" }
  5869. ],
  5870. "outputs": [
  5871. { "name": "output", "type": "Tensor" }
  5872. ]
  5873. },
  5874. {
  5875. "name": "torch.linalg_solve:",
  5876. "attributes": [
  5877. { "name": "input", "type": "Tensor" },
  5878. { "name": "other", "type": "Tensor" }
  5879. ],
  5880. "outputs": [
  5881. { "name": "output", "type": "Tensor" }
  5882. ]
  5883. },
  5884. {
  5885. "name": "torch.linalg_tensorinv:",
  5886. "attributes": [
  5887. { "name": "self", "type": "Tensor" },
  5888. { "name": "ind", "type": "int64", "default": 2 }
  5889. ],
  5890. "outputs": [
  5891. { "name": "output", "type": "Tensor" }
  5892. ]
  5893. },
  5894. {
  5895. "name": "torch.linalg_tensorsolve:",
  5896. "attributes": [
  5897. { "name": "self", "type": "Tensor" },
  5898. { "name": "other", "type": "Tensor" },
  5899. { "name": "dims", "type": "int64[]", "optional": true, "default": null }
  5900. ],
  5901. "outputs": [
  5902. { "name": "output", "type": "Tensor" }
  5903. ]
  5904. },
  5905. {
  5906. "name": "torch.linspace",
  5907. "attributes": [
  5908. { "name": "start", "type": "Scalar" },
  5909. { "name": "end", "type": "Scalar" },
  5910. { "name": "steps", "type": "int64", "optional": true, "default": null },
  5911. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5912. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5913. { "name": "device", "type": "Device", "optional": true, "default": null },
  5914. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5915. ],
  5916. "outputs": [
  5917. { "name": "output", "type": "Tensor" }
  5918. ]
  5919. },
  5920. {
  5921. "name": "torch.sign",
  5922. "inputs": [
  5923. { "name": "self", "type": "Tensor" }
  5924. ],
  5925. "outputs": [
  5926. { "name": "output", "type": "Tensor" }
  5927. ]
  5928. },
  5929. {
  5930. "name": "torch.sign_",
  5931. "inputs": [
  5932. { "name": "self", "type": "Tensor" }
  5933. ],
  5934. "outputs": [
  5935. { "name": "output", "type": "Tensor" }
  5936. ]
  5937. },
  5938. {
  5939. "name": "torch.signbit",
  5940. "inputs": [
  5941. { "name": "self", "type": "Tensor" }
  5942. ],
  5943. "outputs": [
  5944. { "name": "output", "type": "Tensor" }
  5945. ]
  5946. },
  5947. {
  5948. "name": "torch.dist",
  5949. "attributes": [
  5950. { "name": "start", "type": "Scalar", "default": 2 }
  5951. ],
  5952. "inputs": [
  5953. { "name": "self", "type": "Tensor" },
  5954. { "name": "other", "type": "Tensor" }
  5955. ],
  5956. "outputs": [
  5957. { "name": "output", "type": "Tensor" }
  5958. ]
  5959. },
  5960. {
  5961. "name": "torch.ceil:",
  5962. "inputs": [
  5963. { "name": "self", "type": "Tensor" }
  5964. ],
  5965. "outputs": [
  5966. { "name": "output", "type": "Tensor" }
  5967. ]
  5968. },
  5969. {
  5970. "name": "torch.ceil_:",
  5971. "inputs": [
  5972. { "name": "self", "type": "Tensor" }
  5973. ],
  5974. "outputs": [
  5975. { "name": "output", "type": "Tensor" }
  5976. ]
  5977. },
  5978. {
  5979. "name": "torch.einsum",
  5980. "inputs": [
  5981. { "name": "equation", "type": "string" },
  5982. { "name": "tensors", "type": "Tensor[]" }
  5983. ],
  5984. "outputs": [
  5985. { "name": "output", "type": "Tensor" }
  5986. ]
  5987. },
  5988. {
  5989. "name": "torch.dot",
  5990. "inputs": [
  5991. { "name": "self", "type": "Tensor" },
  5992. { "name": "tensor", "type": "Tensor" }
  5993. ],
  5994. "outputs": [
  5995. { "name": "output", "type": "Tensor" }
  5996. ]
  5997. },
  5998. {
  5999. "name": "torch.vdot",
  6000. "inputs": [
  6001. { "name": "self", "type": "Tensor" },
  6002. { "name": "other", "type": "Tensor" }
  6003. ],
  6004. "outputs": [
  6005. { "name": "output", "type": "Tensor" }
  6006. ]
  6007. },
  6008. {
  6009. "name": "torch.var:",
  6010. "inputs": [
  6011. { "name": "self", "type": "Tensor" },
  6012. { "name": "unbiased", "type": "boolean", "default": true }
  6013. ],
  6014. "outputs": [
  6015. { "name": "output", "type": "Tensor" }
  6016. ]
  6017. },
  6018. {
  6019. "name": "torch.var:dim",
  6020. "inputs": [
  6021. { "name": "self", "type": "Tensor" },
  6022. { "name": "dim", "type": "int64[]" },
  6023. { "name": "unbiased", "type": "boolean", "default": true },
  6024. { "name": "keepdim", "type": "boolean", "default": false }
  6025. ],
  6026. "outputs": [
  6027. { "name": "output", "type": "Tensor" }
  6028. ]
  6029. },
  6030. {
  6031. "name": "torch.var:correction",
  6032. "inputs": [
  6033. { "name": "self", "type": "Tensor" },
  6034. { "name": "dim", "type": "int64[]", "optional": true },
  6035. { "name": "correction", "type": "int64", "optional": true },
  6036. { "name": "keepdim", "type": "boolean", "default": false }
  6037. ],
  6038. "outputs": [
  6039. { "name": "output", "type": "Tensor" }
  6040. ]
  6041. },
  6042. {
  6043. "name": "torch.var:names_dim",
  6044. "inputs": [
  6045. { "name": "self", "type": "Tensor" },
  6046. { "name": "dim", "type": "Dimname[]" },
  6047. { "name": "unbiased", "type": "boolean", "default": true },
  6048. { "name": "keepdim", "type": "boolean", "default": false }
  6049. ],
  6050. "outputs": [
  6051. { "name": "output", "type": "Tensor" }
  6052. ]
  6053. },
  6054. {
  6055. "name": "torch.var:correction_names",
  6056. "inputs": [
  6057. { "name": "self", "type": "Tensor" },
  6058. { "name": "dim", "type": "Dimname[]" },
  6059. { "name": "correction", "type": "int64", "default": true },
  6060. { "name": "keepdim", "type": "boolean", "default": false }
  6061. ],
  6062. "outputs": [
  6063. { "name": "output", "type": "Tensor" }
  6064. ]
  6065. },
  6066. {
  6067. "name": "torch.var_mean:",
  6068. "inputs": [
  6069. { "name": "self", "type": "Tensor" },
  6070. { "name": "unbiased", "type": "boolean", "default": true }
  6071. ],
  6072. "outputs": [
  6073. { "name": "output", "type": "Tensor" },
  6074. { "name": "?", "type": "Tensor" }
  6075. ]
  6076. },
  6077. {
  6078. "name": "torch.var_mean:dim",
  6079. "inputs": [
  6080. { "name": "self", "type": "Tensor" },
  6081. { "name": "dim", "type": "int64[]" },
  6082. { "name": "unbiased", "type": "boolean", "default": true },
  6083. { "name": "keepdim", "type": "boolean", "default": false }
  6084. ],
  6085. "outputs": [
  6086. { "name": "output", "type": "Tensor" },
  6087. { "name": "?", "type": "Tensor" }
  6088. ]
  6089. },
  6090. {
  6091. "name": "torch.all:",
  6092. "inputs": [
  6093. { "name": "self", "type": "Tensor" }
  6094. ],
  6095. "outputs": [
  6096. { "name": "output", "type": "Tensor" }
  6097. ]
  6098. },
  6099. {
  6100. "name": "torch.all:dim",
  6101. "inputs": [
  6102. { "name": "self", "type": "Tensor" },
  6103. { "name": "dim", "type": "int64" },
  6104. { "name": "keepdim", "type": "boolean", "default": false }
  6105. ],
  6106. "outputs": [
  6107. { "name": "output", "type": "Tensor" }
  6108. ]
  6109. },
  6110. {
  6111. "name": "torch.any:",
  6112. "inputs": [
  6113. { "name": "self", "type": "Tensor" }
  6114. ],
  6115. "outputs": [
  6116. { "name": "output", "type": "Tensor" }
  6117. ]
  6118. },
  6119. {
  6120. "name": "torch.any:dim",
  6121. "inputs": [
  6122. { "name": "self", "type": "Tensor" },
  6123. { "name": "dim", "type": "int64" },
  6124. { "name": "keepdim", "type": "boolean", "default": false }
  6125. ],
  6126. "outputs": [
  6127. { "name": "output", "type": "Tensor" }
  6128. ]
  6129. },
  6130. {
  6131. "name": "torch.renorm",
  6132. "attributes": [
  6133. { "name": "p", "type": "Scalar" },
  6134. { "name": "dim", "type": "int64" },
  6135. { "name": "maxnorm", "type": "Scalar" }
  6136. ],
  6137. "inputs": [
  6138. { "name": "self", "type": "Tensor" }
  6139. ],
  6140. "outputs": [
  6141. { "name": "output", "type": "Tensor" }
  6142. ]
  6143. },
  6144. {
  6145. "name": "torch.unfold",
  6146. "attributes": [
  6147. { "name": "dimension", "type": "int64" },
  6148. { "name": "size", "type": "int64" },
  6149. { "name": "step", "type": "int64" }
  6150. ],
  6151. "inputs": [
  6152. { "name": "self", "type": "Tensor" }
  6153. ],
  6154. "outputs": [
  6155. { "name": "output", "type": "Tensor" }
  6156. ]
  6157. },
  6158. {
  6159. "name": "torch.fake_quantize_per_channel_affine",
  6160. "inputs": [
  6161. { "name": "self", "type": "Tensor" },
  6162. { "name": "scale", "type": "Tensor" },
  6163. { "name": "zero_point", "type": "Tensor" },
  6164. { "name": "axis", "type": "int64" },
  6165. { "name": "quant_min", "type": "int64" },
  6166. { "name": "quant_max", "type": "int64" }
  6167. ],
  6168. "outputs": [
  6169. { "name": "output", "type": "Tensor" }
  6170. ]
  6171. },
  6172. {
  6173. "name": "torch.fake_quantize_per_tensor_affine",
  6174. "inputs": [
  6175. { "name": "self", "type": "Tensor" },
  6176. { "name": "scale", "type": "float32" },
  6177. { "name": "zero_point", "type": "int64" },
  6178. { "name": "quant_min", "type": "int64" },
  6179. { "name": "quant_max", "type": "int64" }
  6180. ],
  6181. "outputs": [
  6182. { "name": "output", "type": "Tensor" }
  6183. ]
  6184. },
  6185. {
  6186. "name": "torch.addmv",
  6187. "inputs": [
  6188. { "name": "self", "type": "Tensor" },
  6189. { "name": "mat", "type": "Tensor" },
  6190. { "name": "vec", "type": "Tensor" },
  6191. { "name": "beta", "type": "Scalar", "default": 1 },
  6192. { "name": "alpha", "type": "Scalar", "default": 1 }
  6193. ],
  6194. "outputs": [
  6195. { "name": "output", "type": "Tensor" }
  6196. ]
  6197. },
  6198. {
  6199. "name": "torch.addmv_",
  6200. "inputs": [
  6201. { "name": "self", "type": "Tensor" },
  6202. { "name": "mat", "type": "Tensor" },
  6203. { "name": "vec", "type": "Tensor" },
  6204. { "name": "beta", "type": "Scalar", "default": 1 },
  6205. { "name": "alpha", "type": "Scalar", "default": 1 }
  6206. ],
  6207. "outputs": [
  6208. { "name": "output", "type": "Tensor" }
  6209. ]
  6210. },
  6211. {
  6212. "name": "torch.type_as",
  6213. "inputs": [
  6214. { "name": "self", "type": "Tensor" },
  6215. { "name": "other", "type": "Tensor" }
  6216. ],
  6217. "outputs": [
  6218. { "name": "output", "type": "Tensor" }
  6219. ]
  6220. },
  6221. {
  6222. "name": "torch.normal_",
  6223. "inputs": [
  6224. { "name": "self", "type": "Tensor" },
  6225. { "name": "mean", "type": "float32", "default": 0 },
  6226. { "name": "std", "type": "float32", "default": 1 },
  6227. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  6228. ],
  6229. "outputs": [
  6230. { "name": "output", "type": "Tensor" }
  6231. ]
  6232. },
  6233. {
  6234. "name": "torch.normal:Tensor_float",
  6235. "inputs": [
  6236. { "name": "self", "type": "Tensor" },
  6237. { "name": "mean", "type": "float32", "default": 0 },
  6238. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  6239. ],
  6240. "outputs": [
  6241. { "name": "output", "type": "Tensor" }
  6242. ]
  6243. },
  6244. {
  6245. "name": "torch.normal:float_Tensor",
  6246. "inputs": [
  6247. { "name": "mean", "type": "float32" },
  6248. { "name": "std", "type": "Tensor" },
  6249. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  6250. ],
  6251. "outputs": [
  6252. { "name": "output", "type": "Tensor" }
  6253. ]
  6254. },
  6255. {
  6256. "name": "torch.normal:Tensor_Tensor",
  6257. "inputs": [
  6258. { "name": "mean", "type": "Tensor" },
  6259. { "name": "std", "type": "Tensor" },
  6260. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  6261. ],
  6262. "outputs": [
  6263. { "name": "output", "type": "Tensor" }
  6264. ]
  6265. },
  6266. {
  6267. "name": "torch.normal:float_float",
  6268. "inputs": [
  6269. { "name": "mean", "type": "float32", "default": 0 },
  6270. { "name": "std", "type": "float32" },
  6271. { "name": "size", "type": "int64[]" },
  6272. { "name": "generator", "type": "Generator", "optional": true, "default": null },
  6273. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6274. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6275. { "name": "device", "type": "Device", "optional": true, "default": null },
  6276. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  6277. ],
  6278. "outputs": [
  6279. { "name": "output", "type": "Tensor" }
  6280. ]
  6281. },
  6282. {
  6283. "name": "torch.empty_like",
  6284. "inputs": [
  6285. { "name": "self", "type": "Tensor" },
  6286. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6287. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6288. { "name": "device", "type": "Device", "optional": true, "default": null },
  6289. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  6290. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  6291. ],
  6292. "outputs": [
  6293. { "name": "output", "type": "Tensor" }
  6294. ]
  6295. },
  6296. {
  6297. "name": "torch.classes._nnapi.Compilation",
  6298. "inputs": [
  6299. { "name": "serialized_model", "type": "Tensor" },
  6300. { "name": "inputs", "type": "Tensor[]" },
  6301. { "name": "parameter_buffers", "type": "Tensor[]" }
  6302. ],
  6303. "outputs": [
  6304. { "name": "output", "type": "Tensor[]" }
  6305. ]
  6306. },
  6307. {
  6308. "name": "torch.argsort:",
  6309. "inputs": [
  6310. { "name": "self", "type": "Tensor" },
  6311. { "name": "dim", "type": "int64", "default": -1 },
  6312. { "name": "descending", "type": "boolean", "default": false }
  6313. ],
  6314. "outputs": [
  6315. { "name": "output", "type": "Tensor" }
  6316. ]
  6317. },
  6318. {
  6319. "name": "torch.argsort:dimname",
  6320. "inputs": [
  6321. { "name": "self", "type": "Tensor" },
  6322. { "name": "dim", "type": "Dimname" },
  6323. { "name": "descending", "type": "boolean", "default": false }
  6324. ],
  6325. "outputs": [
  6326. { "name": "output", "type": "Tensor" }
  6327. ]
  6328. },
  6329. {
  6330. "name": "ops._caffe2.CopyGPUToCPU",
  6331. "inputs": [
  6332. { "name": "input", "type": "Tensor" },
  6333. { "name": "?" }
  6334. ],
  6335. "outputs": [
  6336. { "name": "output", "type": "Tensor" }
  6337. ]
  6338. },
  6339. {
  6340. "name": "ops._caffe2.CopyCPUToGPU",
  6341. "inputs": [
  6342. { "name": "input", "type": "Tensor" },
  6343. { "name": "?" }
  6344. ],
  6345. "outputs": [
  6346. { "name": "output", "type": "Tensor" }
  6347. ]
  6348. },
  6349. {
  6350. "name": "ops._caffe2.GenerateProposals",
  6351. "inputs": [
  6352. { "name": "scores", "type": "Tensor" },
  6353. { "name": "bbox_deltas", "type": "Tensor" },
  6354. { "name": "im_info", "type": "Tensor" },
  6355. { "name": "anchors", "type": "Tensor" },
  6356. { "name": "spatial_scale", "type": "float32" },
  6357. { "name": "pre_nms_topN", "type": "int64" },
  6358. { "name": "post_nms_topN", "type": "int64" },
  6359. { "name": "nms_thresh", "type": "float32" },
  6360. { "name": "min_size", "type": "float32" },
  6361. { "name": "angle_bound_on", "type": "boolean" },
  6362. { "name": "angle_bound_lo", "type": "int64" },
  6363. { "name": "angle_bound_hi", "type": "int64" },
  6364. { "name": "clip_angle_thresh", "type": "float32" },
  6365. { "name": "legacy_plus_one", "type": "boolean" },
  6366. { "name": "?" }
  6367. ],
  6368. "outputs": [
  6369. { "name": "output_0", "type": "Tensor" },
  6370. { "name": "output_1", "type": "Tensor" }
  6371. ]
  6372. },
  6373. {
  6374. "name": "ops._caffe2.CollectRpnProposals",
  6375. "inputs": [
  6376. { "name": "input_list", "type": "Tensor[]" },
  6377. { "name": "rpn_max_level", "type": "int64" },
  6378. { "name": "rpn_min_level", "type": "int64" },
  6379. { "name": "rpn_post_nms_topN", "type": "int64" },
  6380. { "name": "?" }
  6381. ],
  6382. "outputs": [
  6383. { "name": "rois", "type": "Tensor" }
  6384. ]
  6385. },
  6386. {
  6387. "name": "ops._caffe2.DistributeFpnProposals",
  6388. "inputs": [
  6389. { "name": "rois", "type": "Tensor" },
  6390. { "name": "roi_canonical_scale", "type": "int64" },
  6391. { "name": "roi_canonical_level", "type": "int64" },
  6392. { "name": "roi_max_level", "type": "int64" },
  6393. { "name": "roi_min_level", "type": "int64" },
  6394. { "name": "legacy_plus_one", "type": "boolean" },
  6395. { "name": "?" }
  6396. ],
  6397. "outputs": [
  6398. { "name": "rois_fpn2", "type": "Tensor" },
  6399. { "name": "rois_fpn3", "type": "Tensor" },
  6400. { "name": "rois_fpn4", "type": "Tensor" },
  6401. { "name": "rois_fpn5", "type": "Tensor" },
  6402. { "name": "rois_idx_restore_int32", "type": "Tensor" }
  6403. ]
  6404. },
  6405. {
  6406. "name": "ops._caffe2.RoIAlign",
  6407. "inputs": [
  6408. { "name": "features", "type": "Tensor" },
  6409. { "name": "rois", "type": "Tensor" },
  6410. { "name": "order", "type": "string" },
  6411. { "name": "spatial_scale", "type": "float32" },
  6412. { "name": "pooled_h", "type": "int64" },
  6413. { "name": "pooled_w", "type": "int64" },
  6414. { "name": "sampling_ratio", "type": "int64" },
  6415. { "name": "aligned", "type": "boolean" },
  6416. { "name": "?" }
  6417. ],
  6418. "outputs": [
  6419. { "name": "output", "type": "Tensor" }
  6420. ]
  6421. },
  6422. {
  6423. "name": "ops._caffe2.BatchPermutation",
  6424. "inputs": [
  6425. { "name": "X", "type": "Tensor" },
  6426. { "name": "indices", "type": "Tensor" },
  6427. { "name": "?" }
  6428. ],
  6429. "outputs": [
  6430. { "name": "output", "type": "Tensor" }
  6431. ]
  6432. },
  6433. {
  6434. "name": "ops._caffe2.BBoxTransform",
  6435. "inputs": [
  6436. { "name": "rois", "type": "Tensor" },
  6437. { "name": "deltas", "type": "Tensor" },
  6438. { "name": "im_info", "type": "Tensor" },
  6439. { "name": "weights", "type": "float32[]" },
  6440. { "name": "apply_scale", "type": "boolean" },
  6441. { "name": "rotated", "type": "boolean" },
  6442. { "name": "angle_bound_on", "type": "boolean" },
  6443. { "name": "angle_bound_lo", "type": "int64" },
  6444. { "name": "angle_bound_hi", "type": "int64" },
  6445. { "name": "clip_angle_thresh", "type": "float32" },
  6446. { "name": "legacy_plus_one", "type": "boolean" },
  6447. { "name": "?" }
  6448. ],
  6449. "outputs": [
  6450. { "name": "output_0", "type": "Tensor" },
  6451. { "name": "output_1", "type": "Tensor" }
  6452. ]
  6453. },
  6454. {
  6455. "name": "ops._caffe2.BoxWithNMSLimit",
  6456. "inputs": [
  6457. { "name": "scores", "type": "Tensor" },
  6458. { "name": "boxes", "type": "Tensor" },
  6459. { "name": "batch_splits", "type": "Tensor" },
  6460. { "name": "score_thresh", "type": "float32" },
  6461. { "name": "nms", "type": "float32" },
  6462. { "name": "detections_per_im", "type": "int64" },
  6463. { "name": "soft_nms_enabled", "type": "boolean" },
  6464. { "name": "soft_nms_method", "type": "string" },
  6465. { "name": "soft_nms_sigma", "type": "float32" },
  6466. { "name": "soft_nms_min_score_thres", "type": "float32" },
  6467. { "name": "rotated", "type": "boolean" },
  6468. { "name": "cls_agnostic_bbox_reg", "type": "boolean" },
  6469. { "name": "input_boxes_include_bg_cls", "type": "boolean" },
  6470. { "name": "output_classes_include_bg_cls", "type": "boolean" },
  6471. { "name": "legacy_plus_one", "type": "boolean" },
  6472. { "name": "?" }
  6473. ],
  6474. "outputs": [
  6475. { "name": "scores", "type": "Tensor" },
  6476. { "name": "boxes", "type": "Tensor" },
  6477. { "name": "classes", "type": "Tensor" },
  6478. { "name": "batch_splits", "type": "Tensor" },
  6479. { "name": "keeps", "type": "Tensor" },
  6480. { "name": "keeps_size", "type": "Tensor" }
  6481. ]
  6482. },
  6483. {
  6484. "name": "ops.neuron.forward_v2_1",
  6485. "inputs": [
  6486. { "name": "input", "type": "Tensor[]" },
  6487. { "name": "model" }
  6488. ],
  6489. "outputs": [
  6490. { "name": "output", "type": "Tensor" }
  6491. ]
  6492. },
  6493. {
  6494. "name": "torch.repeat_interleave:Tensor",
  6495. "inputs": [
  6496. { "name": "repeats", "type": "Tensor" }
  6497. ],
  6498. "outputs": [
  6499. { "name": "output", "type": "Tensor" }
  6500. ]
  6501. },
  6502. {
  6503. "name": "torch.repeat_interleave:self_Tensor",
  6504. "inputs": [
  6505. { "name": "self", "type": "Tensor" },
  6506. { "name": "repeats", "type": "Tensor" },
  6507. { "name": "dim", "type": "int64", "optional": true, "default": null }
  6508. ],
  6509. "outputs": [
  6510. { "name": "output", "type": "Tensor" }
  6511. ]
  6512. },
  6513. {
  6514. "name": "torch.repeat_interleave:self_int",
  6515. "inputs": [
  6516. { "name": "self", "type": "Tensor" },
  6517. { "name": "repeats", "type": "int64" },
  6518. { "name": "dim", "type": "int64", "optional": true, "default": null }
  6519. ],
  6520. "outputs": [
  6521. { "name": "output", "type": "Tensor" }
  6522. ]
  6523. },
  6524. {
  6525. "name": "torch.cpu",
  6526. "inputs": [
  6527. { "name": "self", "type": "Tensor" }
  6528. ],
  6529. "outputs": [
  6530. { "name": "output", "type": "Tensor" }
  6531. ]
  6532. },
  6533. {
  6534. "name": "ops.quantized.hardswish",
  6535. "category": "Activation",
  6536. "inputs": [
  6537. { "name": "input", "type": "Tensor" },
  6538. { "name": "output_scale", "type": "float32" },
  6539. { "name": "output_zero_point", "type": "int64" }
  6540. ],
  6541. "outputs": [
  6542. { "name": "output", "type": "Tensor" }
  6543. ]
  6544. },
  6545. {
  6546. "name": "ops.quantized.celu",
  6547. "category": "Activation",
  6548. "inputs": [
  6549. { "name": "self", "type": "Tensor" },
  6550. { "name": "output_scale", "type": "float32" },
  6551. { "name": "output_zero_point", "type": "int64" },
  6552. { "name": "alpha", "type": "Scalar", "default": 1 }
  6553. ],
  6554. "outputs": [
  6555. { "name": "output", "type": "Tensor" }
  6556. ]
  6557. },
  6558. {
  6559. "name": "ops.quantized.mul:",
  6560. "inputs": [
  6561. { "name": "qa", "type": "Tensor" },
  6562. { "name": "qb", "type": "Tensor" },
  6563. { "name": "scale", "type": "float32" },
  6564. { "name": "zero_point", "type": "int64" }
  6565. ],
  6566. "outputs": [
  6567. { "name": "qc", "type": "Tensor" }
  6568. ]
  6569. },
  6570. {
  6571. "name": "ops.quantized.mul:Scalar",
  6572. "inputs": [
  6573. { "name": "qa", "type": "Tensor" },
  6574. { "name": "b", "type": "Scalar" }
  6575. ],
  6576. "outputs": [
  6577. { "name": "qc", "type": "Tensor" }
  6578. ]
  6579. },
  6580. {
  6581. "name": "ops.quantized.mul:Scalar2",
  6582. "inputs": [
  6583. { "name": "b", "type": "Scalar" },
  6584. { "name": "qa", "type": "Tensor" }
  6585. ],
  6586. "outputs": [
  6587. { "name": "qc", "type": "Tensor" }
  6588. ]
  6589. },
  6590. {
  6591. "name": "ops.quantized.mul_relu:Scalar",
  6592. "inputs": [
  6593. { "name": "qa", "type": "Tensor" },
  6594. { "name": "b", "type": "Scalar" }
  6595. ],
  6596. "outputs": [
  6597. { "name": "qc", "type": "Tensor" }
  6598. ]
  6599. },
  6600. {
  6601. "name": "ops.quantized.mul_relu:Scalar2",
  6602. "inputs": [
  6603. { "name": "b", "type": "Scalar" },
  6604. { "name": "qa", "type": "Tensor" }
  6605. ],
  6606. "outputs": [
  6607. { "name": "qc", "type": "Tensor" }
  6608. ]
  6609. },
  6610. {
  6611. "name": "ops.quantized.mul_scalar",
  6612. "inputs": [
  6613. { "name": "qa", "type": "Tensor" },
  6614. { "name": "qb", "type": "Scalar" }
  6615. ],
  6616. "outputs": [
  6617. { "name": "qc", "type": "Tensor" }
  6618. ]
  6619. },
  6620. {
  6621. "name": "ops.quantized.mul_scalar_relu",
  6622. "inputs": [
  6623. { "name": "qa", "type": "Tensor" },
  6624. { "name": "qb", "type": "Scalar" }
  6625. ],
  6626. "outputs": [
  6627. { "name": "qc", "type": "Tensor" }
  6628. ]
  6629. },
  6630. {
  6631. "name": "ops.quantized.relu6",
  6632. "category": "Activation",
  6633. "inputs": [
  6634. { "name": "qx", "type": "Tensor" },
  6635. { "name": "inplace", "type": "boolean", "default": false }
  6636. ],
  6637. "outputs": [
  6638. { "name": "output", "type": "Tensor" }
  6639. ]
  6640. },
  6641. {
  6642. "name": "ops.quantized.leaky_relu",
  6643. "category": "Activation",
  6644. "inputs": [
  6645. { "name": "qx", "type": "Tensor" },
  6646. { "name": "negative_slope", "type": "Scalar", "default": false },
  6647. { "name": "inplace", "type": "boolean" },
  6648. { "name": "output_scale", "type": "float32" },
  6649. { "name": "output_zero_point", "type": "int64" }
  6650. ],
  6651. "outputs": [
  6652. { "name": "output", "type": "Tensor" }
  6653. ]
  6654. },
  6655. {
  6656. "name": "ops.quantized.sigmoid",
  6657. "category": "Activation",
  6658. "inputs": [
  6659. { "name": "qx", "type": "Tensor" },
  6660. { "name": "output_scale", "type": "float32" },
  6661. { "name": "output_zero_point", "type": "int64" }
  6662. ],
  6663. "outputs": [
  6664. { "name": "output", "type": "Tensor" }
  6665. ]
  6666. },
  6667. {
  6668. "name": "torch.full:names",
  6669. "inputs": [
  6670. { "name": "size", "type": "int64[]" },
  6671. { "name": "fill_value", "type": "Scalar" },
  6672. { "name": "names", "type": "Dimname[]", "optional": true },
  6673. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6674. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6675. { "name": "device", "type": "Device", "optional": true, "default": null },
  6676. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  6677. ],
  6678. "outputs": [
  6679. { "name": "output", "type": "Tensor" }
  6680. ]
  6681. },
  6682. {
  6683. "name": "torch.full:",
  6684. "inputs": [
  6685. { "name": "size", "type": "int64[]" },
  6686. { "name": "fill_value", "type": "Scalar" },
  6687. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6688. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6689. { "name": "device", "type": "Device", "optional": true, "default": null },
  6690. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  6691. ],
  6692. "outputs": [
  6693. { "name": "output", "type": "Tensor" }
  6694. ]
  6695. },
  6696. {
  6697. "name": "torch.full:out",
  6698. "inputs": [
  6699. { "name": "size", "type": "int64[]" },
  6700. { "name": "fill_value", "type": "Scalar" }
  6701. ],
  6702. "outputs": [
  6703. { "name": "output", "type": "Tensor" }
  6704. ]
  6705. },
  6706. {
  6707. "name": "torch.full_like:",
  6708. "inputs": [
  6709. { "name": "self", "type": "Tensor" },
  6710. { "name": "fill_value", "type": "Scalar" },
  6711. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6712. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6713. { "name": "device", "type": "Device", "optional": true, "default": null },
  6714. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  6715. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  6716. ],
  6717. "outputs": [
  6718. { "name": "output", "type": "Tensor" }
  6719. ]
  6720. },
  6721. {
  6722. "name": "torch.new_empty:",
  6723. "inputs": [
  6724. { "name": "self", "type": "Tensor" },
  6725. { "name": "size", "type": "int64[]" },
  6726. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6727. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6728. { "name": "device", "type": "Device", "optional": true, "default": null },
  6729. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  6730. ],
  6731. "outputs": [
  6732. { "name": "output", "type": "Tensor" }
  6733. ]
  6734. },
  6735. {
  6736. "name": "torch.new_empty_strided:",
  6737. "inputs": [
  6738. { "name": "self", "type": "Tensor" },
  6739. { "name": "size", "type": "int64[]" },
  6740. { "name": "stride", "type": "int64[]" },
  6741. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  6742. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  6743. { "name": "device", "type": "Device", "optional": true, "default": null },
  6744. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  6745. ],
  6746. "outputs": [
  6747. { "name": "output", "type": "Tensor" }
  6748. ]
  6749. },
  6750. {
  6751. "name": "torch.linear:",
  6752. "category": "Layer",
  6753. "inputs": [
  6754. { "name": "input", "type": "Tensor" },
  6755. { "name": "weight", "type": "Tensor" },
  6756. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  6757. ],
  6758. "outputs": [
  6759. { "name": "output", "type": "Tensor" }
  6760. ]
  6761. },
  6762. {
  6763. "name": "torch._thnn_fused_gru_cell",
  6764. "inputs": [
  6765. { "name": "input_gates", "type": "Tensor" },
  6766. { "name": "hidden_gates", "type": "Tensor" },
  6767. { "name": "hx", "type": "Tensor" },
  6768. { "name": "input_bias", "type": "Tensor", "optional": true, "default": null },
  6769. { "name": "hidden_bias", "type": "Tensor", "optional": true, "default": null }
  6770. ],
  6771. "outputs": [
  6772. { "name": "output", "type": "Tensor" },
  6773. { "name": "output2", "type": "Tensor" }
  6774. ]
  6775. },
  6776. {
  6777. "name": "torch._thnn_fused_lstm_cell",
  6778. "inputs": [
  6779. { "name": "input_gates", "type": "Tensor" },
  6780. { "name": "hidden_gates", "type": "Tensor" },
  6781. { "name": "cx", "type": "Tensor" },
  6782. { "name": "input_bias", "type": "Tensor", "optional": true, "default": null },
  6783. { "name": "hidden_bias", "type": "Tensor", "optional": true, "default": null }
  6784. ],
  6785. "outputs": [
  6786. { "name": "output", "type": "Tensor" },
  6787. { "name": "output2", "type": "Tensor" },
  6788. { "name": "output3", "type": "Tensor" }
  6789. ]
  6790. },
  6791. {
  6792. "name": "torch._unique2:",
  6793. "inputs": [
  6794. { "name": "self", "type": "Tensor" },
  6795. { "name": "sorted", "type": "boolean", "default": false },
  6796. { "name": "return_inverse", "type": "boolean", "default": false },
  6797. { "name": "return_counts", "type": "boolean", "default": false }
  6798. ],
  6799. "outputs": [
  6800. { "name": "output", "type": "Tensor" },
  6801. { "name": "output", "type": "Tensor" },
  6802. { "name": "output", "type": "Tensor" }
  6803. ]
  6804. },
  6805. {
  6806. "name": "torch._weight_norm",
  6807. "inputs": [
  6808. { "name": "v", "type": "Tensor" },
  6809. { "name": "g", "type": "Tensor" },
  6810. { "name": "dim", "type": "int64", "default": 0 }
  6811. ],
  6812. "outputs": [
  6813. { "name": "output", "type": "Tensor" }
  6814. ]
  6815. },
  6816. {
  6817. "name": "torch.requires_grad_",
  6818. "inputs": [
  6819. { "name": "self", "type": "Tensor" },
  6820. { "name": "requires_grad", "type": "boolean" }
  6821. ],
  6822. "outputs": [
  6823. { "name": "output", "type": "Tensor" }
  6824. ]
  6825. },
  6826. {
  6827. "name": "torch.stft",
  6828. "inputs": [
  6829. { "name": "self", "type": "Tensor" },
  6830. { "name": "n_fft", "type": "int64" },
  6831. { "name": "hop_length", "type": "int64", "optional": true, "default": null },
  6832. { "name": "win_length", "type": "int64", "optional": true, "default": null },
  6833. { "name": "window", "type": "Tensor", "optional": true, "default": null },
  6834. { "name": "normalized", "type": "boolean", "default": false },
  6835. { "name": "onesided", "type": "boolean", "optional": true, "default": null },
  6836. { "name": "return_complex", "type": "boolean", "optional": true, "default": null }
  6837. ],
  6838. "outputs": [
  6839. { "name": "output", "type": "Tensor" }
  6840. ]
  6841. },
  6842. {
  6843. "name": "torch.istft",
  6844. "inputs": [
  6845. { "name": "self", "type": "Tensor" },
  6846. { "name": "n_fft", "type": "int64" },
  6847. { "name": "hop_length", "type": "int64", "optional": true, "default": null },
  6848. { "name": "win_length", "type": "int64", "optional": true, "default": null },
  6849. { "name": "window", "type": "Tensor", "optional": true, "default": null },
  6850. { "name": "center", "type": "boolean", "default": true },
  6851. { "name": "normalized", "type": "boolean", "default": false },
  6852. { "name": "onesided", "type": "boolean", "optional": true, "default": null },
  6853. { "name": "length", "type": "boolean", "optional": true, "default": null },
  6854. { "name": "return_complex", "type": "boolean", "default": false }
  6855. ],
  6856. "outputs": [
  6857. { "name": "output", "type": "Tensor" }
  6858. ]
  6859. },
  6860. {
  6861. "name": "torch.unbind:int",
  6862. "inputs": [
  6863. { "name": "self", "type": "Tensor" },
  6864. { "name": "dim", "type": "int64", "default": 0 }
  6865. ],
  6866. "outputs": [
  6867. { "name": "output", "type": "Tensor[]" }
  6868. ]
  6869. },
  6870. {
  6871. "name": "torch.unbind:Dimname",
  6872. "inputs": [
  6873. { "name": "self", "type": "Tensor" },
  6874. { "name": "dim", "type": "Dimname" }
  6875. ],
  6876. "outputs": [
  6877. { "name": "output", "type": "Tensor[]" }
  6878. ]
  6879. },
  6880. {
  6881. "name": "torch.as_tensor",
  6882. "inputs": [
  6883. { "name": "data" },
  6884. { "name": "dtype" },
  6885. { "name": "device" }
  6886. ],
  6887. "outputs": [
  6888. { "name": "output", "type": "Tensor" }
  6889. ]
  6890. },
  6891. {
  6892. "name": "torch.as_strided",
  6893. "inputs": [
  6894. { "name": "self", "type": "Tensor" },
  6895. { "name": "size", "type": "int64[]" },
  6896. { "name": "stride", "type": "int64[]" },
  6897. { "name": "storage_offset", "type": "int64", "optional": true, "default": null }
  6898. ],
  6899. "outputs": [
  6900. { "name": "output", "type": "Tensor" }
  6901. ]
  6902. },
  6903. {
  6904. "name": "torch.tensor",
  6905. "inputs": [
  6906. { "name": "data" },
  6907. { "name": "dtype" },
  6908. { "name": "device" },
  6909. { "name": "requires_grad", "type": "boolean" }
  6910. ],
  6911. "outputs": [
  6912. { "name": "output", "type": "Tensor" }
  6913. ]
  6914. },
  6915. {
  6916. "name": "torch.tensor_split:sections",
  6917. "inputs": [
  6918. { "name": "self", "type": "Tensor" },
  6919. { "name": "sections", "type": "int64" },
  6920. { "name": "dim", "type": "int64", "default": 0 }
  6921. ],
  6922. "outputs": [
  6923. { "name": "output", "type": "Tensor[]" }
  6924. ]
  6925. },
  6926. {
  6927. "name": "torch.tensor_split:indices",
  6928. "inputs": [
  6929. { "name": "self", "type": "Tensor" },
  6930. { "name": "indices", "type": "int64[]" },
  6931. { "name": "dim", "type": "int64", "default": 0 }
  6932. ],
  6933. "outputs": [
  6934. { "name": "output", "type": "Tensor[]" }
  6935. ]
  6936. },
  6937. {
  6938. "name": "torch.tensor_split:tensor_indices_or_sections",
  6939. "inputs": [
  6940. { "name": "self", "type": "Tensor" },
  6941. { "name": "tensor_indices_or_sections", "type": "Tensor" },
  6942. { "name": "dim", "type": "int64", "default": 0 }
  6943. ],
  6944. "outputs": [
  6945. { "name": "output", "type": "Tensor[]" }
  6946. ]
  6947. },
  6948. {
  6949. "name": "torch.__upsample:",
  6950. "category": "Layer",
  6951. "inputs": [
  6952. { "name": "input", "type": "Tensor" },
  6953. { "name": "size", "type": "int64", "optional": true, "default": null },
  6954. { "name": "scale_factor", "type": "int64", "optional": true, "default": null },
  6955. { "name": "mode", "type": "string", "default": "nearest" },
  6956. { "name": "align_corners", "type": "boolean", "optional": true, "default": null }
  6957. ],
  6958. "outputs": [
  6959. { "name": "output", "type": "Tensor" }
  6960. ]
  6961. },
  6962. {
  6963. "name": "torch.__upsample:size_list",
  6964. "category": "Layer",
  6965. "inputs": [
  6966. { "name": "input", "type": "Tensor" },
  6967. { "name": "size", "type": "int64[]", "optional": true, "default": null },
  6968. { "name": "scale_factor", "type": "int64", "optional": true, "default": null },
  6969. { "name": "mode", "type": "string", "default": "nearest" },
  6970. { "name": "align_corners", "type": "boolean", "optional": true, "default": null }
  6971. ],
  6972. "outputs": [
  6973. { "name": "output", "type": "Tensor" }
  6974. ]
  6975. },
  6976. {
  6977. "name": "torch.triu_",
  6978. "category": "Layer",
  6979. "inputs": [
  6980. { "name": "self", "type": "Tensor" },
  6981. { "name": "diagonal", "type": "int64", "default": 0 }
  6982. ],
  6983. "outputs": [
  6984. { "name": "output", "type": "Tensor" }
  6985. ]
  6986. },
  6987. {
  6988. "name": "torch.triu:out",
  6989. "category": "Layer",
  6990. "inputs": [
  6991. { "name": "self", "type": "Tensor" },
  6992. { "name": "diagonal", "type": "int64", "default": 0 }
  6993. ],
  6994. "outputs": [
  6995. { "name": "output", "type": "Tensor" }
  6996. ]
  6997. },
  6998. {
  6999. "name": "torch.tril",
  7000. "category": "Layer",
  7001. "inputs": [
  7002. { "name": "self", "type": "Tensor" },
  7003. { "name": "diagonal", "type": "int64", "default": 0 }
  7004. ],
  7005. "outputs": [
  7006. { "name": "output", "type": "Tensor" }
  7007. ]
  7008. },
  7009. {
  7010. "name": "torch.tril_indices",
  7011. "category": "Layer",
  7012. "inputs": [
  7013. { "name": "row", "type": "int64" },
  7014. { "name": "col", "type": "int64" },
  7015. { "name": "offset", "type": "int64", "default": 0 },
  7016. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  7017. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  7018. { "name": "device", "type": "Device", "optional": true, "default": null },
  7019. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  7020. ],
  7021. "outputs": [
  7022. { "name": "output", "type": "Tensor" }
  7023. ]
  7024. },
  7025. {
  7026. "name": "torch.view_as_real",
  7027. "category": "Layer",
  7028. "inputs": [
  7029. { "name": "self", "type": "Tensor" }
  7030. ],
  7031. "outputs": [
  7032. { "name": "output", "type": "Tensor" }
  7033. ]
  7034. },
  7035. {
  7036. "name": "torch.view_as_complex",
  7037. "category": "Layer",
  7038. "inputs": [
  7039. { "name": "self", "type": "Tensor" }
  7040. ],
  7041. "outputs": [
  7042. { "name": "output", "type": "Tensor" }
  7043. ]
  7044. },
  7045. {
  7046. "name": "ops.torchvision.nms",
  7047. "inputs": [
  7048. { "name": "dets", "type": "Tensor" },
  7049. { "name": "scores", "type": "Tensor" },
  7050. { "name": "iou_threshold", "type": "float32" }
  7051. ],
  7052. "outputs": [
  7053. { "name": "output", "type": "Tensor" }
  7054. ]
  7055. },
  7056. {
  7057. "name": "ops.torchvision.roi_align",
  7058. "inputs": [
  7059. { "name": "input", "type": "Tensor" },
  7060. { "name": "rois", "type": "Tensor" },
  7061. { "name": "spatial_scale", "type": "float32" },
  7062. { "name": "pooled_height", "type": "int64" },
  7063. { "name": "pooled_width", "type": "int64" },
  7064. { "name": "sampling_ratio", "type": "int64" },
  7065. { "name": "aligned", "type": "boolean" }
  7066. ],
  7067. "outputs": [
  7068. { "name": "output", "type": "Tensor" }
  7069. ]
  7070. },
  7071. {
  7072. "name": "torch.__xor__:Scalar",
  7073. "inputs": [
  7074. { "name": "self", "type": "Tensor" },
  7075. { "name": "other", "type": "Scalar" }
  7076. ],
  7077. "outputs": [
  7078. { "name": "output", "type": "Tensor" }
  7079. ]
  7080. },
  7081. {
  7082. "name": "torch.__xor__:Tensor",
  7083. "inputs": [
  7084. { "name": "self", "type": "Tensor" },
  7085. { "name": "other", "type": "Tensor" }
  7086. ],
  7087. "outputs": [
  7088. { "name": "output", "type": "Tensor" }
  7089. ]
  7090. },
  7091. {
  7092. "name": "torch.__lshift__:Scalar",
  7093. "inputs": [
  7094. { "name": "self", "type": "Tensor" },
  7095. { "name": "other", "type": "Scalar" }
  7096. ],
  7097. "outputs": [
  7098. { "name": "output", "type": "Tensor" }
  7099. ]
  7100. },
  7101. {
  7102. "name": "torch.__lshift__:Tensor",
  7103. "inputs": [
  7104. { "name": "self", "type": "Tensor" },
  7105. { "name": "other", "type": "Tensor" }
  7106. ],
  7107. "outputs": [
  7108. { "name": "output", "type": "Tensor" }
  7109. ]
  7110. },
  7111. {
  7112. "name": "torch.__rshift__:Scalar",
  7113. "inputs": [
  7114. { "name": "self", "type": "Tensor" },
  7115. { "name": "other", "type": "Scalar" }
  7116. ],
  7117. "outputs": [
  7118. { "name": "output", "type": "Tensor" }
  7119. ]
  7120. },
  7121. {
  7122. "name": "torch.__rshift__:Tensor",
  7123. "inputs": [
  7124. { "name": "self", "type": "Tensor" },
  7125. { "name": "other", "type": "Tensor" }
  7126. ],
  7127. "outputs": [
  7128. { "name": "output", "type": "Tensor" }
  7129. ]
  7130. },
  7131. {
  7132. "name": "torch.__ixor__:Scalar",
  7133. "inputs": [
  7134. { "name": "self", "type": "Tensor" },
  7135. { "name": "other", "type": "Scalar" }
  7136. ],
  7137. "outputs": [
  7138. { "name": "output", "type": "Tensor" }
  7139. ]
  7140. },
  7141. {
  7142. "name": "torch.__ixor__:Tensor",
  7143. "inputs": [
  7144. { "name": "self", "type": "Tensor" },
  7145. { "name": "other", "type": "Tensor" }
  7146. ],
  7147. "outputs": [
  7148. { "name": "output", "type": "Tensor" }
  7149. ]
  7150. },
  7151. {
  7152. "name": "torch.__ilshift__:Scalar",
  7153. "inputs": [
  7154. { "name": "self", "type": "Tensor" },
  7155. { "name": "other", "type": "Scalar" }
  7156. ],
  7157. "outputs": [
  7158. { "name": "output", "type": "Tensor" }
  7159. ]
  7160. },
  7161. {
  7162. "name": "torch.__ilshift__:Tensor",
  7163. "inputs": [
  7164. { "name": "self", "type": "Tensor" },
  7165. { "name": "other", "type": "Tensor" }
  7166. ],
  7167. "outputs": [
  7168. { "name": "output", "type": "Tensor" }
  7169. ]
  7170. },
  7171. {
  7172. "name": "torch.__irshift__:Scalar",
  7173. "inputs": [
  7174. { "name": "self", "type": "Tensor" },
  7175. { "name": "other", "type": "Scalar" }
  7176. ],
  7177. "outputs": [
  7178. { "name": "output", "type": "Tensor" }
  7179. ]
  7180. },
  7181. {
  7182. "name": "torch.__irshift__:Tensor",
  7183. "inputs": [
  7184. { "name": "self", "type": "Tensor" },
  7185. { "name": "other", "type": "Tensor" }
  7186. ],
  7187. "outputs": [
  7188. { "name": "output", "type": "Tensor" }
  7189. ]
  7190. },
  7191. {
  7192. "name": "torch.one_hot:",
  7193. "inputs": [
  7194. { "name": "self", "type": "Tensor" },
  7195. { "name": "num_classes", "type": "int64", "default": -1 }
  7196. ],
  7197. "outputs": [
  7198. { "name": "output", "type": "Tensor" }
  7199. ]
  7200. },
  7201. {
  7202. "name": "torch.hann_window:",
  7203. "inputs": [
  7204. { "name": "window_length", "type": "int64" },
  7205. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  7206. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  7207. { "name": "device", "type": "Device", "optional": true, "default": null },
  7208. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  7209. ],
  7210. "outputs": [
  7211. { "name": "output", "type": "Tensor" }
  7212. ]
  7213. },
  7214. {
  7215. "name": "torch.hann_window:periodic",
  7216. "inputs": [
  7217. { "name": "window_length", "type": "int64" },
  7218. { "name": "periodic", "type": "boolean" },
  7219. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  7220. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  7221. { "name": "device", "type": "Device", "optional": true, "default": null },
  7222. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  7223. ],
  7224. "outputs": [
  7225. { "name": "output", "type": "Tensor" }
  7226. ]
  7227. },
  7228. {
  7229. "name": "torch.nonzero_numpy",
  7230. "inputs": [
  7231. { "name": "self", "type": "Tensor" }
  7232. ],
  7233. "outputs": [
  7234. { "name": "output", "type": "Tensor" }
  7235. ]
  7236. },
  7237. {
  7238. "name": "torch._dim_arange:",
  7239. "inputs": [
  7240. { "name": "like", "type": "Tensor" },
  7241. { "name": "dim", "type": "int64" }
  7242. ],
  7243. "outputs": [
  7244. { "name": "output", "type": "Tensor" }
  7245. ]
  7246. },
  7247. {
  7248. "name": "torch.cross:",
  7249. "inputs": [
  7250. { "name": "self", "type": "Tensor" },
  7251. { "name": "other", "type": "Tensor" },
  7252. { "name": "dim", "type": "int64", "optional": true, "default": null }
  7253. ],
  7254. "outputs": [
  7255. { "name": "output", "type": "Tensor" }
  7256. ]
  7257. },
  7258. {
  7259. "name": "torch.cross:out",
  7260. "inputs": [
  7261. { "name": "self", "type": "Tensor" },
  7262. { "name": "other", "type": "Tensor" },
  7263. { "name": "dim", "type": "int64", "optional": true, "default": null }
  7264. ],
  7265. "outputs": [
  7266. { "name": "output", "type": "Tensor" }
  7267. ]
  7268. },
  7269. {
  7270. "name": "torch.cross_entropy_loss",
  7271. "inputs": [
  7272. { "name": "self", "type": "Tensor" },
  7273. { "name": "target", "type": "Tensor" },
  7274. { "name": "weight", "type": "Tensor", "optional": true, "default": null },
  7275. { "name": "reduction", "type": "int64" },
  7276. { "name": "ignore_index", "type": "int64", "default": -100 },
  7277. { "name": "label_smoothing", "type": "float32", "default": 0 }
  7278. ],
  7279. "outputs": [
  7280. { "name": "output", "type": "Tensor" }
  7281. ]
  7282. },
  7283. {
  7284. "name": "torch.binary_cross_entropy_with_logits",
  7285. "inputs": [
  7286. { "name": "self", "type": "Tensor" },
  7287. { "name": "target", "type": "Tensor" },
  7288. { "name": "weight", "type": "Tensor", "optional": true, "default": null },
  7289. { "name": "pos_weight", "type": "Tensor", "optional": true, "default": null },
  7290. { "name": "reduction", "type": "int64" }
  7291. ],
  7292. "outputs": [
  7293. { "name": "output", "type": "Tensor" }
  7294. ]
  7295. }
  7296. ]