2
0

pytorch-metadata.json 158 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927
  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. "attributes": [
  377. { "name": "stride", "type": "int64[]", "default": 1 },
  378. { "name": "padding", "type": "int64[]", "default": 0 },
  379. { "name": "dilation", "type": "int64[]", "default": 1 },
  380. { "name": "transposed", "type": "boolean", "default": false },
  381. { "name": "output_padding", "type": "int64[]", "default": 0 },
  382. { "name": "groups", "type": "int64", "default": 1 },
  383. { "name": "benchmark", "type": "boolean", "visible": false },
  384. { "name": "deterministic", "type": "boolean", "visible": false },
  385. { "name": "cudnn_enabled", "type": "boolean", "visible": false },
  386. { "name": "allow_tf32", "type": "boolean", "visible": false }
  387. ],
  388. "inputs": [
  389. { "name": "input", "type": "Tensor" },
  390. { "name": "weight", "type": "Tensor" },
  391. { "name": "bias", "type": "Tensor", "optional": true }
  392. ],
  393. "outputs": [
  394. { "name": "output", "type": "Tensor" }
  395. ]
  396. },
  397. {
  398. "name": "torch._convolution:deprecated",
  399. "category": "Layer",
  400. "attributes": [
  401. { "name": "stride", "type": "int64[]", "default": 1 },
  402. { "name": "padding", "type": "int64[]", "default": 0 },
  403. { "name": "dilation", "type": "int64[]", "default": 1 },
  404. { "name": "transposed", "type": "boolean", "default": false },
  405. { "name": "output_padding", "type": "int64[]", "default": 0 },
  406. { "name": "groups", "type": "int64", "default": 1 },
  407. { "name": "benchmark", "type": "boolean", "visible": false },
  408. { "name": "deterministic", "type": "boolean", "visible": false },
  409. { "name": "cudnn_enabled", "type": "boolean", "visible": false }
  410. ],
  411. "inputs": [
  412. { "name": "input", "type": "Tensor" },
  413. { "name": "weight", "type": "Tensor" },
  414. { "name": "bias", "type": "Tensor", "optional": true }
  415. ],
  416. "outputs": [
  417. { "name": "output", "type": "Tensor" }
  418. ]
  419. },
  420. {
  421. "name": "torch.conv1d",
  422. "category": "Layer",
  423. "attributes": [
  424. { "name": "stride", "type": "int64[]", "default": 1 },
  425. { "name": "padding", "type": "int64[]", "default": 0 },
  426. { "name": "dilation", "type": "int64[]", "default": 1 },
  427. { "name": "groups", "type": "int64", "default": 1 }
  428. ],
  429. "inputs": [
  430. { "name": "input", "type": "Tensor" },
  431. { "name": "weight", "type": "Tensor" },
  432. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  433. ],
  434. "outputs": [
  435. { "name": "output", "type": "Tensor" }
  436. ]
  437. },
  438. {
  439. "name": "torch.conv2d",
  440. "category": "Layer",
  441. "attributes": [
  442. { "name": "stride", "type": "int64[]", "default": 1 },
  443. { "name": "padding", "type": "int64[]", "default": 0 },
  444. { "name": "dilation", "type": "int64[]", "default": 1 },
  445. { "name": "groups", "type": "int64", "default": 1 }
  446. ],
  447. "inputs": [
  448. { "name": "input", "type": "Tensor" },
  449. { "name": "weight", "type": "Tensor" },
  450. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  451. ],
  452. "outputs": [
  453. { "name": "output", "type": "Tensor" }
  454. ]
  455. },
  456. {
  457. "name": "torch.conv3d",
  458. "category": "Layer",
  459. "attributes": [
  460. { "name": "stride", "type": "int64[]", "default": 1 },
  461. { "name": "padding", "type": "int64[]", "default": 0 },
  462. { "name": "dilation", "type": "int64[]", "default": 1 },
  463. { "name": "groups", "type": "int64", "default": 1 }
  464. ],
  465. "inputs": [
  466. { "name": "input", "type": "Tensor" },
  467. { "name": "weight", "type": "Tensor" },
  468. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  469. ],
  470. "outputs": [
  471. { "name": "output", "type": "Tensor" }
  472. ]
  473. },
  474. {
  475. "name": "torch.conv_transpose1d:input",
  476. "category": "Layer",
  477. "attributes": [
  478. { "name": "stride", "type": "int64[]", "default": 1 },
  479. { "name": "padding", "type": "int64[]", "default": 0 },
  480. { "name": "output_padding", "type": "int64[]", "default": 0 },
  481. { "name": "groups", "type": "int64", "default": 1 },
  482. { "name": "dilation", "type": "int64[]", "default": 1 }
  483. ],
  484. "inputs": [
  485. { "name": "input", "type": "Tensor" },
  486. { "name": "weight", "type": "Tensor" },
  487. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  488. ],
  489. "outputs": [
  490. { "name": "output", "type": "Tensor" }
  491. ]
  492. },
  493. {
  494. "name": "torch.conv_transpose2d:input",
  495. "category": "Layer",
  496. "attributes": [
  497. { "name": "stride", "type": "int64[]", "default": 1 },
  498. { "name": "padding", "type": "int64[]", "default": 0 },
  499. { "name": "output_padding", "type": "int64[]", "default": 0 },
  500. { "name": "groups", "type": "int64", "default": 1 },
  501. { "name": "dilation", "type": "int64[]", "default": 1 }
  502. ],
  503. "inputs": [
  504. { "name": "input", "type": "Tensor" },
  505. { "name": "weight", "type": "Tensor" },
  506. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  507. ],
  508. "outputs": [
  509. { "name": "output", "type": "Tensor" }
  510. ]
  511. },
  512. {
  513. "name": "torch.conv_transpose3d:input",
  514. "category": "Layer",
  515. "attributes": [
  516. { "name": "stride", "type": "int64[]", "default": 1 },
  517. { "name": "padding", "type": "int64[]", "default": 0 },
  518. { "name": "output_padding", "type": "int64[]", "default": 0 },
  519. { "name": "groups", "type": "int64", "default": 1 },
  520. { "name": "dilation", "type": "int64[]", "default": 1 }
  521. ],
  522. "inputs": [
  523. { "name": "input", "type": "Tensor" },
  524. { "name": "weight", "type": "Tensor" },
  525. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  526. ],
  527. "outputs": [
  528. { "name": "output", "type": "Tensor" }
  529. ]
  530. },
  531. {
  532. "name": "ops.prepacked.conv2d_clamp_run",
  533. "category": "Layer",
  534. "inputs": [
  535. { "name": "input", "type": "Tensor" },
  536. { "name": "orig_weight_", "type": "Tensor" },
  537. { "name": "orig_bias_", "type": "Tensor", "optional": true },
  538. { "name": "stride_", "type": "int64[]" },
  539. { "name": "padding_", "type": "int64[]" },
  540. { "name": "dilation_", "type": "int64[]" },
  541. { "name": "groups_", "type": "int64" },
  542. { "name": "output_min_", "type": "Scalar", "optional": true },
  543. { "name": "output_max_", "type": "Scalar", "optional": true }
  544. ],
  545. "outputs": [
  546. { "name": "output", "type": "Tensor" }
  547. ]
  548. },
  549. {
  550. "name": "ops.prepacked.linear_clamp_run",
  551. "category": "Layer",
  552. "attributes": [
  553. { "name": "output_min_", "type": "Scalar", "optional": true },
  554. { "name": "output_max_", "type": "Scalar", "optional": true }
  555. ],
  556. "inputs": [
  557. { "name": "input", "type": "Tensor" },
  558. { "name": "orig_weight_", "type": "Tensor" },
  559. { "name": "orig_bias_", "type": "Tensor", "optional": true }
  560. ],
  561. "outputs": [
  562. { "name": "output", "type": "Tensor" }
  563. ]
  564. },
  565. {
  566. "name": "ops.quantized.add",
  567. "attributes": [
  568. { "name": "scale", "type": "float32" },
  569. { "name": "zero_point", "type": "int64" }
  570. ],
  571. "inputs": [
  572. { "name": "A", "type": "Tensor" },
  573. { "name": "B", "type": "Tensor" }
  574. ],
  575. "outputs": [
  576. { "name": "C", "type": "Tensor" }
  577. ]
  578. },
  579. {
  580. "name": "ops.quantized.linear",
  581. "category": "Layer",
  582. "attributes": [
  583. { "name": "Y_scale_i", "type": "float32" },
  584. { "name": "Y_zero_point_i", "type": "int64" }
  585. ],
  586. "inputs": [
  587. { "name": "X", "type": "Tensor" },
  588. { "name": "weight", "type": "Tensor" },
  589. { "name": "bias", "type": "Tensor" }
  590. ],
  591. "outputs": [
  592. { "name": "Y", "type": "Tensor" }
  593. ]
  594. },
  595. {
  596. "name": "ops.quantized.linear_relu",
  597. "category": "Layer",
  598. "attributes": [
  599. { "name": "Y_scale_i", "type": "float32" },
  600. { "name": "Y_zero_point_i", "type": "int64" }
  601. ],
  602. "inputs": [
  603. { "name": "X", "type": "Tensor" },
  604. { "name": "weight", "type": "Tensor" },
  605. { "name": "bias", "type": "Tensor" }
  606. ],
  607. "outputs": [
  608. { "name": "Y", "type": "Tensor" }
  609. ]
  610. },
  611. {
  612. "name": "ops.quantized.linear_dynamic",
  613. "category": "Layer",
  614. "attributes": [
  615. { "name": "reduce_range", "type": "boolean", "default": false }
  616. ],
  617. "inputs": [
  618. { "name": "X", "type": "Tensor" },
  619. { "name": "weight", "type": "Tensor" },
  620. { "name": "bias", "type": "Tensor" }
  621. ],
  622. "outputs": [
  623. { "name": "Y", "type": "Tensor" }
  624. ]
  625. },
  626. {
  627. "name": "ops.quantized.linear_relu_dynamic",
  628. "category": "Layer",
  629. "attributes": [
  630. { "name": "Y_scale_i", "type": "float32" },
  631. { "name": "Y_zero_point_i", "type": "int64" }
  632. ],
  633. "inputs": [
  634. { "name": "X", "type": "Tensor" },
  635. { "name": "weight", "type": "Tensor" },
  636. { "name": "bias", "type": "Tensor" }
  637. ],
  638. "outputs": [
  639. { "name": "Y", "type": "Tensor" }
  640. ]
  641. },
  642. {
  643. "name": "torch.dequantize",
  644. "inputs": [
  645. { "name": "input", "type": "Tensor" }
  646. ],
  647. "outputs": [
  648. { "name": "output", "type": "Tensor" }
  649. ]
  650. },
  651. {
  652. "name": "ops.quantized.conv1d",
  653. "category": "Layer",
  654. "inputs": [
  655. { "name": "input", "type": "Tensor" },
  656. { "name": "weight", "type": "Tensor" },
  657. { "name": "bias", "type": "Tensor" },
  658. { "name": "output_scale", "type": "float32" },
  659. { "name": "output_zero_point", "type": "int64" }
  660. ],
  661. "outputs": [
  662. { "name": "output", "type": "Tensor" }
  663. ]
  664. },
  665. {
  666. "name": "ops.quantized.conv2d:",
  667. "category": "Layer",
  668. "inputs": [
  669. { "name": "input", "type": "Tensor" },
  670. { "name": "weight", "type": "Tensor" },
  671. { "name": "bias", "type": "Tensor" },
  672. { "name": "stride", "type": "int64[]", "default": 1 },
  673. { "name": "padding", "type": "int64[]", "default": 0 },
  674. { "name": "dilation", "type": "int64[]", "default": 1 },
  675. { "name": "groups", "type": "int64", "default": 1 },
  676. { "name": "output_scale", "type": "float32" },
  677. { "name": "output_zero_point", "type": "int64" }
  678. ],
  679. "outputs": [
  680. { "name": "output", "type": "Tensor" }
  681. ]
  682. },
  683. {
  684. "name": "ops.quantized.conv2d:new",
  685. "category": "Layer",
  686. "inputs": [
  687. { "name": "input", "type": "Tensor" },
  688. { "name": "weight", "type": "Tensor" },
  689. { "name": "bias", "type": "Tensor" },
  690. { "name": "output_scale", "type": "float32" },
  691. { "name": "output_zero_point", "type": "int64" }
  692. ],
  693. "outputs": [
  694. { "name": "output", "type": "Tensor" }
  695. ]
  696. },
  697. {
  698. "name": "ops.quantized.conv3d:",
  699. "category": "Layer",
  700. "attributes": [
  701. { "name": "stride", "type": "int64[]", "default": 1 },
  702. { "name": "padding", "type": "int64[]", "default": 0 },
  703. { "name": "dilation", "type": "int64[]", "default": 1 },
  704. { "name": "groups", "type": "int64", "default": 1 },
  705. { "name": "output_scale", "type": "float32" },
  706. { "name": "output_zero_point", "type": "int64" }
  707. ],
  708. "inputs": [
  709. { "name": "input", "type": "Tensor" },
  710. { "name": "weight", "type": "Tensor" },
  711. { "name": "bias", "type": "Tensor" }
  712. ],
  713. "outputs": [
  714. { "name": "output", "type": "Tensor" }
  715. ]
  716. },
  717. {
  718. "name": "ops.quantized.conv3d:new",
  719. "category": "Layer",
  720. "inputs": [
  721. { "name": "input", "type": "Tensor" },
  722. { "name": "weight", "type": "Tensor" },
  723. { "name": "bias", "type": "Tensor" },
  724. { "name": "output_scale", "type": "float32" },
  725. { "name": "output_zero_point", "type": "int64" }
  726. ],
  727. "outputs": [
  728. { "name": "output", "type": "Tensor" }
  729. ]
  730. },
  731. {
  732. "name": "ops.quantized.conv1d_relu",
  733. "category": "Layer",
  734. "inputs": [
  735. { "name": "input", "type": "Tensor" },
  736. { "name": "weight", "type": "Tensor" },
  737. { "name": "bias", "type": "Tensor" },
  738. { "name": "output_scale", "type": "float64" },
  739. { "name": "output_zero_point", "type": "int64" }
  740. ],
  741. "outputs": [
  742. { "name": "output", "type": "Tensor" }
  743. ]
  744. },
  745. {
  746. "name": "ops.quantized.conv2d_relu:",
  747. "category": "Layer",
  748. "inputs": [
  749. { "name": "input", "type": "Tensor" },
  750. { "name": "weight", "type": "Tensor" },
  751. { "name": "bias", "type": "Tensor" },
  752. { "name": "stride", "type": "int64[]", "default": 1 },
  753. { "name": "padding", "type": "int64[]", "default": 0 },
  754. { "name": "dilation", "type": "int64[]", "default": 1 },
  755. { "name": "groups", "type": "int64", "default": 1 },
  756. { "name": "output_scale", "type": "float64" },
  757. { "name": "output_zero_point", "type": "int64" }
  758. ],
  759. "outputs": [
  760. { "name": "output", "type": "Tensor" }
  761. ]
  762. },
  763. {
  764. "name": "ops.quantized.conv2d_relu:new",
  765. "category": "Layer",
  766. "inputs": [
  767. { "name": "input", "type": "Tensor" },
  768. { "name": "weight", "type": "Tensor" },
  769. { "name": "bias", "type": "Tensor" },
  770. { "name": "output_scale", "type": "float64" },
  771. { "name": "output_zero_point", "type": "int64" }
  772. ],
  773. "outputs": [
  774. { "name": "output", "type": "Tensor" }
  775. ]
  776. },
  777. {
  778. "name": "ops.quantized.conv3d_relu:",
  779. "category": "Layer",
  780. "inputs": [
  781. { "name": "input", "type": "Tensor" },
  782. { "name": "weight", "type": "Tensor" },
  783. { "name": "bias", "type": "Tensor" },
  784. { "name": "stride", "type": "int64[]", "default": 1 },
  785. { "name": "padding", "type": "int64[]", "default": 0 },
  786. { "name": "dilation", "type": "int64[]", "default": 1 },
  787. { "name": "groups", "type": "int64", "default": 1 },
  788. { "name": "output_scale", "type": "float64" },
  789. { "name": "output_zero_point", "type": "int64" }
  790. ],
  791. "outputs": [
  792. { "name": "output", "type": "Tensor" }
  793. ]
  794. },
  795. {
  796. "name": "ops.quantized.conv3d_relu:new",
  797. "category": "Layer",
  798. "inputs": [
  799. { "name": "input", "type": "Tensor" },
  800. { "name": "weight", "type": "Tensor" },
  801. { "name": "bias", "type": "Tensor" },
  802. { "name": "output_scale", "type": "float64" },
  803. { "name": "output_zero_point", "type": "int64" }
  804. ],
  805. "outputs": [
  806. { "name": "output", "type": "Tensor" }
  807. ]
  808. },
  809. {
  810. "name": "ops.quantized.add_relu:",
  811. "inputs": [
  812. { "name": "qa", "type": "Tensor" },
  813. { "name": "qb", "type": "Tensor" },
  814. { "name": "scale", "type": "float32" },
  815. { "name": "zero_point", "type": "int64" }
  816. ],
  817. "outputs": [
  818. { "name": "qc", "type": "Tensor" }
  819. ]
  820. },
  821. {
  822. "name": "ops.quantized.add_relu:Scalar",
  823. "inputs": [
  824. { "name": "qa", "type": "Tensor" },
  825. { "name": "b", "type": "Scalar" }
  826. ],
  827. "outputs": [
  828. { "name": "qc", "type": "Tensor" }
  829. ]
  830. },
  831. {
  832. "name": "ops.quantized.cat",
  833. "category": "Tensor",
  834. "inputs": [
  835. { "name": "qx", "type": "Tensor[]" },
  836. { "name": "dim", "type": "int64" },
  837. { "name": "scale", "type": "float32", "optional": true },
  838. { "name": "zero_point", "type": "int64", "optional": true }
  839. ],
  840. "outputs": [
  841. { "name": "output", "type": "Tensor" }
  842. ]
  843. },
  844. {
  845. "name": "ops.quantized.cat_relu",
  846. "category": "Tensor",
  847. "inputs": [
  848. { "name": "qx", "type": "Tensor[]" },
  849. { "name": "dim", "type": "int64" },
  850. { "name": "scale", "type": "float32", "optional": true },
  851. { "name": "zero_point", "type": "int64", "optional": true }
  852. ],
  853. "outputs": [
  854. { "name": "output", "type": "Tensor" }
  855. ]
  856. },
  857. {
  858. "name": "ops.quantized.batch_norm:",
  859. "inputs": [
  860. { "name": "qx", "type": "Tensor" },
  861. { "name": "weight", "type": "Tensor", "optional": true },
  862. { "name": "bias", "type": "Tensor", "optional": true },
  863. { "name": "mean", "type": "Tensor" },
  864. { "name": "var", "type": "Tensor" },
  865. { "name": "eps", "type": "float32" },
  866. { "name": "output_scale", "type": "float32", "optional": true },
  867. { "name": "outputzero_point", "type": "int64", "optional": true }
  868. ],
  869. "outputs": [
  870. { "name": "output", "type": "Tensor" }
  871. ]
  872. },
  873. {
  874. "name": "ops.quantized.batch_norm_relu",
  875. "category": "Normalization",
  876. "inputs": [
  877. { "name": "qx", "type": "Tensor" },
  878. { "name": "weight", "type": "Tensor", "optional": true },
  879. { "name": "bias", "type": "Tensor", "optional": true },
  880. { "name": "mean", "type": "Tensor" },
  881. { "name": "var", "type": "Tensor" },
  882. { "name": "eps", "type": "float32" },
  883. { "name": "output_scale", "type": "float32", "optional": true },
  884. { "name": "outputzero_point", "type": "int64", "optional": true }
  885. ],
  886. "outputs": [
  887. { "name": "output", "type": "Tensor" }
  888. ]
  889. },
  890. {
  891. "name": "ops.quantized.batch_norm1d",
  892. "category": "Normalization",
  893. "inputs": [
  894. { "name": "qx", "type": "Tensor" },
  895. { "name": "weight", "type": "Tensor", "optional": true },
  896. { "name": "bias", "type": "Tensor", "optional": true },
  897. { "name": "mean", "type": "Tensor" },
  898. { "name": "var", "type": "Tensor" },
  899. { "name": "eps", "type": "float32" },
  900. { "name": "output_scale", "type": "float32", "optional": true },
  901. { "name": "outputzero_point", "type": "int64", "optional": true }
  902. ],
  903. "outputs": [
  904. { "name": "output", "type": "Tensor" }
  905. ]
  906. },
  907. {
  908. "name": "ops.quantized.batch_norm1d_relu",
  909. "category": "Normalization",
  910. "inputs": [
  911. { "name": "qx", "type": "Tensor" },
  912. { "name": "weight", "type": "Tensor", "optional": true },
  913. { "name": "bias", "type": "Tensor", "optional": true },
  914. { "name": "mean", "type": "Tensor" },
  915. { "name": "var", "type": "Tensor" },
  916. { "name": "eps", "type": "float32" },
  917. { "name": "output_scale", "type": "float32", "optional": true },
  918. { "name": "outputzero_point", "type": "int64", "optional": true }
  919. ],
  920. "outputs": [
  921. { "name": "output", "type": "Tensor" }
  922. ]
  923. },
  924. {
  925. "name": "ops.quantized.batch_norm2d",
  926. "category": "Normalization",
  927. "inputs": [
  928. { "name": "qx", "type": "Tensor" },
  929. { "name": "weight", "type": "Tensor", "optional": true },
  930. { "name": "bias", "type": "Tensor", "optional": true },
  931. { "name": "mean", "type": "Tensor" },
  932. { "name": "var", "type": "Tensor" },
  933. { "name": "eps", "type": "float32" },
  934. { "name": "output_scale", "type": "float32", "optional": true },
  935. { "name": "outputzero_point", "type": "int64", "optional": true }
  936. ],
  937. "outputs": [
  938. { "name": "output", "type": "Tensor" }
  939. ]
  940. },
  941. {
  942. "name": "ops.quantized.batch_norm2d_relu",
  943. "category": "Normalization",
  944. "inputs": [
  945. { "name": "qx", "type": "Tensor" },
  946. { "name": "weight", "type": "Tensor", "optional": true },
  947. { "name": "bias", "type": "Tensor", "optional": true },
  948. { "name": "mean", "type": "Tensor" },
  949. { "name": "var", "type": "Tensor" },
  950. { "name": "eps", "type": "float32" },
  951. { "name": "output_scale", "type": "float32", "optional": true },
  952. { "name": "outputzero_point", "type": "int64", "optional": true }
  953. ],
  954. "outputs": [
  955. { "name": "output", "type": "Tensor" }
  956. ]
  957. },
  958. {
  959. "name": "ops.quantized.batch_norm3d",
  960. "category": "Normalization",
  961. "inputs": [
  962. { "name": "qx", "type": "Tensor" },
  963. { "name": "weight", "type": "Tensor", "optional": true },
  964. { "name": "bias", "type": "Tensor", "optional": true },
  965. { "name": "mean", "type": "Tensor" },
  966. { "name": "var", "type": "Tensor" },
  967. { "name": "eps", "type": "float32" },
  968. { "name": "output_scale", "type": "float32", "optional": true },
  969. { "name": "outputzero_point", "type": "int64", "optional": true }
  970. ],
  971. "outputs": [
  972. { "name": "output", "type": "Tensor" }
  973. ]
  974. },
  975. {
  976. "name": "ops.quantized.batch_norm3d_relu",
  977. "category": "Normalization",
  978. "inputs": [
  979. { "name": "qx", "type": "Tensor" },
  980. { "name": "weight", "type": "Tensor", "optional": true },
  981. { "name": "bias", "type": "Tensor", "optional": true },
  982. { "name": "mean", "type": "Tensor" },
  983. { "name": "var", "type": "Tensor" },
  984. { "name": "eps", "type": "float32" },
  985. { "name": "output_scale", "type": "float32", "optional": true },
  986. { "name": "outputzero_point", "type": "int64", "optional": true }
  987. ],
  988. "outputs": [
  989. { "name": "output", "type": "Tensor" }
  990. ]
  991. },
  992. {
  993. "name": "torch.max_pool1d",
  994. "category": "Pool",
  995. "attributes": [
  996. { "name": "kernel_size", "type": "int64[]" },
  997. { "name": "stride", "type": "int64[]" },
  998. { "name": "padding", "type": "int64[]", "default": 0 },
  999. { "name": "dilation", "type": "int64[]", "default": 1 },
  1000. { "name": "ceil_mode", "type": "boolean", "default": false }
  1001. ],
  1002. "inputs": [
  1003. { "name": "self", "type": "Tensor" }
  1004. ],
  1005. "outputs": [
  1006. { "name": "output", "type": "Tensor" }
  1007. ]
  1008. },
  1009. {
  1010. "name": "torch.max_pool2d",
  1011. "category": "Pool",
  1012. "attributes": [
  1013. { "name": "kernel_size", "type": "int64[]" },
  1014. { "name": "stride", "type": "int64[]" },
  1015. { "name": "padding", "type": "int64[]", "default": 0 },
  1016. { "name": "dilation", "type": "int64[]", "default": 1 },
  1017. { "name": "ceil_mode", "type": "boolean", "default": false }
  1018. ],
  1019. "inputs": [
  1020. { "name": "self", "type": "Tensor" }
  1021. ],
  1022. "outputs": [
  1023. { "name": "output", "type": "Tensor" }
  1024. ]
  1025. },
  1026. {
  1027. "name": "torch.max_pool3d",
  1028. "category": "Pool",
  1029. "attributes": [
  1030. { "name": "kernel_size", "type": "int64[]" },
  1031. { "name": "stride", "type": "int64[]" },
  1032. { "name": "padding", "type": "int64[]", "default": 0 },
  1033. { "name": "dilation", "type": "int64[]", "default": 1 },
  1034. { "name": "ceil_mode", "type": "boolean", "default": false }
  1035. ],
  1036. "inputs": [
  1037. { "name": "self", "type": "Tensor" }
  1038. ],
  1039. "outputs": [
  1040. { "name": "output", "type": "Tensor" }
  1041. ]
  1042. },
  1043. {
  1044. "name": "torch.max_unpool2d",
  1045. "category": "Pool",
  1046. "inputs": [
  1047. { "name": "self", "type": "Tensor" },
  1048. { "name": "indices", "type": "Tensor" },
  1049. { "name": "output_size", "type": "int64[]" }
  1050. ],
  1051. "outputs": [
  1052. { "name": "output", "type": "Tensor" }
  1053. ]
  1054. },
  1055. {
  1056. "name": "torch.max_unpool3d",
  1057. "category": "Pool",
  1058. "inputs": [
  1059. { "name": "self", "type": "Tensor" },
  1060. { "name": "indices", "type": "Tensor" },
  1061. { "name": "output_size", "type": "int64[]" },
  1062. { "name": "stride", "type": "int64[]" },
  1063. { "name": "padding", "type": "int64[]" }
  1064. ],
  1065. "outputs": [
  1066. { "name": "output", "type": "Tensor" }
  1067. ]
  1068. },
  1069. {
  1070. "name": "torch.max_pool1d_with_indices",
  1071. "category": "Pool",
  1072. "attributes": [
  1073. { "name": "kernel_size", "type": "int64[]" },
  1074. { "name": "stride", "type": "int64[]" },
  1075. { "name": "padding", "type": "int64[]", "default": 0 },
  1076. { "name": "dilation", "type": "int64[]", "default": 1 },
  1077. { "name": "ceil_mode", "type": "boolean", "default": false },
  1078. { "name": "return_indices", "type": "boolean", "default": false }
  1079. ],
  1080. "inputs": [
  1081. { "name": "input", "type": "Tensor" }
  1082. ],
  1083. "outputs": [
  1084. { "name": "output", "type": "Tensor" },
  1085. { "name": "indices", "type": "Tensor" }
  1086. ]
  1087. },
  1088. {
  1089. "name": "torch.max_pool2d_with_indices",
  1090. "category": "Pool",
  1091. "attributes": [
  1092. { "name": "kernel_size", "type": "int64[]" },
  1093. { "name": "stride", "type": "int64[]" },
  1094. { "name": "padding", "type": "int64[]", "default": 0 },
  1095. { "name": "dilation", "type": "int64[]", "default": 1 },
  1096. { "name": "ceil_mode", "type": "boolean", "default": false },
  1097. { "name": "return_indices", "type": "boolean", "default": false }
  1098. ],
  1099. "inputs": [
  1100. { "name": "input", "type": "Tensor" }
  1101. ],
  1102. "outputs": [
  1103. { "name": "output", "type": "Tensor" },
  1104. { "name": "indices", "type": "Tensor" }
  1105. ]
  1106. },
  1107. {
  1108. "name": "torch.adaptive_avg_pool1d",
  1109. "category": "Pool",
  1110. "attributes": [
  1111. { "name": "output_size", "type": "int64[]", "visible": false }
  1112. ],
  1113. "inputs": [
  1114. { "name": "input", "type": "Tensor" }
  1115. ],
  1116. "outputs": [
  1117. { "name": "output", "type": "Tensor" }
  1118. ]
  1119. },
  1120. {
  1121. "name": "torch.adaptive_avg_pool2d",
  1122. "category": "Pool",
  1123. "attributes": [
  1124. { "name": "output_size", "type": "int64[]", "visible": false }
  1125. ],
  1126. "inputs": [
  1127. { "name": "input", "type": "Tensor" }
  1128. ],
  1129. "outputs": [
  1130. { "name": "output", "type": "Tensor" }
  1131. ]
  1132. },
  1133. {
  1134. "name": "torch.adaptive_avg_pool3d",
  1135. "category": "Pool",
  1136. "attributes": [
  1137. { "name": "output_size", "type": "int64[]", "visible": false }
  1138. ],
  1139. "inputs": [
  1140. { "name": "input", "type": "Tensor" }
  1141. ],
  1142. "outputs": [
  1143. { "name": "output", "type": "Tensor" }
  1144. ]
  1145. },
  1146. {
  1147. "name": "torch.adaptive_max_pool1d",
  1148. "category": "Pool",
  1149. "attributes": [
  1150. { "name": "output_size", "type": "int64[]", "visible": false }
  1151. ],
  1152. "inputs": [
  1153. { "name": "input", "type": "Tensor" }
  1154. ],
  1155. "outputs": [
  1156. { "name": "output", "type": "Tensor" },
  1157. { "name": "?", "type": "Tensor" }
  1158. ]
  1159. },
  1160. {
  1161. "name": "torch.adaptive_max_pool2d",
  1162. "category": "Pool",
  1163. "attributes": [
  1164. { "name": "output_size", "type": "int64[]", "visible": false }
  1165. ],
  1166. "inputs": [
  1167. { "name": "input", "type": "Tensor" }
  1168. ],
  1169. "outputs": [
  1170. { "name": "output", "type": "Tensor" },
  1171. { "name": "?", "type": "Tensor" }
  1172. ]
  1173. },
  1174. {
  1175. "name": "torch.adaptive_max_pool3d",
  1176. "category": "Pool",
  1177. "attributes": [
  1178. { "name": "output_size", "type": "int64[]", "visible": false }
  1179. ],
  1180. "inputs": [
  1181. { "name": "input", "type": "Tensor" }
  1182. ],
  1183. "outputs": [
  1184. { "name": "output", "type": "Tensor" },
  1185. { "name": "?", "type": "Tensor" }
  1186. ]
  1187. },
  1188. {
  1189. "name": "torch.avg_pool1d",
  1190. "category": "Pool",
  1191. "attributes": [
  1192. { "name": "kernel_size", "type": "int64[]" },
  1193. { "name": "stride", "type": "int64[]" },
  1194. { "name": "padding", "type": "int64[]", "default": 0 },
  1195. { "name": "ceil_mode", "type": "boolean", "default": false },
  1196. { "name": "count_include_pad", "type": "boolean", "default": true },
  1197. { "name": "divisor_override" }
  1198. ],
  1199. "inputs": [
  1200. { "name": "input", "type": "Tensor" }
  1201. ],
  1202. "outputs": [
  1203. { "name": "output", "type": "Tensor" }
  1204. ]
  1205. },
  1206. {
  1207. "name": "torch.avg_pool2d",
  1208. "category": "Pool",
  1209. "attributes": [
  1210. { "name": "kernel_size", "type": "int64[]" },
  1211. { "name": "stride", "type": "int64[]" },
  1212. { "name": "padding", "type": "int64[]" },
  1213. { "name": "ceil_mode", "type": "boolean", "default": false },
  1214. { "name": "count_include_pad", "type": "boolean", "default": true },
  1215. { "name": "divisor_override", "type": "int64", "optional": true }
  1216. ],
  1217. "inputs": [
  1218. { "name": "input", "type": "Tensor" }
  1219. ],
  1220. "outputs": [
  1221. { "name": "output", "type": "Tensor" }
  1222. ]
  1223. },
  1224. {
  1225. "name": "torch.avg_pool3d",
  1226. "category": "Pool",
  1227. "attributes": [
  1228. { "name": "kernel_size", "type": "int64[]" },
  1229. { "name": "stride", "type": "int64[]" },
  1230. { "name": "padding", "type": "int64[]" },
  1231. { "name": "ceil_mode", "type": "boolean", "default": false },
  1232. { "name": "count_include_pad", "type": "boolean", "default": true },
  1233. { "name": "divisor_override", "type": "int64", "optional": true }
  1234. ],
  1235. "inputs": [
  1236. { "name": "input", "type": "Tensor" }
  1237. ],
  1238. "outputs": [
  1239. { "name": "output", "type": "Tensor" }
  1240. ]
  1241. },
  1242. {
  1243. "name": "torch.batch_norm",
  1244. "category": "Normalization",
  1245. "attributes": [
  1246. { "name": "training", "type": "boolean", "visible": false },
  1247. { "name": "momentum", "type": "float32", "default": 0.1 },
  1248. { "name": "eps", "type": "float32", "default": 0.00001 },
  1249. { "name": "cudnn_enabled", "type": "boolean", "visible": false }
  1250. ],
  1251. "inputs": [
  1252. { "name": "input", "type": "Tensor" },
  1253. { "name": "weight", "type": "Tensor", "optional": true },
  1254. { "name": "bias", "type": "Tensor", "optional": true },
  1255. { "name": "running_mean", "type": "Tensor", "optional": true },
  1256. { "name": "running_var", "type": "Tensor", "optional": true }
  1257. ],
  1258. "outputs": [
  1259. { "name": "output", "type": "Tensor" }
  1260. ]
  1261. },
  1262. {
  1263. "name": "torch.group_norm",
  1264. "category": "Normalization",
  1265. "inputs": [
  1266. { "name": "input", "type": "Tensor" },
  1267. { "name": "num_groups", "type": "int64" },
  1268. { "name": "weight", "type": "Tensor", "optional": true },
  1269. { "name": "bias", "type": "Tensor", "optional": true },
  1270. { "name": "eps", "type": "float32", "default": 0.00001 },
  1271. { "name": "cudnn_enabled", "type": "boolean", "visible": true }
  1272. ],
  1273. "outputs": [
  1274. { "name": "output", "type": "Tensor" }
  1275. ]
  1276. },
  1277. {
  1278. "name": "torch.relu_",
  1279. "category": "Activation",
  1280. "inputs": [
  1281. { "name": "self", "type": "Tensor" }
  1282. ],
  1283. "outputs": [
  1284. { "name": "output", "type": "Tensor" }
  1285. ]
  1286. },
  1287. {
  1288. "name": "torch.relu",
  1289. "category": "Activation",
  1290. "inputs": [
  1291. { "name": "self", "type": "Tensor" }
  1292. ],
  1293. "outputs": [
  1294. { "name": "output", "type": "Tensor" }
  1295. ]
  1296. },
  1297. {
  1298. "name": "torch.prelu",
  1299. "category": "Activation",
  1300. "inputs": [
  1301. { "name": "self", "type": "Tensor" },
  1302. { "name": "weight", "type": "Tensor" }
  1303. ],
  1304. "outputs": [
  1305. { "name": "output", "type": "Tensor" }
  1306. ]
  1307. },
  1308. {
  1309. "name": "torch.hardtanh",
  1310. "category": "Activation",
  1311. "attributes": [
  1312. { "name": "min_value", "type": "float64" },
  1313. { "name": "max_value", "type": "float64" }
  1314. ],
  1315. "inputs": [
  1316. { "name": "self", "type": "Tensor" }
  1317. ],
  1318. "outputs": [
  1319. { "name": "output", "type": "Tensor" }
  1320. ]
  1321. },
  1322. {
  1323. "name": "torch.hardtanh_",
  1324. "category": "Activation",
  1325. "attributes": [
  1326. { "name": "min_value", "type": "float64" },
  1327. { "name": "max_value", "type": "float64" }
  1328. ],
  1329. "inputs": [
  1330. { "name": "self", "type": "Tensor" }
  1331. ],
  1332. "outputs": [
  1333. { "name": "output", "type": "Tensor" }
  1334. ]
  1335. },
  1336. {
  1337. "name": "torch.hardswish",
  1338. "category": "Activation",
  1339. "inputs": [
  1340. { "name": "self", "type": "Tensor" }
  1341. ],
  1342. "outputs": [
  1343. { "name": "output", "type": "Tensor" }
  1344. ]
  1345. },
  1346. {
  1347. "name": "torch.hardswish_",
  1348. "category": "Activation",
  1349. "inputs": [
  1350. { "name": "self", "type": "Tensor" }
  1351. ],
  1352. "outputs": [
  1353. { "name": "output", "type": "Tensor" }
  1354. ]
  1355. },
  1356. {
  1357. "name": "torch.hardsigmoid",
  1358. "category": "Activation",
  1359. "inputs": [
  1360. { "name": "self", "type": "Tensor" }
  1361. ],
  1362. "outputs": [
  1363. { "name": "output", "type": "Tensor" }
  1364. ]
  1365. },
  1366. {
  1367. "name": "torch.hardsigmoid_",
  1368. "category": "Activation",
  1369. "inputs": [
  1370. { "name": "self", "type": "Tensor" }
  1371. ],
  1372. "outputs": [
  1373. { "name": "output", "type": "Tensor" }
  1374. ]
  1375. },
  1376. {
  1377. "name": "torch.sigmoid",
  1378. "category": "Activation",
  1379. "inputs": [
  1380. { "name": "input", "type": "Tensor" }
  1381. ],
  1382. "outputs": [
  1383. { "name": "output", "type": "Tensor" }
  1384. ]
  1385. },
  1386. {
  1387. "name": "torch.sigmoid_",
  1388. "category": "Activation",
  1389. "inputs": [
  1390. { "name": "input", "type": "Tensor" }
  1391. ],
  1392. "outputs": [
  1393. { "name": "output", "type": "Tensor" }
  1394. ]
  1395. },
  1396. {
  1397. "name": "torch.threshold",
  1398. "category": "Activation",
  1399. "attributes": [
  1400. { "name": "threshold", "type": "float64", "default": 0 },
  1401. { "name": "value", "type": "float64", "default": 0 },
  1402. { "name": "inplace", "type": "boolean", "default": false }
  1403. ],
  1404. "inputs": [
  1405. { "name": "input", "type": "Tensor" }
  1406. ],
  1407. "outputs": [
  1408. { "name": "output", "type": "Tensor" }
  1409. ]
  1410. },
  1411. {
  1412. "name": "torch.threshold_",
  1413. "category": "Activation",
  1414. "attributes": [
  1415. { "name": "threshold", "type": "float64" },
  1416. { "name": "value", "type": "float64" },
  1417. { "name": "inplace", "type": "boolean" }
  1418. ],
  1419. "inputs": [
  1420. { "name": "input", "type": "Tensor" }
  1421. ],
  1422. "outputs": [
  1423. { "name": "output", "type": "Tensor" }
  1424. ]
  1425. },
  1426. {
  1427. "name": "torch.softmax:int",
  1428. "category": "Activation",
  1429. "inputs": [
  1430. { "name": "input", "type": "Tensor" },
  1431. { "name": "dim", "type": "int64" },
  1432. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1433. ],
  1434. "outputs": [
  1435. { "name": "output", "type": "Tensor" }
  1436. ]
  1437. },
  1438. {
  1439. "name": "torch.softmax:Dimname",
  1440. "category": "Activation",
  1441. "inputs": [
  1442. { "name": "input", "type": "Tensor" },
  1443. { "name": "dim", "type": "Dimname" },
  1444. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1445. ],
  1446. "outputs": [
  1447. { "name": "output", "type": "Tensor" }
  1448. ]
  1449. },
  1450. {
  1451. "name": "torch.log_softmax:int",
  1452. "category": "Activation",
  1453. "attributes": [
  1454. { "name": "dim", "type": "int64" },
  1455. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1456. ],
  1457. "inputs": [
  1458. { "name": "input", "type": "Tensor" }
  1459. ],
  1460. "outputs": [
  1461. { "name": "output", "type": "Tensor" }
  1462. ]
  1463. },
  1464. {
  1465. "name": "torch.log_softmax:Dimname",
  1466. "category": "Activation",
  1467. "attributes": [
  1468. { "name": "dim", "type": "Dimname" },
  1469. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  1470. ],
  1471. "inputs": [
  1472. { "name": "input", "type": "Tensor" }
  1473. ],
  1474. "outputs": [
  1475. { "name": "output", "type": "Tensor" }
  1476. ]
  1477. },
  1478. {
  1479. "name": "torch.tanh",
  1480. "category": "Activation",
  1481. "inputs": [
  1482. { "name": "input", "type": "Tensor" }
  1483. ],
  1484. "outputs": [
  1485. { "name": "output", "type": "Tensor" }
  1486. ]
  1487. },
  1488. {
  1489. "name": "torch.leaky_relu",
  1490. "category": "Activation",
  1491. "attributes": [
  1492. { "name": "negative_slope", "type": "Scalar", "default": 0.01 }
  1493. ],
  1494. "inputs": [
  1495. { "name": "input", "type": "Tensor" }
  1496. ],
  1497. "outputs": [
  1498. { "name": "output", "type": "Tensor" }
  1499. ]
  1500. },
  1501. {
  1502. "name": "torch.slice",
  1503. "category": "Tensor",
  1504. "inputs": [
  1505. { "name": "input", "type": "Tensor" },
  1506. { "name": "dim", "type": "int64", "default": 0 },
  1507. { "name": "start", "type": "int64", "default": 0 },
  1508. { "name": "end", "type": "int64", "default": 9223372036854776000 },
  1509. { "name": "step", "type": "int64", "default": 1 }
  1510. ],
  1511. "outputs": [
  1512. { "name": "output", "type": "Tensor" }
  1513. ]
  1514. },
  1515. {
  1516. "name": "torch.dropout",
  1517. "category": "Dropout",
  1518. "attributes": [
  1519. { "name": "p", "type": "float64", "default": 0.5 },
  1520. { "name": "train", "type": "boolean", "visible": false }
  1521. ],
  1522. "inputs": [
  1523. { "name": "input", "type": "Tensor" }
  1524. ],
  1525. "outputs": [
  1526. { "name": "output", "type": "Tensor" }
  1527. ]
  1528. },
  1529. {
  1530. "name": "torch.dropout_",
  1531. "category": "Dropout",
  1532. "attributes": [
  1533. { "name": "p", "type": "float64", "default": 0.5 },
  1534. { "name": "train", "type": "boolean", "visible": false }
  1535. ],
  1536. "inputs": [
  1537. { "name": "input", "type": "Tensor" }
  1538. ],
  1539. "outputs": [
  1540. { "name": "output", "type": "Tensor" }
  1541. ]
  1542. },
  1543. {
  1544. "name": "torch.feature_dropout",
  1545. "category": "Dropout",
  1546. "attributes": [
  1547. { "name": "p", "type": "float64" },
  1548. { "name": "train", "type": "boolean" }
  1549. ],
  1550. "inputs": [
  1551. { "name": "input", "type": "Tensor" }
  1552. ],
  1553. "outputs": [
  1554. { "name": "output", "type": "Tensor" }
  1555. ]
  1556. },
  1557. {
  1558. "name": "torch.feature_dropout_",
  1559. "category": "Dropout",
  1560. "attributes": [
  1561. { "name": "p", "type": "float64" },
  1562. { "name": "train", "type": "boolean" }
  1563. ],
  1564. "inputs": [
  1565. { "name": "input", "type": "Tensor" }
  1566. ],
  1567. "outputs": [
  1568. { "name": "output", "type": "Tensor" }
  1569. ]
  1570. },
  1571. {
  1572. "name": "torch.alpha_dropout",
  1573. "category": "Dropout",
  1574. "attributes": [
  1575. { "name": "p", "type": "float64" },
  1576. { "name": "train", "type": "boolean" }
  1577. ],
  1578. "inputs": [
  1579. { "name": "input", "type": "Tensor" }
  1580. ],
  1581. "outputs": [
  1582. { "name": "output", "type": "Tensor" }
  1583. ]
  1584. },
  1585. {
  1586. "name": "torch.alpha_dropout_",
  1587. "category": "Dropout",
  1588. "attributes": [
  1589. { "name": "p", "type": "float64" },
  1590. { "name": "train", "type": "boolean" }
  1591. ],
  1592. "inputs": [
  1593. { "name": "input", "type": "Tensor" }
  1594. ],
  1595. "outputs": [
  1596. { "name": "output", "type": "Tensor" }
  1597. ]
  1598. },
  1599. {
  1600. "name": "torch.feature_alpha_dropout",
  1601. "category": "Dropout",
  1602. "attributes": [
  1603. { "name": "p", "type": "float64" },
  1604. { "name": "train", "type": "boolean" }
  1605. ],
  1606. "inputs": [
  1607. { "name": "input", "type": "Tensor" }
  1608. ],
  1609. "outputs": [
  1610. { "name": "output", "type": "Tensor" }
  1611. ]
  1612. },
  1613. {
  1614. "name": "torch.feature_alpha_dropout_",
  1615. "category": "Dropout",
  1616. "attributes": [
  1617. { "name": "p", "type": "float64" },
  1618. { "name": "train", "type": "boolean" }
  1619. ],
  1620. "inputs": [
  1621. { "name": "input", "type": "Tensor" }
  1622. ],
  1623. "outputs": [
  1624. { "name": "output", "type": "Tensor" }
  1625. ]
  1626. },
  1627. {
  1628. "name": "torch.mm",
  1629. "inputs": [
  1630. { "name": "self", "type": "Tensor" },
  1631. { "name": "mat2", "type": "Tensor" }
  1632. ],
  1633. "outputs": [
  1634. { "name": "output", "type": "Tensor" }
  1635. ]
  1636. },
  1637. {
  1638. "name": "torch.addmm",
  1639. "category": "Layer",
  1640. "attributes": [
  1641. { "name": "alpha", "type": "Scalar", "default": 1 },
  1642. { "name": "beta", "type": "Scalar", "default": 1 }
  1643. ],
  1644. "inputs": [
  1645. { "name": "self", "type": "Tensor" },
  1646. { "name": "mat1", "type": "Tensor" },
  1647. { "name": "mat2", "type": "Tensor" }
  1648. ],
  1649. "outputs": [
  1650. { "name": "output", "type": "Tensor" }
  1651. ]
  1652. },
  1653. {
  1654. "name": "torch.add_:Tensor",
  1655. "inputs": [
  1656. { "name": "self", "type": "Tensor" },
  1657. { "name": "other", "type": "Tensor" },
  1658. { "name": "alpha", "type": "Scalar", "default": 1 }
  1659. ],
  1660. "outputs": [
  1661. { "name": "output", "type": "Tensor" }
  1662. ]
  1663. },
  1664. {
  1665. "name": "torch.add_:Scalar",
  1666. "inputs": [
  1667. { "name": "self", "type": "Tensor" },
  1668. { "name": "other", "type": "Scalar" },
  1669. { "name": "alpha", "type": "Scalar", "default": 1 }
  1670. ],
  1671. "outputs": [
  1672. { "name": "output", "type": "Tensor" }
  1673. ]
  1674. },
  1675. {
  1676. "name": "torch._add_relu:Tensor",
  1677. "attributes": [
  1678. { "name": "alpha", "type": "Scalar", "default": 1 }
  1679. ],
  1680. "inputs": [
  1681. { "name": "self", "type": "Tensor" },
  1682. { "name": "other", "type": "Tensor" }
  1683. ],
  1684. "outputs": [
  1685. { "name": "output", "type": "Tensor" }
  1686. ]
  1687. },
  1688. {
  1689. "name": "torch._add_relu_:Tensor",
  1690. "attributes": [
  1691. { "name": "alpha", "type": "Scalar", "default": 1 }
  1692. ],
  1693. "inputs": [
  1694. { "name": "self", "type": "Tensor" },
  1695. { "name": "other", "type": "Tensor" }
  1696. ],
  1697. "outputs": [
  1698. { "name": "output", "type": "Tensor" }
  1699. ]
  1700. },
  1701. {
  1702. "name": "torch.sub_:Tensor",
  1703. "inputs": [
  1704. { "name": "input", "type": "Tensor" },
  1705. { "name": "other", "type": "Tensor" },
  1706. { "name": "alpha", "type": "Scalar", "default": 1 }
  1707. ],
  1708. "outputs": [
  1709. { "name": "output", "type": "Tensor" }
  1710. ]
  1711. },
  1712. {
  1713. "name": "torch.sub_:Scalar",
  1714. "inputs": [
  1715. { "name": "input", "type": "Tensor" },
  1716. { "name": "other", "type": "Scalar" },
  1717. { "name": "alpha", "type": "Scalar", "default": 1 }
  1718. ],
  1719. "outputs": [
  1720. { "name": "output", "type": "Tensor" }
  1721. ]
  1722. },
  1723. {
  1724. "name": "torch.pow",
  1725. "attributes": [
  1726. { "name": "exponent", "type": "Scalar" }
  1727. ],
  1728. "inputs": [
  1729. { "name": "self", "type": "Tensor" }
  1730. ],
  1731. "outputs": [
  1732. { "name": "output", "type": "Tensor" }
  1733. ]
  1734. },
  1735. {
  1736. "name": "torch.pow_",
  1737. "attributes": [
  1738. { "name": "exponent", "type": "Tensor" }
  1739. ],
  1740. "inputs": [
  1741. { "name": "self", "type": "Tensor" }
  1742. ],
  1743. "outputs": [
  1744. { "name": "output", "type": "Tensor" }
  1745. ]
  1746. },
  1747. {
  1748. "name": "torch.add:Tensor",
  1749. "attributes": [
  1750. { "name": "alpha", "type": "Scalar", "default": 1 }
  1751. ],
  1752. "inputs": [
  1753. { "name": "input", "type": "Tensor" },
  1754. { "name": "other", "type": "Tensor" }
  1755. ],
  1756. "outputs": [
  1757. { "name": "output", "type": "Tensor" }
  1758. ]
  1759. },
  1760. {
  1761. "name": "torch.add:Scalar",
  1762. "attributes": [
  1763. { "name": "other", "type": "Scalar" },
  1764. { "name": "alpha", "type": "Scalar", "default": 1 }
  1765. ],
  1766. "inputs": [
  1767. { "name": "input", "type": "Tensor" }
  1768. ],
  1769. "outputs": [
  1770. { "name": "output", "type": "Tensor" }
  1771. ]
  1772. },
  1773. {
  1774. "name": "torch.sub:Scalar",
  1775. "inputs": [
  1776. { "name": "input", "type": "Tensor" },
  1777. { "name": "other", "type": "Scalar" },
  1778. { "name": "alpha", "type": "Scalar", "default": 1 }
  1779. ],
  1780. "outputs": [
  1781. { "name": "output", "type": "Tensor" }
  1782. ]
  1783. },
  1784. {
  1785. "name": "torch.rsub:Scalar",
  1786. "attributes": [
  1787. { "name": "other", "type": "Scalar" },
  1788. { "name": "alpha", "type": "Scalar", "default": 1 }
  1789. ],
  1790. "inputs": [
  1791. { "name": "input", "type": "Tensor" }
  1792. ],
  1793. "outputs": [
  1794. { "name": "output", "type": "Tensor" }
  1795. ]
  1796. },
  1797. {
  1798. "name": "torch.sub:Tensor",
  1799. "inputs": [
  1800. { "name": "input", "type": "Tensor" },
  1801. { "name": "other", "type": "Tensor" },
  1802. { "name": "alpha", "type": "Scalar", "default": 1 }
  1803. ],
  1804. "outputs": [
  1805. { "name": "output", "type": "Tensor" }
  1806. ]
  1807. },
  1808. {
  1809. "name": "torch.rsub:Tensor",
  1810. "inputs": [
  1811. { "name": "input", "type": "Tensor" },
  1812. { "name": "other", "type": "Tensor" },
  1813. { "name": "alpha", "type": "Scalar", "default": 1 }
  1814. ],
  1815. "outputs": [
  1816. { "name": "output", "type": "Tensor" }
  1817. ]
  1818. },
  1819. {
  1820. "name": "torch.div:Tensor",
  1821. "inputs": [
  1822. { "name": "input", "type": "Tensor" },
  1823. { "name": "other", "type": "Tensor" }
  1824. ],
  1825. "outputs": [
  1826. { "name": "output", "type": "Tensor" }
  1827. ]
  1828. },
  1829. {
  1830. "name": "torch.div:Scalar",
  1831. "inputs": [
  1832. { "name": "input", "type": "Tensor" },
  1833. { "name": "other", "type": "Scalar" }
  1834. ],
  1835. "outputs": [
  1836. { "name": "output", "type": "Tensor" }
  1837. ]
  1838. },
  1839. {
  1840. "name": "torch.floordiv",
  1841. "inputs": [
  1842. { "name": "input", "type": "Tensor" },
  1843. { "name": "other", "type": "Tensor" }
  1844. ],
  1845. "outputs": [
  1846. { "name": "output", "type": "Tensor" }
  1847. ]
  1848. },
  1849. {
  1850. "name": "torch.sum:",
  1851. "inputs": [
  1852. { "name": "self", "type": "Tensor" }
  1853. ],
  1854. "outputs": [
  1855. { "name": "output", "type": "Tensor" }
  1856. ]
  1857. },
  1858. {
  1859. "name": "torch.sum:dim_IntList",
  1860. "inputs": [
  1861. { "name": "self", "type": "Tensor" },
  1862. { "name": "dim", "type": "int64[]" },
  1863. { "name": "keepdim", "type": "boolean", "default": false },
  1864. { "name": "dtype", "type": "ScalarType", "default": null }
  1865. ],
  1866. "outputs": [
  1867. { "name": "output", "type": "Tensor" }
  1868. ]
  1869. },
  1870. {
  1871. "name": "torch.matmul",
  1872. "inputs": [
  1873. { "name": "self", "type": "Tensor" },
  1874. { "name": "other", "type": "Tensor" }
  1875. ],
  1876. "outputs": [
  1877. { "name": "output", "type": "Tensor" }
  1878. ]
  1879. },
  1880. {
  1881. "name": "torch.transpose",
  1882. "inputs": [
  1883. { "name": "input", "type": "Tensor" },
  1884. { "name": "dim0", "type": "int64" },
  1885. { "name": "dim1", "type": "int64" }
  1886. ],
  1887. "outputs": [
  1888. { "name": "output", "type": "Tensor" }
  1889. ]
  1890. },
  1891. {
  1892. "name": "torch.chunk",
  1893. "attributes": [
  1894. { "name": "chunks", "type": "int64" },
  1895. { "name": "dim", "type": "int64", "default": 0 }
  1896. ],
  1897. "inputs": [
  1898. { "name": "self", "type": "Tensor" }
  1899. ],
  1900. "outputs": [
  1901. { "name": "output", "type": "Tensor[]" }
  1902. ]
  1903. },
  1904. {
  1905. "name": "torch.unsafe_chunk",
  1906. "inputs": [
  1907. { "name": "self", "type": "Tensor" },
  1908. { "name": "chunks", "type": "int64" },
  1909. { "name": "dim", "type": "int64", "default": 0 }
  1910. ],
  1911. "outputs": [
  1912. { "name": "output", "type": "Tensor[]" }
  1913. ]
  1914. },
  1915. {
  1916. "name": "torch.view",
  1917. "inputs": [
  1918. { "name": "input", "type": "Tensor" },
  1919. { "name": "size", "type": "int64[]", "visible": false }
  1920. ],
  1921. "outputs": [
  1922. { "name": "output", "type": "Tensor" }
  1923. ]
  1924. },
  1925. {
  1926. "name": "torch.to:dtype_layout",
  1927. "inputs": [
  1928. { "name": "self", "type": "Tensor" },
  1929. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  1930. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  1931. { "name": "device", "type": "Device", "optional": true, "default": null },
  1932. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  1933. { "name": "non_blocking", "type": "boolean", "default": false },
  1934. { "name": "copy", "type": "boolean", "default": false },
  1935. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  1936. ],
  1937. "outputs": [
  1938. { "name": "output", "type": "Tensor" }
  1939. ]
  1940. },
  1941. {
  1942. "name": "torch.to:device",
  1943. "inputs": [
  1944. { "name": "self", "type": "Tensor" },
  1945. { "name": "device", "type": "Device" },
  1946. { "name": "dtype", "type": "ScalarType" },
  1947. { "name": "non_blocking", "type": "boolean", "default": false },
  1948. { "name": "copy", "type": "boolean", "default": false },
  1949. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  1950. ],
  1951. "outputs": [
  1952. { "name": "output", "type": "Tensor" }
  1953. ]
  1954. },
  1955. {
  1956. "name": "torch.to:dtype",
  1957. "inputs": [
  1958. { "name": "self", "type": "Tensor" },
  1959. { "name": "dtype", "type": "ScalarType" },
  1960. { "name": "non_blocking", "type": "boolean", "default": false },
  1961. { "name": "copy", "type": "boolean", "default": false },
  1962. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  1963. ],
  1964. "outputs": [
  1965. { "name": "output", "type": "Tensor" }
  1966. ]
  1967. },
  1968. {
  1969. "name": "torch.to:other",
  1970. "inputs": [
  1971. { "name": "self", "type": "Tensor" },
  1972. { "name": "other", "type": "Tensor" },
  1973. { "name": "non_blocking", "type": "boolean", "default": false },
  1974. { "name": "copy", "type": "boolean", "default": false },
  1975. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  1976. ],
  1977. "outputs": [
  1978. { "name": "output", "type": "Tensor" }
  1979. ]
  1980. },
  1981. {
  1982. "name": "torch.clone",
  1983. "inputs": [
  1984. { "name": "input", "type": "Tensor" }
  1985. ],
  1986. "outputs": [
  1987. { "name": "output", "type": "Tensor" }
  1988. ]
  1989. },
  1990. {
  1991. "name": "torch.expand",
  1992. "attributes": [
  1993. { "name": "size", "type": "int64[]" },
  1994. { "name": "implicit", "type": "boolean", "default": false }
  1995. ],
  1996. "inputs": [
  1997. { "name": "input", "type": "Tensor" }
  1998. ],
  1999. "outputs": [
  2000. { "name": "output", "type": "Tensor" }
  2001. ]
  2002. },
  2003. {
  2004. "name": "torch.copy_",
  2005. "attributes": [
  2006. { "name": "non_blocking", "type": "boolean", "default": false }
  2007. ],
  2008. "inputs": [
  2009. { "name": "self", "type": "Tensor" },
  2010. { "name": "src", "type": "Tensor" }
  2011. ],
  2012. "outputs": [
  2013. { "name": "output", "type": "Tensor" }
  2014. ]
  2015. },
  2016. {
  2017. "name": "torch.reshape",
  2018. "category": "Shape",
  2019. "attributes": [
  2020. { "name": "shape", "type": "int64[]" }
  2021. ],
  2022. "inputs": [
  2023. { "name": "self", "type": "Tensor" }
  2024. ],
  2025. "outputs": [
  2026. { "name": "output", "type": "Tensor" }
  2027. ]
  2028. },
  2029. {
  2030. "name": "torch.round",
  2031. "inputs": [
  2032. { "name": "self", "type": "Tensor" }
  2033. ],
  2034. "outputs": [
  2035. { "name": "output", "type": "Tensor" }
  2036. ]
  2037. },
  2038. {
  2039. "name": "torch.round_",
  2040. "inputs": [
  2041. { "name": "self", "type": "Tensor" }
  2042. ],
  2043. "outputs": [
  2044. { "name": "output", "type": "Tensor" }
  2045. ]
  2046. },
  2047. {
  2048. "name": "torch.select:int",
  2049. "inputs": [
  2050. { "name": "self", "type": "Tensor" },
  2051. { "name": "dim", "type": "int64" },
  2052. { "name": "index", "type": "int64" }
  2053. ],
  2054. "outputs": [
  2055. { "name": "output", "type": "Tensor" }
  2056. ]
  2057. },
  2058. {
  2059. "name": "torch.select:Dimname",
  2060. "inputs": [
  2061. { "name": "self", "type": "Tensor" },
  2062. { "name": "dim", "type": "Dimname" },
  2063. { "name": "index", "type": "int64" }
  2064. ],
  2065. "outputs": [
  2066. { "name": "output", "type": "Tensor" }
  2067. ]
  2068. },
  2069. {
  2070. "name": "torch.repeat",
  2071. "attributes": [
  2072. { "name": "repeats", "type": "int64[]" }
  2073. ],
  2074. "inputs": [
  2075. { "name": "input", "type": "Tensor" }
  2076. ],
  2077. "outputs": [
  2078. { "name": "output", "type": "Tensor" }
  2079. ]
  2080. },
  2081. {
  2082. "name": "torch.mean:",
  2083. "attributes": [
  2084. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  2085. ],
  2086. "inputs": [
  2087. { "name": "self", "type": "Tensor" }
  2088. ],
  2089. "outputs": [
  2090. { "name": "output", "type": "Tensor" }
  2091. ]
  2092. },
  2093. {
  2094. "name": "torch.mean:dim",
  2095. "attributes": [
  2096. { "name": "dim", "type": "int64[]" },
  2097. { "name": "keepdim", "type": "boolean", "default": false },
  2098. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  2099. ],
  2100. "inputs": [
  2101. { "name": "self", "type": "Tensor" }
  2102. ],
  2103. "outputs": [
  2104. { "name": "output", "type": "Tensor" }
  2105. ]
  2106. },
  2107. {
  2108. "name": "torch.median:",
  2109. "inputs": [
  2110. { "name": "self", "type": "Tensor" }
  2111. ],
  2112. "outputs": [
  2113. { "name": "output", "type": "Tensor" }
  2114. ]
  2115. },
  2116. {
  2117. "name": "torch.median:dim",
  2118. "attributes": [
  2119. { "name": "dim", "type": "int64" },
  2120. { "name": "keepdim", "type": "boolean", "default": false }
  2121. ],
  2122. "inputs": [
  2123. { "name": "self", "type": "Tensor" }
  2124. ],
  2125. "outputs": [
  2126. { "name": "values", "type": "Tensor" },
  2127. { "name": "indices", "type": "Tensor" }
  2128. ]
  2129. },
  2130. {
  2131. "name": "torch.lstm:input",
  2132. "category": "Layer",
  2133. "attributes": [
  2134. { "name": "has_biases", "type": "boolean" },
  2135. { "name": "num_layers", "type": "int64" },
  2136. { "name": "dropout", "type": "float64" },
  2137. { "name": "train", "type": "boolean" },
  2138. { "name": "bidirectional", "type": "boolean" },
  2139. { "name": "batch_first", "type": "boolean" }
  2140. ],
  2141. "inputs": [
  2142. { "name": "input", "type": "Tensor" },
  2143. { "name": "hx", "type": "Tensor[]" },
  2144. { "name": "params", "type": "Tensor[]" }
  2145. ],
  2146. "outputs": [
  2147. { "name": "output", "type": "Tensor" },
  2148. { "name": "?", "type": "Tensor" },
  2149. { "name": "?", "type": "Tensor" }
  2150. ]
  2151. },
  2152. {
  2153. "name": "torch.lstm:data",
  2154. "category": "Layer",
  2155. "attributes": [
  2156. { "name": "has_biases", "type": "boolean" },
  2157. { "name": "num_layers", "type": "int64" },
  2158. { "name": "dropout", "type": "float64" },
  2159. { "name": "train", "type": "boolean" },
  2160. { "name": "bidirectional", "type": "boolean" }
  2161. ],
  2162. "inputs": [
  2163. { "name": "data", "type": "Tensor" },
  2164. { "name": "batch_sizes", "type": "Tensor" },
  2165. { "name": "hx", "type": "Tensor[]" },
  2166. { "name": "params", "type": "Tensor[]" }
  2167. ],
  2168. "outputs": [
  2169. { "name": "output", "type": "Tensor" },
  2170. { "name": "?", "type": "Tensor" },
  2171. { "name": "?", "type": "Tensor" }
  2172. ]
  2173. },
  2174. {
  2175. "name": "torch.gru:input",
  2176. "category": "Layer",
  2177. "attributes": [
  2178. { "name": "has_biases", "type": "boolean" },
  2179. { "name": "num_layers", "type": "int64" },
  2180. { "name": "dropout", "type": "float64" },
  2181. { "name": "train", "type": "boolean" },
  2182. { "name": "bidirectional", "type": "boolean" },
  2183. { "name": "first_batch", "type": "boolean" }
  2184. ],
  2185. "inputs": [
  2186. { "name": "input", "type": "Tensor" },
  2187. { "name": "hx", "type": "Tensor" },
  2188. { "name": "params", "type": "Tensor[]" }
  2189. ],
  2190. "outputs": [
  2191. { "name": "output", "type": "Tensor" },
  2192. { "name": "?", "type": "Tensor" }
  2193. ]
  2194. },
  2195. {
  2196. "name": "torch.gru:data",
  2197. "category": "Layer",
  2198. "attributes": [
  2199. { "name": "has_biases", "type": "boolean" },
  2200. { "name": "num_layers", "type": "int64" },
  2201. { "name": "dropout", "type": "float64" },
  2202. { "name": "train", "type": "boolean" },
  2203. { "name": "bidirectional", "type": "boolean" }
  2204. ],
  2205. "inputs": [
  2206. { "name": "data", "type": "Tensor" },
  2207. { "name": "batch_sizes", "type": "Tensor" },
  2208. { "name": "hx", "type": "Tensor" },
  2209. { "name": "params", "type": "Tensor[]" }
  2210. ],
  2211. "outputs": [
  2212. { "name": "output", "type": "Tensor" },
  2213. { "name": "?", "type": "Tensor" }
  2214. ]
  2215. },
  2216. {
  2217. "name": "torch.contiguous",
  2218. "attributes": [
  2219. { "name": "memory_format", "type": "MemoryFormat", "default": null }
  2220. ],
  2221. "inputs": [
  2222. { "name": "input", "type": "Tensor" }
  2223. ],
  2224. "outputs": [
  2225. { "name": "output", "type": "Tensor" }
  2226. ]
  2227. },
  2228. {
  2229. "name": "torch.squeeze:",
  2230. "category": "Transform",
  2231. "inputs": [
  2232. { "name": "self", "type": "Tensor" }
  2233. ],
  2234. "outputs": [
  2235. { "name": "output", "type": "Tensor" }
  2236. ]
  2237. },
  2238. {
  2239. "name": "torch.squeeze:dim",
  2240. "category": "Transform",
  2241. "inputs": [
  2242. { "name": "self", "type": "Tensor" },
  2243. { "name": "dim", "type": "int64" }
  2244. ],
  2245. "outputs": [
  2246. { "name": "output", "type": "Tensor" }
  2247. ]
  2248. },
  2249. {
  2250. "name": "torch.squeeze:dimname",
  2251. "category": "Transform",
  2252. "inputs": [
  2253. { "name": "self", "type": "Tensor" },
  2254. { "name": "dim", "type": "Dimname" }
  2255. ],
  2256. "outputs": [
  2257. { "name": "output", "type": "Tensor" }
  2258. ]
  2259. },
  2260. {
  2261. "name": "torch.squeeze_:",
  2262. "category": "Transform",
  2263. "inputs": [
  2264. { "name": "self", "type": "Tensor" }
  2265. ],
  2266. "outputs": [
  2267. { "name": "output", "type": "Tensor" }
  2268. ]
  2269. },
  2270. {
  2271. "name": "torch.squeeze_:dim",
  2272. "category": "Transform",
  2273. "inputs": [
  2274. { "name": "self", "type": "Tensor" },
  2275. { "name": "dim", "type": "int64" }
  2276. ],
  2277. "outputs": [
  2278. { "name": "output", "type": "Tensor" }
  2279. ]
  2280. },
  2281. {
  2282. "name": "torch.squeeze_:dimname",
  2283. "category": "Transform",
  2284. "inputs": [
  2285. { "name": "self", "type": "Tensor" },
  2286. { "name": "dim", "type": "Dimname" }
  2287. ],
  2288. "outputs": [
  2289. { "name": "output", "type": "Tensor" }
  2290. ]
  2291. },
  2292. {
  2293. "name": "torch.unsqueeze",
  2294. "category": "Transform",
  2295. "inputs": [
  2296. { "name": "self", "type": "Tensor" },
  2297. { "name": "dim", "type": "int64" }
  2298. ],
  2299. "outputs": [
  2300. { "name": "output", "type": "Tensor" }
  2301. ]
  2302. },
  2303. {
  2304. "name": "torch.unsqueeze_",
  2305. "category": "Transform",
  2306. "inputs": [
  2307. { "name": "self", "type": "Tensor" },
  2308. { "name": "dim", "type": "int64" }
  2309. ],
  2310. "outputs": [
  2311. { "name": "output", "type": "Tensor" }
  2312. ]
  2313. },
  2314. {
  2315. "name": "torch.max:",
  2316. "inputs": [
  2317. { "name": "self", "type": "Tensor" }
  2318. ],
  2319. "outputs": [
  2320. { "name": "output", "type": "Tensor" }
  2321. ]
  2322. },
  2323. {
  2324. "name": "torch.max:dim",
  2325. "attributes": [
  2326. { "name": "dim", "type": "int64", "default": 0 },
  2327. { "name": "keepdim", "type": "boolean", "default": false }
  2328. ],
  2329. "inputs": [
  2330. { "name": "self", "type": "Tensor" }
  2331. ],
  2332. "outputs": [
  2333. { "name": "values", "type": "Tensor" },
  2334. { "name": "indices", "type": "Tensor" }
  2335. ]
  2336. },
  2337. {
  2338. "name": "torch.max:other",
  2339. "inputs": [
  2340. { "name": "self", "type": "Tensor" },
  2341. { "name": "other", "type": "Tensor" }
  2342. ],
  2343. "outputs": [
  2344. { "name": "output", "type": "Tensor" }
  2345. ]
  2346. },
  2347. {
  2348. "name": "torch.maximum",
  2349. "inputs": [
  2350. { "name": "self", "type": "Tensor" },
  2351. { "name": "other", "type": "Tensor" }
  2352. ],
  2353. "outputs": [
  2354. { "name": "output", "type": "Tensor" }
  2355. ]
  2356. },
  2357. {
  2358. "name": "torch.min:",
  2359. "inputs": [
  2360. { "name": "self", "type": "Tensor" }
  2361. ],
  2362. "outputs": [
  2363. { "name": "output", "type": "Tensor" }
  2364. ]
  2365. },
  2366. {
  2367. "name": "torch.min:dim",
  2368. "attributes": [
  2369. { "name": "dim", "type": "int64", "default": 0 },
  2370. { "name": "keepdim", "type": "boolean", "default": false }
  2371. ],
  2372. "inputs": [
  2373. { "name": "self", "type": "Tensor" }
  2374. ],
  2375. "outputs": [
  2376. { "name": "values", "type": "Tensor" },
  2377. { "name": "indices", "type": "Tensor" }
  2378. ]
  2379. },
  2380. {
  2381. "name": "torch.min:other",
  2382. "inputs": [
  2383. { "name": "self", "type": "Tensor" },
  2384. { "name": "other", "type": "Tensor" }
  2385. ],
  2386. "outputs": [
  2387. { "name": "output", "type": "Tensor" }
  2388. ]
  2389. },
  2390. {
  2391. "name": "torch.minimum",
  2392. "inputs": [
  2393. { "name": "self", "type": "Tensor" },
  2394. { "name": "other", "type": "Tensor" }
  2395. ],
  2396. "outputs": [
  2397. { "name": "output", "type": "Tensor" }
  2398. ]
  2399. },
  2400. {
  2401. "name": "torch.cat:",
  2402. "category": "Tensor",
  2403. "inputs": [
  2404. { "name": "tensors", "type": "Tensor[]" },
  2405. { "name": "dim", "type": "int64", "default": 0 }
  2406. ],
  2407. "outputs": [
  2408. { "name": "output", "type": "Tensor" }
  2409. ]
  2410. },
  2411. {
  2412. "name": "torch.cat:names",
  2413. "category": "Tensor",
  2414. "inputs": [
  2415. { "name": "tensors", "type": "Tensor[]" },
  2416. { "name": "dim", "type": "Dimname" }
  2417. ],
  2418. "outputs": [
  2419. { "name": "output", "type": "Tensor" }
  2420. ]
  2421. },
  2422. {
  2423. "name": "torch.broadcast_tensors",
  2424. "inputs": [
  2425. { "name": "tensors", "type": "Tensor[]" }
  2426. ],
  2427. "outputs": [
  2428. { "name": "output", "type": "Tensor[]" }
  2429. ]
  2430. },
  2431. {
  2432. "name": "torch.meshgrid",
  2433. "category": "Tensor",
  2434. "inputs": [
  2435. { "name": "tensors", "type": "Tensor[]" }
  2436. ],
  2437. "outputs": [
  2438. { "name": "output", "type": "Tensor[]" }
  2439. ]
  2440. },
  2441. {
  2442. "name": "torch.stack",
  2443. "category": "Tensor",
  2444. "attributes": [
  2445. { "name": "dim", "type": "int64", "default": 0 }
  2446. ],
  2447. "inputs": [
  2448. { "name": "inputs", "type": "Tensor[]" }
  2449. ],
  2450. "outputs": [
  2451. { "name": "output", "type": "Tensor" }
  2452. ]
  2453. },
  2454. {
  2455. "name": "torch.expand_as",
  2456. "inputs": [
  2457. { "name": "inputs", "type": "Tensor" },
  2458. { "name": "other", "type": "Tensor" }
  2459. ],
  2460. "outputs": [
  2461. { "name": "output", "type": "Tensor" }
  2462. ]
  2463. },
  2464. {
  2465. "name": "torch.permute",
  2466. "category": "Shape",
  2467. "attributes": [
  2468. { "name": "dims", "type": "int64[]" }
  2469. ],
  2470. "inputs": [
  2471. { "name": "inputs", "type": "Tensor" }
  2472. ],
  2473. "outputs": [
  2474. { "name": "output", "type": "Tensor" }
  2475. ]
  2476. },
  2477. {
  2478. "name": "torch.embedding",
  2479. "category": "Transform",
  2480. "attributes": [
  2481. { "name": "padding_idx", "type": "int64" },
  2482. { "name": "scale_grad_by_freq", "type": "boolean", "default": false },
  2483. { "name": "sparse", "type": "boolean", "default": false }
  2484. ],
  2485. "inputs": [
  2486. { "name": "inputs", "type": "Tensor" },
  2487. { "name": "weight", "type": "Tensor" }
  2488. ],
  2489. "outputs": [
  2490. { "name": "output", "type": "Tensor" }
  2491. ]
  2492. },
  2493. {
  2494. "name": "torch.embedding_bag",
  2495. "category": "Transform",
  2496. "inputs": [
  2497. { "name": "weight", "type": "Tensor" },
  2498. { "name": "indices", "type": "Tensor" },
  2499. { "name": "offsets", "type": "Tensor" },
  2500. { "name": "scale_grad_by_freq", "type": "boolean", "default": false },
  2501. { "name": "mode", "type": "int64", "default": 0 },
  2502. { "name": "sparse", "type": "boolean", "default": false },
  2503. { "name": "per_sample_weights", "type": "Tensor", "optional": true, "default": null },
  2504. { "name": "include_last_offset", "type": "boolean", "default": false }
  2505. ],
  2506. "outputs": [
  2507. { "name": "output1", "type": "Tensor" },
  2508. { "name": "output2", "type": "Tensor" },
  2509. { "name": "output3", "type": "Tensor" },
  2510. { "name": "output4", "type": "Tensor" }
  2511. ]
  2512. },
  2513. {
  2514. "name": "torch.flatten",
  2515. "category": "Shape",
  2516. "attributes": [
  2517. { "name": "start_dim", "type": "int64" },
  2518. { "name": "end_dim", "type": "int64" }
  2519. ],
  2520. "inputs": [
  2521. { "name": "self", "type": "Tensor" }
  2522. ],
  2523. "outputs": [
  2524. { "name": "output", "type": "Tensor" }
  2525. ]
  2526. },
  2527. {
  2528. "name": "torch._pack_padded_sequence",
  2529. "attributes": [
  2530. { "name": "batch_first", "type": "boolean" }
  2531. ],
  2532. "inputs": [
  2533. { "name": "input", "type": "Tensor" },
  2534. { "name": "lengths", "type": "Tensor" }
  2535. ],
  2536. "outputs": [
  2537. { "name": "output", "type": "Tensor" },
  2538. { "name": "?", "type": "Tensor" }
  2539. ]
  2540. },
  2541. {
  2542. "name": "torch._pad_packed_sequence",
  2543. "attributes": [
  2544. { "name": "batch_first", "type": "boolean" },
  2545. { "name": "padding_value", "type": "Scalar" },
  2546. { "name": "total_length", "type": "int64" }
  2547. ],
  2548. "inputs": [
  2549. { "name": "data", "type": "Tensor" },
  2550. { "name": "batch_sizes", "type": "Tensor" }
  2551. ],
  2552. "outputs": [
  2553. { "name": "output", "type": "Tensor" },
  2554. { "name": "?", "type": "Tensor" }
  2555. ]
  2556. },
  2557. {
  2558. "name": "torch.quantize_per_tensor:",
  2559. "inputs": [
  2560. { "name": "self", "type": "Tensor" },
  2561. { "name": "scale", "type": "float32" },
  2562. { "name": "zero_point", "type": "int64" },
  2563. { "name": "dtype", "type": "ScalarType" }
  2564. ],
  2565. "outputs": [
  2566. { "name": "output", "type": "Tensor" }
  2567. ]
  2568. },
  2569. {
  2570. "name": "torch.quantize_per_tensor:qparams",
  2571. "inputs": [
  2572. { "name": "self", "type": "Tensor" },
  2573. { "name": "scale", "type": "Tensor" },
  2574. { "name": "zero_point", "type": "Tensor" },
  2575. { "name": "dtype", "type": "ScalarType" }
  2576. ],
  2577. "outputs": [
  2578. { "name": "output", "type": "Tensor" }
  2579. ]
  2580. },
  2581. {
  2582. "name": "torch.quantize_per_tensor:tensors",
  2583. "inputs": [
  2584. { "name": "tensors", "type": "Tensor[]" },
  2585. { "name": "scales", "type": "Tensor" },
  2586. { "name": "zero_points", "type": "Tensor" },
  2587. { "name": "dtype", "type": "ScalarType" }
  2588. ],
  2589. "outputs": [
  2590. { "name": "outputs", "type": "Tensor[]" }
  2591. ]
  2592. },
  2593. {
  2594. "name": "torch.floor_divide",
  2595. "inputs": [
  2596. { "name": "self", "type": "Tensor" },
  2597. { "name": "other", "type": "Tensor" }
  2598. ],
  2599. "outputs": [
  2600. { "name": "output", "type": "Tensor" }
  2601. ]
  2602. },
  2603. {
  2604. "name": "torch.index",
  2605. "inputs": [
  2606. { "name": "self", "type": "Tensor" },
  2607. { "name": "indices", "type": "Tensor[]", "optional": true }
  2608. ],
  2609. "outputs": [
  2610. { "name": "output", "type": "Tensor" }
  2611. ]
  2612. },
  2613. {
  2614. "name": "torch._cast_Byte",
  2615. "attributes": [
  2616. { "name": "non_blocking", "type": "boolean", "default": false }
  2617. ],
  2618. "inputs": [
  2619. { "name": "input", "type": "Tensor" }
  2620. ],
  2621. "outputs": [
  2622. { "name": "output", "type": "Tensor" }
  2623. ]
  2624. },
  2625. {
  2626. "name": "torch._cast_Char",
  2627. "attributes": [
  2628. { "name": "non_blocking", "type": "boolean", "default": false }
  2629. ],
  2630. "inputs": [
  2631. { "name": "input", "type": "Tensor" }
  2632. ],
  2633. "outputs": [
  2634. { "name": "output", "type": "Tensor" }
  2635. ]
  2636. },
  2637. {
  2638. "name": "torch._cast_Short",
  2639. "attributes": [
  2640. { "name": "non_blocking", "type": "boolean", "default": false }
  2641. ],
  2642. "inputs": [
  2643. { "name": "input", "type": "Tensor" }
  2644. ],
  2645. "outputs": [
  2646. { "name": "output", "type": "Tensor" }
  2647. ]
  2648. },
  2649. {
  2650. "name": "torch._cast_Int",
  2651. "attributes": [
  2652. { "name": "non_blocking", "type": "boolean", "default": false }
  2653. ],
  2654. "inputs": [
  2655. { "name": "input", "type": "Tensor" }
  2656. ],
  2657. "outputs": [
  2658. { "name": "output", "type": "Tensor" }
  2659. ]
  2660. },
  2661. {
  2662. "name": "torch._cast_Long",
  2663. "attributes": [
  2664. { "name": "non_blocking", "type": "boolean", "default": false }
  2665. ],
  2666. "inputs": [
  2667. { "name": "input", "type": "Tensor" }
  2668. ],
  2669. "outputs": [
  2670. { "name": "output", "type": "Tensor" }
  2671. ]
  2672. },
  2673. {
  2674. "name": "torch._cast_Half",
  2675. "attributes": [
  2676. { "name": "non_blocking", "type": "boolean", "default": false }
  2677. ],
  2678. "inputs": [
  2679. { "name": "input", "type": "Tensor" }
  2680. ],
  2681. "outputs": [
  2682. { "name": "output", "type": "Tensor" }
  2683. ]
  2684. },
  2685. {
  2686. "name": "torch._cast_Float",
  2687. "attributes": [
  2688. { "name": "non_blocking", "type": "boolean", "default": false }
  2689. ],
  2690. "inputs": [
  2691. { "name": "input", "type": "Tensor" }
  2692. ],
  2693. "outputs": [
  2694. { "name": "output", "type": "Tensor" }
  2695. ]
  2696. },
  2697. {
  2698. "name": "torch._cast_Double",
  2699. "attributes": [
  2700. { "name": "non_blocking", "type": "boolean", "default": false }
  2701. ],
  2702. "inputs": [
  2703. { "name": "input", "type": "Tensor" }
  2704. ],
  2705. "outputs": [
  2706. { "name": "output", "type": "Tensor" }
  2707. ]
  2708. },
  2709. {
  2710. "name": "torch.frobenius_norm:",
  2711. "category": "Normalization",
  2712. "inputs": [
  2713. { "name": "self", "type": "Tensor" }
  2714. ],
  2715. "outputs": [
  2716. { "name": "output", "type": "Tensor" }
  2717. ]
  2718. },
  2719. {
  2720. "name": "torch.frobenius_norm:dim",
  2721. "category": "Normalization",
  2722. "attributes": [
  2723. { "name": "dim", "type": "int64[]" },
  2724. { "name": "keepdim", "type": "boolean", "default": false }
  2725. ],
  2726. "inputs": [
  2727. { "name": "self", "type": "Tensor" }
  2728. ],
  2729. "outputs": [
  2730. { "name": "output", "type": "Tensor" }
  2731. ]
  2732. },
  2733. {
  2734. "name": "torch.norm:ScalarOpt_dtype",
  2735. "attributes": [
  2736. { "name": "p", "type": "Scalar", "optional": true },
  2737. { "name": "dtype", "type": "ScalarType" }
  2738. ],
  2739. "inputs": [
  2740. { "name": "self", "type": "Tensor" }
  2741. ],
  2742. "outputs": [
  2743. { "name": "output", "type": "Tensor" }
  2744. ]
  2745. },
  2746. {
  2747. "name": "torch.norm:ScalarOpt_dim_dtype",
  2748. "attributes": [
  2749. { "name": "p", "type": "Scalar", "optional": true },
  2750. { "name": "dim", "type": "int64[1]" },
  2751. { "name": "keepdim", "type": "boolean" },
  2752. { "name": "dtype", "type": "ScalarType" }
  2753. ],
  2754. "inputs": [
  2755. { "name": "self", "type": "Tensor" }
  2756. ],
  2757. "outputs": [
  2758. { "name": "output", "type": "Tensor" }
  2759. ]
  2760. },
  2761. {
  2762. "name": "torch.norm:ScalarOpt_dim",
  2763. "attributes": [
  2764. { "name": "p", "type": "Scalar", "optional": true },
  2765. { "name": "dim", "type": "int64[1]" },
  2766. { "name": "keepdim", "type": "boolean", "default": false }
  2767. ],
  2768. "inputs": [
  2769. { "name": "self", "type": "Tensor" }
  2770. ],
  2771. "outputs": [
  2772. { "name": "output", "type": "Tensor" }
  2773. ]
  2774. },
  2775. {
  2776. "name": "torch.norm:dtype_out",
  2777. "attributes": [
  2778. { "name": "p", "type": "Scalar", "optional": true },
  2779. { "name": "dim", "type": "int64[1]" },
  2780. { "name": "keepdim", "type": "boolean", "default": false },
  2781. { "name": "dtype", "type": "ScalarType" }
  2782. ],
  2783. "inputs": [
  2784. { "name": "self", "type": "Tensor" }
  2785. ],
  2786. "outputs": [
  2787. { "name": "output", "type": "Tensor" }
  2788. ]
  2789. },
  2790. {
  2791. "name": "torch.norm:Scalar",
  2792. "attributes": [
  2793. { "name": "p", "type": "Scalar", "default": 2 }
  2794. ],
  2795. "inputs": [
  2796. { "name": "self", "type": "Tensor" }
  2797. ],
  2798. "outputs": [
  2799. { "name": "output", "type": "Tensor" }
  2800. ]
  2801. },
  2802. {
  2803. "name": "torch.glu",
  2804. "category": "Activation",
  2805. "attributes": [
  2806. { "name": "dim", "type": "int64", "default": -1 }
  2807. ],
  2808. "inputs": [
  2809. { "name": "self", "type": "Tensor" }
  2810. ],
  2811. "outputs": [
  2812. { "name": "output", "type": "Tensor" }
  2813. ]
  2814. },
  2815. {
  2816. "name": "torch.gelu",
  2817. "category": "Activation",
  2818. "inputs": [
  2819. { "name": "self", "type": "Tensor" }
  2820. ],
  2821. "outputs": [
  2822. { "name": "output", "type": "Tensor" }
  2823. ]
  2824. },
  2825. {
  2826. "name": "torch.celu",
  2827. "category": "Activation",
  2828. "attributes": [
  2829. { "name": "alpha", "type": "Scalar", "default": 1 }
  2830. ],
  2831. "inputs": [
  2832. { "name": "self", "type": "Tensor" }
  2833. ],
  2834. "outputs": [
  2835. { "name": "output", "type": "Tensor" }
  2836. ]
  2837. },
  2838. {
  2839. "name": "torch.elu",
  2840. "category": "Activation",
  2841. "attributes": [
  2842. { "name": "alpha", "type": "Scalar", "default": 1 },
  2843. { "name": "scale", "type": "Scalar", "default": 1 },
  2844. { "name": "input_scale", "type": "Scalar", "default": 1 }
  2845. ],
  2846. "inputs": [
  2847. { "name": "self", "type": "Tensor" }
  2848. ],
  2849. "outputs": [
  2850. { "name": "output", "type": "Tensor" }
  2851. ]
  2852. },
  2853. {
  2854. "name": "torch.elu_",
  2855. "category": "Activation",
  2856. "attributes": [
  2857. { "name": "alpha", "type": "Scalar", "default": 1 },
  2858. { "name": "scale", "type": "Scalar", "default": 1 },
  2859. { "name": "input_scale", "type": "Scalar", "default": 1 }
  2860. ],
  2861. "inputs": [
  2862. { "name": "self", "type": "Tensor" }
  2863. ],
  2864. "outputs": [
  2865. { "name": "output", "type": "Tensor" }
  2866. ]
  2867. },
  2868. {
  2869. "name": "torch.selu",
  2870. "category": "Activation",
  2871. "inputs": [
  2872. { "name": "self", "type": "Tensor" }
  2873. ],
  2874. "outputs": [
  2875. { "name": "output", "type": "Tensor" }
  2876. ]
  2877. },
  2878. {
  2879. "name": "torch.selu_",
  2880. "category": "Activation",
  2881. "inputs": [
  2882. { "name": "self", "type": "Tensor" }
  2883. ],
  2884. "outputs": [
  2885. { "name": "output", "type": "Tensor" }
  2886. ]
  2887. },
  2888. {
  2889. "name": "torch.silu",
  2890. "category": "Activation",
  2891. "inputs": [
  2892. { "name": "self", "type": "Tensor" }
  2893. ],
  2894. "outputs": [
  2895. { "name": "output", "type": "Tensor" }
  2896. ]
  2897. },
  2898. {
  2899. "name": "torch.silu_",
  2900. "category": "Activation",
  2901. "inputs": [
  2902. { "name": "self", "type": "Tensor" }
  2903. ],
  2904. "outputs": [
  2905. { "name": "output", "type": "Tensor" }
  2906. ]
  2907. },
  2908. {
  2909. "name": "torch.hardswish",
  2910. "category": "Activation",
  2911. "inputs": [
  2912. { "name": "self", "type": "Tensor" }
  2913. ],
  2914. "outputs": [
  2915. { "name": "output", "type": "Tensor" }
  2916. ]
  2917. },
  2918. {
  2919. "name": "torch.hardswish_",
  2920. "category": "Activation",
  2921. "inputs": [
  2922. { "name": "self", "type": "Tensor" }
  2923. ],
  2924. "outputs": [
  2925. { "name": "output", "type": "Tensor" }
  2926. ]
  2927. },
  2928. {
  2929. "name": "torch.pixel_shuffle",
  2930. "attributes": [
  2931. { "name": "upscale_factor", "type": "int64" }
  2932. ],
  2933. "inputs": [
  2934. { "name": "self", "type": "Tensor" }
  2935. ],
  2936. "outputs": [
  2937. { "name": "output", "type": "Tensor" }
  2938. ]
  2939. },
  2940. {
  2941. "name": "torch.nonzero",
  2942. "inputs": [
  2943. { "name": "self", "type": "Tensor" }
  2944. ],
  2945. "outputs": [
  2946. { "name": "output", "type": "Tensor" }
  2947. ]
  2948. },
  2949. {
  2950. "name": "torch.neg",
  2951. "inputs": [
  2952. { "name": "self", "type": "Tensor" }
  2953. ],
  2954. "outputs": [
  2955. { "name": "output", "type": "Tensor" }
  2956. ]
  2957. },
  2958. {
  2959. "name": "torch.bitwise_not",
  2960. "inputs": [
  2961. { "name": "self", "type": "Tensor" }
  2962. ],
  2963. "outputs": [
  2964. { "name": "output", "type": "Tensor" }
  2965. ]
  2966. },
  2967. {
  2968. "name": "torch.bitwise_not_",
  2969. "inputs": [
  2970. { "name": "self", "type": "Tensor" }
  2971. ],
  2972. "outputs": [
  2973. { "name": "output", "type": "Tensor" }
  2974. ]
  2975. },
  2976. {
  2977. "name": "torch.rsqrt",
  2978. "inputs": [
  2979. { "name": "self", "type": "Tensor" }
  2980. ],
  2981. "outputs": [
  2982. { "name": "output", "type": "Tensor" }
  2983. ]
  2984. },
  2985. {
  2986. "name": "torch.grid_sampler",
  2987. "attributes": [
  2988. { "name": "interpolation_mode", "type": "int64" },
  2989. { "name": "padding_mode", "type": "int64" },
  2990. { "name": "align_corners", "type": "boolean" }
  2991. ],
  2992. "inputs": [
  2993. { "name": "input", "type": "Tensor" },
  2994. { "name": "grid", "type": "Tensor" }
  2995. ],
  2996. "outputs": [
  2997. { "name": "output", "type": "Tensor" }
  2998. ]
  2999. },
  3000. {
  3001. "name": "torch.sort",
  3002. "attributes": [
  3003. { "name": "dim", "type": "int64", "default": -1 },
  3004. { "name": "descending", "type": "boolean", "default": false }
  3005. ],
  3006. "inputs": [
  3007. { "name": "self", "type": "Tensor" }
  3008. ],
  3009. "outputs": [
  3010. { "name": "values", "type": "Tensor" },
  3011. { "name": "indices", "type": "Tensor" }
  3012. ]
  3013. },
  3014. {
  3015. "name": "torch.gt:Tensor",
  3016. "inputs": [
  3017. { "name": "input", "type": "Tensor" },
  3018. { "name": "other", "type": "Tensor" }
  3019. ],
  3020. "outputs": [
  3021. { "name": "output", "type": "Tensor" }
  3022. ]
  3023. },
  3024. {
  3025. "name": "torch.gt:Scalar",
  3026. "attributes": [
  3027. { "name": "other", "type": "Scalar" }
  3028. ],
  3029. "inputs": [
  3030. { "name": "input", "type": "Tensor" }
  3031. ],
  3032. "outputs": [
  3033. { "name": "output", "type": "Tensor" }
  3034. ]
  3035. },
  3036. {
  3037. "name": "torch.mul:Tensor",
  3038. "inputs": [
  3039. { "name": "input", "type": "Tensor" },
  3040. { "name": "other", "type": "Tensor" }
  3041. ],
  3042. "outputs": [
  3043. { "name": "output", "type": "Tensor" }
  3044. ]
  3045. },
  3046. {
  3047. "name": "torch.mul:Scalar",
  3048. "attributes": [
  3049. { "name": "other", "type": "Scalar" }
  3050. ],
  3051. "inputs": [
  3052. { "name": "input", "type": "Tensor" }
  3053. ],
  3054. "outputs": [
  3055. { "name": "output", "type": "Tensor" }
  3056. ]
  3057. },
  3058. {
  3059. "name": "torch.mul_:Tensor",
  3060. "inputs": [
  3061. { "name": "input", "type": "Tensor" },
  3062. { "name": "other", "type": "Tensor" }
  3063. ],
  3064. "outputs": [
  3065. { "name": "output", "type": "Tensor" }
  3066. ]
  3067. },
  3068. {
  3069. "name": "torch.mul_:Scalar",
  3070. "attributes": [
  3071. { "name": "other", "type": "Scalar" }
  3072. ],
  3073. "inputs": [
  3074. { "name": "input", "type": "Tensor" }
  3075. ],
  3076. "outputs": [
  3077. { "name": "output", "type": "Tensor" }
  3078. ]
  3079. },
  3080. {
  3081. "name": "torch.div_:Tensor",
  3082. "inputs": [
  3083. { "name": "input", "type": "Tensor" },
  3084. { "name": "other", "type": "Tensor" }
  3085. ],
  3086. "outputs": [
  3087. { "name": "output", "type": "Tensor" }
  3088. ]
  3089. },
  3090. {
  3091. "name": "torch.div_:Scalar",
  3092. "attributes": [
  3093. { "name": "other", "type": "Scalar" }
  3094. ],
  3095. "inputs": [
  3096. { "name": "input", "type": "Tensor" }
  3097. ],
  3098. "outputs": [
  3099. { "name": "output", "type": "Tensor" }
  3100. ]
  3101. },
  3102. {
  3103. "name": "torch.bmm",
  3104. "inputs": [
  3105. { "name": "input", "type": "Tensor" },
  3106. { "name": "mat2", "type": "Tensor" }
  3107. ],
  3108. "outputs": [
  3109. { "name": "output", "type": "Tensor" }
  3110. ]
  3111. },
  3112. {
  3113. "name": "torch.constant_pad_nd",
  3114. "category": "Tensor",
  3115. "attributes": [
  3116. { "name": "pad", "type": "int64[]" },
  3117. { "name": "value", "type": "Scalar", "default": 0 }
  3118. ],
  3119. "inputs": [
  3120. { "name": "input", "type": "Tensor" }
  3121. ],
  3122. "outputs": [
  3123. { "name": "output", "type": "Tensor" }
  3124. ]
  3125. },
  3126. {
  3127. "name": "torch.ones",
  3128. "attributes": [
  3129. { "name": "size", "type": "int64[]" },
  3130. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3131. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3132. { "name": "device", "type": "Device", "optional": true, "default": null },
  3133. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3134. ],
  3135. "outputs": [
  3136. { "name": "output", "type": "Tensor" }
  3137. ]
  3138. },
  3139. {
  3140. "name": "torch.empty:names",
  3141. "attributes": [
  3142. { "name": "size", "type": "int64[]" },
  3143. { "name": "names", "type": "Dimname[]", "optional": true, "default": null },
  3144. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3145. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3146. { "name": "device", "type": "Device", "optional": true, "default": null },
  3147. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3148. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3149. ],
  3150. "outputs": [
  3151. { "name": "output", "type": "Tensor" }
  3152. ]
  3153. },
  3154. {
  3155. "name": "torch.empty:memory_format",
  3156. "attributes": [
  3157. { "name": "size", "type": "int64[]" },
  3158. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3159. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3160. { "name": "device", "type": "Device", "optional": true, "default": null },
  3161. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3162. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3163. ],
  3164. "outputs": [
  3165. { "name": "output", "type": "Tensor" }
  3166. ]
  3167. },
  3168. {
  3169. "name": "torch.zeros",
  3170. "attributes": [
  3171. { "name": "size", "type": "int64[]" },
  3172. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3173. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3174. { "name": "device", "type": "Device", "optional": true, "default": null },
  3175. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3176. ],
  3177. "outputs": [
  3178. { "name": "output", "type": "Tensor" }
  3179. ]
  3180. },
  3181. {
  3182. "name": "torch.zeros_like",
  3183. "attributes": [
  3184. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3185. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3186. { "name": "device", "type": "Device", "optional": true, "default": null },
  3187. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3188. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3189. ],
  3190. "inputs": [
  3191. { "name": "self", "type": "Tensor" }
  3192. ],
  3193. "outputs": [
  3194. { "name": "output", "type": "Tensor" }
  3195. ]
  3196. },
  3197. {
  3198. "name": "torch.ones_like",
  3199. "attributes": [
  3200. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3201. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3202. { "name": "device", "type": "Device", "optional": true, "default": null },
  3203. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3204. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3205. ],
  3206. "inputs": [
  3207. { "name": "input", "type": "Tensor" }
  3208. ],
  3209. "outputs": [
  3210. { "name": "output", "type": "Tensor" }
  3211. ]
  3212. },
  3213. {
  3214. "name": "torch.arange:",
  3215. "attributes": [
  3216. { "name": "end", "type": "Scalar" },
  3217. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3218. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3219. { "name": "device", "type": "Device", "optional": true, "default": null },
  3220. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3221. ],
  3222. "outputs": [
  3223. { "name": "output", "type": "Tensor" }
  3224. ]
  3225. },
  3226. {
  3227. "name": "torch.arange:start",
  3228. "attributes": [
  3229. { "name": "start", "type": "Scalar" },
  3230. { "name": "end", "type": "Scalar" },
  3231. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3232. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3233. { "name": "device", "type": "Device", "optional": true, "default": null },
  3234. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3235. ],
  3236. "outputs": [
  3237. { "name": "output", "type": "Tensor" }
  3238. ]
  3239. },
  3240. {
  3241. "name": "torch.arange:start_step",
  3242. "attributes": [
  3243. { "name": "start", "type": "Scalar" },
  3244. { "name": "end", "type": "Scalar" },
  3245. { "name": "step", "type": "Scalar" },
  3246. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3247. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3248. { "name": "device", "type": "Device", "optional": true, "default": null },
  3249. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3250. ],
  3251. "outputs": [
  3252. { "name": "output", "type": "Tensor" }
  3253. ]
  3254. },
  3255. {
  3256. "name": "torch.arange:start_out",
  3257. "attributes": [
  3258. { "name": "start", "type": "Scalar" },
  3259. { "name": "end", "type": "Scalar" },
  3260. { "name": "step", "type": "Scalar", "default": 1 }
  3261. ],
  3262. "outputs": [
  3263. { "name": "output", "type": "Tensor" }
  3264. ]
  3265. },
  3266. {
  3267. "name": "torch.clamp",
  3268. "inputs": [
  3269. { "name": "self", "type": "Tensor" },
  3270. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  3271. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  3272. ],
  3273. "outputs": [
  3274. { "name": "output", "type": "Tensor" }
  3275. ]
  3276. },
  3277. {
  3278. "name": "torch.clamp_",
  3279. "inputs": [
  3280. { "name": "self", "type": "Tensor" },
  3281. { "name": "min", "type": "Scalar", "optional": true, "default": null },
  3282. { "name": "max", "type": "Scalar", "optional": true, "default": null }
  3283. ],
  3284. "outputs": [
  3285. { "name": "output", "type": "Tensor" }
  3286. ]
  3287. },
  3288. {
  3289. "name": "torch.clamp_min",
  3290. "inputs": [
  3291. { "name": "self", "type": "Tensor" },
  3292. { "name": "min", "type": "Scalar" }
  3293. ],
  3294. "outputs": [
  3295. { "name": "output", "type": "Tensor" }
  3296. ]
  3297. },
  3298. {
  3299. "name": "torch.clamp_min_",
  3300. "inputs": [
  3301. { "name": "self", "type": "Tensor" },
  3302. { "name": "min", "type": "Scalar" }
  3303. ],
  3304. "outputs": [
  3305. { "name": "output", "type": "Tensor" }
  3306. ]
  3307. },
  3308. {
  3309. "name": "torch.clamp_max",
  3310. "inputs": [
  3311. { "name": "self", "type": "Tensor" },
  3312. { "name": "max", "type": "Scalar" }
  3313. ],
  3314. "outputs": [
  3315. { "name": "output", "type": "Tensor" }
  3316. ]
  3317. },
  3318. {
  3319. "name": "torch.clamp_max_",
  3320. "inputs": [
  3321. { "name": "self", "type": "Tensor" },
  3322. { "name": "max", "type": "Scalar" }
  3323. ],
  3324. "outputs": [
  3325. { "name": "output", "type": "Tensor" }
  3326. ]
  3327. },
  3328. {
  3329. "name": "torch.fill_:Scalar",
  3330. "inputs": [
  3331. { "name": "self", "type": "Tensor" },
  3332. { "name": "value", "type": "Scalar" }
  3333. ],
  3334. "outputs": [
  3335. { "name": "output", "type": "Tensor" }
  3336. ]
  3337. },
  3338. {
  3339. "name": "torch.fill_:Tensor",
  3340. "inputs": [
  3341. { "name": "self", "type": "Tensor" },
  3342. { "name": "value", "type": "Tensor" }
  3343. ],
  3344. "outputs": [
  3345. { "name": "output", "type": "Tensor" }
  3346. ]
  3347. },
  3348. {
  3349. "name": "torch.floor",
  3350. "inputs": [
  3351. { "name": "self", "type": "Tensor" }
  3352. ],
  3353. "outputs": [
  3354. { "name": "output", "type": "Tensor" }
  3355. ]
  3356. },
  3357. {
  3358. "name": "torch.floor_",
  3359. "inputs": [
  3360. { "name": "self", "type": "Tensor" }
  3361. ],
  3362. "outputs": [
  3363. { "name": "output", "type": "Tensor" }
  3364. ]
  3365. },
  3366. {
  3367. "name": "torch.floor_divide",
  3368. "inputs": [
  3369. { "name": "self", "type": "Tensor" },
  3370. { "name": "other", "type": "Tensor" }
  3371. ],
  3372. "outputs": [
  3373. { "name": "output", "type": "Tensor" }
  3374. ]
  3375. },
  3376. {
  3377. "name": "torch.floor_divide_:Tensor",
  3378. "inputs": [
  3379. { "name": "self", "type": "Tensor" },
  3380. { "name": "other", "type": "Tensor" }
  3381. ],
  3382. "outputs": [
  3383. { "name": "output", "type": "Tensor" }
  3384. ]
  3385. },
  3386. {
  3387. "name": "torch.floor_divide_:Scalar",
  3388. "inputs": [
  3389. { "name": "self", "type": "Tensor" },
  3390. { "name": "other", "type": "Scalar" }
  3391. ],
  3392. "outputs": [
  3393. { "name": "output", "type": "Tensor" }
  3394. ]
  3395. },
  3396. {
  3397. "name": "torch.abs",
  3398. "inputs": [
  3399. { "name": "self", "type": "Tensor" }
  3400. ],
  3401. "outputs": [
  3402. { "name": "output", "type": "Tensor" }
  3403. ]
  3404. },
  3405. {
  3406. "name": "torch.abs_",
  3407. "inputs": [
  3408. { "name": "self", "type": "Tensor" }
  3409. ],
  3410. "outputs": [
  3411. { "name": "output", "type": "Tensor" }
  3412. ]
  3413. },
  3414. {
  3415. "name": "torch.leaky_relu_",
  3416. "category": "Activation",
  3417. "attributes": [
  3418. { "name": "negative_slope", "type": "Scalar", "default": 0.01 }
  3419. ],
  3420. "inputs": [
  3421. { "name": "input", "type": "Tensor" }
  3422. ],
  3423. "outputs": [
  3424. { "name": "output", "type": "Tensor" }
  3425. ]
  3426. },
  3427. {
  3428. "name": "torch.instance_norm",
  3429. "category": "Normalization",
  3430. "attributes": [
  3431. { "name": "use_input_stats", "type": "boolean" },
  3432. { "name": "momentum", "type": "float32" },
  3433. { "name": "eps", "type": "float32" },
  3434. { "name": "cudnn_enabled", "type": "boolean" }
  3435. ],
  3436. "inputs": [
  3437. { "name": "input" },
  3438. { "name": "weight", "optional": true },
  3439. { "name": "bias", "optional": true },
  3440. { "name": "running_mean", "optional": true },
  3441. { "name": "running_var", "optional": true }
  3442. ],
  3443. "outputs": [
  3444. { "name": "output", "type": "Tensor" }
  3445. ]
  3446. },
  3447. {
  3448. "name": "torch.reflection_pad1d",
  3449. "category": "Tensor",
  3450. "attributes": [
  3451. { "name": "padding", "type": "int64[]" }
  3452. ],
  3453. "inputs": [
  3454. { "name": "input", "type": "Tensor" }
  3455. ],
  3456. "outputs": [
  3457. { "name": "output", "type": "Tensor" }
  3458. ]
  3459. },
  3460. {
  3461. "name": "torch.replication_pad1d",
  3462. "category": "Tensor",
  3463. "attributes": [
  3464. { "name": "padding", "type": "int64[]" }
  3465. ],
  3466. "inputs": [
  3467. { "name": "input", "type": "Tensor" }
  3468. ],
  3469. "outputs": [
  3470. { "name": "output", "type": "Tensor" }
  3471. ]
  3472. },
  3473. {
  3474. "name": "torch.replication_pad2d",
  3475. "category": "Tensor",
  3476. "attributes": [
  3477. { "name": "padding", "type": "int64[]" }
  3478. ],
  3479. "inputs": [
  3480. { "name": "input", "type": "Tensor" }
  3481. ],
  3482. "outputs": [
  3483. { "name": "output", "type": "Tensor" }
  3484. ]
  3485. },
  3486. {
  3487. "name": "torch.replication_pad3d",
  3488. "category": "Tensor",
  3489. "attributes": [
  3490. { "name": "padding", "type": "int64[]" }
  3491. ],
  3492. "inputs": [
  3493. { "name": "input", "type": "Tensor" }
  3494. ],
  3495. "outputs": [
  3496. { "name": "output", "type": "Tensor" }
  3497. ]
  3498. },
  3499. {
  3500. "name": "torch.reflection_pad2d",
  3501. "category": "Tensor",
  3502. "attributes": [
  3503. { "name": "padding", "type": "int64[]" }
  3504. ],
  3505. "inputs": [
  3506. { "name": "input", "type": "Tensor" }
  3507. ],
  3508. "outputs": [
  3509. { "name": "output", "type": "Tensor" }
  3510. ]
  3511. },
  3512. {
  3513. "name": "torch.reflection_pad3d",
  3514. "category": "Tensor",
  3515. "attributes": [
  3516. { "name": "padding", "type": "int64[]" }
  3517. ],
  3518. "inputs": [
  3519. { "name": "input", "type": "Tensor" }
  3520. ],
  3521. "outputs": [
  3522. { "name": "output", "type": "Tensor" }
  3523. ]
  3524. },
  3525. {
  3526. "name": "torch.rnn_relu:input",
  3527. "category": "Layer",
  3528. "attributes": [
  3529. { "name": "has_biases", "type": "boolean" },
  3530. { "name": "num_layers", "type": "int64" },
  3531. { "name": "dropout", "type": "float32" },
  3532. { "name": "train", "type": "boolean" },
  3533. { "name": "bidirectional", "type": "boolean" },
  3534. { "name": "batch_first", "type": "boolean" }
  3535. ],
  3536. "inputs": [
  3537. { "name": "input", "type": "Tensor" },
  3538. { "name": "hx", "type": "Tensor" },
  3539. { "name": "params", "type": "Tensor[]" }
  3540. ],
  3541. "outputs": [
  3542. { "name": "output", "type": "Tensor" },
  3543. { "name": "?", "type": "Tensor" }
  3544. ]
  3545. },
  3546. {
  3547. "name": "torch.rnn_tanh:input",
  3548. "category": "Layer",
  3549. "attributes": [
  3550. { "name": "has_biases", "type": "boolean" },
  3551. { "name": "num_layers", "type": "int64" },
  3552. { "name": "dropout", "type": "float32" },
  3553. { "name": "train", "type": "boolean" },
  3554. { "name": "bidirectional", "type": "boolean" },
  3555. { "name": "batch_first", "type": "boolean" }
  3556. ],
  3557. "inputs": [
  3558. { "name": "input", "type": "Tensor" },
  3559. { "name": "hx", "type": "Tensor" },
  3560. { "name": "params", "type": "Tensor[]" }
  3561. ],
  3562. "outputs": [
  3563. { "name": "output" },
  3564. { "name": "?" }
  3565. ]
  3566. },
  3567. {
  3568. "name": "torch.detach",
  3569. "inputs": [
  3570. { "name": "self", "type": "Tensor" }
  3571. ],
  3572. "outputs": [
  3573. { "name": "output", "type": "Tensor" }
  3574. ]
  3575. },
  3576. {
  3577. "name": "torch.upsample_bicubic2d",
  3578. "category": "Layer",
  3579. "attributes": [
  3580. { "name": "output_size", "type": "int64[]" },
  3581. { "name": "align_corners", "type": "boolean" },
  3582. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  3583. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  3584. ],
  3585. "inputs": [
  3586. { "name": "self", "type": "Tensor" }
  3587. ],
  3588. "outputs": [
  3589. { "name": "output", "type": "Tensor" }
  3590. ]
  3591. },
  3592. {
  3593. "name": "torch.upsample_linear1d:",
  3594. "inputs": [
  3595. { "name": "self", "type": "Tensor" },
  3596. { "name": "output_size", "type": "int64[]" },
  3597. { "name": "align_corners", "type": "boolean" },
  3598. { "name": "scales", "type": "float32", "optional": true, "default": null }
  3599. ],
  3600. "outputs": [
  3601. { "name": "output", "type": "Tensor" }
  3602. ]
  3603. },
  3604. {
  3605. "name": "torch.upsample_linear1d:vec",
  3606. "inputs": [
  3607. { "name": "self", "type": "Tensor" },
  3608. { "name": "output_size", "type": "int64[]", "optional": true },
  3609. { "name": "align_corners", "type": "boolean" },
  3610. { "name": "scale_factors", "type": "float32[]", "optional": true }
  3611. ],
  3612. "outputs": [
  3613. { "name": "output", "type": "Tensor" }
  3614. ]
  3615. },
  3616. {
  3617. "name": "torch.upsample_bilinear2d:",
  3618. "category": "Layer",
  3619. "inputs": [
  3620. { "name": "self", "type": "Tensor" },
  3621. { "name": "output_size", "type": "int64[]" },
  3622. { "name": "align_corners", "type": "boolean" },
  3623. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  3624. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  3625. ],
  3626. "outputs": [
  3627. { "name": "output", "type": "Tensor" }
  3628. ]
  3629. },
  3630. {
  3631. "name": "torch.upsample_bilinear2d:vec",
  3632. "category": "Layer",
  3633. "inputs": [
  3634. { "name": "input", "type": "Tensor" },
  3635. { "name": "output_size", "type": "int64[]", "optional": true },
  3636. { "name": "align_corners", "type": "boolean" },
  3637. { "name": "scale_factors", "type": "float32[]", "optional": true }
  3638. ],
  3639. "outputs": [
  3640. { "name": "output", "type": "Tensor" }
  3641. ]
  3642. },
  3643. {
  3644. "name": "torch.upsample_nearest1d:",
  3645. "category": "Layer",
  3646. "inputs": [
  3647. { "name": "self", "type": "Tensor" },
  3648. { "name": "output_size", "type": "int64[]" },
  3649. { "name": "scales", "type": "float32", "optional": true, "default": null }
  3650. ],
  3651. "outputs": [
  3652. { "name": "output", "type": "Tensor" }
  3653. ]
  3654. },
  3655. {
  3656. "name": "torch.upsample_nearest1d:vec",
  3657. "category": "Layer",
  3658. "inputs": [
  3659. { "name": "input", "type": "Tensor" },
  3660. { "name": "output_size", "type": "int64[]", "optional": true },
  3661. { "name": "scales_factors", "type": "float32[]", "optional": true }
  3662. ],
  3663. "outputs": [
  3664. { "name": "output", "type": "Tensor" }
  3665. ]
  3666. },
  3667. {
  3668. "name": "torch.upsample_nearest2d:",
  3669. "category": "Layer",
  3670. "inputs": [
  3671. { "name": "self", "type": "Tensor" },
  3672. { "name": "output_size", "type": "int64[]" },
  3673. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  3674. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  3675. ],
  3676. "outputs": [
  3677. { "name": "output", "type": "Tensor" }
  3678. ]
  3679. },
  3680. {
  3681. "name": "torch.upsample_nearest2d:vec",
  3682. "category": "Layer",
  3683. "inputs": [
  3684. { "name": "input", "type": "Tensor" },
  3685. { "name": "output_size", "type": "int64[]", "optional": true },
  3686. { "name": "scales_factors", "type": "float32[]", "optional": true }
  3687. ],
  3688. "outputs": [
  3689. { "name": "output", "type": "Tensor" }
  3690. ]
  3691. },
  3692. {
  3693. "name": "torch.upsample_nearest3d:",
  3694. "category": "Layer",
  3695. "inputs": [
  3696. { "name": "self", "type": "Tensor" },
  3697. { "name": "output_size", "type": "int64[]" },
  3698. { "name": "scales_d", "type": "float32", "optional": true, "default": null },
  3699. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  3700. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  3701. ],
  3702. "outputs": [
  3703. { "name": "output", "type": "Tensor" }
  3704. ]
  3705. },
  3706. {
  3707. "name": "torch.upsample_nearest3d:vec",
  3708. "category": "Layer",
  3709. "inputs": [
  3710. { "name": "input", "type": "Tensor" },
  3711. { "name": "output_size", "type": "int64[]", "optional": true },
  3712. { "name": "scales_factors", "type": "float32[]", "optional": true }
  3713. ],
  3714. "outputs": [
  3715. { "name": "output", "type": "Tensor" }
  3716. ]
  3717. },
  3718. {
  3719. "name": "torch.upsample_trilinear3d:",
  3720. "inputs": [
  3721. { "name": "self", "type": "Tensor" },
  3722. { "name": "output_size", "type": "int64[]" },
  3723. { "name": "align_corners", "type": "boolean" },
  3724. { "name": "scales_d", "type": "float32", "optional": true, "default": null },
  3725. { "name": "scales_h", "type": "float32", "optional": true, "default": null },
  3726. { "name": "scales_w", "type": "float32", "optional": true, "default": null }
  3727. ],
  3728. "outputs": [
  3729. { "name": "output", "type": "int64" }
  3730. ]
  3731. },
  3732. {
  3733. "name": "torch.upsample_trilinear3d:vec",
  3734. "inputs": [
  3735. { "name": "grad_output", "type": "Tensor" },
  3736. { "name": "output_size", "type": "int64[]", "optional": true },
  3737. { "name": "input_size", "type": "int64[]" },
  3738. { "name": "align_corners", "type": "boolean" },
  3739. { "name": "scale_factors", "type": "float32[]", "optional": true }
  3740. ],
  3741. "outputs": [
  3742. { "name": "output", "type": "int64" }
  3743. ]
  3744. },
  3745. {
  3746. "name": "torch.size",
  3747. "inputs": [
  3748. { "name": "self", "type": "Tensor" },
  3749. { "name": "dim", "type": "int64" }
  3750. ],
  3751. "outputs": [
  3752. { "name": "output", "type": "int64" }
  3753. ]
  3754. },
  3755. {
  3756. "name": "torch.exp",
  3757. "inputs": [
  3758. { "name": "self", "type": "Tensor" }
  3759. ],
  3760. "outputs": [
  3761. { "name": "output", "type": "Tensor" }
  3762. ]
  3763. },
  3764. {
  3765. "name": "torch.reciprocal",
  3766. "inputs": [
  3767. { "name": "self", "type": "Tensor" }
  3768. ],
  3769. "outputs": [
  3770. { "name": "output", "type": "Tensor" }
  3771. ]
  3772. },
  3773. {
  3774. "name": "torch.square",
  3775. "inputs": [
  3776. { "name": "self", "type": "Tensor" }
  3777. ],
  3778. "outputs": [
  3779. { "name": "output", "type": "Tensor" }
  3780. ]
  3781. },
  3782. {
  3783. "name": "torch.square_",
  3784. "inputs": [
  3785. { "name": "self", "type": "Tensor" }
  3786. ],
  3787. "outputs": [
  3788. { "name": "output", "type": "Tensor" }
  3789. ]
  3790. },
  3791. {
  3792. "name": "torch.sqrt",
  3793. "inputs": [
  3794. { "name": "self", "type": "Tensor" }
  3795. ],
  3796. "outputs": [
  3797. { "name": "output", "type": "Tensor" }
  3798. ]
  3799. },
  3800. {
  3801. "name": "torch.std:",
  3802. "attributes": [
  3803. { "name": "unbiased", "type": "boolean", "default": true }
  3804. ],
  3805. "inputs": [
  3806. { "name": "self", "type": "Tensor" }
  3807. ],
  3808. "outputs": [
  3809. { "name": "output", "type": "Tensor" }
  3810. ]
  3811. },
  3812. {
  3813. "name": "torch.std:dim",
  3814. "attributes": [
  3815. { "name": "dim", "type": "int64[]" },
  3816. { "name": "unbiased", "type": "boolean", "default": true },
  3817. { "name": "keepdim", "type": "boolean", "default": false }
  3818. ],
  3819. "inputs": [
  3820. { "name": "self", "type": "Tensor" }
  3821. ],
  3822. "outputs": [
  3823. { "name": "output", "type": "Tensor" }
  3824. ]
  3825. },
  3826. {
  3827. "name": "torch.std_mean:",
  3828. "attributes": [
  3829. { "name": "unbiased", "type": "boolean", "default": true }
  3830. ],
  3831. "inputs": [
  3832. { "name": "self", "type": "Tensor" }
  3833. ],
  3834. "outputs": [
  3835. { "name": "output", "type": "Tensor" }
  3836. ]
  3837. },
  3838. {
  3839. "name": "torch.std_mean:dim",
  3840. "attributes": [
  3841. { "name": "dim", "type": "int64[]" },
  3842. { "name": "unbiased", "type": "boolean", "default": true },
  3843. { "name": "keepdim", "type": "boolean", "default": false }
  3844. ],
  3845. "inputs": [
  3846. { "name": "self", "type": "Tensor" }
  3847. ],
  3848. "outputs": [
  3849. { "name": "output", "type": "Tensor" }
  3850. ]
  3851. },
  3852. {
  3853. "name": "torch.erf",
  3854. "inputs": [
  3855. { "name": "input", "type": "Tensor" }
  3856. ],
  3857. "outputs": [
  3858. { "name": "output", "type": "Tensor" }
  3859. ]
  3860. },
  3861. {
  3862. "name": "torch.argmin",
  3863. "attributes": [
  3864. { "name": "dim", "type": "int64" },
  3865. { "name": "keepdim", "type": "boolean", "default": false }
  3866. ],
  3867. "inputs": [
  3868. { "name": "input", "type": "Tensor" }
  3869. ],
  3870. "outputs": [
  3871. { "name": "output", "type": "Tensor" }
  3872. ]
  3873. },
  3874. {
  3875. "name": "torch.argmax",
  3876. "attributes": [
  3877. { "name": "dim", "type": "int64" },
  3878. { "name": "keepdim", "type": "boolean", "default": false }
  3879. ],
  3880. "inputs": [
  3881. { "name": "input", "type": "Tensor" }
  3882. ],
  3883. "outputs": [
  3884. { "name": "output", "type": "Tensor" }
  3885. ]
  3886. },
  3887. {
  3888. "name": "torch.log",
  3889. "inputs": [
  3890. { "name": "self", "type": "Tensor" }
  3891. ],
  3892. "outputs": [
  3893. { "name": "output", "type": "Tensor" }
  3894. ]
  3895. },
  3896. {
  3897. "name": "torch.log10",
  3898. "inputs": [
  3899. { "name": "self", "type": "Tensor" }
  3900. ],
  3901. "outputs": [
  3902. { "name": "output", "type": "Tensor" }
  3903. ]
  3904. },
  3905. {
  3906. "name": "torch.log1p",
  3907. "inputs": [
  3908. { "name": "self", "type": "Tensor" }
  3909. ],
  3910. "outputs": [
  3911. { "name": "output", "type": "Tensor" }
  3912. ]
  3913. },
  3914. {
  3915. "name": "torch.log2",
  3916. "inputs": [
  3917. { "name": "self", "type": "Tensor" }
  3918. ],
  3919. "outputs": [
  3920. { "name": "output", "type": "Tensor" }
  3921. ]
  3922. },
  3923. {
  3924. "name": "torch.logdet",
  3925. "inputs": [
  3926. { "name": "self", "type": "Tensor" }
  3927. ],
  3928. "outputs": [
  3929. { "name": "output", "type": "Tensor" }
  3930. ]
  3931. },
  3932. {
  3933. "name": "torch.affine_grid_generator",
  3934. "attributes": [
  3935. { "name": "size", "type": "int64[]" },
  3936. { "name": "align_corners", "type": "boolean" }
  3937. ],
  3938. "inputs": [
  3939. { "name": "theta", "type": "Tensor" }
  3940. ],
  3941. "outputs": [
  3942. { "name": "output", "type": "Tensor" }
  3943. ]
  3944. },
  3945. {
  3946. "name": "torch.__interpolate",
  3947. "attributes": [
  3948. { "name": "size", "type": "int64" },
  3949. { "name": "scale_factor", "type": "float32[]" },
  3950. { "name": "mode", "type": "string" },
  3951. { "name": "align_corners", "type": "boolean" },
  3952. { "name": "recompute_scale_factor", "type": "boolean" }
  3953. ],
  3954. "inputs": [
  3955. { "name": "input", "type": "Tensor" }
  3956. ],
  3957. "outputs": [
  3958. { "name": "output", "type": "Tensor" }
  3959. ]
  3960. },
  3961. {
  3962. "name": "torch.rand:",
  3963. "attributes": [
  3964. { "name": "size", "type": "int64[]" },
  3965. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3966. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3967. { "name": "device", "type": "Device", "optional": true, "default": null },
  3968. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3969. ],
  3970. "outputs": [
  3971. { "name": "output", "type": "Tensor" }
  3972. ]
  3973. },
  3974. {
  3975. "name": "torch.rand:generator",
  3976. "attributes": [
  3977. { "name": "high", "type": "int64", "optional": true, "default": null },
  3978. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3979. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3980. { "name": "device", "type": "Device", "optional": true, "default": null },
  3981. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  3982. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  3983. ],
  3984. "outputs": [
  3985. { "name": "output", "type": "Tensor" }
  3986. ]
  3987. },
  3988. {
  3989. "name": "torch.rand_link",
  3990. "attributes": [
  3991. { "name": "size", "type": "int64[]" },
  3992. { "name": "generator", "type": "Generator", "optional": true, "default": null },
  3993. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  3994. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  3995. { "name": "device", "type": "Device", "optional": true, "default": null },
  3996. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  3997. ],
  3998. "outputs": [
  3999. { "name": "output", "type": "Tensor" }
  4000. ]
  4001. },
  4002. {
  4003. "name": "torch.randint:generator",
  4004. "inputs": [
  4005. { "name": "high", "type": "int64", "optional": true, "default": null },
  4006. { "name": "size", "type": "int64[]" },
  4007. { "name": "generator", "type": "Generator" },
  4008. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4009. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4010. { "name": "device", "type": "Device", "optional": true, "default": null },
  4011. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4012. ],
  4013. "outputs": [
  4014. { "name": "output", "type": "Tensor" }
  4015. ]
  4016. },
  4017. {
  4018. "name": "torch.randint:low",
  4019. "inputs": [
  4020. { "name": "low", "type": "int64", "optional": true, "default": null },
  4021. { "name": "high", "type": "int64", "optional": true, "default": null },
  4022. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4023. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4024. { "name": "device", "type": "Device", "optional": true, "default": null },
  4025. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4026. ],
  4027. "outputs": [
  4028. { "name": "output", "type": "Tensor" }
  4029. ]
  4030. },
  4031. {
  4032. "name": "torch.randn",
  4033. "attributes": [
  4034. { "name": "size", "type": "int64[]" },
  4035. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4036. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4037. { "name": "device", "type": "Device", "optional": true, "default": null },
  4038. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4039. ],
  4040. "outputs": [
  4041. { "name": "output", "type": "Tensor" }
  4042. ]
  4043. },
  4044. {
  4045. "name": "torch.unsafe_split:Tensor",
  4046. "category": "Tensor",
  4047. "attributes": [
  4048. { "name": "split_size", "type": "int64" },
  4049. { "name": "dim", "type": "int64" }
  4050. ],
  4051. "inputs": [
  4052. { "name": "input", "type": "Tensor" }
  4053. ],
  4054. "outputs": [
  4055. { "name": "outputs", "type": "Tensor[]" }
  4056. ]
  4057. },
  4058. {
  4059. "name": "torch.split:Tensor",
  4060. "category": "Tensor",
  4061. "attributes": [
  4062. { "name": "split_size", "type": "int64" },
  4063. { "name": "dim", "type": "int64" }
  4064. ],
  4065. "inputs": [
  4066. { "name": "self", "type": "Tensor" }
  4067. ],
  4068. "outputs": [
  4069. { "name": "outputs", "type": "Tensor[]" }
  4070. ]
  4071. },
  4072. {
  4073. "name": "torch.split_with_sizes",
  4074. "category": "Tensor",
  4075. "attributes": [
  4076. { "name": "split_sizes", "type": "int64[]" },
  4077. { "name": "dim", "type": "int64" }
  4078. ],
  4079. "inputs": [
  4080. { "name": "self", "type": "Tensor" }
  4081. ],
  4082. "outputs": [
  4083. { "name": "outputs", "type": "Tensor[]" }
  4084. ]
  4085. },
  4086. {
  4087. "name": "torch.remainder:Tensor",
  4088. "inputs": [
  4089. { "name": "self", "type": "Tensor" },
  4090. { "name": "other", "type": "Tensor" }
  4091. ],
  4092. "outputs": [
  4093. { "name": "output", "type": "Tensor" }
  4094. ]
  4095. },
  4096. {
  4097. "name": "torch.remainder:Scalar",
  4098. "inputs": [
  4099. { "name": "self", "type": "Tensor" },
  4100. { "name": "other", "type": "Scalar" }
  4101. ],
  4102. "outputs": [
  4103. { "name": "output", "type": "Tensor" }
  4104. ]
  4105. },
  4106. {
  4107. "name": "torch.remainder_:Tensor",
  4108. "inputs": [
  4109. { "name": "self", "type": "Tensor" },
  4110. { "name": "other", "type": "Tensor" }
  4111. ],
  4112. "outputs": [
  4113. { "name": "output", "type": "Tensor" }
  4114. ]
  4115. },
  4116. {
  4117. "name": "torch.remainder_:Scalar",
  4118. "inputs": [
  4119. { "name": "self", "type": "Tensor" },
  4120. { "name": "other", "type": "Scalar" }
  4121. ],
  4122. "outputs": [
  4123. { "name": "output", "type": "Tensor" }
  4124. ]
  4125. },
  4126. {
  4127. "name": "torch.layer_norm",
  4128. "category": "Normalization",
  4129. "inputs": [
  4130. { "name": "input", "type": "Tensor" },
  4131. { "name": "normalized_shape", "type": "int64[]" },
  4132. { "name": "weight", "type": "Tensor", "optional": true, "default": null },
  4133. { "name": "bias", "type": "Tensor", "optional": true, "default": null },
  4134. { "name": "eps", "type": "float32", "default": 0.00001 },
  4135. { "name": "cudnn_enable", "type": "boolean", "default": true }
  4136. ],
  4137. "outputs": [
  4138. { "name": "output", "type": "Tensor" }
  4139. ]
  4140. },
  4141. {
  4142. "name": "torch.scatter_:src",
  4143. "inputs": [
  4144. { "name": "self", "type": "Tensor" },
  4145. { "name": "dim", "type": "int64" },
  4146. { "name": "index", "type": "Tensor" },
  4147. { "name": "src", "type": "Tensor" }
  4148. ],
  4149. "outputs": [
  4150. { "name": "output", "type": "Tensor" }
  4151. ]
  4152. },
  4153. {
  4154. "name": "torch.scatter_:value",
  4155. "inputs": [
  4156. { "name": "self", "type": "Tensor" },
  4157. { "name": "dim", "type": "int64" },
  4158. { "name": "index", "type": "Tensor" },
  4159. { "name": "src", "type": "Scalar" }
  4160. ],
  4161. "outputs": [
  4162. { "name": "output", "type": "Tensor" }
  4163. ]
  4164. },
  4165. {
  4166. "name": "torch.scatter_add_",
  4167. "inputs": [
  4168. { "name": "self", "type": "Tensor" },
  4169. { "name": "dim", "type": "int64"},
  4170. { "name": "index", "type": "Tensor" },
  4171. { "name": "src", "type": "Tensor" }
  4172. ],
  4173. "outputs": [
  4174. { "name": "output", "type": "Tensor" }
  4175. ]
  4176. },
  4177. {
  4178. "name": "torch.sin",
  4179. "inputs": [
  4180. { "name": "self", "type": "Tensor" }
  4181. ],
  4182. "outputs": [
  4183. { "name": "output", "type": "Tensor" }
  4184. ]
  4185. },
  4186. {
  4187. "name": "torch.sinh",
  4188. "inputs": [
  4189. { "name": "self", "type": "Tensor" }
  4190. ],
  4191. "outputs": [
  4192. { "name": "output", "type": "Tensor" }
  4193. ]
  4194. },
  4195. {
  4196. "name": "torch.cos",
  4197. "inputs": [
  4198. { "name": "self", "type": "Tensor" }
  4199. ],
  4200. "outputs": [
  4201. { "name": "output", "type": "Tensor" }
  4202. ]
  4203. },
  4204. {
  4205. "name": "torch.cosh",
  4206. "inputs": [
  4207. { "name": "self", "type": "Tensor" }
  4208. ],
  4209. "outputs": [
  4210. { "name": "output", "type": "Tensor" }
  4211. ]
  4212. },
  4213. {
  4214. "name": "torch.tan",
  4215. "inputs": [
  4216. { "name": "self", "type": "Tensor" }
  4217. ],
  4218. "outputs": [
  4219. { "name": "output", "type": "Tensor" }
  4220. ]
  4221. },
  4222. {
  4223. "name": "torch.atan2:Tensor",
  4224. "inputs": [
  4225. { "name": "input", "type": "Tensor" },
  4226. { "name": "other", "type": "Tensor" }
  4227. ],
  4228. "outputs": [
  4229. { "name": "output", "type": "Tensor" }
  4230. ]
  4231. },
  4232. {
  4233. "name": "torch.tan_",
  4234. "inputs": [
  4235. { "name": "self", "type": "Tensor" }
  4236. ],
  4237. "outputs": [
  4238. { "name": "output", "type": "Tensor" }
  4239. ]
  4240. },
  4241. {
  4242. "name": "torch.tanh",
  4243. "inputs": [
  4244. { "name": "self", "type": "Tensor" }
  4245. ],
  4246. "outputs": [
  4247. { "name": "output", "type": "Tensor" }
  4248. ]
  4249. },
  4250. {
  4251. "name": "torch.tanh_",
  4252. "inputs": [
  4253. { "name": "self", "type": "Tensor" }
  4254. ],
  4255. "outputs": [
  4256. { "name": "output", "type": "Tensor" }
  4257. ]
  4258. },
  4259. {
  4260. "name": "torch.narrow:",
  4261. "attributes": [
  4262. { "name": "dim", "type": "int64" },
  4263. { "name": "start", "type": "int64" },
  4264. { "name": "length", "type": "int64" }
  4265. ],
  4266. "inputs": [
  4267. { "name": "self", "type": "Tensor" }
  4268. ],
  4269. "outputs": [
  4270. { "name": "output", "type": "Tensor" }
  4271. ]
  4272. },
  4273. {
  4274. "name": "torch.narrow:Tensor",
  4275. "inputs": [
  4276. { "name": "self", "type": "Tensor" },
  4277. { "name": "dim", "type": "int64" },
  4278. { "name": "start", "type": "Tensor" },
  4279. { "name": "length", "type": "int64" }
  4280. ],
  4281. "outputs": [
  4282. { "name": "output", "type": "Tensor" }
  4283. ]
  4284. },
  4285. {
  4286. "name": "torch.narrow_copy",
  4287. "attributes": [
  4288. { "name": "dim", "type": "int64" },
  4289. { "name": "start", "type": "int64" },
  4290. { "name": "length", "type": "int64" }
  4291. ],
  4292. "inputs": [
  4293. { "name": "self", "type": "Tensor" }
  4294. ],
  4295. "outputs": [
  4296. { "name": "output", "type": "Tensor" }
  4297. ]
  4298. },
  4299. {
  4300. "name": "torch.index_select:",
  4301. "inputs": [
  4302. { "name": "self", "type": "Tensor" },
  4303. { "name": "dim", "type": "int64" },
  4304. { "name": "index", "type": "Tensor" }
  4305. ],
  4306. "outputs": [
  4307. { "name": "output", "type": "Tensor" }
  4308. ]
  4309. },
  4310. {
  4311. "name": "torch.index_put",
  4312. "attributes": [
  4313. { "name": "accumulate", "type": "boolean", "default": false }
  4314. ],
  4315. "inputs": [
  4316. { "name": "self", "type": "Tensor" },
  4317. { "name": "indices", "type": "Tensor[]", "optional": true },
  4318. { "name": "values", "type": "Tensor" }
  4319. ],
  4320. "outputs": [
  4321. { "name": "output", "type": "Tensor" }
  4322. ]
  4323. },
  4324. {
  4325. "name": "torch.index_put_",
  4326. "attributes": [
  4327. { "name": "accumulate", "type": "boolean", "default": false }
  4328. ],
  4329. "inputs": [
  4330. { "name": "self", "type": "Tensor" },
  4331. { "name": "indices", "type": "Tensor[]", "optional": true },
  4332. { "name": "values", "type": "Tensor" }
  4333. ],
  4334. "outputs": [
  4335. { "name": "output", "type": "Tensor" }
  4336. ]
  4337. },
  4338. {
  4339. "name": "torch.new_zeros",
  4340. "attributes": [
  4341. { "name": "size", "type": "int64[]" },
  4342. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4343. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4344. { "name": "device", "type": "Device", "optional": true, "default": null },
  4345. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4346. ],
  4347. "inputs": [
  4348. { "name": "self", "type": "Tensor" }
  4349. ],
  4350. "outputs": [
  4351. { "name": "output", "type": "Tensor" }
  4352. ]
  4353. },
  4354. {
  4355. "name": "torch.new_full",
  4356. "attributes": [
  4357. { "name": "size", "type": "int64[]" },
  4358. { "name": "fill_value", "type": "Scalar" },
  4359. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4360. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4361. { "name": "device", "type": "Device", "optional": true, "default": null },
  4362. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4363. ],
  4364. "inputs": [
  4365. { "name": "self", "type": "Tensor" }
  4366. ],
  4367. "outputs": [
  4368. { "name": "output", "type": "Tensor" }
  4369. ]
  4370. },
  4371. {
  4372. "name": "torch.softplus",
  4373. "category": "Activation",
  4374. "attributes": [
  4375. { "name": "beta", "type": "Scalar", "default": 1 },
  4376. { "name": "threshold", "type": "Scalar", "default": 20 }
  4377. ],
  4378. "inputs": [
  4379. { "name": "self", "type": "Tensor" }
  4380. ],
  4381. "outputs": [
  4382. { "name": "output", "type": "Tensor" }
  4383. ]
  4384. },
  4385. {
  4386. "name": "torch.zero_",
  4387. "inputs": [
  4388. { "name": "self", "type": "Tensor" }
  4389. ],
  4390. "outputs": [
  4391. { "name": "output", "type": "Tensor" }
  4392. ]
  4393. },
  4394. {
  4395. "name": "torch.cumsum",
  4396. "inputs": [
  4397. { "name": "self", "type": "Tensor" },
  4398. { "name": "dim", "type": "int64" },
  4399. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }
  4400. ],
  4401. "outputs": [
  4402. { "name": "output", "type": "Tensor" }
  4403. ]
  4404. },
  4405. {
  4406. "name": "torch.masked_fill:Scalar",
  4407. "inputs": [
  4408. { "name": "self", "type": "Tensor" },
  4409. { "name": "mask", "type": "Tensor" },
  4410. { "name": "value", "type": "Scalar" }
  4411. ],
  4412. "outputs": [
  4413. { "name": "output", "type": "Tensor" }
  4414. ]
  4415. },
  4416. {
  4417. "name": "torch.masked_fill:Tensor",
  4418. "inputs": [
  4419. { "name": "self", "type": "Tensor" },
  4420. { "name": "mask", "type": "Tensor" },
  4421. { "name": "value", "type": "Tensor" }
  4422. ],
  4423. "outputs": [
  4424. { "name": "output", "type": "Tensor" }
  4425. ]
  4426. },
  4427. {
  4428. "name": "torch.argmax",
  4429. "attributes": [
  4430. { "name": "dim", "type": "int64", "optional": true, "default": null },
  4431. { "name": "keepdim", "type": "boolean", "default": false }
  4432. ],
  4433. "inputs": [
  4434. { "name": "self", "type": "Tensor" }
  4435. ],
  4436. "outputs": [
  4437. { "name": "output", "type": "Tensor" }
  4438. ]
  4439. },
  4440. {
  4441. "name": "torch.argmin",
  4442. "attributes": [
  4443. { "name": "dim", "type": "int64", "optional": true, "default": null },
  4444. { "name": "keepdim", "type": "boolean", "default": false }
  4445. ],
  4446. "inputs": [
  4447. { "name": "self", "type": "Tensor" }
  4448. ],
  4449. "outputs": [
  4450. { "name": "output", "type": "Tensor" }
  4451. ]
  4452. },
  4453. {
  4454. "name": "torch.fft",
  4455. "attributes": [
  4456. { "name": "signal_ndim", "type": "int64" },
  4457. { "name": "normalized", "type": "boolean", "default": false }
  4458. ],
  4459. "inputs": [
  4460. { "name": "self", "type": "Tensor" }
  4461. ],
  4462. "outputs": [
  4463. { "name": "output", "type": "Tensor" }
  4464. ]
  4465. },
  4466. {
  4467. "name": "torch.topk:",
  4468. "inputs": [
  4469. { "name": "self", "type": "Tensor" },
  4470. { "name": "k", "type": "int64" },
  4471. { "name": "dim", "type": "int64", "default": -1 },
  4472. { "name": "largest", "type": "int64", "default": true },
  4473. { "name": "sorted", "type": "boolean", "default": false }
  4474. ],
  4475. "outputs": [
  4476. { "name": "values", "type": "Tensor" },
  4477. { "name": "indices", "type": "Tensor" }
  4478. ]
  4479. },
  4480. {
  4481. "name": "torch.random_:",
  4482. "attributes": [
  4483. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4484. ],
  4485. "inputs": [
  4486. { "name": "self", "type": "Tensor" }
  4487. ],
  4488. "outputs": [
  4489. { "name": "output", "type": "Tensor" }
  4490. ]
  4491. },
  4492. {
  4493. "name": "torch.random_:to",
  4494. "attributes": [
  4495. { "name": "to", "type": "int64" },
  4496. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4497. ],
  4498. "inputs": [
  4499. { "name": "self", "type": "Tensor" }
  4500. ],
  4501. "outputs": [
  4502. { "name": "output", "type": "Tensor" }
  4503. ]
  4504. },
  4505. {
  4506. "name": "torch.random_:from",
  4507. "attributes": [
  4508. { "name": "from", "type": "int64" },
  4509. { "name": "to", "type": "int64", "optional": true },
  4510. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4511. ],
  4512. "inputs": [
  4513. { "name": "self", "type": "Tensor" }
  4514. ],
  4515. "outputs": [
  4516. { "name": "output", "type": "Tensor" }
  4517. ]
  4518. },
  4519. {
  4520. "name": "torch.uniform_",
  4521. "attributes": [
  4522. { "name": "from", "type": "float32", "default": 0 },
  4523. { "name": "to", "type": "float32", "default": 1 },
  4524. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4525. ],
  4526. "inputs": [
  4527. { "name": "self", "type": "Tensor" }
  4528. ],
  4529. "outputs": [
  4530. { "name": "output", "type": "Tensor" }
  4531. ]
  4532. },
  4533. {
  4534. "name": "torch.cauchy_",
  4535. "attributes": [
  4536. { "name": "median", "type": "float32", "default": 0 },
  4537. { "name": "sigma", "type": "float32", "default": 1 },
  4538. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4539. ],
  4540. "inputs": [
  4541. { "name": "self", "type": "Tensor" }
  4542. ],
  4543. "outputs": [
  4544. { "name": "output", "type": "Tensor" }
  4545. ]
  4546. },
  4547. {
  4548. "name": "torch.log_normal_",
  4549. "attributes": [
  4550. { "name": "mean", "type": "float32", "default": 1 },
  4551. { "name": "std", "type": "float32", "default": 2 },
  4552. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4553. ],
  4554. "inputs": [
  4555. { "name": "self", "type": "Tensor" }
  4556. ],
  4557. "outputs": [
  4558. { "name": "output", "type": "Tensor" }
  4559. ]
  4560. },
  4561. {
  4562. "name": "torch.exponential_",
  4563. "attributes": [
  4564. { "name": "lambd", "type": "float32", "default": 1 },
  4565. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4566. ],
  4567. "inputs": [
  4568. { "name": "self", "type": "Tensor" }
  4569. ],
  4570. "outputs": [
  4571. { "name": "output", "type": "Tensor" }
  4572. ]
  4573. },
  4574. {
  4575. "name": "torch.geometric_",
  4576. "attributes": [
  4577. { "name": "p", "type": "float32" },
  4578. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  4579. ],
  4580. "inputs": [
  4581. { "name": "self", "type": "Tensor" }
  4582. ],
  4583. "outputs": [
  4584. { "name": "output", "type": "Tensor" }
  4585. ]
  4586. },
  4587. {
  4588. "name": "torch.mv",
  4589. "inputs": [
  4590. { "name": "self", "type": "Tensor" },
  4591. { "name": "vec", "type": "Tensor" }
  4592. ],
  4593. "outputs": [
  4594. { "name": "output", "type": "Tensor" }
  4595. ]
  4596. },
  4597. {
  4598. "name": "torch.mvlgamma",
  4599. "attributes": [
  4600. { "name": "p", "type": "int64" }
  4601. ],
  4602. "inputs": [
  4603. { "name": "self", "type": "Tensor" }
  4604. ],
  4605. "outputs": [
  4606. { "name": "output", "type": "Tensor" }
  4607. ]
  4608. },
  4609. {
  4610. "name": "torch.mvlgamma_",
  4611. "attributes": [
  4612. { "name": "p", "type": "int64" }
  4613. ],
  4614. "inputs": [
  4615. { "name": "self", "type": "Tensor" }
  4616. ],
  4617. "outputs": [
  4618. { "name": "output", "type": "Tensor" }
  4619. ]
  4620. },
  4621. {
  4622. "name": "torch.quantize_per_channel",
  4623. "attributes": [
  4624. { "name": "axis", "type": "int64" },
  4625. { "name": "dtype", "type": "ScalarType" }
  4626. ],
  4627. "inputs": [
  4628. { "name": "self", "type": "Tensor" },
  4629. { "name": "scales", "type": "Tensor" },
  4630. { "name": "zero_points", "type": "Tensor" }
  4631. ],
  4632. "outputs": [
  4633. { "name": "output", "type": "Tensor" }
  4634. ]
  4635. },
  4636. {
  4637. "name": "torch._shape_as_tensor",
  4638. "inputs": [
  4639. { "name": "self", "type": "Tensor" }
  4640. ],
  4641. "outputs": [
  4642. { "name": "output", "type": "Tensor" }
  4643. ]
  4644. },
  4645. {
  4646. "name": "torch.clone",
  4647. "attributes": [
  4648. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  4649. ],
  4650. "inputs": [
  4651. { "name": "self", "type": "Tensor" }
  4652. ],
  4653. "outputs": [
  4654. { "name": "output", "type": "Tensor" }
  4655. ]
  4656. },
  4657. {
  4658. "name": "torch.im2col",
  4659. "attributes": [
  4660. { "name": "kernel_size", "type": "int64[]" },
  4661. { "name": "dilation", "type": "int64[]" },
  4662. { "name": "padding", "type": "int64[]" },
  4663. { "name": "stride", "type": "int64[]" }
  4664. ],
  4665. "inputs": [
  4666. { "name": "self", "type": "Tensor" }
  4667. ],
  4668. "outputs": [
  4669. { "name": "output", "type": "Tensor" }
  4670. ]
  4671. },
  4672. {
  4673. "name": "torch.col2im",
  4674. "attributes": [
  4675. { "name": "output_size", "type": "int64[]" },
  4676. { "name": "kernel_size", "type": "int64[]" },
  4677. { "name": "dilation", "type": "int64[]" },
  4678. { "name": "padding", "type": "int64[]" },
  4679. { "name": "stride", "type": "int64[]" }
  4680. ],
  4681. "inputs": [
  4682. { "name": "self", "type": "Tensor" }
  4683. ],
  4684. "outputs": [
  4685. { "name": "output", "type": "Tensor" }
  4686. ]
  4687. },
  4688. {
  4689. "name": "torch.eq:Tensor",
  4690. "inputs": [
  4691. { "name": "self", "type": "Tensor" },
  4692. { "name": "other", "type": "Tensor" }
  4693. ],
  4694. "outputs": [
  4695. { "name": "output", "type": "Tensor" }
  4696. ]
  4697. },
  4698. {
  4699. "name": "torch.eq:Scalar",
  4700. "inputs": [
  4701. { "name": "self", "type": "Tensor" },
  4702. { "name": "other", "type": "Scalar" }
  4703. ],
  4704. "outputs": [
  4705. { "name": "output", "type": "Tensor" }
  4706. ]
  4707. },
  4708. {
  4709. "name": "torch.where:self",
  4710. "inputs": [
  4711. { "name": "condition", "type": "Tensor" },
  4712. { "name": "self", "type": "Tensor" },
  4713. { "name": "other", "type": "Tensor" }
  4714. ],
  4715. "outputs": [
  4716. { "name": "output", "type": "Tensor" }
  4717. ]
  4718. },
  4719. {
  4720. "name": "torch.__and__:Tensor",
  4721. "inputs": [
  4722. { "name": "self", "type": "Tensor" },
  4723. { "name": "other", "type": "Tensor" }
  4724. ],
  4725. "outputs": [
  4726. { "name": "output", "type": "Tensor" }
  4727. ]
  4728. },
  4729. {
  4730. "name": "torch.__or__:Tensor",
  4731. "inputs": [
  4732. { "name": "self", "type": "Tensor" },
  4733. { "name": "other", "type": "Tensor" }
  4734. ],
  4735. "outputs": [
  4736. { "name": "output", "type": "Tensor" }
  4737. ]
  4738. },
  4739. {
  4740. "name": "torch.ge:Tensor",
  4741. "inputs": [
  4742. { "name": "self", "type": "Tensor" },
  4743. { "name": "other", "type": "Tensor" }
  4744. ],
  4745. "outputs": [
  4746. { "name": "output", "type": "Tensor" }
  4747. ]
  4748. },
  4749. {
  4750. "name": "torch.ge:Scalar",
  4751. "attributes": [
  4752. { "name": "other", "type": "Scalar" }
  4753. ],
  4754. "inputs": [
  4755. { "name": "self", "type": "Tensor" }
  4756. ],
  4757. "outputs": [
  4758. { "name": "output", "type": "Tensor" }
  4759. ]
  4760. },
  4761. {
  4762. "name": "torch.lt:Tensor",
  4763. "inputs": [
  4764. { "name": "self", "type": "Tensor" },
  4765. { "name": "other", "type": "Tensor" }
  4766. ],
  4767. "outputs": [
  4768. { "name": "output", "type": "Tensor" }
  4769. ]
  4770. },
  4771. {
  4772. "name": "torch.lt:Scalar",
  4773. "attributes": [
  4774. { "name": "other", "type": "Scalar" }
  4775. ],
  4776. "inputs": [
  4777. { "name": "self", "type": "Tensor" }
  4778. ],
  4779. "outputs": [
  4780. { "name": "output", "type": "Tensor" }
  4781. ]
  4782. },
  4783. {
  4784. "name": "torch.flip",
  4785. "attributes": [
  4786. { "name": "dims", "type": "int64[]" }
  4787. ],
  4788. "inputs": [
  4789. { "name": "self", "type": "Tensor" }
  4790. ],
  4791. "outputs": [
  4792. { "name": "output", "type": "Tensor" }
  4793. ]
  4794. },
  4795. {
  4796. "name": "torch.linspace",
  4797. "attributes": [
  4798. { "name": "start", "type": "Scalar" },
  4799. { "name": "end", "type": "Scalar" },
  4800. { "name": "steps", "type": "int64", "optional": true, "default": null },
  4801. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  4802. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  4803. { "name": "device", "type": "Device", "optional": true, "default": null },
  4804. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  4805. ],
  4806. "outputs": [
  4807. { "name": "output", "type": "Tensor" }
  4808. ]
  4809. },
  4810. {
  4811. "name": "torch.sign",
  4812. "inputs": [
  4813. { "name": "self", "type": "Tensor" }
  4814. ],
  4815. "outputs": [
  4816. { "name": "output", "type": "Tensor" }
  4817. ]
  4818. },
  4819. {
  4820. "name": "torch.sign_",
  4821. "inputs": [
  4822. { "name": "self", "type": "Tensor" }
  4823. ],
  4824. "outputs": [
  4825. { "name": "output", "type": "Tensor" }
  4826. ]
  4827. },
  4828. {
  4829. "name": "torch.signbit",
  4830. "inputs": [
  4831. { "name": "self", "type": "Tensor" }
  4832. ],
  4833. "outputs": [
  4834. { "name": "output", "type": "Tensor" }
  4835. ]
  4836. },
  4837. {
  4838. "name": "torch.dist",
  4839. "attributes": [
  4840. { "name": "start", "type": "Scalar", "default": 2 }
  4841. ],
  4842. "inputs": [
  4843. { "name": "self", "type": "Tensor" },
  4844. { "name": "other", "type": "Tensor" }
  4845. ],
  4846. "outputs": [
  4847. { "name": "output", "type": "Tensor" }
  4848. ]
  4849. },
  4850. {
  4851. "name": "torch.ceil",
  4852. "inputs": [
  4853. { "name": "self", "type": "Tensor" }
  4854. ],
  4855. "outputs": [
  4856. { "name": "output", "type": "Tensor" }
  4857. ]
  4858. },
  4859. {
  4860. "name": "torch.ceil_",
  4861. "inputs": [
  4862. { "name": "self", "type": "Tensor" }
  4863. ],
  4864. "outputs": [
  4865. { "name": "output", "type": "Tensor" }
  4866. ]
  4867. },
  4868. {
  4869. "name": "torch.einsum",
  4870. "inputs": [
  4871. { "name": "equation", "type": "string" },
  4872. { "name": "tensors", "type": "Tensor[]" }
  4873. ],
  4874. "outputs": [
  4875. { "name": "output", "type": "Tensor" }
  4876. ]
  4877. },
  4878. {
  4879. "name": "torch.dot",
  4880. "inputs": [
  4881. { "name": "self", "type": "Tensor" },
  4882. { "name": "tensor", "type": "Tensor" }
  4883. ],
  4884. "outputs": [
  4885. { "name": "output", "type": "Tensor" }
  4886. ]
  4887. },
  4888. {
  4889. "name": "torch.vdot",
  4890. "inputs": [
  4891. { "name": "self", "type": "Tensor" },
  4892. { "name": "other", "type": "Tensor" }
  4893. ],
  4894. "outputs": [
  4895. { "name": "output", "type": "Tensor" }
  4896. ]
  4897. },
  4898. {
  4899. "name": "torch.var_mean:",
  4900. "attributes": [
  4901. { "name": "unbiased", "type": "boolean", "default": true }
  4902. ],
  4903. "inputs": [
  4904. { "name": "self", "type": "Tensor" }
  4905. ],
  4906. "outputs": [
  4907. { "name": "output", "type": "Tensor" },
  4908. { "name": "?", "type": "Tensor" }
  4909. ]
  4910. },
  4911. {
  4912. "name": "torch.var_mean:dim",
  4913. "attributes": [
  4914. { "name": "dim", "type": "int64[]" },
  4915. { "name": "unbiased", "type": "boolean", "default": true },
  4916. { "name": "keepdim", "type": "boolean", "default": false }
  4917. ],
  4918. "inputs": [
  4919. { "name": "self", "type": "Tensor" }
  4920. ],
  4921. "outputs": [
  4922. { "name": "output", "type": "Tensor" },
  4923. { "name": "?", "type": "Tensor" }
  4924. ]
  4925. },
  4926. {
  4927. "name": "torch.all",
  4928. "inputs": [
  4929. { "name": "self", "type": "Tensor" }
  4930. ],
  4931. "outputs": [
  4932. { "name": "output", "type": "Tensor" }
  4933. ]
  4934. },
  4935. {
  4936. "name": "torch.any",
  4937. "inputs": [
  4938. { "name": "self", "type": "Tensor" }
  4939. ],
  4940. "outputs": [
  4941. { "name": "output", "type": "Tensor" }
  4942. ]
  4943. },
  4944. {
  4945. "name": "torch.renorm",
  4946. "attributes": [
  4947. { "name": "p", "type": "Scalar" },
  4948. { "name": "dim", "type": "int64" },
  4949. { "name": "maxnorm", "type": "Scalar" }
  4950. ],
  4951. "inputs": [
  4952. { "name": "self", "type": "Tensor" }
  4953. ],
  4954. "outputs": [
  4955. { "name": "output", "type": "Tensor" }
  4956. ]
  4957. },
  4958. {
  4959. "name": "torch.unfold",
  4960. "attributes": [
  4961. { "name": "dimension", "type": "int64" },
  4962. { "name": "size", "type": "int64" },
  4963. { "name": "step", "type": "int64" }
  4964. ],
  4965. "inputs": [
  4966. { "name": "self", "type": "Tensor" }
  4967. ],
  4968. "outputs": [
  4969. { "name": "output", "type": "Tensor" }
  4970. ]
  4971. },
  4972. {
  4973. "name": "torch.fake_quantize_per_channel_affine",
  4974. "inputs": [
  4975. { "name": "self", "type": "Tensor" },
  4976. { "name": "scale", "type": "Tensor" },
  4977. { "name": "zero_point", "type": "Tensor" },
  4978. { "name": "axis", "type": "int64" },
  4979. { "name": "quant_min", "type": "int64" },
  4980. { "name": "quant_max", "type": "int64" }
  4981. ],
  4982. "outputs": [
  4983. { "name": "output", "type": "Tensor" }
  4984. ]
  4985. },
  4986. {
  4987. "name": "torch.fake_quantize_per_tensor_affine",
  4988. "inputs": [
  4989. { "name": "self", "type": "Tensor" },
  4990. { "name": "scale", "type": "float32" },
  4991. { "name": "zero_point", "type": "int64" },
  4992. { "name": "quant_min", "type": "int64" },
  4993. { "name": "quant_max", "type": "int64" }
  4994. ],
  4995. "outputs": [
  4996. { "name": "output", "type": "Tensor" }
  4997. ]
  4998. },
  4999. {
  5000. "name": "torch.addmv",
  5001. "inputs": [
  5002. { "name": "self", "type": "Tensor" },
  5003. { "name": "mat", "type": "Tensor" },
  5004. { "name": "vec", "type": "Tensor" },
  5005. { "name": "beta", "type": "Scalar", "default": 1 },
  5006. { "name": "alpha", "type": "Scalar", "default": 1 }
  5007. ],
  5008. "outputs": [
  5009. { "name": "output", "type": "Tensor" }
  5010. ]
  5011. },
  5012. {
  5013. "name": "torch.addmv_",
  5014. "inputs": [
  5015. { "name": "self", "type": "Tensor" },
  5016. { "name": "mat", "type": "Tensor" },
  5017. { "name": "vec", "type": "Tensor" },
  5018. { "name": "beta", "type": "Scalar", "default": 1 },
  5019. { "name": "alpha", "type": "Scalar", "default": 1 }
  5020. ],
  5021. "outputs": [
  5022. { "name": "output", "type": "Tensor" }
  5023. ]
  5024. },
  5025. {
  5026. "name": "torch.type_as",
  5027. "inputs": [
  5028. { "name": "self", "type": "Tensor" },
  5029. { "name": "other", "type": "Tensor" }
  5030. ],
  5031. "outputs": [
  5032. { "name": "output", "type": "Tensor" }
  5033. ]
  5034. },
  5035. {
  5036. "name": "torch.normal_",
  5037. "inputs": [
  5038. { "name": "self", "type": "Tensor" },
  5039. { "name": "mean", "type": "float32", "default": 0 },
  5040. { "name": "std", "type": "float32", "default": 1 },
  5041. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5042. ],
  5043. "outputs": [
  5044. { "name": "output", "type": "Tensor" }
  5045. ]
  5046. },
  5047. {
  5048. "name": "torch.normal:Tensor_float",
  5049. "inputs": [
  5050. { "name": "self", "type": "Tensor" },
  5051. { "name": "mean", "type": "float32", "default": 0 },
  5052. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5053. ],
  5054. "outputs": [
  5055. { "name": "output", "type": "Tensor" }
  5056. ]
  5057. },
  5058. {
  5059. "name": "torch.normal:float_Tensor",
  5060. "inputs": [
  5061. { "name": "mean", "type": "float32" },
  5062. { "name": "std", "type": "Tensor" },
  5063. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5064. ],
  5065. "outputs": [
  5066. { "name": "output", "type": "Tensor" }
  5067. ]
  5068. },
  5069. {
  5070. "name": "torch.normal:Tensor_Tensor",
  5071. "inputs": [
  5072. { "name": "mean", "type": "Tensor" },
  5073. { "name": "std", "type": "Tensor" },
  5074. { "name": "generator", "type": "Generator", "optional": true, "default": null }
  5075. ],
  5076. "outputs": [
  5077. { "name": "output", "type": "Tensor" }
  5078. ]
  5079. },
  5080. {
  5081. "name": "torch.normal:float_float",
  5082. "inputs": [
  5083. { "name": "mean", "type": "float32", "default": 0 },
  5084. { "name": "std", "type": "float32" },
  5085. { "name": "size", "type": "int64[]" },
  5086. { "name": "generator", "type": "Generator", "optional": true, "default": null },
  5087. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5088. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5089. { "name": "device", "type": "Device", "optional": true, "default": null },
  5090. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5091. ],
  5092. "outputs": [
  5093. { "name": "output", "type": "Tensor" }
  5094. ]
  5095. },
  5096. {
  5097. "name": "torch.empty_like",
  5098. "inputs": [
  5099. { "name": "self", "type": "Tensor" },
  5100. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5101. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5102. { "name": "device", "type": "Device", "optional": true, "default": null },
  5103. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  5104. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  5105. ],
  5106. "outputs": [
  5107. { "name": "output", "type": "Tensor" }
  5108. ]
  5109. },
  5110. {
  5111. "name": "torch.classes._nnapi.Compilation",
  5112. "inputs": [
  5113. { "name": "serialized_model", "type": "Tensor" },
  5114. { "name": "inputs", "type": "Tensor[]" },
  5115. { "name": "parameter_buffers", "type": "Tensor[]" }
  5116. ],
  5117. "outputs": [
  5118. { "name": "output", "type": "Tensor[]" }
  5119. ]
  5120. },
  5121. {
  5122. "name": "torch.argsort:",
  5123. "inputs": [
  5124. { "name": "self", "type": "Tensor" },
  5125. { "name": "dim", "type": "int64", "default": -1 },
  5126. { "name": "descending", "type": "boolean", "default": false }
  5127. ],
  5128. "outputs": [
  5129. { "name": "output", "type": "Tensor" }
  5130. ]
  5131. },
  5132. {
  5133. "name": "torch.argsort:dimname",
  5134. "inputs": [
  5135. { "name": "self", "type": "Tensor" },
  5136. { "name": "dim", "type": "Dimname" },
  5137. { "name": "descending", "type": "boolean", "default": false }
  5138. ],
  5139. "outputs": [
  5140. { "name": "output", "type": "Tensor" }
  5141. ]
  5142. },
  5143. {
  5144. "name": "ops._caffe2.CopyGPUToCPU",
  5145. "inputs": [
  5146. { "name": "input", "type": "Tensor" },
  5147. { "name": "?" }
  5148. ],
  5149. "outputs": [
  5150. { "name": "output", "type": "Tensor" }
  5151. ]
  5152. },
  5153. {
  5154. "name": "ops._caffe2.CopyCPUToGPU",
  5155. "inputs": [
  5156. { "name": "input", "type": "Tensor" },
  5157. { "name": "?" }
  5158. ],
  5159. "outputs": [
  5160. { "name": "output", "type": "Tensor" }
  5161. ]
  5162. },
  5163. {
  5164. "name": "ops._caffe2.GenerateProposals",
  5165. "inputs": [
  5166. { "name": "scores", "type": "Tensor" },
  5167. { "name": "bbox_deltas", "type": "Tensor" },
  5168. { "name": "im_info", "type": "Tensor" },
  5169. { "name": "anchors", "type": "Tensor" },
  5170. { "name": "spatial_scale", "type": "float32" },
  5171. { "name": "pre_nms_topN", "type": "int64" },
  5172. { "name": "post_nms_topN", "type": "int64" },
  5173. { "name": "nms_thresh", "type": "float32" },
  5174. { "name": "min_size", "type": "float32" },
  5175. { "name": "angle_bound_on", "type": "boolean" },
  5176. { "name": "angle_bound_lo", "type": "int64" },
  5177. { "name": "angle_bound_hi", "type": "int64" },
  5178. { "name": "clip_angle_thresh", "type": "float32" },
  5179. { "name": "legacy_plus_one", "type": "boolean" },
  5180. { "name": "?" }
  5181. ],
  5182. "outputs": [
  5183. { "name": "output_0", "type": "Tensor" },
  5184. { "name": "output_1", "type": "Tensor" }
  5185. ]
  5186. },
  5187. {
  5188. "name": "ops._caffe2.CollectRpnProposals",
  5189. "inputs": [
  5190. { "name": "input_list", "type": "Tensor[]" },
  5191. { "name": "rpn_max_level", "type": "int64" },
  5192. { "name": "rpn_min_level", "type": "int64" },
  5193. { "name": "rpn_post_nms_topN", "type": "int64" },
  5194. { "name": "?" }
  5195. ],
  5196. "outputs": [
  5197. { "name": "rois", "type": "Tensor" }
  5198. ]
  5199. },
  5200. {
  5201. "name": "ops._caffe2.DistributeFpnProposals",
  5202. "inputs": [
  5203. { "name": "rois", "type": "Tensor" },
  5204. { "name": "roi_canonical_scale", "type": "int64" },
  5205. { "name": "roi_canonical_level", "type": "int64" },
  5206. { "name": "roi_max_level", "type": "int64" },
  5207. { "name": "roi_min_level", "type": "int64" },
  5208. { "name": "legacy_plus_one", "type": "boolean" },
  5209. { "name": "?" }
  5210. ],
  5211. "outputs": [
  5212. { "name": "rois_fpn2", "type": "Tensor" },
  5213. { "name": "rois_fpn3", "type": "Tensor" },
  5214. { "name": "rois_fpn4", "type": "Tensor" },
  5215. { "name": "rois_fpn5", "type": "Tensor" },
  5216. { "name": "rois_idx_restore_int32", "type": "Tensor" }
  5217. ]
  5218. },
  5219. {
  5220. "name": "ops._caffe2.RoIAlign",
  5221. "inputs": [
  5222. { "name": "features", "type": "Tensor" },
  5223. { "name": "rois", "type": "Tensor" },
  5224. { "name": "order", "type": "string" },
  5225. { "name": "spatial_scale", "type": "float32" },
  5226. { "name": "pooled_h", "type": "int64" },
  5227. { "name": "pooled_w", "type": "int64" },
  5228. { "name": "sampling_ratio", "type": "int64" },
  5229. { "name": "aligned", "type": "boolean" },
  5230. { "name": "?" }
  5231. ],
  5232. "outputs": [
  5233. { "name": "output", "type": "Tensor" }
  5234. ]
  5235. },
  5236. {
  5237. "name": "ops._caffe2.BatchPermutation",
  5238. "inputs": [
  5239. { "name": "X", "type": "Tensor" },
  5240. { "name": "indices", "type": "Tensor" },
  5241. { "name": "?" }
  5242. ],
  5243. "outputs": [
  5244. { "name": "output", "type": "Tensor" }
  5245. ]
  5246. },
  5247. {
  5248. "name": "ops._caffe2.BBoxTransform",
  5249. "inputs": [
  5250. { "name": "rois", "type": "Tensor" },
  5251. { "name": "deltas", "type": "Tensor" },
  5252. { "name": "im_info", "type": "Tensor" },
  5253. { "name": "weights", "type": "float32[]" },
  5254. { "name": "apply_scale", "type": "boolean" },
  5255. { "name": "rotated", "type": "boolean" },
  5256. { "name": "angle_bound_on", "type": "boolean" },
  5257. { "name": "angle_bound_lo", "type": "int64" },
  5258. { "name": "angle_bound_hi", "type": "int64" },
  5259. { "name": "clip_angle_thresh", "type": "float32" },
  5260. { "name": "legacy_plus_one", "type": "boolean" },
  5261. { "name": "?" }
  5262. ],
  5263. "outputs": [
  5264. { "name": "output_0", "type": "Tensor" },
  5265. { "name": "output_1", "type": "Tensor" }
  5266. ]
  5267. },
  5268. {
  5269. "name": "ops._caffe2.BoxWithNMSLimit",
  5270. "inputs": [
  5271. { "name": "scores", "type": "Tensor" },
  5272. { "name": "boxes", "type": "Tensor" },
  5273. { "name": "batch_splits", "type": "Tensor" },
  5274. { "name": "score_thresh", "type": "float32" },
  5275. { "name": "nms", "type": "float32" },
  5276. { "name": "detections_per_im", "type": "int64" },
  5277. { "name": "soft_nms_enabled", "type": "boolean" },
  5278. { "name": "soft_nms_method", "type": "string" },
  5279. { "name": "soft_nms_sigma", "type": "float32" },
  5280. { "name": "soft_nms_min_score_thres", "type": "float32" },
  5281. { "name": "rotated", "type": "boolean" },
  5282. { "name": "cls_agnostic_bbox_reg", "type": "boolean" },
  5283. { "name": "input_boxes_include_bg_cls", "type": "boolean" },
  5284. { "name": "output_classes_include_bg_cls", "type": "boolean" },
  5285. { "name": "legacy_plus_one", "type": "boolean" },
  5286. { "name": "?" }
  5287. ],
  5288. "outputs": [
  5289. { "name": "scores", "type": "Tensor" },
  5290. { "name": "boxes", "type": "Tensor" },
  5291. { "name": "classes", "type": "Tensor" },
  5292. { "name": "batch_splits", "type": "Tensor" },
  5293. { "name": "keeps", "type": "Tensor" },
  5294. { "name": "keeps_size", "type": "Tensor" }
  5295. ]
  5296. },
  5297. {
  5298. "name": "torch.repeat_interleave:Tensor",
  5299. "inputs": [
  5300. { "name": "repeats", "type": "Tensor" }
  5301. ],
  5302. "outputs": [
  5303. { "name": "output", "type": "Tensor" }
  5304. ]
  5305. },
  5306. {
  5307. "name": "torch.repeat_interleave:self_Tensor",
  5308. "inputs": [
  5309. { "name": "self", "type": "Tensor" },
  5310. { "name": "repeats", "type": "Tensor" },
  5311. { "name": "dim", "type": "int64", "optional": true, "default": null }
  5312. ],
  5313. "outputs": [
  5314. { "name": "output", "type": "Tensor" }
  5315. ]
  5316. },
  5317. {
  5318. "name": "torch.repeat_interleave:self_int",
  5319. "inputs": [
  5320. { "name": "self", "type": "Tensor" },
  5321. { "name": "repeats", "type": "int64" },
  5322. { "name": "dim", "type": "int64", "optional": true, "default": null }
  5323. ],
  5324. "outputs": [
  5325. { "name": "output", "type": "Tensor" }
  5326. ]
  5327. },
  5328. {
  5329. "name": "torch.cpu",
  5330. "inputs": [
  5331. { "name": "self", "type": "Tensor" }
  5332. ],
  5333. "outputs": [
  5334. { "name": "output", "type": "Tensor" }
  5335. ]
  5336. },
  5337. {
  5338. "name": "ops.quantized.hardswish",
  5339. "category": "Activation",
  5340. "inputs": [
  5341. { "name": "input", "type": "Tensor" },
  5342. { "name": "output_scale", "type": "float32" },
  5343. { "name": "output_zero_point", "type": "int64" }
  5344. ],
  5345. "outputs": [
  5346. { "name": "output", "type": "Tensor" }
  5347. ]
  5348. },
  5349. {
  5350. "name": "ops.quantized.celu",
  5351. "category": "Activation",
  5352. "inputs": [
  5353. { "name": "self", "type": "Tensor" },
  5354. { "name": "output_scale", "type": "float32" },
  5355. { "name": "output_zero_point", "type": "int64" },
  5356. { "name": "alpha", "type": "Scalar", "default": 1 }
  5357. ],
  5358. "outputs": [
  5359. { "name": "output", "type": "Tensor" }
  5360. ]
  5361. },
  5362. {
  5363. "name": "ops.quantized.mul:",
  5364. "inputs": [
  5365. { "name": "qa", "type": "Tensor" },
  5366. { "name": "qb", "type": "Tensor" },
  5367. { "name": "scale", "type": "float32" },
  5368. { "name": "zero_point", "type": "int64" }
  5369. ],
  5370. "outputs": [
  5371. { "name": "qc", "type": "Tensor" }
  5372. ]
  5373. },
  5374. {
  5375. "name": "ops.quantized.mul:Scalar",
  5376. "inputs": [
  5377. { "name": "qa", "type": "Tensor" },
  5378. { "name": "b", "type": "Scalar" }
  5379. ],
  5380. "outputs": [
  5381. { "name": "qc", "type": "Tensor" }
  5382. ]
  5383. },
  5384. {
  5385. "name": "ops.quantized.mul:Scalar2",
  5386. "inputs": [
  5387. { "name": "b", "type": "Scalar" },
  5388. { "name": "qa", "type": "Tensor" }
  5389. ],
  5390. "outputs": [
  5391. { "name": "qc", "type": "Tensor" }
  5392. ]
  5393. },
  5394. {
  5395. "name": "ops.quantized.mul_relu:Scalar",
  5396. "inputs": [
  5397. { "name": "qa", "type": "Tensor" },
  5398. { "name": "b", "type": "Scalar" }
  5399. ],
  5400. "outputs": [
  5401. { "name": "qc", "type": "Tensor" }
  5402. ]
  5403. },
  5404. {
  5405. "name": "ops.quantized.mul_relu:Scalar2",
  5406. "inputs": [
  5407. { "name": "b", "type": "Scalar" },
  5408. { "name": "qa", "type": "Tensor" }
  5409. ],
  5410. "outputs": [
  5411. { "name": "qc", "type": "Tensor" }
  5412. ]
  5413. },
  5414. {
  5415. "name": "ops.quantized.mul_scalar",
  5416. "inputs": [
  5417. { "name": "qa", "type": "Tensor" },
  5418. { "name": "qb", "type": "Scalar" }
  5419. ],
  5420. "outputs": [
  5421. { "name": "qc", "type": "Tensor" }
  5422. ]
  5423. },
  5424. {
  5425. "name": "ops.quantized.mul_scalar_relu",
  5426. "inputs": [
  5427. { "name": "qa", "type": "Tensor" },
  5428. { "name": "qb", "type": "Scalar" }
  5429. ],
  5430. "outputs": [
  5431. { "name": "qc", "type": "Tensor" }
  5432. ]
  5433. },
  5434. {
  5435. "name": "ops.quantized.relu6",
  5436. "category": "Activation",
  5437. "inputs": [
  5438. { "name": "qx", "type": "Tensor" },
  5439. { "name": "inplace", "type": "boolean", "default": false }
  5440. ],
  5441. "outputs": [
  5442. { "name": "output", "type": "Tensor" }
  5443. ]
  5444. },
  5445. {
  5446. "name": "ops.quantized.leaky_relu",
  5447. "category": "Activation",
  5448. "inputs": [
  5449. { "name": "qx", "type": "Tensor" },
  5450. { "name": "negative_slope", "type": "Scalar", "default": false },
  5451. { "name": "inplace", "type": "boolean" },
  5452. { "name": "output_scale", "type": "float32" },
  5453. { "name": "output_zero_point", "type": "int64" }
  5454. ],
  5455. "outputs": [
  5456. { "name": "output", "type": "Tensor" }
  5457. ]
  5458. },
  5459. {
  5460. "name": "ops.quantized.sigmoid",
  5461. "category": "Activation",
  5462. "inputs": [
  5463. { "name": "qx", "type": "Tensor" },
  5464. { "name": "output_scale", "type": "float32" },
  5465. { "name": "output_zero_point", "type": "int64" }
  5466. ],
  5467. "outputs": [
  5468. { "name": "output", "type": "Tensor" }
  5469. ]
  5470. },
  5471. {
  5472. "name": "torch.full:names",
  5473. "inputs": [
  5474. { "name": "size", "type": "int64[]" },
  5475. { "name": "fill_value", "type": "Scalar" },
  5476. { "name": "names", "type": "Dimname[]", "optional": true },
  5477. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5478. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5479. { "name": "device", "type": "Device", "optional": true, "default": null },
  5480. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5481. ],
  5482. "outputs": [
  5483. { "name": "output", "type": "Tensor" }
  5484. ]
  5485. },
  5486. {
  5487. "name": "torch.full:",
  5488. "inputs": [
  5489. { "name": "size", "type": "int64[]" },
  5490. { "name": "fill_value", "type": "Scalar" },
  5491. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5492. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5493. { "name": "device", "type": "Device", "optional": true, "default": null },
  5494. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5495. ],
  5496. "outputs": [
  5497. { "name": "output", "type": "Tensor" }
  5498. ]
  5499. },
  5500. {
  5501. "name": "torch.full:out",
  5502. "inputs": [
  5503. { "name": "size", "type": "int64[]" },
  5504. { "name": "fill_value", "type": "Scalar" }
  5505. ],
  5506. "outputs": [
  5507. { "name": "output", "type": "Tensor" }
  5508. ]
  5509. },
  5510. {
  5511. "name": "torch.full_like:",
  5512. "inputs": [
  5513. { "name": "self", "type": "Tensor" },
  5514. { "name": "fill_value", "type": "Scalar" },
  5515. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5516. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5517. { "name": "device", "type": "Device", "optional": true, "default": null },
  5518. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null },
  5519. { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null }
  5520. ],
  5521. "outputs": [
  5522. { "name": "output", "type": "Tensor" }
  5523. ]
  5524. },
  5525. {
  5526. "name": "torch.new_empty:",
  5527. "inputs": [
  5528. { "name": "self", "type": "Tensor" },
  5529. { "name": "size", "type": "int64[]" },
  5530. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5531. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5532. { "name": "device", "type": "Device", "optional": true, "default": null },
  5533. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5534. ],
  5535. "outputs": [
  5536. { "name": "output", "type": "Tensor" }
  5537. ]
  5538. },
  5539. {
  5540. "name": "torch.new_empty_strided:",
  5541. "inputs": [
  5542. { "name": "self", "type": "Tensor" },
  5543. { "name": "size", "type": "int64[]" },
  5544. { "name": "stride", "type": "int64[]" },
  5545. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5546. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5547. { "name": "device", "type": "Device", "optional": true, "default": null },
  5548. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5549. ],
  5550. "outputs": [
  5551. { "name": "output", "type": "Tensor" }
  5552. ]
  5553. },
  5554. {
  5555. "name": "torch.linear:",
  5556. "category": "Layer",
  5557. "inputs": [
  5558. { "name": "input", "type": "Tensor" },
  5559. { "name": "weight", "type": "Tensor" },
  5560. { "name": "bias", "type": "Tensor", "optional": true, "default": null }
  5561. ],
  5562. "outputs": [
  5563. { "name": "output", "type": "Tensor" }
  5564. ]
  5565. },
  5566. {
  5567. "name": "torch._weight_norm",
  5568. "inputs": [
  5569. { "name": "v", "type": "Tensor" },
  5570. { "name": "g", "type": "Tensor" },
  5571. { "name": "dim", "type": "int64", "default": 0 }
  5572. ],
  5573. "outputs": [
  5574. { "name": "output", "type": "Tensor" }
  5575. ]
  5576. },
  5577. {
  5578. "name": "torch.requires_grad_",
  5579. "inputs": [
  5580. { "name": "self", "type": "Tensor" },
  5581. { "name": "requires_grad", "type": "boolean" }
  5582. ],
  5583. "outputs": [
  5584. { "name": "output", "type": "Tensor" }
  5585. ]
  5586. },
  5587. {
  5588. "name": "torch.stft",
  5589. "inputs": [
  5590. { "name": "self", "type": "Tensor" },
  5591. { "name": "n_fft", "type": "int64" },
  5592. { "name": "hop_length", "type": "int64", "optional": true, "default": null },
  5593. { "name": "win_length", "type": "int64", "optional": true, "default": null },
  5594. { "name": "window", "type": "Tensor", "optional": true, "default": null },
  5595. { "name": "normalized", "type": "boolean", "default": false },
  5596. { "name": "onesided", "type": "boolean", "optional": true, "default": null },
  5597. { "name": "return_complex", "type": "boolean", "optional": true, "default": null }
  5598. ],
  5599. "outputs": [
  5600. { "name": "output", "type": "Tensor" }
  5601. ]
  5602. },
  5603. {
  5604. "name": "torch.istft",
  5605. "inputs": [
  5606. { "name": "self", "type": "Tensor" },
  5607. { "name": "n_fft", "type": "int64" },
  5608. { "name": "hop_length", "type": "int64", "optional": true, "default": null },
  5609. { "name": "win_length", "type": "int64", "optional": true, "default": null },
  5610. { "name": "window", "type": "Tensor", "optional": true, "default": null },
  5611. { "name": "center", "type": "boolean", "default": true },
  5612. { "name": "normalized", "type": "boolean", "default": false },
  5613. { "name": "onesided", "type": "boolean", "optional": true, "default": null },
  5614. { "name": "length", "type": "boolean", "optional": true, "default": null },
  5615. { "name": "return_complex", "type": "boolean", "default": false }
  5616. ],
  5617. "outputs": [
  5618. { "name": "output", "type": "Tensor" }
  5619. ]
  5620. },
  5621. {
  5622. "name": "torch.unbind:int",
  5623. "inputs": [
  5624. { "name": "self", "type": "Tensor" },
  5625. { "name": "dim", "type": "int64", "default": 0 }
  5626. ],
  5627. "outputs": [
  5628. { "name": "output", "type": "Tensor[]" }
  5629. ]
  5630. },
  5631. {
  5632. "name": "torch.unbind:Dimname",
  5633. "inputs": [
  5634. { "name": "self", "type": "Tensor" },
  5635. { "name": "dim", "type": "Dimname" }
  5636. ],
  5637. "outputs": [
  5638. { "name": "output", "type": "Tensor[]" }
  5639. ]
  5640. },
  5641. {
  5642. "name": "torch.as_tensor",
  5643. "inputs": [
  5644. { "name": "data" },
  5645. { "name": "dtype" },
  5646. { "name": "device" }
  5647. ],
  5648. "outputs": [
  5649. { "name": "output", "type": "Tensor" }
  5650. ]
  5651. },
  5652. {
  5653. "name": "torch.tensor",
  5654. "inputs": [
  5655. { "name": "data" },
  5656. { "name": "dtype" },
  5657. { "name": "device" },
  5658. { "name": "requires_grad", "type": "boolean" }
  5659. ],
  5660. "outputs": [
  5661. { "name": "output", "type": "Tensor" }
  5662. ]
  5663. },
  5664. {
  5665. "name": "torch.__upsample:",
  5666. "category": "Layer",
  5667. "inputs": [
  5668. { "name": "input", "type": "Tensor" },
  5669. { "name": "size", "type": "int64", "optional": true, "default": null },
  5670. { "name": "scale_factor", "type": "int64", "optional": true, "default": null },
  5671. { "name": "mode", "type": "string", "default": "nearest" },
  5672. { "name": "align_corners", "type": "boolean", "optional": true, "default": null }
  5673. ],
  5674. "outputs": [
  5675. { "name": "output", "type": "Tensor" }
  5676. ]
  5677. },
  5678. {
  5679. "name": "torch.__upsample:size_list",
  5680. "category": "Layer",
  5681. "inputs": [
  5682. { "name": "input", "type": "Tensor" },
  5683. { "name": "size", "type": "int64[]", "optional": true, "default": null },
  5684. { "name": "scale_factor", "type": "int64", "optional": true, "default": null },
  5685. { "name": "mode", "type": "string", "default": "nearest" },
  5686. { "name": "align_corners", "type": "boolean", "optional": true, "default": null }
  5687. ],
  5688. "outputs": [
  5689. { "name": "output", "type": "Tensor" }
  5690. ]
  5691. },
  5692. {
  5693. "name": "torch.triu_",
  5694. "category": "Layer",
  5695. "inputs": [
  5696. { "name": "self", "type": "Tensor" },
  5697. { "name": "diagonal", "type": "int64", "default": 0 }
  5698. ],
  5699. "outputs": [
  5700. { "name": "output", "type": "Tensor" }
  5701. ]
  5702. },
  5703. {
  5704. "name": "torch.triu:out",
  5705. "category": "Layer",
  5706. "inputs": [
  5707. { "name": "self", "type": "Tensor" },
  5708. { "name": "diagonal", "type": "int64", "default": 0 }
  5709. ],
  5710. "outputs": [
  5711. { "name": "output", "type": "Tensor" }
  5712. ]
  5713. },
  5714. {
  5715. "name": "torch.view_as_real",
  5716. "category": "Layer",
  5717. "inputs": [
  5718. { "name": "self", "type": "Tensor" }
  5719. ],
  5720. "outputs": [
  5721. { "name": "output", "type": "Tensor" }
  5722. ]
  5723. },
  5724. {
  5725. "name": "torch.view_as_complex",
  5726. "category": "Layer",
  5727. "inputs": [
  5728. { "name": "self", "type": "Tensor" }
  5729. ],
  5730. "outputs": [
  5731. { "name": "output", "type": "Tensor" }
  5732. ]
  5733. },
  5734. {
  5735. "name": "ops.torchvision.nms",
  5736. "inputs": [
  5737. { "name": "dets", "type": "Tensor" },
  5738. { "name": "scores", "type": "Tensor" },
  5739. { "name": "iou_threshold", "type": "float32" }
  5740. ],
  5741. "outputs": [
  5742. { "name": "output", "type": "Tensor" }
  5743. ]
  5744. },
  5745. {
  5746. "name": "ops.torchvision.roi_align",
  5747. "inputs": [
  5748. { "name": "input", "type": "Tensor" },
  5749. { "name": "rois", "type": "Tensor" },
  5750. { "name": "spatial_scale", "type": "float32" },
  5751. { "name": "pooled_height", "type": "int64" },
  5752. { "name": "pooled_width", "type": "int64" },
  5753. { "name": "sampling_ratio", "type": "int64" },
  5754. { "name": "aligned", "type": "boolean" }
  5755. ],
  5756. "outputs": [
  5757. { "name": "output", "type": "Tensor" }
  5758. ]
  5759. },
  5760. {
  5761. "name": "torch.__xor__:Scalar",
  5762. "inputs": [
  5763. { "name": "self", "type": "Tensor" },
  5764. { "name": "other", "type": "Scalar" }
  5765. ],
  5766. "outputs": [
  5767. { "name": "output", "type": "Tensor" }
  5768. ]
  5769. },
  5770. {
  5771. "name": "torch.__xor__:Tensor",
  5772. "inputs": [
  5773. { "name": "self", "type": "Tensor" },
  5774. { "name": "other", "type": "Tensor" }
  5775. ],
  5776. "outputs": [
  5777. { "name": "output", "type": "Tensor" }
  5778. ]
  5779. },
  5780. {
  5781. "name": "torch.__lshift__:Scalar",
  5782. "inputs": [
  5783. { "name": "self", "type": "Tensor" },
  5784. { "name": "other", "type": "Scalar" }
  5785. ],
  5786. "outputs": [
  5787. { "name": "output", "type": "Tensor" }
  5788. ]
  5789. },
  5790. {
  5791. "name": "torch.__lshift__:Tensor",
  5792. "inputs": [
  5793. { "name": "self", "type": "Tensor" },
  5794. { "name": "other", "type": "Tensor" }
  5795. ],
  5796. "outputs": [
  5797. { "name": "output", "type": "Tensor" }
  5798. ]
  5799. },
  5800. {
  5801. "name": "torch.__rshift__:Scalar",
  5802. "inputs": [
  5803. { "name": "self", "type": "Tensor" },
  5804. { "name": "other", "type": "Scalar" }
  5805. ],
  5806. "outputs": [
  5807. { "name": "output", "type": "Tensor" }
  5808. ]
  5809. },
  5810. {
  5811. "name": "torch.__rshift__:Tensor",
  5812. "inputs": [
  5813. { "name": "self", "type": "Tensor" },
  5814. { "name": "other", "type": "Tensor" }
  5815. ],
  5816. "outputs": [
  5817. { "name": "output", "type": "Tensor" }
  5818. ]
  5819. },
  5820. {
  5821. "name": "torch.__ixor__:Scalar",
  5822. "inputs": [
  5823. { "name": "self", "type": "Tensor" },
  5824. { "name": "other", "type": "Scalar" }
  5825. ],
  5826. "outputs": [
  5827. { "name": "output", "type": "Tensor" }
  5828. ]
  5829. },
  5830. {
  5831. "name": "torch.__ixor__:Tensor",
  5832. "inputs": [
  5833. { "name": "self", "type": "Tensor" },
  5834. { "name": "other", "type": "Tensor" }
  5835. ],
  5836. "outputs": [
  5837. { "name": "output", "type": "Tensor" }
  5838. ]
  5839. },
  5840. {
  5841. "name": "torch.__ilshift__:Scalar",
  5842. "inputs": [
  5843. { "name": "self", "type": "Tensor" },
  5844. { "name": "other", "type": "Scalar" }
  5845. ],
  5846. "outputs": [
  5847. { "name": "output", "type": "Tensor" }
  5848. ]
  5849. },
  5850. {
  5851. "name": "torch.__ilshift__:Tensor",
  5852. "inputs": [
  5853. { "name": "self", "type": "Tensor" },
  5854. { "name": "other", "type": "Tensor" }
  5855. ],
  5856. "outputs": [
  5857. { "name": "output", "type": "Tensor" }
  5858. ]
  5859. },
  5860. {
  5861. "name": "torch.__irshift__:Scalar",
  5862. "inputs": [
  5863. { "name": "self", "type": "Tensor" },
  5864. { "name": "other", "type": "Scalar" }
  5865. ],
  5866. "outputs": [
  5867. { "name": "output", "type": "Tensor" }
  5868. ]
  5869. },
  5870. {
  5871. "name": "torch.__irshift__:Tensor",
  5872. "inputs": [
  5873. { "name": "self", "type": "Tensor" },
  5874. { "name": "other", "type": "Tensor" }
  5875. ],
  5876. "outputs": [
  5877. { "name": "output", "type": "Tensor" }
  5878. ]
  5879. },
  5880. {
  5881. "name": "torch.one_hot:",
  5882. "inputs": [
  5883. { "name": "self", "type": "Tensor" },
  5884. { "name": "num_classes", "type": "int64", "default": -1 }
  5885. ],
  5886. "outputs": [
  5887. { "name": "output", "type": "Tensor" }
  5888. ]
  5889. },
  5890. {
  5891. "name": "torch.hann_window:",
  5892. "inputs": [
  5893. { "name": "window_length", "type": "int64" },
  5894. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5895. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5896. { "name": "device", "type": "Device", "optional": true, "default": null },
  5897. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5898. ],
  5899. "outputs": [
  5900. { "name": "output", "type": "Tensor" }
  5901. ]
  5902. },
  5903. {
  5904. "name": "torch.hann_window:periodic",
  5905. "inputs": [
  5906. { "name": "window_length", "type": "int64" },
  5907. { "name": "periodic", "type": "boolean" },
  5908. { "name": "dtype", "type": "ScalarType", "optional": true, "default": null },
  5909. { "name": "layout", "type": "Layout", "optional": true, "default": null },
  5910. { "name": "device", "type": "Device", "optional": true, "default": null },
  5911. { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }
  5912. ],
  5913. "outputs": [
  5914. { "name": "output", "type": "Tensor" }
  5915. ]
  5916. },
  5917. {
  5918. "name": "torch.nonzero_numpy",
  5919. "inputs": [
  5920. { "name": "self", "type": "Tensor" }
  5921. ],
  5922. "outputs": [
  5923. { "name": "output", "type": "Tensor" }
  5924. ]
  5925. }
  5926. ]