2021-09-11 03:21:29 +02:00
/ *
Human
homepage : < https : //github.com/vladmandic/human>
author : < https : //github.com/vladmandic>'
* /
2021-03-10 00:32:35 +01:00
2021-10-04 22:29:15 +02:00
var Human = ( ( ) => { var nle = Object . create ; var N1 = Object . defineProperty ; var rle = Object . getOwnPropertyDescriptor ; var sle = Object . getOwnPropertyNames ; var ole = Object . getPrototypeOf , ale = Object . prototype . hasOwnProperty ; var ile = ( e , t , n ) => t in e ? N1 ( e , t , { enumerable : ! 0 , configurable : ! 0 , writable : ! 0 , value : n } ) : e [ t ] = n ; var aP = e => N1 ( e , "__esModule" , { value : ! 0 } ) ; var _c = ( e => typeof require != "undefined" ? require : typeof Proxy != "undefined" ? new Proxy ( e , { get : ( t , n ) => ( typeof require != "undefined" ? require : t ) [ n ] } ) : e ) ( function ( e ) { if ( typeof require != "undefined" ) return require . apply ( this , arguments ) ; throw new Error ( 'Dynamic require of "' + e + '" is not supported' ) } ) ; var Gr = ( e , t ) => ( ) => ( t || e ( ( t = { exports : { } } ) . exports , t ) , t . exports ) , dn = ( e , t ) => { aP ( e ) ; for ( var n in t ) N1 ( e , n , { get : t [ n ] , enumerable : ! 0 } ) } , lle = ( e , t , n ) => { if ( t && typeof t == "object" || typeof t == "function" ) for ( let r of sle ( t ) ) ! ale . call ( e , r ) && r !== "default" && N1 ( e , r , { get : ( ) => t [ r ] , enumerable : ! ( n = rle ( t , r ) ) || n . enumerable } ) ; return e } , T1 = e => lle ( aP ( N1 ( e != null ? nle ( ole ( e ) ) : { } , "default" , e && e . _ _esModule && "default" in e ? { get : ( ) => e . default , enumerable : ! 0 } : { value : e , enumerable : ! 0 } ) ) , e ) ; var nt = ( e , t , n ) => ( ile ( e , typeof t != "symbol" ? t + "" : t , n ) , n ) , iP = ( e , t , n ) => { if ( ! t . has ( e ) ) throw TypeError ( "Cannot " + n ) } ; var _1 = ( e , t , n ) => ( iP ( e , t , "read from private field" ) , n ? n . call ( e ) : t . get ( e ) ) , E1 = ( e , t , n ) => { if ( t . has ( e ) ) throw TypeError ( "Cannot add the same private member more than once" ) ; t instanceof WeakSet ? t . add ( e ) : t . set ( e , n ) } , R1 = ( e , t , n , r ) => ( iP ( e , t , "write to private field" ) , r ? r . call ( e , n ) : t . set ( e , n ) , n ) ; var jj = Gr ( ( ltt , Uj ) => { Uj . exports = jn ; var Ga = null ; try { Ga = new WebAssembly . Instance ( new WebAssembly . Module ( new Uint8Array ( [ 0 , 97 , 115 , 109 , 1 , 0 , 0 , 0 , 1 , 13 , 2 , 96 , 0 , 1 , 127 , 96 , 4 , 127 , 127 , 127 , 127 , 1 , 127 , 3 , 7 , 6 , 0 , 1 , 1 , 1 , 1 , 1 , 6 , 6 , 1 , 127 , 1 , 65 , 0 , 11 , 7 , 50 , 6 , 3 , 109 , 117 , 108 , 0 , 1 , 5 , 100 , 105 , 118 , 95 , 115 , 0 , 2 , 5 , 100 , 105 , 118 , 95 , 117 , 0 , 3 , 5 , 114 , 101 , 109 , 95 , 115 , 0 , 4 , 5 , 114 , 101 , 109 , 95 , 117 , 0 , 5 , 8 , 103 , 101 , 116 , 95 , 104 , 105 , 103 , 104 , 0 , 0 , 10 , 191 , 1 , 6 , 4 , 0 , 35 , 0 , 11 , 36 , 1 , 1 , 126 , 32 , 0 , 173 , 32 , 1 , 173 , 66 , 32 , 134 , 132 , 32 , 2 , 173 , 32 , 3 , 173 , 66 , 32 , 134 , 132 , 126 , 34 , 4 , 66 , 32 , 135 , 167 , 36 , 0 , 32 , 4 , 167 , 11 , 36 , 1 , 1 , 126 , 32 , 0 , 173 , 32 , 1 , 173 , 66 , 32 , 134 , 132 , 32 , 2 , 173 , 32 , 3 , 173 , 66 , 32 , 134 , 132 , 127 , 34 , 4 , 66 , 32 , 135 , 167 , 36 , 0 , 32 , 4 , 167 , 11 , 36 , 1 , 1 , 126 , 32 , 0 , 173 , 32 , 1 , 173 , 66 , 32 , 134 , 132 , 32 , 2 , 173 , 32 , 3 , 173 , 66 , 32 , 134 , 132 , 128 , 34 , 4 , 66 , 32 , 135 , 167 , 36 , 0 , 32 , 4 , 167 , 11 , 36 , 1 , 1 , 126 , 32 , 0 , 173 , 32 , 1 , 173 , 66 , 32 , 134 , 132 , 32 , 2 , 173 , 32 , 3 , 173 , 66 , 32 , 134 , 132 , 129 , 34 , 4 , 66 , 32 , 135 , 167 , 36 , 0 , 32 , 4 , 167 , 11 , 36 , 1 , 1 , 126 , 32 , 0 , 173 , 32 , 1 , 173 , 66 , 32 , 134 , 132 , 32 , 2 , 173 , 32 , 3 , 173 , 66 , 32 , 134 , 132 , 130 , 34 , 4 , 66 , 32 , 135 , 167 , 36 , 0 , 32 , 4 , 167 , 11 ] ) ) , { } ) . exports } catch ( e ) { } function jn ( e , t , n ) { this . low = e | 0 , this . high = t | 0 , this . unsigned = ! ! n } jn . prototype . _ _isLong _ _ ; Object . defineProperty ( jn . prototype , "__isLong__" , { value : ! 0 } ) ; function Qo ( e ) { return ( e && e . _ _isLong _ _ ) === ! 0 } jn . isLong = Qo ; var Oj = { } , Pj = { } ; function Ph ( e , t ) { var n , r , s ; return t ? ( e >>>= 0 , ( s = 0 <= e && e < 256 ) && ( r = Pj [ e ] , r ) ? r : ( n = Hn ( e , ( e | 0 ) < 0 ? - 1 : 0 , ! 0 ) , s && ( Pj [ e ] = n ) , n ) ) : ( e |= 0 , ( s = - 128 <= e && e < 128 ) && ( r = Oj [ e ] , r ) ? r : ( n = Hn ( e , e < 0 ? - 1 : 0 , ! 1 ) , s && ( Oj [ e ] = n ) , n ) ) } jn . fromInt = Ph ; function Ua ( e , t ) { if ( isNaN ( e ) ) return t ? Mh : ja ; if ( t ) { if ( e < 0 ) return Mh ; if ( e >= zj ) return Gj } else { if ( e <= - Lj ) return ea ; if ( e + 1 >= Lj ) return Wj } return e < 0 ? Ua ( - e , t ) . neg ( ) : Hn ( e % fy | 0 , e / fy | 0 , t ) } jn . fromNumber = Ua ; function Hn ( e , t , n ) { return new jn ( e , t , n ) } jn . fromBits = Hn ; var x3 = Math . pow ; function z7 ( e , t , n ) { if ( e . length === 0 ) throw Error ( "empty string" ) ; if ( e === "NaN" || e === "Infinity" || e === "+Infinity" || e === "-Infinity" ) return ja ; if ( typeof t == "number" ? ( n = t , t = ! 1 ) : t = ! ! t , n = n || 10 , n < 2 || 36 < n ) throw RangeError ( "radix" ) ; var r ; if ( ( r = e . indexOf ( "-" ) ) > 0 ) throw Error ( "interior hyphen" ) ; if ( r === 0 ) return z7 ( e . substring ( 1 ) , t , n ) . neg ( ) ; for ( var s = Ua ( x3 ( n , 8 ) ) , o = ja , a = 0 ; a < e . length ; a += 8 ) { var i = Math . min ( 8 , e . length - a ) , l = parseInt ( e . substring ( a , a + i ) , n ) ; if ( i < 8 ) { var u = Ua ( x3 ( n , i ) ) ; o = o . mul ( u ) . add ( Ua ( l ) ) } else o = o . mul ( s ) , o = o . add ( Ua ( l ) ) } return o . unsigned = t , o } jn . fromString = z7 ; function Di ( e , t ) { return typeof e == "number" ? Ua ( e , t ) : typeof e == "string" ? z7 ( e , t ) : Hn ( e . low , e . high , typeof t == "boolean" ? t : e . unsigned ) } jn . fromValue = Di ; var Mj = 1 << 16 , tMe = 1 << 24 , fy = Mj * Mj , zj = fy * fy , Lj = zj / 2 , Bj = Ph ( tMe ) , ja = Ph ( 0 ) ; jn . ZERO = ja ; var Mh = Ph ( 0 , ! 0 ) ; jn . UZERO = Mh ; var hy = Ph ( 1 ) ; jn . ONE = hy ; var Vj = Ph ( 1 , ! 0 ) ; jn . UONE = Vj ; var L7 = Ph ( - 1 ) ; jn . NEG _ONE = L7 ; var Wj = Hn ( 4294967295 | 0 , 21474
2021-09-11 16:29:31 +02:00
` )),l.join( `
2021-10-04 22:29:15 +02:00
` )}function wue(e,t,n,r){let s=wr(t),o=r[r.length-1],a=new Array(o).fill(0),i=t.length,l=n==="complex64"?nb(e):e;if(i>1)for(let u=0;u<s/o;u++){let c=u*o;for(let p=0;p<o;p++)a[p]=Math.max(a[p],tb(l[c+p],0,n).length)}return a}function tb(e,t,n){let r;return Array.isArray(e)?r= ` $ { parseFloat ( e [ 0 ] . toFixed ( FT ) ) } + $ { parseFloat ( e [ 1 ] . toFixed ( FT ) ) } j ` :Rc(e)?r= ` '${e}' ` :n==="bool"?r=FP(e):r=parseFloat(e.toFixed(FT)).toString(),O1(r,t)}function FP(e){return e===0?"false":"true"}function Vw(e,t,n,r,s,o=!0){let a=n==="complex64"?2:1,i=t[0],l=t.length;if(l===0){if(n==="complex64"){let m=nb(e);return[tb(m[0],0,n)]}return n==="bool"?[FP(e[0])]:[e[0].toString()]}if(l===1){if(i> $ P){let g=eb*a,x=Array.from(e.slice(0,g)),y=Array.from(e.slice((i-eb)*a,i*a));return n==="complex64"&&(x=nb(x),y=nb(y)),["["+x.map((b,A)=>tb(b,s[A],n)).join(", ")+", ..., "+y.map((b,A)=>tb(b,s[i-eb+A],n)).join(", ")+"]"]}let m=n==="complex64"?nb(e):Array.from(e);return["["+m.map((g,x)=>tb(g,s[x],n)).join(", ")+"]"]}let u=t.slice(1),c=r.slice(1),p=r[0]*a,d=[];if(i> $ P){for(let m=0;m<eb;m++){let g=m*p,x=g+p;d.push(...Vw(e.slice(g,x),u,n,c,s,!1))}d.push("...");for(let m=i-eb;m<i;m++){let g=m*p,x=g+p;d.push(...Vw(e.slice(g,x),u,n,c,s,m===i-1))}}else for(let m=0;m<i;m++){let g=m*p,x=g+p;d.push(...Vw(e.slice(g,x),u,n,c,s,m===i-1))}let f=l===2?",":"";d[0]="["+d[0]+f;for(let m=1;m<d.length-1;m++)d[m]=" "+d[m]+f;let h= ` ,
` ;for(let m=2;m<l;m++)h+= `
` ;return d[d.length-1]=" "+d[d.length-1]+"]"+(o?"":h),d}function nb(e){let t=[];for(let n=0;n<e.length;n+=2)t.push([e[n],e[n+1]]);return t}var Ur=class{constructor(e,t,n){if(this.dtype=t,this.shape=e.slice(),this.size=wr(e),n!=null){let r=n.length;H(r===this.size,()=> ` Length of values '${r}' does not match the size inferred by the shape '${this.size}' . ` )}if(t==="complex64")throw new Error("complex64 dtype TensorBuffers are not supported. Please create a TensorBuffer for the real and imaginary parts separately and call tf.complex(real, imag).");this.values=n||gP(t,this.size),this.strides=lg(e)}set(e,...t){t.length===0&&(t=[0]),H(t.length===this.rank,()=> ` The number of provided coordinates ( $ { t . length } ) must match the rank ( $ { this . rank } ) ` );let n=this.locToIndex(t);this.values[n]=e}get(...e){e.length===0&&(e=[0]);let t=0;for(let r of e){if(r<0||r>=this.shape[t]){let s= ` Requested out of range element at $ { e } . Buffer shape = $ { this . shape } ` ;throw new Error(s)}t++}let n=e[e.length-1];for(let r=0;r<e.length-1;++r)n+=this.strides[r]*e[r];return this.values[n]}locToIndex(e){if(this.rank===0)return 0;if(this.rank===1)return e[0];let t=e[e.length-1];for(let n=0;n<e.length-1;++n)t+=this.strides[n]*e[n];return t}indexToLoc(e){if(this.rank===0)return[];if(this.rank===1)return[e];let t=new Array(this.shape.length);for(let n=0;n<t.length-1;++n)t[n]=Math.floor(e/this.strides[n]),e-=t[n]*this.strides[n];return t[t.length-1]=e,t}get rank(){return this.shape.length}toTensor(){return nl().makeTensor(this.values,this.shape,this.dtype)}},nl=null,d0=null,kue=null;function Iue(e){nl=e}function Cue(e){d0=e}function Sue(e){kue=e}var Vt=class{constructor(e,t,n,r){this.kept=!1,this.isDisposedInternal=!1,this.shape=e.slice(),this.dtype=t||"float32",this.size=wr(e),this.strides=lg(e),this.dataId=n,this.id=r,this.rankType=this.rank<5?this.rank.toString():"higher"}get rank(){return this.shape.length}async buffer(){let e=await this.data();return d0.buffer(this.shape,this.dtype,e)}bufferSync(){return d0.buffer(this.shape,this.dtype,this.dataSync())}async array(){let e=await this.data();return ug(this.shape,e,this.dtype==="complex64")}arraySync(){return ug(this.shape,this.dataSync(),this.dtype==="complex64")}async data(){this.throwIfDisposed();let e=nl().read(this.dataId);if(this.dtype==="string"){let t=await e;try{return t.map(n=>Bw(n))}catch(n){throw new Error("Failed to decode the string bytes into utf-8. To get the original bytes, call tensor.bytes().")}}return e}dataSync(){this.throwIfDisposed();let e=nl().readSync(this.dataId);if(this.dtype==="string")try{return e.map(t=>Bw(t))}catch(t){throw new Error("Failed to decode the string bytes into utf-8. To get the original bytes, call tensor.bytes().")}return e}async bytes(){this.throwIfDisposed();let e=await nl().read(this.dataId);return this.dtype==="string"?e:new Uint8Array(e.buffer)}dispose(){this.isDisposed||(nl().disposeTensor(this),this.isDisposedInternal=!0)}get isDisposed(){return this.isDisposedInternal}throwIfDisposed(){if(this.isDisposed)throw new Error("Tensor is disposed.")}print(e=!1){return d0.print(this,e)}clone(){return this.throwIfDisposed(),d0.clone(this)}toString(e=!1){let t=this.dataSync();return vue(t,this.shape,this.dtype,e)}cast(e){return this.throwIfDisposed(),d0.cast(this,e)}variable(e=!0,t,n){return this.throwIfDisposed(),nl().makeVariable(this,e,t,n)}};Object.defineProperty(Vt,Symbol.hasInstance,{value:e=>!!e&&e.data!=null&&e.dataSync!=null&&e.throwIfDisposed!=null});function ve(){return _T("Tensor",()=>Vt)}ve();var rb=class extends Vt{constructor(e,t,n,r){super(e.shape,e.dtype,e.dataId,r);this.trainable=t,this.name=n}assign(e){if(e.dtype!==this.dtype)throw new Error( ` dtype of the new value ( $ { e . dtype } ) and previous value ( $ { this . dtype } ) must match ` );if(!Ml(e.shape,this.shape))throw new Error( ` shape of the new value ( $ { e . shape } ) and previous value ( $ { this . shape } ) must match ` );nl().disposeTensor(this),this.dataId=e.dataId,nl().incRef(this,null)}dispose(){nl().disposeVariable(this),this.isDisposedInternal=!0}};Object.defineProperty(rb,Symbol.hasInstance,{value:e=>e instanceof Vt
Manifest JSON has weights with names : $ { i . join ( ", " ) } . ` )}let l=s.reduce((f,h,m)=>(h&&f.push(m),f),[]),u=[];l.forEach(f=>{t[f].paths.forEach(h=>{let m=n+(n.endsWith("/")?"":"/")+h;u.push(m)})});let c=await e(u),p={},d=0;return l.forEach(f=>{let h=t[f].paths.length,m=0;for(let A=0;A<h;A++)m+=c[d+A].byteLength;let g=new ArrayBuffer(m),x=new Uint8Array(g),y=0;for(let A=0;A<h;A++){let v=new Uint8Array(c[d+A]);x.set(v,y),y+=v.byteLength}o[f].forEach(A=>{let v=g.slice(A.groupOffset,A.groupOffset+A.sizeBytes),w=GP(v,[A.manifestEntry]);for(let k in w)p[k]=w[k]}),d+=h}),p}}var wce="application/octet-stream",kce="application/json",ZT=class{constructor(e,t){if(this.DEFAULT_METHOD="POST",t==null&&(t={}),this.weightPathPrefix=t.weightPathPrefix,this.onProgress=t.onProgress,this.weightUrlConverter=t.weightUrlConverter,t.fetchFunc!=null?(H(typeof t.fetchFunc=="function",()=>"Must pass a function that matches the signature of ` fetch ` (see https://developer.mozilla.org/en-US/docs/Web/API/Fetch_API)"),this.fetch=t.fetchFunc):this.fetch=be().platform.fetch,H(e!=null&&e.length>0,()=>"URL path for http must not be null, undefined or empty."),Array.isArray(e)&&H(e.length===2,()=> ` URL paths for http must have a length of 2 , ( actual length is $ { e . length } ) . ` ),this.path=e,t.requestInit!=null&&t.requestInit.body!=null)throw new Error("requestInit is expected to have no pre-existing body, but has one.");this.requestInit=t.requestInit||{}}async save(e){if(e.modelTopology instanceof ArrayBuffer)throw new Error("BrowserHTTPRequest.save() does not support saving model topology in binary formats yet.");let t=Object.assign({method:this.DEFAULT_METHOD},this.requestInit);t.body=new FormData;let n=[{paths:["./model.weights.bin"],weights:e.weightSpecs}],r=HP(e,n);t.body.append("model.json",new Blob([JSON.stringify(r)],{type:kce}),"model.json"),e.weightData!=null&&t.body.append("model.weights.bin",new Blob([e.weightData],{type:wce}),"model.weights.bin");let s=await this.fetch(this.path,t);if(s.ok)return{modelArtifactsInfo:ib(e),responses:[s]};throw new Error( ` BrowserHTTPRequest . save ( ) failed due to HTTP response status $ { s . status } . ` )}async load(){let e=await this.fetch(this.path,this.requestInit);if(!e.ok)throw new Error( ` Request to $ { this . path } failed with status code $ { e . status } . Please verify this URL points to the model JSON of the model to load . ` );let t;try{t=await e.json()}catch(s){let o= ` Failed to parse model JSON of response from $ { this . path } . ` ;throw this.path.endsWith(".pb")?o+=" Your path contains a .pb file extension. Support for .pb models have been removed in TensorFlow.js 1.0 in favor of .json models. You can re-convert your Python TensorFlow model using the TensorFlow.js 1.0 conversion scripts or you can convert your.pb models with the 'pb2json'NPM script in the tensorflow/tfjs-converter repository.":o+=" Please make sure the server is serving valid JSON for this request.",new Error(o)}let n=t.modelTopology,r=t.weightsManifest;if(n==null&&r==null)throw new Error( ` The JSON from HTTP path $ { this . path } contains neither model topology or manifest for weights . ` );return jT(t,s=>this.loadWeights(s))}async loadWeights(e){let t=Array.isArray(this.path)?this.path[1]:this.path,[n,r]=Ice(t),s=this.weightPathPrefix||n,o=[];for(let u of e)o.push(...u.weights);let a=[],i=[];for(let u of e)for(let c of u.paths)this.weightUrlConverter!=null?i.push(this.weightUrlConverter(c)):a.push(s+c+r);this.weightUrlConverter&&a.push(...await Promise.all(i));let l=await r9(a,{requestInit:this.requestInit,fetchFunc:this.fetch,onProgress:this.onProgress});return[o,UT(l)]}};ZT.URL_SCHEME_REGEX=/^https?: \/ \/ /;function Ice(e){let t=e.lastIndexOf("/"),n=e.lastIndexOf("?"),r=e.substring(0,t),s=n>t?e.substring(n):"";return[r+"/",s]}function YT(e){return e.match(ZT.URL_SCHEME_REGEX)!=null}var o9=(e,t)=>{if(typeof fetch=="undefined"&&(t==null||t.fetchFunc==null))return null;{let n=!0;if(Array.isArray(e)?n=e.every(r=>YT(r)):n=YT(e),n)return JT(e,t)}return null};gr.registerSaveRouter(o9);gr.registerLoadRouter(o9);function JT(e,t){return new ZT(e,t)}function Cce(e,t){return JT(e,t)}var Q
Actual : $ { s } .
Expected : $ { o } . ` );for(let a=0;a<o.length;++a){let i=s[a],l=o[a];if(!n(i,l))throw new Error( ` Arrays differ : actual [ $ { a } ] = $ { i } , expected [ $ { a } ] = $ { l } .
Actual : $ { s } .
Expected : $ { o } . ` )}}function qce(e,t){e().then(()=>t.fail(),()=>t())}function Kce(e,t){let n=typeof t=="string"||typeof t=="number"||typeof t=="boolean"?[t]:t;return Rc(e)||Rc(e[0])||Rc(t)||Rc(t[0])?o_(e,n,(r,s)=>r==s):o_(e,t,(r,s)=>a_(r,s,0))}function Xce(e,t,n){if(n==null&&(n=s_()),!a_(e,t,n))throw new Error( ` Numbers differ : actual === $ { e } , expected === $ { t } ` )}function a_(e,t,n){return!isFinite(e)&&!isFinite(t)?!0:!(isNaN(e)||isNaN(t)||Math.abs(e-t)>n)}function Zce(e,t,n){for(let r=0;r<e.length;r++)if(e[r]<t||e[r]>n)throw new Error( ` Value out of range : $ { e [ r ] } low : $ { t } , high : $ { n } ` )}function Yce(e,t){expect(new Float32Array(e)).toEqual(new Float32Array(t))}function I9(e){for(let t=0;t<e.length;t++){let n=e[t];Array.isArray(n)?I9(n):e[t]=Q1(n)}return e}var qw="3.9.0";function C9(){be().set("PROD",!0)}function Jce(){be().set("DEBUG",!0)}function Qce(){be().set("DEPRECATION_WARNINGS_ENABLED",!1),console.warn("TensorFlow.js deprecation warnings have been disabled.")}function i_(e){be().getBool("DEPRECATION_WARNINGS_ENABLED")&&console.warn(e+" You can disable deprecation warnings with tf.disableDeprecationWarnings().")}Sue(i_);function epe(){se.disposeVariables()}function zo(){return se}function Kw(){return se.memory()}function tpe(e){return se.profile(e)}function ce(e,t){return se.tidy(e,t)}function ye(e){BT(e).forEach(n=>n.dispose())}function us(e){return se.keep(e)}function npe(e){return se.time(e)}function S9(e){return se.setBackend(e)}function Xw(){return se.ready()}function gi(){return se.backendName}function rpe(e){se.removeBackend(e)}function l_(e){return se.findBackend(e)}function spe(e){return se.findBackendFactory(e)}function y0(e,t,n=1){return se.registerBackend(e,t,n)}function sl(){return se.backend}function ope(e,t){be().setPlatform(e,t)}function ape(e,t){let n=U(e,"a","add"),r=U(t,"b","add");[n,r]=ur(n,r);let s={a:n,b:r};return se.runKernel( $ c,s)}var Re=ae({add_:ape});function ipe(e,t){let n=U(e,"a","floorDiv"),r=U(t,"b","floorDiv");[n,r]=ur(n,r);let s={a:n,b:r};return se.runKernel(tf,s)}var Zw=ae({floorDiv_:ipe});function lpe(e,t){let n=U(e,"a","div"),r=U(t,"b","div");if([n,r]=ur(n,r),n.dtype==="int32"&&r.dtype==="int32")return Zw(n,r);let s={a:n,b:r},o={};return se.runKernel(Yd,s,o)}var Me=ae({div_:lpe});function upe(e,t){let n=U(e,"a","mul"),r=U(t,"b","mul");[n,r]=ur(n,r);let s={a:n,b:r};return se.runKernel(mf,s)}var te=ae({mul_:upe});function cpe(e){let t=U(e,"x","abs");if(t.dtype==="complex64"){let n={x:t};return se.runKernel(z1,n)}else{let n={x:t};return se.runKernel(cg,n)}}var Tr=ae({abs_:cpe});function ppe(e){let n={x:U(e,"x","acos")};return se.runKernel(pg,n)}var u_=ae({acos_:ppe});function dpe(e){let n={x:U(e,"x","acosh")};return se.runKernel(dg,n)}var c_=ae({acosh_:dpe});function fpe(e){H(Array.isArray(e),()=>"The argument passed to tf.addN() must be a list of tensors"),H(e.length>=1,()=> ` Must pass at least one tensor to tf . addN ( ) , but got $ { e . length } ` );let t=e.map((s,o)=>U(s, ` tensors$ { o } ` ,"addN")),n=t[0];t.forEach(s=>{if(s.dtype!==n.dtype)throw new Error("All tensors passed to tf.addN() must have the same dtype")}),t.forEach(s=>{if(!Ml(s.shape,n.shape))throw new Error("All tensors passed to tf.addN() must have the same shape")});let r=t;return se.runKernel(Ld,r)}var Yw=ae({addN_:fpe});function hpe(e,t=null,n=!1){let s={x:U(e,"x","all","bool")},o={axis:t,keepDims:n};return se.runKernel(fg,s,o)}var Jw=ae({all_:hpe});function mpe(e,t=null,n=!1){let s={x:U(e,"x","any","bool")},o={axis:t,keepDims:n};return se.runKernel(hg,s,o)}var lb=ae({any_:mpe});function gpe(e,t=0){let r={x:U(e,"x","argMax")},s={axis:t};return se.runKernel(Bd,r,s)}var xi=ae({argMax_:gpe});function xpe(e,t=0){let r={x:U(e,"x","argMin")},s={axis:t};return se.runKernel(P1,r,s)}var p_=ae({argMin_:xpe});function ype(e){let n={x:U(e,"x","asin")};return se.runKernel(mg,n)}var d_=ae({asin_:ype});function bpe(e){let n={x:U(e,"x","asinh")};return se.runKernel(gg,n)}var f_=ae({asinh_:bpe});function Ape(e){let n={x:U(e,"x","atan")};return se.runKernel(xg,n)}var h_=ae({atan_:Ape});function vpe(e,t){let n=U(e,"a","atan2"),r=U(t,"b","atan2");[n,r
with dtype $ { o . dtype } . ` )}),n.length===1)return mi(n[0]);let r=n,s={axis:t};return se.runKernel(vg,r,s)}var En=ae({concat_:Dpe});function $ pe(e){let n={x:U(e,"x","sigmoid")};return se.runKernel(Nf,n)}var xo=ae({sigmoid_: $ pe});function Fpe(e,t,n){let r=U(e,"x","slice","string_or_numeric");if(r.rank===0)throw new Error("Slicing scalar is not possible");let s={x:r},o={begin:t,size:n};return se.runKernel(Qg,s,o)}var gt=ae({slice_:Fpe});function Ope(e){let n={x:U(e,"x","tanh")};return se.runKernel(Ff,n)}var Hf=ae({tanh_:Ope});function Ppe(e,t,n,r,s,o){let a=U(e,"forgetBias","basicLSTMCell"),i=U(t,"lstmKernel","basicLSTMCell"),l=U(n,"lstmBias","basicLSTMCell"),u=U(r,"data","basicLSTMCell"),c=U(s,"c","basicLSTMCell"),p=U(o,"h","basicLSTMCell"),d=En([u,p],1),f=Lt(d,i),h=Re(f,l),m=h.shape[0],g=h.shape[1]/4,x=[m,g],y=gt(h,[0,0],x),b=gt(h,[0,g],x),A=gt(h,[0,g*2],x),v=gt(h,[0,g*3],x),w=Re(te(xo(y),Hf(b)),te(c,xo(Re(a,A)))),k=te(Hf(w),xo(v));return[w,k]}var Mpe=ae({basicLSTMCell_:Ppe});function zpe(e,t,n){let r=U(e,"x","batchToSpaceND"),s=t.reduce((i,l)=>i*l);H(r.rank>=1+t.length,()=> ` input rank is $ { r . rank } but should be > than blockShape . length $ { t . length } ` ),H(n.length===t.length,()=> ` crops . length is $ { n . length } but should be equal to blockShape . length $ { t . length } ` ),H(r.shape[0]%s==0,()=> ` input tensor batch is $ { r . shape [ 0 ] } but is not divisible by the product of the elements of blockShape $ { t . join ( " * " ) } === $ { s } ` );let o={x:r},a={blockShape:t,crops:n};return se.runKernel(Ag,o,a)}var pb=ae({batchToSpaceND_:zpe});function Lpe(e){let t;return e.rank===0||e.rank===1?t=ie(e,[1,1,1,e.size]):e.rank===2?t=ie(e,[1,1,e.shape[0],e.shape[1]]):e.rank===3?t=ie(e,[1,e.shape[0],e.shape[1],e.shape[2]]):t=e,t}function Bpe(e,t,n,r,s,o){o==null&&(o=.001);let a=U(e,"x","batchNorm"),i=U(t,"mean","batchNorm"),l=U(n,"variance","batchNorm"),u;s!=null&&(u=U(s,"scale","batchNorm"));let c;r!=null&&(c=U(r,"offset","batchNorm")),H(i.rank===l.rank,()=>"Batch normalization gradient requires mean and variance to have equal ranks."),H(c==null||i.rank===c.rank,()=>"Batch normalization gradient requires mean and offset to have equal ranks."),H(u==null||i.rank===u.rank,()=>"Batch normalization gradient requires mean and scale to have equal ranks.");let d={x:Lpe(a),scale:u,offset:c,mean:i,variance:l},f={varianceEpsilon:o},h=se.runKernel(nf,d,f);return ie(h,a.shape)}var qf=ae({batchNorm_:Bpe});function Vpe(e,t,n,r,s,o){let a=U(e,"x","batchNorm"),i=U(t,"mean","batchNorm"),l=U(n,"variance","batchNorm"),u;s!=null&&(u=U(s,"scale","batchNorm"));let c;return r!=null&&(c=U(r,"offset","batchNorm")),H(a.rank===2,()=> ` Error in batchNorm2D : x must be rank 2 but got rank $ { a . rank } . ` ),H(i.rank===2||i.rank===1,()=> ` Error in batchNorm2D : mean must be rank 2 or rank 1 but got rank $ { i . rank } . ` ),H(l.rank===2||l.rank===1,()=> ` Error in batchNorm2D : variance must be rank 2 or rank 1 but got rank $ { l . rank } . ` ),u!=null&&H(u.rank===2||u.rank===1,()=> ` Error in batchNorm2D : scale must be rank 2 or rank 1 but got rank $ { u . rank } . ` ),c!=null&&H(c.rank===2||c.rank===1,()=> ` Error in batchNorm2D : offset must be rank 2 or rank 1 but got rank $ { c . rank } . ` ),qf(a,i,l,c,u,o)}var E9=ae({batchNorm2d_:Vpe});function Wpe(e,t,n,r,s,o){let a=U(e,"x","batchNorm"),i=U(t,"mean","batchNorm"),l=U(n,"variance","batchNorm"),u;s!=null&&(u=U(s,"scale","batchNorm"));let c;return r!=null&&(c=U(r,"offset","batchNorm")),H(a.rank===3,()=> ` Error in batchNorm3D : x must be rank 3 but got rank $ { a . rank } . ` ),H(i.rank===3||i.rank===1,()=> ` Error in batchNorm3D : mean must be rank 3 or rank 1 but got rank $ { i . rank } . ` ),H(l.rank===3||l.rank===1,()=> ` Error in batchNorm3D : variance must be rank 3 or rank 1 but got rank $ { l . rank } . ` ),u!=null&&H(u.rank===3||u.rank===1,()=> ` Error in batchNorm3D : scale must be rank 3 or rank 1 but got rank $ { u . rank } . ` ),c!=null&&H(c.rank===3||c.rank===1,()=> ` Error in batchNorm3D : offset must be rank 3 or rank 1 but got rank $ { c . rank } . ` ),qf(a,i,l,c,u,o)}var R9=ae({batchNorm3d_:Wpe});function Gpe(e,t,n,r,s,o){let a=U(e,"x","batchNorm"),i=U(t,"mean","batchNorm"),l=U(n,"variance","batchNorm"),u;s!=null&&(u
$ { s } and $ { t } for depthToSpace with input shape
$ { r . shape } ` ),H(o*t>=0,()=> ` Negative dimension size caused by overflow when multiplying
$ { o } and $ { t } for depthToSpace with input shape
$ { r . shape } ` ),H(a%(t*t)==0,()=> ` Dimension size must be evenly divisible by $ { t * t } but is $ { a } for depthToSpace with input shape $ { r . shape } ` );let i={x:r},l={blockSize:t,dataFormat:n};return se.runKernel(kg,i,l)}var I_=ae({depthToSpace_:cde});function pde(e,t,n,r,s="NHWC",o=[1,1],a){let i=U(e,"x","depthwiseConv2d"),l=U(t,"filter","depthwiseConv2d"),u=i,c=!1;i.rank===3&&(c=!0,u=ie(i,[1,i.shape[0],i.shape[1],i.shape[2]])),H(u.rank===4,()=> ` Error in depthwiseConv2d : input must be rank 4 , but got rank $ { u . rank } . ` ),H(l.rank===4,()=> ` Error in depthwiseConv2d : filter must be rank 4 , but got rank $ { l . rank } . ` ),H(u.shape[3]===l.shape[2],()=> ` Error in depthwiseConv2d : number of input channels ( $ { u . shape [ 3 ] } ) must match the inChannels dimension in filter $ { l . shape [ 2 ] } . ` ),a!=null&&H(as(r),()=> ` Error in depthwiseConv2d : pad must be an integer when using , dimRoundingMode $ { a } but got pad $ { r } . ` );let p={x:u,filter:l},d={strides:n,pad:r,dataFormat:s,dilations:o,dimRoundingMode:a},f=se.runKernel(Zd,p,d);return c?ie(f,[f.shape[1],f.shape[2],f.shape[3]]):f}var w0=ae({depthwiseConv2d_:pde});function dde(e){let n={x:U(e,"x","diag")};return se.runKernel(cw,n)}var fde=ae({diag_:dde});function hde(e,t,n,r,s=[1,1],o="NHWC"){let a=U(e,"x","dilation2d"),i=U(t,"filter","dilation2d");H(a.rank===3||a.rank===4,()=> ` Error in dilation2d : input must be rank 3 or 4 , but got rank $ { a . rank } . ` ),H(i.rank===3,()=> ` Error in dilation2d : filter must be rank 3 , but got rank $ { i . rank } . ` ),H(o==="NHWC",()=> ` Error in dilation2d : Only NHWC is currently supported , but got dataFormat of $ { o } ` );let l=a,u=!1;a.rank===3&&(l=ie(a,[1,a.shape[0],a.shape[1],a.shape[2]]),u=!0);let c={x:l,filter:i},p={strides:n,pad:r,dilations:s},d=se.runKernel(B1,c,p);return u?ie(d,[d.shape[1],d.shape[2],d.shape[3]]):d}var C_=ae({dilation2d_:hde});function mde(e,t){let n=e.length,r=[];for(let s=0;s<n;s++){let o=n-1-s,a=e[o]||1;(t[t.length-1-s]||1)>1&&a===1&&r.unshift(o)}return r}function jr(e,t){let n=[];for(let r=0;r<t.length;r++){let s=e[e.length-r-1],o=t.length-r-1,a=t[o];(s==null||s===1&&a>1)&&n.unshift(o)}return n}function Wn(e,t){let n=[],r=Math.max(e.length,t.length);for(let s=0;s<r;s++){let o=e[e.length-s-1];o==null&&(o=1);let a=t[t.length-s-1];if(a==null&&(a=1),o===1)n.unshift(a);else if(a===1)n.unshift(o);else if(o!==a){let i= ` Operands could not be broadcast together with shapes $ { e } and $ { t } . ` ;throw Error(i)}else n.unshift(o)}return n}function gde(e,t){let n=U(e,"a","equal","string_or_numeric"),r=U(t,"b","equal","string_or_numeric");[n,r]=ur(n,r),Wn(n.shape,r.shape);let s={a:n,b:r};return se.runKernel(Cg,s)}var Lo=ae({equal_:gde});function xde(e,t,n){let r=U(t,"a","where"),s=U(n,"b","where"),o=U(e,"condition","where","bool"),a=Wn(Wn(o.shape,r.shape),s.shape),i=A0(o,a),l=A0(r,a),u=A0(s,a),c={condition:i,t:l,e:u};return se.runKernel(Yg,c)}var $ s=ae({where_:xde});function yde(e){let n={x:U(e,"x","zerosLike")};return se.runKernel(u0,n)}var Xt=ae({zerosLike_:yde});function bde(e,t){let n=U(e,"a","div"),r=U(t,"b","div");[n,r]=ur(n,r);let s=Me(n,r),o=Xt(s),a=Lo(r,o);return $ s(a,o,s)}var S_=ae({divNoNan_:bde});function Ade(e,t){let n=U(e,"t1","dot"),r=U(t,"t2","dot");H((n.rank===1||n.rank===2)&&(r.rank===1||r.rank===2),()=> ` Error in dot : inputs must all be rank 1 or 2 , but got ranks $ { n . rank } and $ { r . rank } . ` );let s=n.rank===1?n.size:n.shape[1],o=r.rank===1?r.size:r.shape[0];if(H(s===o,()=> ` Error in dot : inner dimensions of inputs must match , but got $ { s } and $ { o } . ` ),n.rank===1&&r.rank===1){let a=ie(n,[1,-1]),i=ie(r,[-1,1]),l=Lt(a,i);return ie(l,[])}else if(n.rank===1&&r.rank===2){let a=ie(n,[1,-1]),i=ie(r,[r.shape[0],r.shape[1]]),l=Lt(a,i);return ie(l,[l.size])}else if(n.rank===2&&r.rank===1){let a=ie(r,[-1,1]),i=Lt(n,a);return ie(i,[i.size])}else{let a=ie(r,[r.shape[0],r.shape[1]]);return Lt(n,a)}}var B9=ae({dot_:Ade});function vde(e,...t){let n=t.map((s,o)=>U(s, ` tensors$ { o } ` ,"einsum")),r={equation:e};return se.runKernel(fw,n,r)}var V9=ae({einsum_:vde});function wde(e){let n={x:U(e,"x","elu")};return se.runKernel(Jd,n)}var k0=ae({elu_:wde});function kde(e){let t=U(e,"x","erf");H(t.dtype
rank $ { o . rank } . ` ),H(as(t),()=> ` Error in localResponseNormalization : depthRadius must be an integer but got depthRadius $ { t } . ` );let a=o,i=!1;o.rank===3&&(i=!0,a=ie(o,[1,o.shape[0],o.shape[1],o.shape[2]]));let l={x:a},u={depthRadius:t,bias:n,alpha:r,beta:s},c=se.runKernel(U1,l,u);return i?ie(c,[c.shape[1],c.shape[2],c.shape[3]]):c}var R_=ae({localResponseNormalization_:Bde});function Vde(e){let n={x:U(e,"x","log")};return se.runKernel(af,n)}var Vo=ae({log_:Vde});function Wde(e){let n={x:U(e,"x","log1p")};return se.runKernel(Mg,n)}var hb=ae({log1p_:Wde});function Gde(e){return H(Dc(e),()=>"The f passed in grad(f) must be a function"),(t,n)=>{let r=U(t,"x","tf.grad","string_or_numeric"),s=n!=null?U(n,"dy","tf.grad"):null;return se.tidy(()=>{let{value:o,grads:a}=se.gradients(()=>e(r),[r],s);return s!=null&&zs(o.shape,s.shape,"The shape of dy passed in grad(f)(x, dy) must match the shape returned by f(x)"),ak(a),a[0]})}}function Ude(e){return H(Dc(e),()=>"The f passed in grads(f) must be a function"),(t,n)=>{H(Array.isArray(t),()=>"The args passed in grads(f)(args) must be an array of ` Tensor ` s or ` TensorLike ` s");let r=ab(t,"args","tf.grads","string_or_numeric"),s=n!=null?U(n,"dy","tf.grads"):null;return se.tidy(()=>{let{value:o,grads:a}=se.gradients(()=>e(...r),r,s);return s!=null&&zs(o.shape,s.shape,"The shape of dy passed in grads(f)([x1,...], dy) must match the shape returned by f([x1,...])"),ak(a),a})}}function jde(e){return H(Dc(e),()=>"The f passed in valueAndGrad(f) must be a function"),(t,n)=>{H(t instanceof Vt,()=>"The x passed in valueAndGrad(f)(x) must be a tensor"),H(n==null||n instanceof Vt,()=>"The dy passed in valueAndGrad(f)(x, dy) must be a tensor");let{grads:r,value:s}=se.gradients(()=>e(t),[t],n);return ak(r),{grad:r[0],value:s}}}function Hde(e){return H(Dc(e),()=>"The f passed in valueAndGrads(f) must be a function"),(t,n)=>{H(Array.isArray(t)&&t.every(s=>s instanceof Vt),()=>"The args passed in valueAndGrads(f)(args) must be array of tensors"),H(n==null||n instanceof Vt,()=>"The dy passed in valueAndGrads(f)(args, dy) must be a tensor");let r=se.gradients(()=>e(...t),t,n);return n!=null&&zs(r.value.shape,n.shape,"The shape of dy passed in valueAndGrads(f)([x1,...], dy) must match the shape returned by f([x1,...])"),ak(r.grads),r}}function j9(e,t){H(Dc(e),()=>"The f passed in variableGrads(f) must be a function"),H(t==null||Array.isArray(t)&&t.every(u=>u instanceof rb),()=>"The varList passed in variableGrads(f, varList) must be an array of variables");let n=t!=null;if(!n){t=[];for(let u in se.registeredVariables)t.push(se.registeredVariables[u])}let r=n?t.filter(u=>!u.trainable):null,s=t.length;t=t.filter(u=>u.trainable),H(t.length>0,()=> ` variableGrads ( ) expects at least one of the input variables to be trainable , but none of the $ { s } variables is trainable . ` );let o=!0,{value:a,grads:i}=se.gradients(e,t,null,o);H(i.some(u=>u!=null),()=>"Cannot find a connection between any variable and the result of the loss function y=f(x). Please make sure the operations that use variables are inside the function f passed to minimize()."),H(a.rank===0,()=> ` The f passed in variableGrads ( f ) must return a scalar , but it returned a rank - $ { a . rank } tensor ` );let l={};return t.forEach((u,c)=>{i[c]!=null&&(l[u.name]=i[c])}),r!=null&&r.forEach(u=>l[u.name]=null),{value:a,grads:l}}function al(e){return se.customGrad(e)}function ak(e){if(e.filter(n=>n==null).length>0)throw new Error( ` Cannot compute gradient of y = f ( x ) with respect to x . Make sure that
the f you passed encloses all operations that lead from x to y . ` )}function qde(e){let n={x:U(e,"x","neg")};return se.runKernel(Bg,n)}var er=ae({neg_:qde});function Kde(e){let n={x:U(e,"x","softplus")};return se.runKernel(n0,n)}var Xf=ae({softplus_:Kde});function Xde(e){let t=U(e,"x","logSigmoid");return al(r=>({value:er(Xf(er(r))),gradFunc:a=>te(a,xo(er(r)))}))(t)}var H9=ae({logSigmoid_:Xde});function Zde(e,t=null,n=!1){let s={x:U(e,"x","max")},o={reductionIndices:t,keepDims:n};return se.runKernel(lf,s,o)}var Wo=ae({max_:Zde});function Yde(e,t){let n=U(e,"a","sub"),r=U(t,"b","sub");[n,r]=ur(n,r);let s={a:n,b:r};return se.runKernel(Df,s)}var Ge=ae({sub_:Yde});function Jde(e,t=null,n=!1){let r=U(e,"x","sum");r.dtype==="bool"&&(r=Fe(r,"int32"));let s={x:r},o={axis:t,keepDims:n};return se.runKernel(_f,s,o)}var Je=ae({sum_:Jde});function Qde(e,t=-1){let n=U(e,"logits","logSoftmax");if(t===-1&&(t=n.rank-1),t!==n.rank-1)throw Error( ` Log Softmax along a non - last dimension is not yet supported . Logits was rank $ { n . rank } and axis was $ { t } ` );return al((s,o)=>{let a=!0,i=Wo(s,t,!0),l=Ge(s,i),u=Ge(Fe(l,"float32"),Vo(Je(Bo(l),t,a)));return o([u]),{value:u,gradFunc:(p,d)=>{let[f]=d,h=!0,m=Bo(f);return Ge(p,te(Je(p,t,h),m))}}})(n)}var ik=ae({logSoftmax_:Qde});function D_(e,t){for(let n=0;n<e.length;++n)if(e[e.length-n-1]!==t-1-n)return!1;return!0}function q9(e,t,n){let r=e.length+t.length,s=[],o=0,a=0;for(let i=0;i<r;i++)n.indexOf(i)===-1?s.push(e[o++]):s.push(t[a++]);return s}function K9(e,t){let n=[],r=e.length;for(let o=0;o<r;o++)t.indexOf(o)===-1&&n.push(e[o]);let s=t.map(o=>e[o]);return[n,s]}function Zf(e,t){let n=t.map(r=>1);return q9(e,n,t)}function efe(e,t,n){H(D_(t,n),()=> ` $ { e } supports only inner - most axes for now . Got axes $ { t } and rank - $ { n } input . ` )}function X9(e,t){if(D_(e,t))return null;let n=[];for(let r=0;r<t;++r)e.indexOf(r)===-1&&n.push(r);return e.forEach(r=>n.push(r)),n}function $ _(e){return e.map((t,n)=>[n,t]).sort((t,n)=>t[1]-n[1]).map(t=>t[0])}function tfe(e,t){let n=[];for(let r=t-e;r<t;++r)n.push(r);return n}function nfe(e,t=null,n=!1){let r=U(e,"x","logSumExp"),s=Ra(t,r.shape),o=Wo(r,s,!0),a=Ge(r,o),i=Bo(a),l=Je(i,s),u=Vo(l),c=Re(ie(o,u.shape),u);if(n){let p=Zf(c.shape,s);return ie(c,p)}return c}var F_=ae({logSumExp_:nfe});function rfe(e,t){let n=U(e,"a","logicalAnd","bool"),r=U(t,"b","logicalAnd","bool");Wn(n.shape,r.shape);let s={a:n,b:r};return se.runKernel(zg,s)}var Fa=ae({logicalAnd_:rfe});function sfe(e){let n={x:U(e,"x","logicalNot","bool")};return se.runKernel(W1,n)}var mb=ae({logicalNot_:sfe});function ofe(e,t){let n=U(e,"a","logicalOr","bool"),r=U(t,"b","logicalOr","bool");Wn(n.shape,r.shape);let s={a:n,b:r};return se.runKernel(G1,s)}var lk=ae({logicalOr_:ofe});function afe(e,t){let n=U(e,"a","logicalXor","bool"),r=U(t,"b","logicalXor","bool");return Wn(n.shape,r.shape),Fa(lk(e,t),mb(Fa(e,t)))}var Z9=ae({logicalXor_:afe});function ife(e,t,n,r,s){let o=U(e,"x","maxPool"),a=1,i=o,l=!1;o.rank===3&&(l=!0,i=ie(o,[1,o.shape[0],o.shape[1],o.shape[2]])),H(i.rank===4,()=> ` Error in maxPool : input must be rank 4 but got rank $ { i . rank } . ` ),H(ol(n,a),()=> ` Error in maxPool : Either strides or dilations must be 1. Got strides $ { n } and dilations '${a}' ` ),s!=null&&H(as(r),()=> ` Error in maxPool : pad must be an integer when using , dimRoundingMode $ { s } but got pad $ { r } . ` );let u={x:i},c={filterSize:t,strides:n,pad:r,dimRoundingMode:s},p=se.runKernel(cf,u,c);return l?ie(p,[p.shape[1],p.shape[2],p.shape[3]]):p}var gb=ae({maxPool_:ife});function lfe(e,t=[1,1,1],n,r,s,o="NDHWC"){let a=U(e,"x","maxPool3d"),i=a,l=!1;a.rank===4&&(l=!0,i=ie(a,[1,a.shape[0],a.shape[1],a.shape[2],a.shape[3]])),H(i.rank===5,()=> ` Error in maxPool3d : x must be rank 5 but got rank $ { i . rank } . ` ),H(o==="NDHWC",()=> ` Error in maxPool3d : Only NDHWC is currently supported , but got dataFormat of $ { o } ` ),s!=null&&H(as(r),()=> ` Error in maxPool3d : pad must be an integer when using , dimRoundingMode $ { s } but got pad $ { r } . ` );let u={x:i},c={filterSize:t,strides:n,pad:r,dimRoundingMode:s,dataFormat:o},p=se.runKernel(j1,u,c);return l?ie(p,[p.shape[1],p.shape[2],p.sha
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Values should be Tensor1D but received shape $ { o . shape } ` );if(a.rank!==1)throw new Error( ` Dense shape should be Tensor1D but received shape $ { a . shape } ` );if(i.rank!==0)throw new Error( ` Default value should be a scalar but received shape $ { i . shape } ` );let l={indices:s,values:o,denseShape:a,defaultValue:i},u=se.runKernel(Nw,l);return{outputIndices:u[0],outputValues:u[1],emptyRowIndicator:u[2],reverseIndexMap:u[3]}}var wge=ae({sparseFillEmptyRows_:vge});function kge(e,t,n){let r=U(e,"inputIndices","sparseReshape"),s=U(t,"inputShape","sparseReshape"),o=U(n,"newShape","sparseReshape");if(r.rank!==2)throw new Error( ` Input indices should be Tensor2D but received shape
$ { r . shape } ` );if(s.rank!==1)throw new Error( ` Input shape should be Tensor1D but received shape $ { s . shape } ` );if(o.rank!==1)throw new Error( ` New shape should be Tensor1D but received shape $ { o . shape } ` );let a={inputIndices:r,inputShape:s,newShape:o},i=se.runKernel(Tw,a);return{outputIndices:i[0],outputShape:i[1]}}var Ige=ae({sparseReshape_:kge});function Cge(e,t,n){let r=U(e,"data","sparseSegmentMean"),s=U(t,"indices","sparseSegmentMean"),o=U(n,"segmentIds","sparseSegmentMean");if(r.rank<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.rank!==1)throw new Error( ` Indices should be Tensor1D but received shape
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Segment ids should be Tensor1D but received shape
$ { o . shape } ` );let a={data:r,indices:s,segmentIds:o};return se.runKernel(_w,a)}var Sge=ae({sparseSegmentMean_:Cge});function Nge(e,t,n){let r=U(e,"data","sparseSegmentSum"),s=U(t,"indices","sparseSegmentSum"),o=U(n,"segmentIds","sparseSegmentSum");if(r.rank<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.rank!==1)throw new Error( ` Indices should be Tensor1D but received shape
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Segment ids should be Tensor1D but received shape
$ { o . shape } ` );let a={data:r,indices:s,segmentIds:o};return se.runKernel(Ew,a)}var Tge=ae({sparseSegmentSum_:Nge});function _ge(e,t,n,r,s,o,a,i){let l=U(e,"data","stringNGrams","string");if(l.dtype!=="string")throw new Error("Data must be of datatype string");if(l.shape.length!==1)throw new Error( ` Data must be a vector , saw : $ { l . shape } ` );let u=U(t,"dataSplits","stringNGrams");if(u.dtype!=="int32")throw new Error("Data splits must be of datatype int32");let c={separator:n,nGramWidths:r,leftPad:s,rightPad:o,padWidth:a,preserveShortSequences:i},p={data:l,dataSplits:u},d=se.runKernel(Dw,p,c);return{nGrams:d[0],nGramsSplits:d[1]}}var Ege=ae({stringNGrams_:_ge});function Rge(e,t,n=!0){let r=U(e,"input","stringSplit","string"),s=U(t,"delimiter","stringSplit","string");if(r.rank!==1)throw new Error( ` Input should be Tensor1D but received shape $ { r . shape } ` );if(s.rank!==0)throw new Error( ` Delimiter should be a scalar but received shape $ { s . shape } ` );let o={skipEmpty:n},a={input:r,delimiter:s},i=se.runKernel( $ w,a,o);return{indices:i[0],values:i[1],shape:i[2]}}var Dge=ae({stringSplit_:Rge});function $ ge(e,t){let n=U(e,"input","stringToHashBucketFast","string"),r={numBuckets:t};if(t<=0)throw new Error("Number of buckets must be at least 1");let s={input:n};return se.runKernel(Fw,s,r)}var Fge=ae({stringToHashBucketFast_: $ ge}),Oge={fft:wb,ifft:_0,rfft:kb,irfft:bk},Pge={hammingWindow:dme,hannWindow:pM,frame:dM,stft:gme},ht={flipLeftRight:Ame,grayscaleToRGB:wme,resizeNearestNeighbor:yM,resizeBilinear:xM,rotateWithOffset:Ime,cropAndResize:yme,nonMaxSuppression:Sme,nonMaxSuppressionAsync:Fme,nonMaxSuppressionWithScore:Pme,nonMaxSuppressionWithScoreAsync:zme,nonMaxSuppressionPadded:Bme,nonMaxSuppressionPaddedAsync:Wme,threshold:qme,transform:Xme},AM={bandPart:Yme,gramSchmidt:Qme,qr:tge},Mge={absoluteDifference:sge,computeWeightedLoss:Gl,cosineDistance:age,hingeLoss:lge,huberLoss:cge,logLoss:dge,meanSquaredError:hge,sigmoidCrossEntropy:xge,softmaxCrossEntropy:Age},Ib={sparseFillEmptyRows:wge,sparseReshape:Ige,sparseSegmentMean:Sge,sparseSegmentSum:Tge},Tk={stringNGrams:Ege,stringSplit:Dge,stringToHashBucketFast:Fge},Ul=class extends v9{minimize(e,t=!1,n){let{value:r,grads:s}=this.computeGradients(e,n);if(n!=null){let o=n.map(a=>({name:a.name,tensor:s[a.name]}));this.applyGradients(o)}else this.applyGradients(s);return ye(s),t?r:(r.dispose(),null)}get iterations(){return this.iterations_==null&&(this.iterations_=0),this.iterations_}incrementIterations(){this.iterations_=this.iterations+1}computeGradients(e,t){return j9(e,t)}dispose(){this.iterations_!=null&&ye(this.iterations_)}async saveIterations(){return this.iterations_==null&&(this.iterations_=0),{name:"iter",tensor:at(this.iterations_,"int32")}}async getWeights(){throw new Error("getWeights() is not implemented for this optimizer yet.")}async setWeights(e){throw new Error( ` setWeights ( ) is not implemented for this optimizer class $ { this . getClassName ( ) } ` )}async extractIterations(e){return this.iterations_=(await e[0].tensor.data())[0],e.slice(1)}};Object.defineProperty(Ul,Symbol.hasInstance,{value:e=>e.minimize!=null&&e.computeGradients!=null&&e.applyGradients!=null});var _k=class extends Ul{constructor(e,t,n=null){super();this.learningRate=e,this.rho=t,this.epsilon=n,this.accumulatedGrads=[],this.accumulatedUpdates=[],n==null&&(this.epsilon=se.backend.epsilon())}applyGradients(e){(Array.isArray(e)?e.map(n=>n.name):Object.keys(e)).forEach((n,r)=>{let s=se.registeredVariables[n],o=!1;this.accumulatedGrads[r]==null&&(this.accumulatedGrads[r]={originalName: ` $ { n } / accum _grad ` ,variable:ce(()=>Xt(s).variable(o))}),this.accumulatedUpdates[r]==null&&(this.accumulatedUpdates[r]={originalName: ` $ { n } / accum _var ` ,variable:ce(()=>Xt(s).variable(o))});let a=Array.isArray(e)?e[r].tensor:e[n];if(a==null)return;let i=this.accumulatedGrads[r].variable,l=this.accumulatedUpdates[r].variable;ce(()=>{let u=Re(te(i,this.rho),te(Tn(a),1-this.rho)),c=te(Me(As(Re(l,this.epsilon)),As(Re(i,this.epsilon))),a),p=Re(te(l,this.rho),te(Tn(c),1-this.rho));i.assign(u),l.assign(p);let d=Re(te(c,-this.learningRate),s);s.a
$ { o } ) . ` );if(n<r)throw new Error( ` batchDims ( $ { r } ) must be less than or equal to axis ( $ { n } ) . ` );for(let p=0;p<r;++p)if(e.shape[p]!==t.shape[p])throw new Error( ` x . shape [ $ { p } ] : $ { e . shape [ p ] } should be equal to indices . shape [ $ { p } ] : $ { t . shape [ p ] } . ` );let a=e.shape[n],i=[],l=1,u=1,c=1;for(let p=0;p<r;++p)i.push(e.shape[p]),l*=e.shape[p];for(let p=r;p<n;p++)i.push(e.shape[p]),u*=e.shape[p];for(let p=r;p<s;p++)i.push(t.shape[p]);for(let p=n+1;p<o;p++)i.push(e.shape[p]),c*=e.shape[p];return{batchSize:l,sliceSize:c,outerSize:u,dimSize:a,outputShape:i}}function b0e(e){try{return e.map(t=>Bw(t))}catch(t){throw new Error( ` Failed to decode encoded string bytes into utf - 8 , error : $ { t } ` )}}function A0e(e){return e.map(t=>Q1(t))}var ll={};Rt(ll,{nonMaxSuppressionV3Impl:()=>fM,nonMaxSuppressionV4Impl:()=>hM,nonMaxSuppressionV5Impl:()=>mM,whereImpl:()=>rM});var SM={kernelName:cg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>te(e,E0(Fe(n,"float32"),-1))}}},v0e={kernelName:pg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=Tn(Fe(n,"float32")),s=As(Ge(at(1),r));return er(Me(e,s))}}}},w0e={kernelName:dg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=As(Ge(Tn(Fe(n,"float32")),1));return Me(e,r)}}}},k0e={kernelName: $ c,inputsToSave:["a","b"],gradFunc:(e,t)=>{let[n,r]=t,s=Wn(n.shape,r.shape);return{a:()=>{let i=e,l=jr(n.shape,s);return l.length>0&&(i=Je(i,l)),ie(i,n.shape)},b:()=>{let i=e,l=jr(r.shape,s);return l.length>0&&(i=Je(i,l)),ie(i,r.shape)}}}},I0e={kernelName:Ld,saveAllInputs:!0,gradFunc:(e,t)=>{let n={};return t.forEach((r,s)=>{n[s]=()=>e.clone()}),n}},C0e={kernelName:Bd,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Xt(n)}}},S0e={kernelName:P1,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Xt(n)}}},N0e={kernelName:mg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Me(e,As(Ge(at(1),Tn(Fe(n,"float32")))))}}},T0e={kernelName:gg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=As(Re(at(1),Tn(Fe(n,"float32"))));return Me(e,r)}}}},_0e={kernelName:bg,inputsToSave:["a","b"],gradFunc:(e,t)=>{let[n,r]=t,s=Wn(n.shape,r.shape);return{a:()=>{let i=Re(Tn(n),Tn(r)),l=te(e,Me(r,i)),u=jr(n.shape,s);return u.length>0&&(l=Je(l,u)),ie(l,n.shape)},b:()=>{let i=Re(Tn(n),Tn(r)),l=er(te(e,Me(n,i))),u=jr(r.shape,s);return u.length>0&&(l=Je(l,u)),ie(l,r.shape)}}}},E0e={kernelName:xg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Me(e,Re(Tn(Fe(n,"float32")),1))}}},R0e={kernelName:yg,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Me(e,Ge(at(1),Tn(Fe(n,"float32"))))}}};function D0e(e,t,n,r,s,o){let a=U(e,"dy","avgPool3dGrad"),i=U(t,"input","avgPool3dGrad"),l=a,u=i,c=!1;i.rank===4&&(c=!0,l=ie(a,[1,a.shape[0],a.shape[1],a.shape[2],a.shape[3]]),u=ie(i,[1,i.shape[0],i.shape[1],i.shape[2],i.shape[3]])),H(l.rank===5,()=> ` Error in avgPool3dGrad : dy must be rank 5 but got rank $ { l . rank } . ` ),H(u.rank===5,()=> ` Error in avgPool3dGrad : input must be rank 5 but got rank $ { u . rank } . ` ),o!=null&&H(as(s),()=> ` Error in avgPool3dGrad : pad must be an integer when using , dimRoundingMode $ { o } but got pad $ { s } . ` );let p={dy:l,input:u},d={filterSize:n,strides:r,pad:s,dimRoundingMode:o},f=se.runKernel(tw,p,d);return c?ie(f,[f.shape[1],f.shape[2],f.shape[3],f.shape[4]]):f}var $ 0e=ae({avgPool3dGrad_:D0e}),F0e={kernelName:M1,inputsToSave:["x"],gradFunc:(e,t,n)=>{let[r]=t,{filterSize:s,strides:o,pad:a,dimRoundingMode:i}=n;return{x:()=> $ 0e(e,r,s,o,a,i)}}};function O0e(e,t,n,r,s){let o=U(e,"dy","avgPoolGrad"),a=U(t,"input","avgPoolGrad");H(a.rank===o.rank,()=> ` Rank of input ( $ { a . rank } ) does not match rank of dy ( $ { o . rank } ) ` );let i=a,l=o,u=!1;a.rank===3&&(u=!0,i=ie(a,[1,a.shape[0],a.shape[1],a.shape[2]]),l=ie(o,[1,o.shape[0],o.shape[1],o.shape[2]])),H(l.rank===4,()=> ` Error in avgPoolGrad : dy must be rank 4 but got rank $ { l . rank } . ` ),H(i.rank===4,()=> ` Error in avgPoolGrad : input must be rank 4 but got rank $ { i . rank } . ` );let c={dy:l,input:i},p={filterSize:n,strides:r,pad:s},d=se.runKernel(ew,c,p);return u?ie(d,[d.shape[1],d.shape[2],d.shape[3]]):d}var P0e=ae({avgPoolGrad_:O0e}),M0e={ke
1. The $ { r } is defined in Python , in which case it needs to be ported to TensorFlow . js or your JavaScript code .
2. The custom $ { r } is defined in JavaScript , but is not registered properly with tf . serialization . registerClass ( ) . ` );return a}else{let o=e;if(o.className==null||o.config==null)throw new pe( ` $ { r } : Improper config format : $ { JSON . stringify ( o ) } .
'className' and 'config' must set . ` );let a=o.className,i,l;if(a in n?[i,l]=n[a]:a in Oa?[i,l]=Oa.className:a in t&&([i,l]=t[a]),i==null)throw new pe( ` Unknown $ { r } : $ { a } . This may be due to one of the following reasons :
1. The $ { r } is defined in Python , in which case it needs to be ported to TensorFlow . js or your JavaScript code .
2. The custom $ { r } is defined in JavaScript , but is not registered properly with tf . serialization . registerClass ( ) . ` );if(l!=null){let u={};for(let f of Object.keys(Oa))u[f]=Oa[f];for(let f of Object.keys(n))u[f]=n[f];let c=o.config;c.customObjects=u;let p=Object.assign({},Oa);for(let f of Object.keys(n))Oa[f]=n[f];rE(o.config);let d=l(i,o.config,n,s);return Oa=Object.assign({},p),d}else{let u=Object.assign({},Oa);for(let p of Object.keys(n))Oa[p]=n[p];let c=new i(o.config);return Oa=Object.assign({},u),c}}}function wye(e,t){return e<t?-1:e>t?1:0}function Pk(e,t){return-1*wye(e,t)}function qc(e){if(e==null)return e;let t=[];for(let n of e)t.indexOf(n)===-1&&t.push(n);return t}function kye(e){if(e==null)throw new pe( ` Invalid value in obj : $ { JSON . stringify ( e ) } ` );for(let t in e)if(e.hasOwnProperty(t))return!1;return!0}function rh(e,t,n){if(n!=null&&e.indexOf(n)<0)throw new pe( ` $ { n } is not a valid $ { t } . Valid values are $ { e } or null / undefined . ` )}function sE(e,t,n=0,r=1/0){return ul(n>=0),ul(r>=n),Array.isArray(e)&&e.length>=n&&e.length<=r&&e.every(s=>typeof s===t)}function cs(e,t){Array.isArray(e)?(E.assert(e.length>0,()=> ` $ { t } is unexpectedly an empty array . ` ),e.forEach((n,r)=>cs(n, ` element $ { r + 1 } of $ { t } ` ))):E.assert(Number.isInteger(e)&&e>0,()=> ` Expected $ { t } to be a positive integer , but got $ { MM ( e ) } . ` )}function MM(e){return e===null?"null":Array.isArray(e)?"["+e.map(t=>MM(t)).join(",")+"]":typeof e=="string"? ` "${e}" ` : ` $ { e } ` }function Iye(e,t){let n=E.now(),r;return(...o)=>{let a=E.now();return a-n<t||(n=a,r=e(...o)),r}}function zM(e){return e==="relu"?"relu":e==="linear"?"linear":e==="elu"?"elu":null}function oE(e,t){return ce(()=>As(Je(te(e,e),t,!0)))}var Nb=class extends De.Serializable{getConfig(){return{}}},aE=class extends Nb{constructor(e){super();this.defaultMaxValue=2,this.defaultAxis=0,this.maxValue=e.maxValue!=null?e.maxValue:this.defaultMaxValue,this.axis=e.axis!=null?e.axis:this.defaultAxis}apply(e){return ce(()=>{let t=oE(e,this.axis),n=yo(t,0,this.maxValue);return te(e,Me(n,Re(Hr(),t)))})}getConfig(){return{maxValue:this.maxValue,axis:this.axis}}};aE.className="MaxNorm";De.registerClass(aE);var iE=class extends Nb{constructor(e){super();this.defaultAxis=0,this.axis=e.axis!=null?e.axis:this.defaultAxis}apply(e){return ce(()=>Me(e,Re(Hr(),oE(e,this.axis))))}getConfig(){return{axis:this.axis}}};iE.className="UnitNorm";De.registerClass(iE);var lE=class extends Nb{apply(e){return yi(e)}};lE.className="NonNeg";De.registerClass(lE);var uE=class extends Nb{constructor(e){super();this.defaultMinValue=0,this.defaultMaxValue=1,this.defaultRate=1,this.defaultAxis=0,this.minValue=e.minValue!=null?e.minValue:this.defaultMinValue,this.maxValue=e.maxValue!=null?e.maxValue:this.defaultMaxValue,this.rate=e.rate!=null?e.rate:this.defaultRate,this.axis=e.axis!=null?e.axis:this.defaultAxis}apply(e){return ce(()=>{let t=oE(e,this.axis),n=Re(te(this.rate,yo(t,this.minValue,this.maxValue)),te(1-this.rate,t));return te(e,Me(n,Re(Hr(),t)))})}getConfig(){return{minValue:this.minValue,maxValue:this.maxValue,rate:this.rate,axis:this.axis}}};uE.className="MinMaxNorm";De.registerClass(uE);var LM={maxNorm:"MaxNorm",minMaxNorm:"MinMaxNorm",nonNeg:"NonNeg",unitNorm:"UnitNorm"};function qr(e){return nE(e)}function BM(e,t={}){return Sb(e,De.SerializationMap.getMap().classNameMap,t,"constraint")}function Kr(e){if(e==null)return null;if(typeof e=="string"){let n={className:e in LM?LM[e]:e,config:{}};return BM(n)}else return e instanceof Nb?e:BM(e)}function Cye(e){return new aE(e)}function Sye(e){return new iE(e)}function Nye(){return new lE}function Tye(e){return new uE(e)}var VM={};Rt(VM,{constant:()=>Yye,glorotNormal:()=>s1e,glorotUniform:()=>r1e,heNormal:()=>o1e,heUniform:()=>a1e,identity:()=>t1e,leCunNormal:()=>i1e,leCunUniform:()=>l1e,ones:()=>Zye,orthogonal:()=>u1e,randomNormal:()=>Qye,randomUniform:()=>Jye,truncatedNormal:()=>e1e,varianceScaling:()=>n1e,zeros:()=>Xye});var _ye=["channelsFirst","channelsLast"],Eye=["nearest","bilinear"],Rye=["valid","same","causal"],Dye=["max","avg"], $ ye=["sum","mul","concat","ave"],D0=new Map;function Ir
because the value dtype is $ { t . dtype } , but TensorArray dtype is $ { this . dtype } . ` );if(this.size()===0&&(this.elementShape==null||this.elementShape.length===0)&&(this.elementShape=t.shape),za(this.elementShape,t.shape, ` TensorArray $ { this . name } : Could not write to TensorArray index $ { e } . ` ),n.read)throw new Error( ` TensorArray $ { this . name } : Could not write to TensorArray index $ { e } , because it has already been read . ` );if(n.written)throw new Error( ` TensorArray $ { this . name } : Could not write to TensorArray index $ { e } , because it has already been written . ` );n.tensor=t,us(t),n.written=!0,this.tensors[e]=n}writeMany(e,t){if(e.length!==t.length)throw new Error( ` TensorArray $ { this . name } : could not write multiple tensors , because the index size : $ { e . length } is not the same as tensors size : $ { t . length } . ` );e.forEach((n,r)=>this.write(n,t[r]))}gather(e,t){if(!!t&&t!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but gather requested dtype $ { t } ` );if(e)e=e.slice(0,this.size());else{e=[];for(let r=0;r<this.size();r++)e.push(r)}if(e.length===0)return ls([],[0].concat(this.elementShape));let n=this.readMany(e);return za(this.elementShape,n[0].shape,"TensorArray shape mismatch: "),vs(n,0)}concat(e){if(!!e&&e!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but concat requested dtype $ { e } ` );if(this.size()===0)return ls([],[0].concat(this.elementShape));let t=[];for(let r=0;r<this.size();r++)t.push(r);let n=this.readMany(t);return za(this.elementShape,n[0].shape, ` TensorArray shape mismatch : tensor array shape ( $ { this . elementShape } ) vs first tensor shape ( $ { n [ 0 ] . shape } ) ` ),En(n,0)}scatter(e,t){if(t.dtype!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but tensor has dtype $ { t . dtype } ` );if(e.length!==t.shape[0])throw new Error( ` Expected len ( indices ) == tensor . shape [ 0 ] , but saw : $ { e . length } vs . $ { t . shape [ 0 ] } ` );let n=Math.max(...e);if(!this.dynamicSize&&n>=this.maxSize)throw new Error( ` Max index must be < array size ( $ { n } vs . $ { this . maxSize } ) ` );this.writeMany(e,Vs(t,0))}split(e,t){if(t.dtype!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but tensor has dtype $ { t . dtype } ` );let n=0,r=e.map(i=>(n+=i,n));if(n!==t.shape[0])throw new Error( ` Expected sum of lengths to be equal to
2021-09-11 16:29:31 +02:00
tensor . shape [ 0 ] , but sum of lengths is
2021-10-04 22:29:15 +02:00
$ { n } , and tensor 's shape is: ${t.shape}`);if(!this.dynamicSize&&e.length!==this.maxSize)throw new Error(`TensorArray' s size is not equal to the size of lengths ( $ { this . maxSize } vs . $ { e . length } ) , and the TensorArray is not marked as dynamically resizeable ` );let s=n===0?0:t.size/n,o=[];ce(()=>{t=ie(t,[1,n,s]);for(let i=0;i<e.length;++i){let l=i===0?0:r[i-1],u=[0,l,0],c=[1,e[i],s];o[i]=ie(gt(t,u,c),this.elementShape)}return o});let a=[];for(let i=0;i<e.length;i++)a[i]=i;this.writeMany(a,o)}},jb=class{constructor(e,t,n,r=-1){this.tensors=e,this.elementShape=t,this.elementDtype=n,e!=null&&e.forEach(s=>{if(n!==s.dtype)throw new Error( ` Invalid data types ; op elements $ { n } , but list elements $ { s . dtype } ` );za(t,s.shape,"TensorList shape mismatch: "),us(s)}),this.idTensor=at(0),this.maxNumElements=r,us(this.idTensor)}get id(){return this.idTensor.id}copy(){return new jb([...this.tensors],this.elementShape,this.elementDtype)}clearAndClose(e){this.tensors.forEach(t=>{(e==null||!e.has(t.id))&&t.dispose()}),this.tensors.length=0,this.idTensor.dispose()}size(){return this.tensors.length}stack(e,t,n=-1){if(t!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { t } , but list elements $ { this . elementDtype } ` );if(n!==-1&&this.tensors.length!==n)throw new Error( ` Operation expected a list with $ { n } elements but got a list with $ { this . tensors . length } elements . ` );za(e,this.elementShape,"TensorList shape mismatch: ");let r=Ub(this.elementShape,this.tensors,e);return ce(()=>{let s=this.tensors.map(o=>ie(o,r));return vs(s,0)})}popBack(e,t){if(t!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { t } , but list elements $ { this . elementDtype } ` );if(this.size()===0)throw new Error("Trying to pop from an empty list.");let n=Ub(this.elementShape,this.tensors,e),r=this.tensors.pop();return za(r.shape,e,"TensorList shape mismatch: "),ie(r,n)}pushBack(e){if(e.dtype!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { e . dtype } , but list elements $ { this . elementDtype } ` );if(za(e.shape,this.elementShape,"TensorList shape mismatch: "),this.maxNumElements===this.size())throw new Error("Trying to push element into a full list.");us(e),this.tensors.push(e)}resize(e){if(e<0)throw new Error( ` TensorListResize expects size to be non - negative . Got : $ { e } ` );if(this.maxNumElements!==-1&&e>this.maxNumElements)throw new Error( ` TensorListResize input size $ { e } is greater maxNumElement $ { this . maxNumElements } . ` );this.tensors.length=e}getItem(e,t,n){if(n!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { n } , but list elements $ { this . elementDtype } ` );if(e<0||e>this.tensors.length)throw new Error( ` Trying to access element $ { e } in a list with $ { this . tensors . length } elements . ` );if(this.tensors[e]==null)throw new Error( ` element at index $ { e } is null . ` );za(this.tensors[e].shape,t,"TensorList shape mismatch: ");let r=Ub(this.elementShape,this.tensors,t);return ie(this.tensors[e],r)}setItem(e,t){if(t.dtype!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { t . dtype } , but list elements $ { this . elementDtype } ` );if(e<0||this.maxNumElements!==-1&&e>=this.maxNumElements)throw new Error( ` Trying to set element $ { e } in a list with max $ { this . maxNumElements } elements . ` );za(this.elementShape,t.shape,"TensorList shape mismatch: "),us(t),this.tensors[e]=t}gather(e,t,n){if(t!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { t } , but list elements $ { this . elementDtype } ` );za(this.elementShape,n,"TensorList shape mismatch: "),e=e.slice(0,this.size());let r=Ub(this.elementShape,this.tensors,n);return e.length===0?ls([],[0].concat(r)):ce(()=>{let s=e.map(o=>ie(this.tensors[o],r));return vs(s,0)})}concat(e,t){if(!!e&&e!==this.elementDtype)throw new Error( ` TensorList dtype is $ { this . elementDtype } but concat requested dtype $ { e } ` );za(this.elementShape,t,"TensorList shape mismatch: ");let n=Ub(this.elementShape,this.tensors,t);return this.size()===0?ls([],[0].concat(n)):ce(()=>{let r=this.tensors.map(s=>ie(s,n));return En(r,0)})}};function D2e(e,t,n){let r=e.dtype;if(e.shape.length<1)throw new Err
2021-09-11 16:29:31 +02:00
tensor . shape [ 0 ] , but sum of lengths is
2021-10-04 22:29:15 +02:00
$ { r } , and tensor ' s shape is : $ { e . shape } ` );let o=e.shape.slice(1),a=Q4(o,n),i=r===0?0:e.size/r,l=ce(()=>{let c=[];e=ie(e,[1,r,i]);for(let p=0;p<t.length;++p){let d=p===0?0:s[p-1],f=[0,d,0],h=[1,t[p],i];c[p]=ie(gt(e,f,h),a)}return e.dispose(),c}),u=new jb([],n,e.dtype,t.length);for(let c=0;c<l.length;c++)u.setItem(c,l[c]);return u}var P2e=async(e,t,n)=>{switch(e.op){case"If":case"StatelessIf":{let r= $ ("thenBranch",e,t,n),s= $ ("elseBranch",e,t,n),o= $ ("cond",e,t,n),a= $ ("args",e,t,n);return(await o.data())[0]?n.functionMap[r].executeFunctionAsync(a,n.tensorArrayMap,n.tensorListMap):n.functionMap[s].executeFunctionAsync(a,n.tensorArrayMap,n.tensorListMap)}case"While":case"StatelessWhile":{let r= $ ("body",e,t,n),s= $ ("cond",e,t,n),o= $ ("args",e,t,n),a=await n.functionMap[s].executeFunctionAsync(o,n.tensorArrayMap,n.tensorListMap),i=o.map(c=>c.id),l=await a[0].data();a.forEach(c=>{!c.kept&&i.indexOf(c.id)===-1&&c.dispose()});let u=o;for(;l[0];){let c=u;u=await n.functionMap[r].executeFunctionAsync(u,n.tensorArrayMap,n.tensorListMap);let p=u.map(f=>f.id);c.forEach(f=>{!f.kept&&i.indexOf(f.id)===-1&&p.indexOf(f.id)===-1&&f.dispose()});let d=await n.functionMap[s].executeFunctionAsync(u,n.tensorArrayMap,n.tensorListMap);l=await d[0].data(),d.forEach(f=>{!f.kept&&i.indexOf(f.id)===-1&&p.indexOf(f.id)===-1&&f.dispose()})}return u}case"LoopCond":{let r= $ ("pred",e,t,n);return[Xl(r)]}case"Switch":{let r= $ ("pred",e,t,n),s= $ ("data",e,t,n);return s.kept||(s=Xl(s)),(await r.data())[0]?[void 0,s]:[s,void 0]}case"Merge":{let r=e.inputNames.find(s=>Gs(s,t,n)!==void 0);if(r){let s=Gs(r,t,n);return[Xl(s)]}return}case"Enter":{let r= $ ("frameName",e,t,n),s= $ ("tensor",e,t,n);return n.enterFrame(r),[Xl(s)]}case"Exit":{let r= $ ("tensor",e,t,n);return n.exitFrame(),[Xl(r)]}case"NextIteration":{let r= $ ("tensor",e,t,n);return n.nextIteration(),[Xl(r)]}case"TensorArrayV3":{let r= $ ("size",e,t,n),s= $ ("dtype",e,t,n),o= $ ("elementShape",e,t,n),a= $ ("dynamicSize",e,t,n),i= $ ("clearAfterRead",e,t,n),l= $ ("identicalElementShapes",e,t,n),u= $ ("name",e,t,n),c=new R2e(u,s,r,o,l,a,i);return n.addTensorArray(c),[c.idTensor,at(1)]}case"TensorArrayWriteV3":{let r= $ ("tensorArrayId",e,t,n),s= $ ("index",e,t,n),o= $ ("tensor",e,t,n),a=n.getTensorArray(r.id);return a.write(s,o),[a.idTensor]}case"TensorArrayReadV3":{let r= $ ("tensorArrayId",e,t,n),s= $ ("index",e,t,n);return[n.getTensorArray(r.id).read(s)]}case"TensorArrayGatherV3":{let r= $ ("tensorArrayId",e,t,n),s= $ ("indices",e,t,n),o= $ ("dtype",e,t,n);return[n.getTensorArray(r.id).gather(s,o)]}case"TensorArrayScatterV3":{let r= $ ("tensorArrayId",e,t,n),s= $ ("indices",e,t,n),o= $ ("tensor",e,t,n),a=n.getTensorArray(r.id);return a.scatter(s,o),[a.idTensor]}case"TensorArrayConcatV3":{let r= $ ("tensorArrayId",e,t,n),s=n.getTensorArray(r.id),o= $ ("dtype",e,t,n);return[s.concat(o)]}case"TensorArraySplitV3":{let r= $ ("tensorArrayId",e,t,n),s= $ ("tensor",e,t,n),o= $ ("lengths",e,t,n),a=n.getTensorArray(r.id);return a.split(o,s),[a.idTensor]}case"TensorArraySizeV3":{let r= $ ("tensorArrayId",e,t,n),s=n.getTensorArray(r.id);return[at(s.size(),"int32")]}case"TensorArrayCloseV3":{let r= $ ("tensorArrayId",e,t,n),s=n.getTensorArray(r.id);return s.clearAndClose(),[s.idTensor]}case"TensorListSetItem":{let r= $ ("tensorListId",e,t,n),s= $ ("index",e,t,n),o= $ ("tensor",e,t,n),a=n.getTensorList(r.id);return a.setItem(s,o),[a.idTensor]}case"TensorListGetItem":{let r= $ ("tensorListId",e,t,n),s= $ ("index",e,t,n),o= $ ("elementShape",e,t,n),a= $ ("elementDType",e,t,n);return[n.getTensorList(r.id).getItem(s,o,a)]}case"TensorListScatterV2":case"TensorListScatter":{let r= $ ("indices",e,t,n),s= $ ("tensor",e,t,n),o= $ ("elementShape",e,t,n),a= $ ("numElements",e,t,n),i=F2e(s,r,o,a);return n.addTensorList(i),[i.idTensor]}case"TensorListReserve":case"EmptyTensorList":{let r= $ ("elementShape",e,t,n),s= $ ("elementDType",e,t,n),o;e.op==="TensorListReserve"?o="numElements":o="maxNumElements";let a= $ (o,e,t,n),i= $ 2e(r,s,a);return n.addTensorList(i),[i.idTensor]}case"TensorListGather":{let r= $ ("tensorListId",e,t,n),s= $ ("indices",e,t,n),o= $ ("elementShape",e,t,n),a= $ ("elementDType",e,t,n);return[n.ge
$ { e } ` );let r;return this.size===1/0||this.size==null?r=this.size:t?r=Math.ceil(this.size/e):r=Math.floor(this.size/e),qo(async()=>(await n.iterator()).columnMajorBatch(e,t,Fve),r)}concatenate(e){let t=this,n;return this.size===1/0||e.size===1/0?n=1/0:this.size!=null&&e.size!=null?n=this.size+e.size:n=null,qo(async()=>(await t.iterator()).concatenate(await e.iterator()),n)}filter(e){let t=this,n;return this.size===1/0?n=1/0:n=null,qo(async()=>(await t.iterator()).filter(r=>ce(()=>e(r))),n)}async forEachAsync(e){return(await this.iterator()).forEachAsync(e)}map(e){let t=this;return qo(async()=>(await t.iterator()).map(n=>ce(()=>e(n))),this.size)}mapAsync(e){let t=this;return qo(async()=>(await t.iterator()).mapAsync(e),this.size)}prefetch(e){if(e==null)throw new RangeError(" ` Dataset . prefetch ( ) ` requires bufferSize to be specified.");let t=this;return qo(async()=>(await t.iterator()).prefetch(e),this.size)}repeat(e){let t=this,n;return this.size!=null&&e>0?n=this.size*e:e===0?n=0:this.size!=null&&(e===void 0||e<0)?n=1/0:n=null,qo(async()=>{let r=rR(async()=>({value:await t.iterator(),done:!1}));return yve(r.take(e))},n)}skip(e){let t=this,n;return this.size!=null&&e>=0&&this.size>=e?n=this.size-e:this.size!=null&&(this.size<e||e===void 0||e<0)?n=0:n=null,qo(async()=>(await t.iterator()).skip(e),n)}shuffle(e,t,n=!0){if(e==null||e<0)throw this.size==null?new RangeError(" ` Dataset . shuffle ( ) ` requires bufferSize to be specified."):new RangeError( ` \ ` Dataset.shuffle() \` requires bufferSize to be specified. If your data fits in main memory (for regular JS objects), and/or GPU memory (for \` tf.Tensor \` s), consider setting bufferSize to the dataset size ( ${ this . size } elements) ` ) ; let r = this , s = cve . alea ( t || E . now ( ) . toString ( ) ) ; return qo ( async ( ) => { let o = s . int32 ( ) ; return n && ( o += s . int32 ( ) ) , ( await r . iterator ( ) ) . shuffle ( e , o . toString ( ) ) } , this . size ) } take ( e ) { let t = this , n ; return this . size != null && this . size > e ? n = e : this . size != null && this . size <= e ? n = this . size : n = null , qo ( async ( ) => ( await t . iterator ( ) ) . take ( e ) , n ) } async toArray ( ) { if ( this . size === 1 / 0 ) throw new Error ( "Can not convert infinite data stream to array." ) ; return ( await this . iterator ( ) ) . toArray ( ) } async toArrayForTest ( ) { if ( this . size === 1 / 0 ) throw new Error ( "Can not convert infinite data stream to array." ) ; return ( await this . iterator ( ) ) . toArrayForTest ( ) } } ; B0 . MAX _BUFFER _SIZE = 1e4 ; function qo ( e , t = null ) { return new class extends B0 { constructor ( ) { super ( ... arguments ) ; this . size = t } async iterator ( ) { return e ( ) } } } function Dve ( e ) { return qo ( async ( ) => eB ( e ) , e . length ) } function $ve ( e ) { if ( ! L0 ( e ) ) throw new Error ( "The argument to zip() must be an object or array." ) ; let t ; if ( Array . isArray ( e ) ) for ( let n = 0 ; n < e . length ; n ++ ) t = t == null ? e [ n ] . size : Math . min ( t , e [ n ] . size ) ; else if ( e instanceof Object ) for ( let n in e ) t = t == null ? e [ n ] . size : Math . min ( t , e [ n ] . size ) ; return qo ( async ( ) => { let n = await JL ( e , r => { if ( r instanceof B0 ) return { value : r . iterator ( ) , recurse : ! 1 } ; if ( L0 ( r ) ) return { value : null , recurse : ! 0 } ; throw new Error ( "Leaves of the structure passed to zip() must be Datasets, not primitives." ) } ) ; return bve ( n , ep . SHORTEST ) } , t ) } function Fve ( e ) { if ( e === null ) return null ; let t = e [ 0 ] ; return hve ( t ) ? { value : Ove ( e ) , recurse : ! 1 } : { value : null , recurse : ! 0 } } function Ove ( e ) { if ( e . length === 0 ) throw new Error ( "Can't make a batch of zero elements." ) ; return e [ 0 ] instanceof Vt ? vs ( e ) : ls ( e ) } var sB = class extends B0 { constructor ( e ) { super ( ) ; this . input = e } async iterator ( ) { return ( await this . input . iterator ( ) ) . decodeUTF8 ( ) . split ( `
` ).map(r=>(r.endsWith(" \r ")&&(r=r.slice(0,-1)),r))}},x5='"',Hb=Symbol("out"),oB=Symbol("field"),y5=Symbol("quote"),oR=Symbol("quoteafterquote"),aB=Symbol("quoteinquote"),iB=class extends B0{constructor(e,t){super();this.input=e,this.hasHeader=!0,this.fullColumnNames=null,this.columnNamesValidated=!1,this.columnConfigs=null,this.configuredColumnsOnly=!1,this.delimiter=",",this.delimWhitespace=!1,this.base=new sB(e),t||(t={}),this.hasHeader=t.hasHeader!==!1,this.fullColumnNames=t.columnNames,this.columnConfigs=t.columnConfigs,this.configuredColumnsOnly=t.configuredColumnsOnly,t.delimWhitespace?(E.assert(t.delimiter==null,()=>"Delimiter should not be provided when delimWhitespace is true."),this.delimWhitespace=!0,this.delimiter=" "):this.delimiter=t.delimiter?t.delimiter:","}async columnNames(){return this.columnNamesValidated||await this.setColumnNames(),this.configuredColumnsOnly?Object.keys(this.columnConfigs):this.fullColumnNames}async setColumnNames(){let e=await this.maybeReadHeaderLine();if(!this.fullColumnNames&&!e)throw new Error("Column names must be provided if there is no header line.");this.fullColumnNames&&e&&E.assert(e.length===this.fullColumnNames.length,()=>"The length of provided columnNames ("+this.fullColumnNames.length.toString()+") does not match the length of the header line read from file ("+e.length.toString()+")."),this.fullColumnNames||(this.fullColumnNames=e);let t=this.fullColumnNames.reduce((r,s)=>(r[s]=r[s]+1||1,r),{}),n=Object.keys(t).filter(r=>t[r]>1);if(E.assert(n.length===0,()=>"Duplicate column names found: "+n.toString()),this.columnConfigs){for(let r of Object.keys(this.columnConfigs))if(this.fullColumnNames.indexOf(r)===-1)throw new Error('The key "'+r+'" provided in columnConfigs does not match any of the column names ('+this.fullColumnNames.toString()+").")}this.columnNamesValidated=!0}async maybeReadHeaderLine(){if(this.hasHeader){let t=await(await this.base.iterator()).next();if(t.done)throw new Error("No data was found for CSV parsing.");let n=t.value;return this.parseRow(n,!1)}else return null}async iterator(){this.columnNamesValidated||await this.setColumnNames();let e=await this.base.iterator();return this.hasHeader&&(e=e.skip(1)),e.map(t=>this.makeDataElement(t))}makeDataElement(e){let t=this.parseRow(e),n={},r={};for(let s=0;s<this.fullColumnNames.length;s++){let o=this.fullColumnNames[s],a=this.columnConfigs?this.columnConfigs[o]:null;if(!(this.configuredColumnsOnly&&!a)){let i=t[s],l=null;if(i==="")if(a&&a.default!==void 0)l=a.default;else{if(a&&(a.required||a.isLabel))throw new Error( ` Required column $ { o } is empty in this line : $ { e } ` );l=void 0}else{let u=Number(i);if(isNaN(u))a&&a.dtype==="bool"?l=this.getBoolean(i):l=i;else if(!a||!a.dtype)l=u;else switch(a.dtype){case"float32":l=u;break;case"int32":l=Math.floor(u);break;case"bool":l=this.getBoolean(i);break;default:l=u}}a&&a.isLabel?r[o]=l:n[o]=l}}return Object.keys(r).length===0?n:{xs:n,ys:r}}getBoolean(e){return e==="1"||e.toLowerCase()==="true"?1:0}parseRow(e,t=!0){let n=[],r=0,s=e.length,o=Hb;for(let a=0;a<s;a++)switch(o){case Hb:switch(e.charAt(a)){case x5:r=a+1,o=y5;break;case this.delimiter:if(r=a+1,this.delimiter===" "&&this.delimWhitespace)break;n.push(""),o=Hb;break;default:o=oB,r=a;break}break;case oB:switch(e.charAt(a)){case this.delimiter:n.push(e.substring(r,a)),o=Hb,r=a+1;break;default:}break;case y5:switch(e.charAt(a)){case x5:o=oR;break;default:}break;case oR:switch(e.charAt(a)){case this.delimiter:n.push(e.substring(r,a-1)),o=Hb,r=a+1;break;case x5:o=y5;break;default:o=aB;break}break;case aB:switch(e.charAt(a)){case x5:o=y5;break;default:}break;default:}if(o===oR?n.push(e.substring(r,s-1)):n.push(e.substring(r)),t&&n.length!==this.fullColumnNames.length)throw new Error( ` Invalid row in csv file . Should have $ { this . fullColumnNames . length } elements in a row , but got $ { n } ` );return n}},lB=class extends ps{constructor(e){super();this.microphoneConfig=e,this.isClosed=!1,this.fftSize=e.fftSize||1024;let t=Math.log2(this.fftSize);if(this.fftSize<0||t<4||t>14||!Number.isInteger(t))throw new Error( ` Invalid fftSi
2021-09-11 16:29:31 +02:00
=== === === === === === === === === =
Hi there \ u { 1 F44B } . Looks like you are running TensorFlow . js in Node . js . To speed things up dramatically , install our node backend , which binds to TensorFlow C ++ , by running npm i @ tensorflow / tfjs - node , or npm i @ tensorflow / tfjs - node - gpu if you have CUDA . Then call require ( '@tensorflow/tfjs-node' ) ; ( - gpu suffix for CUDA ) at the start of your program . Visit https : //github.com/tensorflow/tfjs-node for more details.
2021-10-04 22:29:15 +02:00
=== === === === === === === === === = ` ));let r={id:this.nextDataId()};return this.data.set(r,{values:e,dtype:n,refCount:1}),r}makeTensorInfo(e,t,n){let r;if(t==="string"&&n!=null&&n.length>0&&E.isString(n[0])){let s=n.map(o=>E.encodeString(o));r=this.write(s,e,t)}else r=this.write(n,e,t);return{dataId:r,shape:e,dtype:t}}refCount(e){return this.data.has(e)?this.data.get(e).refCount:0}incRef(e){let t=this.data.get(e);t.refCount++}decRef(e){if(this.data.has(e)){let t=this.data.get(e);t.refCount--}}move(e,t,n,r,s){this.data.set(e,{values:t,dtype:r,refCount:s})}numDataIds(){return this.data.numDataIds()}async read(e){return this.readSync(e)}readSync(e){let{dtype:t,complexTensorInfos:n}=this.data.get(e);if(t==="complex64"){let r=this.readSync(n.real.dataId),s=this.readSync(n.imag.dataId);return G.mergeRealAndImagArrays(r,s)}return this.data.get(e).values}bufferSync(e){let t=this.readSync(e.dataId),n=t;if(e.dtype==="string")try{n=t.map(r=>E.decodeString(r))}catch(r){throw new Error("Failed to decode encoded string bytes into utf-8")}return Wt(e.shape,e.dtype,n)}makeOutput(e,t,n){let r=this.write(e,t,n);return zo().makeTensorFromDataId(r,t,n,this)}disposeData(e,t=!1){if(this.data.has(e)){if(this.data.get(e).refCount--,!t&&this.data.get(e).refCount>0)return!1;let{complexTensorInfos:n}=this.data.get(e);n!=null&&(this.disposeData(n.real.dataId,!0),this.disposeData(n.imag.dataId,!0)),this.data.delete(e)}return!0}disposeIntermediateTensorInfo(e){this.disposeData(e.dataId)}async time(e){let t=E.now();return e(),{kernelMs:E.now()-t}}memory(){return{unreliable:!0,reasons:["The reported memory is an upper bound. Due to automatic garbage collection, the true allocated memory may be less."]}}where(e){rt([e],"where");let t=this.readSync(e.dataId);return Xve(e.shape,t)}dispose(){}floatPrecision(){return 32}epsilon(){return super.epsilon()}};aR.nextDataId=0;var gB={};Rt(gB,{addImpl:()=>yB,bincountImpl:()=>lR,bincountReduceImpl:()=>bB,ceilImpl:()=>AB,concatImpl:()=>uR,equalImpl:()=>vB,expImpl:()=>kB,expm1Impl:()=>CB,floorImpl:()=>SB,gatherNdImpl:()=>NB,gatherV2Impl:()=>TB,greaterEqualImpl:()=>EB,greaterImpl:()=>_B,lessEqualImpl:()=>DB,lessImpl:()=>RB,linSpaceImpl:()=> $ B,logImpl:()=>FB,maxImpl:()=>OB,maximumImpl:()=>PB,minimumImpl:()=>MB,multiplyImpl:()=>cR,negImpl:()=>zB,notEqualImpl:()=>LB,prodImpl:()=>BB,rangeImpl:()=>dR,rsqrtImpl:()=>VB,sigmoidImpl:()=>Mwe,simpleAbsImpl:()=>xB,sliceImpl:()=>v5,sparseFillEmptyRowsImpl:()=>GB,sparseReshapeImpl:()=>UB,sparseSegmentReductionImpl:()=>fR,sqrtImpl:()=>Bwe,squaredDifferenceImpl:()=>jB,stridedSliceImpl:()=>HB,stringNGramsImpl:()=>qB,stringSplitImpl:()=>KB,stringToHashBucketFastImpl:()=>XB,subImpl:()=>ZB,tileImpl:()=>YB,topKImpl:()=>QB,transposeImpl:()=>pR,uniqueImpl:()=>eV});function xB(e){let t=new Float32Array(e.length);for(let n=0;n<e.length;++n)t[n]=Math.abs(e[n]);return t}var Zve=e=>{let{x:t}=e.inputs,n=e.backend;rt(t,"abs");let r=new Float32Array(E.sizeFromShape(t.shape)),s=n.data.get(t.dataId).values;return r=xB(s),n.makeOutput(r,t.shape,"float32")},Yve={kernelName:cg,backendName:"cpu",kernelFunc:Zve};function Dr(e){return(t,n,r,s,o)=>{let a=G.assertAndGetBroadcastShape(t,n),i=a.length,l=E.computeStrides(a),u=E.sizeFromShape(a),c=E.getTypedArrayFromDType(o,u),p=t.length,d=n.length,f=E.computeStrides(t),h=E.computeStrides(n),m=G.getBroadcastDims(t,a),g=G.getBroadcastDims(n,a);if(m.length+g.length===0)for(let x=0;x<c.length;++x)c[x]=e(r[x%r.length],s[x%s.length]);else for(let x=0;x<c.length;++x){let y=E.indexToLoc(x,i,l),b=y.slice(-p);m.forEach(k=>b[k]=0);let A=E.locToIndex(b,p,f),v=y.slice(-d);g.forEach(k=>v[k]=0);let w=E.locToIndex(v,d,h);c[x]=e(r[A],s[w])}return[c,a]}}function Ko(e){let{inputs:t,backend:n}=e,{real:r,imag:s}=t,o=n.data.get(r.dataId).values,a=n.data.get(s.dataId).values,i=n.makeTensorInfo(r.shape,"complex64"),l=n.data.get(i.dataId);return l.complexTensorInfos={real:n.makeTensorInfo(r.shape,"float32",o),imag:n.makeTensorInfo(s.shape,"float32",a)},i}var Jve={kernelName:rw,backendName:"cpu",kernelFunc:Ko};function b5(e,t,n="float32"){if(n==="complex64"){let s=b5(e,t,"float32"),o=b5(e,t,"
indices . shape [ 0 ] = $ { i } ` );let g=E.getArrayFromDType(n,0),x=E.getArrayFromDType(s,0);return[g,[0,p],x,u,c]}let d=!0,f=0,h=new Array(l).fill(0);for(let g=0;g<i;++g){let x=e[g*p];if(x<0)throw new Error( ` indices ( $ { g } , 0 ) is invalid : $ { x } < 0 ` );if(x>=l)throw new Error( ` indices ( $ { g } , 0 ) is invalid : $ { x } >= $ { l } ` );++h[x],d=d&&x>=f,f=x}let m=!0;for(let g=0;g<l;++g){let x=h[g]===0;u[g]=x,m=m&&!x,h[g]=Math.max(h[g],1),g>0&&(h[g]+=h[g-1])}if(m&&d){let g=e,x=r;for(let y=0;y<i;++y)c[y]=y;return[g,[i,p],x,u,c]}else{let g=h[l-1],x=E.getArrayFromDType(n,g*p),y=E.getArrayFromDType(s,g),b=new Array(l).fill(0);for(let A=0;A<i;++A){let v=e[A*p],w=b[v],k=(v===0?0:h[v-1])+w;b[v]++;for(let I=0;I<p;++I)x[k*p+I]=e[A*p+I];y[k]=r[A],c[A]=k}for(let A=0;A<l;++A)if(b[A]===0){let w=A===0?0:h[A-1];x[w*p+0]=A;for(let k=1;k<p;++k)x[w*p+k]=0;y[w]=a}return[x,[g,p],y,u,c]}}function UB(e,t,n,r,s){let o=E.sizeFromShape(r),a=t[0],i=s.length,l=[],u=1,c=-1;for(let g=0;g<i;++g){let x=s[g];if(x===-1){if(c!==-1)throw new Error( ` only one output dimension may be - 1 , not both $ { c } and $ { g } ` );c=g,l.push(1)}else{if(x<0)throw new Error( ` size $ { g } must be non - negative , not $ { x } ` );u*=x,l.push(x)}}if(c!==-1){if(u<=0)throw new Error("reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero");let g=Math.trunc(o/u);if(u*g!==o)throw new Error( ` Input to reshape is a SparseTensor with $ { o }
dense values , but the requested shape requires a multiple of $ { u } . inputShape = $ { r } outputShape = $ { l } ` );l[c]=g}let p=E.sizeFromShape(l);if(p!==o)throw new Error( ` Input to reshape is a tensor with $ { o } dense values , but the requested shape has $ { p } . inputShape = $ { r } outputShape = $ { l } ` );let d=r.length,f=[];if(d>0){f[d-1]=1;for(let g=d-2;g>=0;--g)f[g]=f[g+1]*r[g+1]}let h=[];if(i>0){h[i-1]=1;for(let g=i-2;g>=0;--g)h[g]=h[g+1]*l[g+1]}let m=E.getArrayFromDType(n,a*i);for(let g=0;g<a;++g){let x=0;for(let y=0;y<d;++y)x+=e[g*d+y]*f[y];for(let y=0;y<i;++y)m[g*i+y]=Math.trunc(x/h[y]),x%=h[y]}return[m,[a,i],l]}function fR(e,t,n,r,s,o=!1,a=0){let i=r.length;if(i!==s.length)throw new Error("segmentIds and indices should have same size.");let l=[t[0],e.length/t[0]],u=l[1],p=i>0?s[i-1]+1:0;if(p<0)throw new Error("segment ids must be >= 0");let d=t.slice();d[0]=p;let f=d.reduce((b,A)=>b*A,1),h=E.getArrayFromDType(n,f);if(i===0)return p>0&&h.fill(a),[h,d];if(p<=0)throw new Error("segment ids must be >= 0");let m=0,g=1,x=0,y=s[m];for(;;){let b=0;if(g<i){if(b=s[g],y===b){++g;continue}if(y>=b)throw new Error("segment ids are not increasing")}if(y<0||y>=p)throw new Error( ` Segment id $ { y } out of range [ 0 , $ { p } ) , possibly because segmentIds input is not sorted . ` );y>x&&h.fill(a,x*u,y*u);for(let A=m;A<g;++A){let v=r[A];if(v<0||v>=l[0])throw new Error( ` Bad : indices [ $ { A } ] == $ { r [ A ] } out of range [ 0 , $ { l [ 0 ] } ) ` );for(let w=0;w<u;w++)h[y*u+w]+=e[v*u+w]}if(o)for(let A=0;A<u;A++)h[y*u+A]/=g-m;if(m=g,++g,x=y+1,y=b,g>i)break}return x<p&&h.fill(a,x*u,p*u),[h,d]}var Bwe=np(e=>Math.sqrt(e)),Vwe=In(Tf,e=>Math.sqrt(e)),Wwe={kernelName:Tf,backendName:"cpu",kernelFunc:Vwe},jB=Dr((e,t)=>{let n=e-t;return n*n}),Gwe=ds(Rf,jB),Uwe={kernelName:Rf,backendName:"cpu",kernelFunc:Gwe};function HB(e,t,n,r){let s=Wt(e,t.dtype);for(let o=0;o<s.size;o++){let a=s.indexToLoc(o),i=new Array(a.length);for(let l=0;l<i.length;l++)i[l]=a[l]*n[l]+r[l];s.set(t.get(...i),...a)}return s}var jwe=class{constructor(e,t,n,r,s,o){this.separator=E.encodeString(e),this.nGramWidths=t,this.leftPad=E.encodeString(n),this.rightPad=E.encodeString(r),this.padWidth=s,this.preserveShort=o}getPadWidth(e){return Math.min(this.padWidth<0?e-1:this.padWidth,e-1)}getNumNGrams(e,t){let n=this.getPadWidth(t);return Math.max(0,e+2*n-t+1)}createNGrams(e,t,n,r,s,o){for(let a=0;a<s;++a){let i=this.getPadWidth(o),l=Math.max(0,i-a),u=Math.max(0,i-(s-(a+1))),c=o-(l+u),p=t+(l>0?0:a-i),d=0;d+=l*this.leftPad.length;for(let x=0;x<c;++x)d+=e[p+x].length;d+=u*this.rightPad.length,d+=(l+u+c-1)*this.separator.length,n[r+a]=new Uint8Array(d);let h=n[r+a],m=0,g=x=>x.forEach(y=>h[m++]=y);for(let x=0;x<l;++x)g(this.leftPad),g(this.separator);for(let x=0;x<c-1;++x)g(e[p+x]),g(this.separator);if(c>0){g(e[p+c-1]);for(let x=0;x<u;++x)g(this.separator),g(this.rightPad)}else{for(let x=0;x<u-1;++x)g(this.rightPad),g(this.separator);g(this.rightPad)}}}compute(e,t){let n=e.length,r=t.length;if(r>0){let i=t[0];if(i!==0)throw new Error( ` First split value must be 0 , got $ { i } ` );for(let l=1;l<r;++l){let u=t[l]>=i;if(u=u&&t[l]<=n,!u)throw new Error( ` Invalid split value $ { t [ l ] } , must be in [ $ { i } , $ { n } ] ` );i=t[l]}if(i!==n)throw new Error( ` Last split value must be data size . Expected $ { n } , got $ { i } ` )}let s=r-1,o=E.getArrayFromDType("int32",r);if(n===0||r===0){let i=new Array(n);for(let l=0;l<=s;++l)o[l]=0;return[i,o]}o[0]=0;for(let i=1;i<=s;++i){let l=t[i]-t[i-1],u=0;this.nGramWidths.forEach(c=>{u+=this.getNumNGrams(l,c)}),this.preserveShort&&l>0&&u===0&&(u=1),o[i]=o[i-1]+u}let a=new Array(o[s]);for(let i=0;i<s;++i){let l=t[i],u=o[i];if(this.nGramWidths.forEach(c=>{let p=t[i+1]-t[i],d=this.getNumNGrams(p,c);this.createNGrams(e,l,a,u,d,c),u+=d}),this.preserveShort&&u===o[i]){let c=t[i+1]-t[i];if(c===0)continue;let p=c+2*this.padWidth,d=1;this.createNGrams(e,l,a,u,d,p)}}return[a,o]}};function qB(e,t,n,r,s,o,a,i){return new jwe(n,r,s,o,a,i).compute(e,t)}function Hwe(e,t,n,r){if(!e.length)return;if(t.length===0){for(let o=0;o<e.length;++o)r.push(e.subarray(o,o+1));return}if(t.length===1){let o=t[0],a=e.indexOf(o);for(;a!==-1;){let i=e.subarray
$ { o . shape } ` );if(r.shape.length!==2)throw new Error( ` Indices must be a matrix , saw :
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Values must be a vector , saw :
$ { s . shape } ` );if(a.shape.length!==0)throw new Error( ` Default value must be a scalar , saw :
$ { a . shape } ` );let i=n.data.get(r.dataId).values,l=n.data.get(s.dataId).values,u=n.data.get(o.dataId).values,c=n.data.get(a.dataId).values[0],[p,d,f,h,m]=GB(i,r.shape,r.dtype,l,s.dtype,u,c);return[n.makeTensorInfo(d,r.dtype,p),n.makeTensorInfo([d[0]],s.dtype,f),n.makeTensorInfo([h.length],"bool",new Uint8Array(h.map(g=>Number(g)))),n.makeTensorInfo([m.length],r.dtype,new Int32Array(m))]}var iCe={kernelName:Nw,backendName:"cpu",kernelFunc:aCe};function lCe(e){let{inputs:t,backend:n}=e,{inputIndices:r,inputShape:s,newShape:o}=t;if(r.shape.length!==2)throw new Error( ` Input indices should be a matrix but received shape
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Input shape should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Target shape should be a vector but received shape $ { o . shape } ` );let a=Array.from(n.data.get(s.dataId).values),i=n.data.get(r.dataId).values,l=Array.from(n.data.get(o.dataId).values),[u,c,p]=UB(i,r.shape,r.dtype,a,l);return[n.makeTensorInfo(c,r.dtype,u),n.makeTensorInfo([p.length],o.dtype,new Int32Array(p))]}var uCe={kernelName:Tw,backendName:"cpu",kernelFunc:lCe};function cCe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.data.get(r.dataId).values,i=n.data.get(s.dataId).values,l=n.data.get(o.dataId).values,[u,c]=fR(a,r.shape,r.dtype,i,l,!0);return n.makeTensorInfo(c,r.dtype,u)}var pCe={kernelName:_w,backendName:"cpu",kernelFunc:cCe};function dCe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.data.get(r.dataId).values,i=n.data.get(s.dataId).values,l=n.data.get(o.dataId).values,[u,c]=fR(a,r.shape,r.dtype,i,l);return n.makeTensorInfo(c,r.dtype,u)}var fCe={kernelName:Ew,backendName:"cpu",kernelFunc:dCe};function hCe(e){let{inputs:t,backend:n,attrs:r}=e,{sparseIndices:s,sparseValues:o,defaultValue:a}=t,{outputShape:i}=r,{sliceRank:l,numUpdates:u,sliceSize:c,strides:p,outputSize:d}=G.calculateShapes(o,s,i),f=!1,h=n.bufferSync(s),m=n.bufferSync(o),g=n.data.get(a.dataId).values[0],x=xV(h,m,i,d,c,u,l,p,g,f);return n.makeTensorInfo(i,x.dtype,x.values)}var mCe={kernelName:Rw,backendName:"cpu",kernelFunc:hCe};function gCe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{numOrSizeSplits:o,axis:a}=r,i=E.parseAxisParam(a,s.shape)[0],l=G.prepareSplitSize(s,o,i),u=new Array(s.shape.length).fill(0),c=s.shape.slice();return l.map(p=>{let d=[...c];d[i]=p;let f=ph({inputs:{x:s},backend:n,attrs:{begin:u,size:d}});return u[i]+=p,f})}var xCe={kernelName:s0,backendName:"cpu",kernelFunc:gCe},yCe={kernelName:K1,backendName:"cpu",kernelFunc:({inputs:e,backend:t})=>{let{x:n}=e,r=t;rt(n,"square");let s=r.data.get(n.dataId).values,o=new Float32Array(s.length);for(let i=0;i<s.length;++i){let l=s[i];o[i]=l*l}return{dataId:r.write(o,n.shape,n.dtype),shape:n.shape,dtype:n.dtype}}},bCe=In(Pc,(e,t)=>{let n=t;return isNaN(e)?NaN:e>0?1:n.alpha}),ACe={kernelName:Pc,backendName:"cpu",kernelFunc:bCe};function vCe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,end:a,strides:i,beginMask:l,endMask:u,ellipsisMask:c,newAxisMask:p,shrinkAxisMask:d}=r;rt(s,"stridedSlice");let{nonStrided:f, $ begin:h, $ strides:m,size:g,newShape:x,outShape:y}=Bs.sliceInfo(s.shape,o,a,i,l,u,c,p,d),b=Un({inputs:{x:s},backend:n,attrs:{shape:x}}),A;if(f){let w=ph({inputs:{x:b},backend:n,attrs:{begin:h,size:g}});A=Un({inputs:{x:w},backend:n,attrs:{shape:y}}),n.disposeIntermediateTensorInfo(w)}else if(y.some(w=>w===0))A=n.makeTensorInfo(y,s.dtype,[]);else{let w=n.bufferSync(b),k=HB(y,w,m,h);A=n.makeTensorInfo(k.shape,k.dtype,k.values)}let v=Un({inputs:{x:A},backend:n,attrs:{shape:y}});return n.disposeIntermediateTensorInfo(b),n.disposeIntermediateTensorInfo(A),v}var wCe={kernelName:o0,backendName:"cpu",kernelFunc:vCe};function kCe(e){let{inputs:t,backend:n,attrs:r}=e,{separator:s,nGramWidths:o,leftPad:a,rightPad:i,padWidth:l,preserveShortSequences:u}=r,{data:c,dataSplits:p}=t,d=n.data.get(c.dataId).values,f=n.data.get(p.dataId).values,[h,m]=qB(d,f,s,o,a,i,l,u);return[n.makeTensorInfo([h.length],"string",h),n.makeTensorInfo(p.shape,"int32",m)]}var ICe={kernelName:Dw,backendName:"cpu",kernelFunc:kCe};function CCe(e){let{inputs:t,backend:n,attrs:r}=e,{skipEmpty:s}=r,{input:o,delimiter:a}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(o.shape.length!==1)throw new Error( ` Input must be a vector , got shape : $ { o . shape } ` );if(a.shape.length!==0)throw new Error( ` Delimiter must be a scalar , got shape : $ { a . shape } ` );let i=n.data.get(o.dataId).values,l=n.data.get(a.dataId).values[0],[u,c,p]=KB(i,l,s),d=c.length;return[n.makeTensorInfo([d,2],"int32",u),n.makeTensorInfo([d],"string",c),n.makeTensorInfo([2],"int32",new Int32Array(p))]}var SCe={kernelName: $ w,backendName:"cpu",kernelFunc:CCe};function NCe(e){let{inputs:t,backend:n,attrs:r}=e,{numBuckets:s}=r,{input:o}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(s<=0)throw new Error("Number of buckets must be at least 1");let a=n.data.get(o.dataId).values,i=XB(a,s);return n.makeTensorInfo(o.shape,"int32",i)}var TCe={kernelName:Fw,backendName:"cpu",kernelFunc:NCe},_Ce=In( $ f,e=>Math.tan(e)),ECe={kernelName: $ f,backendName:"cpu",kernelFunc:_Ce},RCe=In(Ff,e=>Math.tanh(e)),DCe={kernelName:Ff,backendName:"cpu",kernelFunc:RCe};function $ Ce(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reps:o}=r;rt(s,"tile");let a=YB(n.bufferSync(s),o);return n.makeTensorInfo(a.shape,a.dtype,a.values)}var FCe={kernelName:Oc,backendName:"cpu",kernelFunc: $ Ce};function OCe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{k:o,sorted:a}=r;rt(s,"topk");let i=n.data.get(s.dataId).values,[l
` ),o=s.length.toString().length+2,a=s.map((p,d)=>E.rightPad((d+1).toString(),o)+p),i=0;for(let p=0;p<a.length;p++)i=Math.max(a[p].length,i);let l=a.slice(0,r-1),u=a.slice(r-1,r),c=a.slice(r);console.log(l.join( `
2021-09-11 16:29:31 +02:00
` )),console.log(t.split( `
2021-10-04 22:29:15 +02:00
` )[0]),console.log( ` % c $ { E . rightPad ( u [ 0 ] , i ) } ` ,"border:1px solid red; background-color:#e3d2d2; color:#a61717"),console.log(c.join( `
` ))}function CV(e){return Zl(e,()=>e.createProgram(),"Unable to create WebGLProgram.")}function SV(e,t){if(Qe(e,()=>e.linkProgram(t)),e.getProgramParameter(t,e.LINK_STATUS)===!1)throw console.log(e.getProgramInfoLog(t)),new Error("Failed to link vertex and fragment shaders.")}function S5(e,t){if(Qe(e,()=>e.validateProgram(t)),e.getProgramParameter(t,e.VALIDATE_STATUS)===!1)throw console.log(e.getProgramInfoLog(t)),new Error("Shader program validation failed.")}function NV(e,t){let n=Zl(e,()=>e.createBuffer(),"Unable to create WebGLBuffer");return Qe(e,()=>e.bindBuffer(e.ARRAY_BUFFER,n)),Qe(e,()=>e.bufferData(e.ARRAY_BUFFER,t,e.STATIC_DRAW)),n}function TV(e,t){let n=Zl(e,()=>e.createBuffer(),"Unable to create WebGLBuffer");return Qe(e,()=>e.bindBuffer(e.ELEMENT_ARRAY_BUFFER,n)),Qe(e,()=>e.bufferData(e.ELEMENT_ARRAY_BUFFER,t,e.STATIC_DRAW)),n}function iSe(){return be().getNumber("WEBGL_VERSION")===2?1:4}function _V(e){return Zl(e,()=>e.createTexture(),"Unable to create WebGLTexture.")}function EV(e,t){let n=be().getNumber("WEBGL_MAX_TEXTURE_SIZE");if(e<=0||t<=0){let r= ` [ $ { e } x$ { t } ] ` ;throw new Error("Requested texture size "+r+" is invalid.")}if(e>n||t>n){let r= ` [ $ { e } x$ { t } ] ` ,s= ` [ $ { n } x$ { n } ] ` ;throw new Error("Requested texture size "+r+" greater than WebGL maximum on this browser / GPU "+s+".")}}function RV(e){return Zl(e,()=>e.createFramebuffer(),"Unable to create WebGLFramebuffer.")}function kR(e,t,n,r,s,o,a){let i=e.getAttribLocation(t,n);return i===-1?!1:(Qe(e,()=>e.bindBuffer(e.ARRAY_BUFFER,r)),Qe(e,()=>e.vertexAttribPointer(i,s,e.FLOAT,!1,o,a)),Qe(e,()=>e.enableVertexAttribArray(i)),!0)}function DV(e,t,n){MV(e,n),Qe(e,()=>e.activeTexture(e.TEXTURE0+n)),Qe(e,()=>e.bindTexture(e.TEXTURE_2D,t))}function lSe(e,t){MV(e,t),Qe(e,()=>e.activeTexture(e.TEXTURE0+t)),Qe(e,()=>e.bindTexture(e.TEXTURE_2D,null))}function $ V(e,t,n){return Zl(e,()=>e.getUniformLocation(t,n),'uniform "'+n+'" not present in program.')}function FV(e,t,n){return e.getUniformLocation(t,n)}function OV(e,t,n,r){Qe(e,()=>DV(e,t,r)),Qe(e,()=>e.uniform1i(n,r))}function uSe(e){Qe(e,()=>e.bindFramebuffer(e.FRAMEBUFFER,null)),Qe(e,()=>e.viewport(0,0,e.canvas.width,e.canvas.height)),Qe(e,()=>e.scissor(0,0,e.canvas.width,e.canvas.height))}function N5(e,t,n){Qe(e,()=>e.bindFramebuffer(e.FRAMEBUFFER,n)),Qe(e,()=>e.framebufferTexture2D(e.FRAMEBUFFER,e.COLOR_ATTACHMENT0,e.TEXTURE_2D,t,0))}function IR(e,t){Qe(e,()=>e.bindFramebuffer(e.FRAMEBUFFER,t)),Qe(e,()=>e.framebufferTexture2D(e.FRAMEBUFFER,e.COLOR_ATTACHMENT0,e.TEXTURE_2D,null,0))}function eA(e){let t=e.checkFramebufferStatus(e.FRAMEBUFFER);if(t!==e.FRAMEBUFFER_COMPLETE)throw new Error("Error binding framebuffer: "+PV(e,t))}function PV(e,t){switch(t){case e.FRAMEBUFFER_INCOMPLETE_ATTACHMENT:return"FRAMEBUFFER_INCOMPLETE_ATTACHMENT";case e.FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT:return"FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT";case e.FRAMEBUFFER_INCOMPLETE_DIMENSIONS:return"FRAMEBUFFER_INCOMPLETE_DIMENSIONS";case e.FRAMEBUFFER_UNSUPPORTED:return"FRAMEBUFFER_UNSUPPORTED";default:return ` unknown error $ { t } ` }}function Zl(e,t,n){let r=Qe(e,()=>t());if(r==null)throw new Error(n);return r}function MV(e,t){let n=e.MAX_COMBINED_TEXTURE_IMAGE_UNITS-1,r=t+e.TEXTURE0;if(r<e.TEXTURE0||r>n){let s= ` [ gl . TEXTURE0 , gl . TEXTURE$ { n } ] ` ;throw new Error( ` textureUnit must be in $ { s } . ` )}}function fh(e,t=2){return E.sizeFromShape(e.slice(0,e.length-t))}function hh(e){if(e.length===0)throw Error("Cannot get rows and columns of an empty shape array.");return[e.length>1?e[e.length-2]:1,e[e.length-1]]}function T5(e){let t=[1,1,1];return e.length===0||e.length===1&&e[0]===1||(t=[fh(e),...hh(e)]),t}function zV(e,t=!1){let n=be().getNumber("WEBGL_MAX_TEXTURE_SIZE");t&&(n=n*2,e=e.map((s,o)=>o>=e.length-2?E.nearestLargerEven(e[o]):e[o]),e.length===1&&(e=[2,e[0]])),e.length!==2&&(e=E.squeezeShape(e).newShape);let r=E.sizeFromShape(e);if(e.length<=1&&r<=n)return[1,r];if(e.length===2&&e[0]<=n&&e[1]<=n)return e;if(e.length===3&&e[0]*e[1]<=n&&e[2]<=n)return[e[0]*e[1],e[2]];if(e.length===3&&e[0]<=n&&e[1]*e[2]<=n)return[e[0],e[1]*e[2]];if(e.length===4&&e[0
2021-09-11 16:29:31 +02:00
bool isnan _custom ( float val ) {
return ( val > 0.0 || val < 0.0 ) ? false : val != 0.0 ;
}
bvec4 isnan _custom ( vec4 val ) {
return bvec4 ( isnan _custom ( val . x ) ,
isnan _custom ( val . y ) , isnan _custom ( val . z ) , isnan _custom ( val . w ) ) ;
}
# define isnan ( value ) isnan _custom ( value )
` ,l="",u= `
# define round ( value ) newRound ( value )
int newRound ( float value ) {
return int ( floor ( value + 0.5 ) ) ;
}
ivec4 newRound ( vec4 value ) {
return ivec4 ( floor ( value + vec4 ( 0.5 ) ) ) ;
}
2021-10-04 22:29:15 +02:00
` ):(e="",t="attribute",n="varying",r="varying",s="texture2D",o="gl_FragColor",a="",i= `
2021-09-11 16:29:31 +02:00
# define isnan ( value ) isnan _custom ( value )
bool isnan _custom ( float val ) {
return ( val > 0. || val < 1. || val == 0. ) ? false : true ;
}
bvec4 isnan _custom ( vec4 val ) {
return bvec4 ( isnan ( val . x ) , isnan ( val . y ) , isnan ( val . z ) , isnan ( val . w ) ) ;
}
` ,l= `
uniform float INFINITY ;
bool isinf ( float val ) {
return abs ( val ) == INFINITY ;
}
bvec4 isinf ( vec4 val ) {
return equal ( abs ( val ) , vec4 ( INFINITY ) ) ;
}
` ,u= `
int round ( float value ) {
return int ( floor ( value + 0.5 ) ) ;
}
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
ivec4 round ( vec4 value ) {
return ivec4 ( floor ( value + vec4 ( 0.5 ) ) ) ;
}
2021-10-04 22:29:15 +02:00
` ),{version:e,attribute:t,varyingVs:n,varyingFs:r,texture2D:s,output:o,defineOutput:a,defineSpecialNaN:i,defineSpecialInf:l,defineRound:u}}function mh(e,t,n="index"){let r=E.computeStrides(t);return r.map((s,o)=>{let a= ` int $ { e [ o ] } = $ { n } / $ { s } ` ,i=o===r.length-1? ` int $ { e [ o + 1 ] } = $ { n } - $ { e [ o ] } * $ { s } ` : ` index -= $ { e [ o ] } * $ { s } ` ;return ` $ { a } ; $ { i } ; ` }).join("")}function D5(e,t,n="index"){let r=E.computeStrides(t);return r.map((s,o)=>{let a= ` int $ { e [ o ] } = $ { n } / outShapeStrides [ $ { o } ] ` ,i=o===r.length-1? ` int $ { e [ o + 1 ] } = $ { n } - $ { e [ o ] } * outShapeStrides [ $ { o } ] ` : ` index -= $ { e [ o ] } * outShapeStrides [ $ { o } ] ` ;return ` $ { a } ; $ { i } ; ` }).join("")}function fSe(e,t){let n=e.length,r=e.map(o=> ` $ { t } [ $ { o } ] ` ),s=new Array(n-1);s[n-2]=r[n-1];for(let o=n-3;o>=0;--o)s[o]= ` ( $ { s [ o + 1 ] } * $ { r [ o + 1 ] } ) ` ;return s}function hSe(e,t,n="index"){let r=e.map((o,a)=>a),s=fSe(r,t);return s.map((o,a)=>{let i= ` int $ { e [ a ] } = $ { n } / $ { s [ a ] } ` ,l=a===s.length-1? ` int $ { e [ a + 1 ] } = $ { n } - $ { e [ a ] } * $ { s [ a ] } ` : ` index -= $ { e [ a ] } * $ { s [ a ] } ` ;return ` $ { i } ; $ { l } ; ` }).join("")}function NR(e){let t=E.computeStrides(e).map(n=>n.toString());return `
2021-09-11 16:29:31 +02:00
int getFlatIndex ( ivec3 coords ) {
return coords . x * $ { t [ 0 ] } + coords . y * $ { t [ 1 ] } + coords . z ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function TR(){return `
2021-09-11 16:29:31 +02:00
int getFlatIndex ( ivec3 coords ) {
return coords . x * outShapeStrides [ 0 ] + coords . y * outShapeStrides [ 1 ] + coords . z ;
2021-08-31 19:00:06 +02:00
}
2021-10-04 22:29:15 +02:00
` }var jV= `
2021-09-11 16:29:31 +02:00
const float FLOAT _MAX = 1.70141184 e38 ;
const float FLOAT _MIN = 1.17549435 e - 38 ;
lowp vec4 encode _float ( highp float v ) {
if ( isnan ( v ) ) {
return vec4 ( 255 , 255 , 255 , 255 ) ;
}
highp float av = abs ( v ) ;
if ( av < FLOAT _MIN ) {
return vec4 ( 0.0 , 0.0 , 0.0 , 0.0 ) ;
} else if ( v > FLOAT _MAX ) {
return vec4 ( 0.0 , 0.0 , 128.0 , 127.0 ) / 255.0 ;
} else if ( v < - FLOAT _MAX ) {
return vec4 ( 0.0 , 0.0 , 128.0 , 255.0 ) / 255.0 ;
}
highp vec4 c = vec4 ( 0 , 0 , 0 , 0 ) ;
highp float e = floor ( log2 ( av ) ) ;
highp float m = exp2 ( fract ( log2 ( av ) ) ) - 1.0 ;
c [ 2 ] = floor ( 128.0 * m ) ;
m -= c [ 2 ] / 128.0 ;
c [ 1 ] = floor ( 32768.0 * m ) ;
m -= c [ 1 ] / 32768.0 ;
c [ 0 ] = floor ( 8388608.0 * m ) ;
highp float ebias = e + 127.0 ;
c [ 3 ] = floor ( ebias / 2.0 ) ;
ebias -= c [ 3 ] * 2.0 ;
c [ 2 ] += floor ( ebias ) * 128.0 ;
c [ 3 ] += 128.0 * step ( 0.0 , - v ) ;
return c / 255.0 ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` ,{getBroadcastDims:HV}=G;function mSe(e,t,n){let r=[];if(e.forEach(f=>{let h=E.sizeFromShape(f.shapeInfo.logicalShape);if(f.shapeInfo.isUniform?r.push( ` uniform float $ { f . name } $ { h > 1 ? ` [ ${ h } ] ` : "" } ; ` ):(r.push( ` uniform sampler2D $ { f . name } ; ` ),r.push( ` uniform int offset$ { f . name } ; ` )),n.enableShapeUniforms){let{uniformShape:m}=_R(n.packedInputs,f.shapeInfo.logicalShape,f.shapeInfo.texShape);switch(m.length){case 1:r.push( ` uniform int $ { f . name } Shape ; ` );break;case 2:r.push( ` uniform ivec2 $ { f . name } Shape ; ` );break;case 3:r.push( ` uniform ivec3 $ { f . name } Shape ; ` );break;case 4:r.push( ` uniform ivec4 $ { f . name } Shape ; ` );break;default:break}r.push( ` uniform ivec2 $ { f . name } TexShape ; ` )}}),n.enableShapeUniforms){switch(t.logicalShape.length){case 1:r.push("uniform int outShape;");break;case 2:r.push("uniform ivec2 outShape;"),r.push("uniform int outShapeStrides;");break;case 3:r.push("uniform ivec3 outShape;"),r.push("uniform ivec2 outShapeStrides;");break;case 4:r.push("uniform ivec4 outShape;"),r.push("uniform ivec3 outShapeStrides;");break;default:break}r.push("uniform ivec2 outTexShape;")}n.customUniforms&&n.customUniforms.forEach(f=>{r.push( ` uniform $ { f . type } $ { f . name } $ { f . arrayIndex ? ` [ ${ f . arrayIndex } ] ` : "" } ; ` )});let s=r.join( `
` ),o=e.map(f=>gSe(f,t,n.packedInputs,n.enableShapeUniforms)).join( `
` ),a=t.texShape,i=Us(),l=bSe(i),u,c,p=wSe(i);return t.isPacked?(u=xSe(t.logicalShape,a,n.enableShapeUniforms),c=vSe(i)):(u=ySe(t.logicalShape,a,n.enableShapeUniforms),c=ASe(i)),n.packedInputs&&(p+=SSe),[p,l,c,s,u,o,n.userCode].join( `
` )}function H0(e,t=!1){let n=e.shapeInfo.logicalShape;switch(n.length){case 0:return zSe(e,t);case 1:return BSe(e,t);case 2:return WSe(e,t);case 3:return USe(e,t);case 4:return HSe(e,t);case 5:return qSe(e);case 6:return KSe(e);default:throw new Error( ` $ { n . length } - D input sampling is not yet supported ` )}}function qV(e,t){switch(e.shapeInfo.logicalShape.length){case 0:return MSe(e);case 1:return LSe(e,t);case 2:return VSe(e,t);case 3:return GSe(e,t);default:return jSe(e,t)}}function gSe(e,t,n=!1,r){let s="";n?s+=qV(e,r):s+=H0(e,r);let o=e.shapeInfo.logicalShape,a=t.logicalShape;return o.length<=a.length&&(n?s+=XSe(e,t):s+=ZSe(e,t)),s}function xSe(e,t,n){switch(e.length){case 0:return KV();case 1:return NSe(e,t,n);case 2:return OSe(e,t,n);case 3:return _Se(e,t,n);default:return RSe(e,t,n)}}function ySe(e,t,n){switch(e.length){case 0:return KV();case 1:return TSe(e,t,n);case 2:return PSe(e,t,n);case 3:return ESe(e,t,n);case 4:return DSe(e,t,n);case 5:return $ Se(e,t);case 6:return FSe(e,t);default:throw new Error( ` $ { e . length } - D output sampling is not yet supported ` )}}function bSe(e){return `
2021-09-11 16:29:31 +02:00
float sampleTexture ( sampler2D textureSampler , vec2 uv ) {
return $ { e . texture2D } ( textureSampler , uv ) . r ;
}
2021-10-04 22:29:15 +02:00
` }function ASe(e){return `
2021-09-11 16:29:31 +02:00
void setOutput ( float val ) {
$ { e . output } = vec4 ( val , 0 , 0 , 0 ) ;
}
2021-10-04 22:29:15 +02:00
` }function vSe(e){return `
2021-09-11 16:29:31 +02:00
void setOutput ( vec4 val ) {
$ { e . output } = val ;
}
2021-10-04 22:29:15 +02:00
` }function wSe(e){return ` $ { e . version }
2021-09-11 16:29:31 +02:00
precision highp float ;
precision highp int ;
precision highp sampler2D ;
$ { e . varyingFs } vec2 resultUV ;
$ { e . defineOutput }
const vec2 halfCR = vec2 ( 0.5 , 0.5 ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
struct ivec5
{
int x ;
int y ;
int z ;
int w ;
int u ;
} ;
struct ivec6
{
int x ;
int y ;
int z ;
int w ;
int u ;
int v ;
} ;
uniform float NAN ;
$ { e . defineSpecialNaN }
$ { e . defineSpecialInf }
$ { e . defineRound }
int imod ( int x , int y ) {
return x - y * ( x / y ) ;
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
int idiv ( int a , int b , float sign ) {
int res = a / b ;
int mod = imod ( a , b ) ;
if ( sign < 0. && mod != 0 ) {
res -= 1 ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
return res ;
}
//Based on the work of Dave Hoskins
//https://www.shadertoy.com/view/4djSRW
# define HASHSCALE1 443.8975
float random ( float seed ) {
vec2 p = resultUV * seed ;
vec3 p3 = fract ( vec3 ( p . xyx ) * HASHSCALE1 ) ;
p3 += dot ( p3 , p3 . yzx + 19.19 ) ;
return fract ( ( p3 . x + p3 . y ) * p3 . z ) ;
2021-06-05 21:10:28 +02:00
}
2021-09-05 22:42:11 +02:00
2021-10-04 22:29:15 +02:00
$ { kSe }
$ { ISe }
$ { CSe }
` }var kSe= `
2021-09-11 16:29:31 +02:00
vec2 uvFromFlat ( int texNumR , int texNumC , int index ) {
int texR = index / texNumC ;
int texC = index - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
vec2 packedUVfrom1D ( int texNumR , int texNumC , int index ) {
int texelIndex = index / 2 ;
int texR = texelIndex / texNumC ;
int texC = texelIndex - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
2021-10-04 22:29:15 +02:00
` ,ISe= `
2021-09-11 16:29:31 +02:00
vec2 packedUVfrom2D ( int texelsInLogicalRow , int texNumR ,
int texNumC , int row , int col ) {
int texelIndex = ( row / 2 ) * texelsInLogicalRow + ( col / 2 ) ;
int texR = texelIndex / texNumC ;
int texC = texelIndex - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
2021-10-04 22:29:15 +02:00
` ,CSe= `
2021-09-11 16:29:31 +02:00
vec2 packedUVfrom3D ( int texNumR , int texNumC ,
int texelsInBatch , int texelsInLogicalRow , int b ,
int row , int col ) {
int index = b * texelsInBatch + ( row / 2 ) * texelsInLogicalRow + ( col / 2 ) ;
int texR = index / texNumC ;
int texC = index - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
2021-10-04 22:29:15 +02:00
` ,SSe= `
2021-09-11 16:29:31 +02:00
float getChannel ( vec4 frag , vec2 innerDims ) {
vec2 modCoord = mod ( innerDims , 2. ) ;
return modCoord . x == 0. ?
( modCoord . y == 0. ? frag . r : frag . g ) :
( modCoord . y == 0. ? frag . b : frag . a ) ;
}
float getChannel ( vec4 frag , int dim ) {
float modCoord = mod ( float ( dim ) , 2. ) ;
return modCoord == 0. ? frag . r : frag . g ;
}
2021-10-04 22:29:15 +02:00
` ;function KV(){return `
2021-09-11 16:29:31 +02:00
int getOutputCoords ( ) {
return 0 ;
2021-06-05 21:10:28 +02:00
}
2021-10-04 22:29:15 +02:00
` }function NSe(e,t,n){let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)];return r[0]===1?n? `
2021-09-11 16:29:31 +02:00
int getOutputCoords ( ) {
return 2 * int ( resultUV . x * ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
int getOutputCoords ( ) {
2021-10-04 22:29:15 +02:00
return 2 * int ( resultUV . x * $ { r [ 1 ] } . 0 ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` :r[1]===1?n? `
2021-09-11 16:29:31 +02:00
int getOutputCoords ( ) {
return 2 * int ( resultUV . y * ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
int getOutputCoords ( ) {
2021-10-04 22:29:15 +02:00
return 2 * int ( resultUV . y * $ { r [ 0 ] } . 0 ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` :n? `
int getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
return 2 * ( resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ) ;
}
` : `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
2021-10-04 22:29:15 +02:00
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
return 2 * ( resTexRC . x * $ { r [ 1 ] } + resTexRC . y ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function TSe(e,t,n){return t[0]===1?n? `
2021-09-11 16:29:31 +02:00
int getOutputCoords ( ) {
return int ( resultUV . x * float ( outTexShape [ 1 ] ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
int getOutputCoords ( ) {
return int ( resultUV . x * $ { t [ 1 ] } . 0 ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` :t[1]===1?n? `
int getOutputCoords ( ) {
return int ( resultUV . y * float ( outTexShape [ 0 ] ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
int getOutputCoords ( ) {
return int ( resultUV . y * $ { t [ 0 ] } . 0 ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` :n? `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
return resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
2021-07-29 15:53:13 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
return resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
2021-06-05 21:10:28 +02:00
}
2021-10-04 22:29:15 +02:00
` }function _Se(e,t,n){if(n)return `
2021-09-11 16:29:31 +02:00
ivec3 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatch = texelsInLogicalRow * int ( ceil ( float ( outShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int b = index / texelsInBatch ;
index -= b * texelsInBatch ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec3 ( b , r , c ) ;
2021-06-05 21:10:28 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)],s=Math.ceil(e[2]/2),o=s*Math.ceil(e[1]/2);return `
2021-09-11 16:29:31 +02:00
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
2021-10-04 22:29:15 +02:00
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
int b = index / $ { o } ;
index -= b * $ { o } ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
2021-09-11 16:29:31 +02:00
return ivec3 ( b , r , c ) ;
2021-07-29 15:53:13 +02:00
}
2021-10-04 22:29:15 +02:00
` }function ESe(e,t,n){if(n)return `
2021-09-11 16:29:31 +02:00
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
2021-10-04 22:29:15 +02:00
$ { D5 ( [ "r" , "c" , "d" ] , e ) }
2021-09-11 16:29:31 +02:00
return ivec3 ( r , c , d ) ;
}
2021-10-04 22:29:15 +02:00
` ;let r=mh(["r","c","d"],e);return `
2021-09-11 16:29:31 +02:00
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
2021-10-04 22:29:15 +02:00
$ { r }
2021-09-11 16:29:31 +02:00
return ivec3 ( r , c , d ) ;
2021-06-05 21:10:28 +02:00
}
2021-10-04 22:29:15 +02:00
` }function RSe(e,t,n){if(n)return `
2021-09-11 16:29:31 +02:00
ivec4 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 3 ] ) / 2.0 ) ) ;
int texelsInBatch = texelsInLogicalRow * int ( ceil ( float ( outShape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatchN = texelsInBatch * outShape [ 1 ] ;
int b2 = index / texelsInBatchN ;
index -= b2 * texelsInBatchN ;
int b = index / texelsInBatch ;
index -= b * texelsInBatch ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec4 ( b2 , b , r , c ) ;
2021-07-29 15:53:13 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)],s=Math.ceil(e[e.length-1]/2),o=s*Math.ceil(e[e.length-2]/2),a=o,i="",l="b, r, c";for(let u=2;u<e.length-1;u++)a*=e[e.length-u-1],i= `
int b$ { u } = index / $ { a } ;
index -= b$ { u } * $ { a } ;
2021-09-11 17:17:13 +02:00
` +i,l= ` b$ { u } , ` +l;return `
2021-09-11 16:29:31 +02:00
ivec$ { e . length } getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
2021-10-04 22:29:15 +02:00
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
2021-09-11 16:29:31 +02:00
2021-09-11 17:17:13 +02:00
$ { i }
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
int b = index / $ { o } ;
index -= b * $ { o } ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
2021-09-11 16:29:31 +02:00
return ivec$ { e . length } ( $ { l } ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function DSe(e,t,n){if(n)return `
2021-09-11 16:29:31 +02:00
ivec4 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
2021-10-04 22:29:15 +02:00
$ { D5 ( [ "r" , "c" , "d" , "d2" ] , e ) }
2021-09-11 16:29:31 +02:00
return ivec4 ( r , c , d , d2 ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let r=mh(["r","c","d","d2"],e);return `
2021-09-11 16:29:31 +02:00
ivec4 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
2021-10-04 22:29:15 +02:00
$ { r }
2021-09-11 16:29:31 +02:00
return ivec4 ( r , c , d , d2 ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function $ Se(e,t){let n=mh(["r","c","d","d2","d3"],e);return `
2021-09-11 16:29:31 +02:00
ivec5 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( $ { t [ 0 ] } ,
$ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { n }
ivec5 outShape = ivec5 ( r , c , d , d2 , d3 ) ;
return outShape ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` }function FSe(e,t){let n=mh(["r","c","d","d2","d3","d4"],e);return `
2021-09-11 16:29:31 +02:00
ivec6 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { n }
ivec6 result = ivec6 ( r , c , d , d2 , d3 , d4 ) ;
return result ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function OSe(e,t,n){let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)];if(E.arraysEqual(e,t))return n? `
2021-09-11 16:29:31 +02:00
ivec2 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
return 2 * ivec2 ( resultUV . yx * vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
}
` : `
ivec2 getOutputCoords ( ) {
2021-10-04 22:29:15 +02:00
return 2 * ivec2 ( resultUV . yx * vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let s=Math.ceil(e[1]/2);return n? `
2021-09-11 16:29:31 +02:00
ivec2 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec2 ( r , c ) ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
2021-10-04 22:29:15 +02:00
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
2021-09-11 16:29:31 +02:00
return ivec2 ( r , c ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function PSe(e,t,n){return E.arraysEqual(e,t)?n? `
2021-09-11 16:29:31 +02:00
ivec2 getOutputCoords ( ) {
return ivec2 ( resultUV . yx * vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
}
` : `
ivec2 getOutputCoords ( ) {
return ivec2 ( resultUV . yx * vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
}
` :e[1]===1?n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
return ivec2 ( index , 0 ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
return ivec2 ( index , 0 ) ;
}
` :e[0]===1?n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
return ivec2 ( 0 , index ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
return ivec2 ( 0 , index ) ;
}
` :n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
int r = index / outShape [ 1 ] ;
int c = index - r * outShape [ 1 ] ;
return ivec2 ( r , c ) ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
int r = index / $ { e [ 1 ] } ;
int c = index - r * $ { e [ 1 ] } ;
return ivec2 ( r , c ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function gh(e){return ` offset$ { e } ` }function MSe(e){let t=e.name,n="get"+t.charAt(0).toUpperCase()+t.slice(1),r=Us();return `
2021-09-11 16:29:31 +02:00
vec4 $ { n } ( ) {
2021-10-04 22:29:15 +02:00
return $ { r . texture2D } ( $ { t } , halfCR ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }function zSe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1);if(e.shapeInfo.isUniform)return ` float $ { r } ( ) { return $ { n } ; } ` ;let[s,o]=e.shapeInfo.texShape;if(s===1&&o===1)return `
float $ { r } ( ) {
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , halfCR ) ;
}
2021-10-04 22:29:15 +02:00
` ;let a=gh(n);if(t)return `
float $ { r } ( ) {
vec2 uv = uvFromFlat ( $ { n } TexShape [ 0 ] , $ { n } TexShape [ 1 ] , $ { a } ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
2021-09-01 00:24:30 +02:00
}
2021-09-11 17:17:13 +02:00
` ;let[i,l]=e.shapeInfo.texShape;return `
2021-10-04 22:29:15 +02:00
float $ { r } ( ) {
vec2 uv = uvFromFlat ( $ { i } , $ { l } , $ { a } ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` }function LSe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=e.shapeInfo.texShape,o=Us();if(t)return `
vec4 $ { r } ( int index ) {
2021-09-11 16:29:31 +02:00
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { n } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { n } TexShape [ 1 ] ) / 2.0 ) ) ;
vec2 uv = packedUVfrom1D (
packedTexShape [ 0 ] , packedTexShape [ 1 ] , index ) ;
2021-10-04 22:29:15 +02:00
return $ { o . texture2D } ( $ { n } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let a=[Math.ceil(s[0]/2),Math.ceil(s[1]/2)];return `
vec4 $ { r } ( int index ) {
2021-09-11 16:29:31 +02:00
vec2 uv = packedUVfrom1D (
2021-10-04 22:29:15 +02:00
$ { a [ 0 ] } , $ { a [ 1 ] } , index ) ;
return $ { o . texture2D } ( $ { n } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function BSe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1);if(e.shapeInfo.isUniform)return `
float $ { r } ( int index ) {
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let s=e.shapeInfo.texShape,o=s[0],a=s[1];if(a===1&&o===1)return `
float $ { r } ( int index ) {
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , halfCR ) ;
}
2021-10-04 22:29:15 +02:00
` ;let i=gh(n);return a===1?t? `
float $ { r } ( int index ) {
2021-09-11 17:17:13 +02:00
vec2 uv = vec2 ( 0.5 , ( float ( index + $ { i } ) + 0.5 ) / float ( $ { n } TexShape [ 0 ] ) ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
` : `
2021-10-04 22:29:15 +02:00
float $ { r } ( int index ) {
vec2 uv = vec2 ( 0.5 , ( float ( index + $ { i } ) + 0.5 ) / $ { o } . 0 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` :o===1?t? `
float $ { r } ( int index ) {
2021-09-11 17:17:13 +02:00
vec2 uv = vec2 ( ( float ( index + $ { i } ) + 0.5 ) / float ( $ { n } TexShape [ 1 ] ) , 0.5 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
` : `
2021-10-04 22:29:15 +02:00
float $ { r } ( int index ) {
vec2 uv = vec2 ( ( float ( index + $ { i } ) + 0.5 ) / $ { a } . 0 , 0.5 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
` :t? `
2021-10-04 22:29:15 +02:00
float $ { r } ( int index ) {
2021-09-11 17:17:13 +02:00
vec2 uv = uvFromFlat ( $ { n } TexShape [ 0 ] , $ { n } TexShape [ 1 ] , index + $ { i } ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
` : `
2021-10-04 22:29:15 +02:00
float $ { r } ( int index ) {
vec2 uv = uvFromFlat ( $ { o } , $ { a } , index + $ { i } ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` }function VSe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape,a=o[0],i=o[1],l=Us();if(o!=null&&E.arraysEqual(n,o))return t? `
vec4 $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
return $ { l . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
vec4 $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { i } . 0 , $ { a } . 0 ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
return $ { l . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` ;if(t)return `
2021-10-04 22:29:15 +02:00
vec4 $ { s } ( int row , int col ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { r } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { r } TexShape [ 1 ] ) / 2.0 ) ) ;
int valuesPerRow = int ( ceil ( float ( $ { r } Shape [ 1 ] ) / 2.0 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = packedUVfrom2D ( valuesPerRow , packedTexShape [ 0 ] , packedTexShape [ 1 ] , row , col ) ;
2021-10-04 22:29:15 +02:00
return $ { l . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let u=[Math.ceil(o[0]/2),Math.ceil(o[1]/2)],c=Math.ceil(n[1]/2);return `
vec4 $ { s } ( int row , int col ) {
2021-09-11 17:17:13 +02:00
vec2 uv = packedUVfrom2D ( $ { c } , $ { u [ 0 ] } , $ { u [ 1 ] } , row , col ) ;
2021-10-04 22:29:15 +02:00
return $ { l . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function WSe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape;if(o!=null&&E.arraysEqual(n,o)){if(t)return `
float $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;let d=o[0],f=o[1];return `
float $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }let{newShape:a,keptDims:i}=E.squeezeShape(n),l=a;if(l.length<n.length){let d=K0(e,l),f=["row","col"];return `
$ { H0 ( d , t ) }
float $ { s } ( int row , int col ) {
return $ { s } ( $ { X0 ( f , i ) } ) ;
2021-09-11 16:29:31 +02:00
}
` }if(e.shapeInfo.isUniform)return `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
2021-09-11 16:29:31 +02:00
int index = round ( dot ( vec2 ( row , col ) , vec2 ( $ { n [ 1 ] } , 1 ) ) ) ;
2021-10-04 22:29:15 +02:00
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let u=o[0],c=o[1],p=gh(r);return c===1?t? `
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { r } Shape [ 1 ] , 1 , 1 ) ) ;
vec2 uv = vec2 ( 0.5 , ( index + 0.5 ) / float ( $ { r } TexShape [ 0 ] ) ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { n [ 1 ] } , 1 , 1 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = vec2 ( 0.5 , ( index + 0.5 ) / $ { u } . 0 ) ;
2021-10-04 22:29:15 +02:00
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` :u===1?t? `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { r } Shape [ 1 ] , 1 , 1 ) ) ;
vec2 uv = vec2 ( ( index + 0.5 ) / float ( $ { r } TexShape [ 1 ] ) , 0.5 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { n [ 1 ] } , 1 , 1 ) ) ;
2021-09-11 17:17:13 +02:00
vec2 uv = vec2 ( ( index + 0.5 ) / $ { c } . 0 , 0.5 ) ;
2021-10-04 22:29:15 +02:00
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` :t? `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int index = row * $ { r } Shape [ 1 ] + col + $ { p } ;
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int index = row * $ { n [ 1 ] } + col + $ { p } ;
2021-09-11 17:17:13 +02:00
vec2 uv = uvFromFlat ( $ { u } , $ { c } , index ) ;
2021-10-04 22:29:15 +02:00
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function GSe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape,a=[Math.ceil(o[0]/2),Math.ceil(o[1]/2)];if(n[0]===1){let d=n.slice(1),f=[1,2],h=K0(e,d),m=["b","row","col"];return `
$ { qV ( h , t ) }
vec4 $ { s } ( int b , int row , int col ) {
return $ { s } ( $ { X0 ( m , f ) } ) ;
}
` }let i=Us();if(t)return `
vec4 $ { s } ( int b , int row , int col ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { r } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { r } TexShape [ 1 ] ) / 2.0 ) ) ;
int valuesPerRow = int ( ceil ( float ( $ { r } Shape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatch = valuesPerRow * int ( ceil ( float ( $ { r } Shape [ 1 ] ) / 2.0 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = packedUVfrom3D (
packedTexShape [ 0 ] , packedTexShape [ 1 ] , texelsInBatch , valuesPerRow , b , row , col ) ;
2021-10-04 22:29:15 +02:00
return $ { i . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let l=a[0],u=a[1],c=Math.ceil(n[2]/2),p=c*Math.ceil(n[1]/2);return `
vec4 $ { s } ( int b , int row , int col ) {
2021-09-11 16:29:31 +02:00
vec2 uv = packedUVfrom3D (
2021-10-04 22:29:15 +02:00
$ { l } , $ { u } , $ { p } , $ { c } , b , row , col ) ;
return $ { i . texture2D } ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function USe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=n[1]*n[2],a=n[2],{newShape:i,keptDims:l}=E.squeezeShape(n),u=i;if(u.length<n.length){let m=K0(e,u),g=["row","col","depth"];return `
$ { H0 ( m , t ) }
float $ { s } ( int row , int col , int depth ) {
return $ { s } ( $ { X0 ( g , l ) } ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
` }if(e.shapeInfo.isUniform)return `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth ) {
2021-09-11 16:29:31 +02:00
int index = round ( dot ( vec3 ( row , col , depth ) ,
2021-10-04 22:29:15 +02:00
vec3 ( $ { o } , $ { a } , 1 ) ) ) ;
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let c=e.shapeInfo.texShape,p=c[0],d=c[1],f=e.shapeInfo.flatOffset;if(d===o&&f==null)return t? `
float $ { s } ( int row , int col , int depth ) {
int stride1 = $ { r } Shape [ 2 ] ;
2021-09-11 16:29:31 +02:00
float texR = float ( row ) ;
float texC = dot ( vec2 ( col , depth ) , vec2 ( stride1 , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth ) {
2021-09-11 16:29:31 +02:00
float texR = float ( row ) ;
2021-10-04 22:29:15 +02:00
float texC = dot ( vec2 ( col , depth ) , vec2 ( $ { a } , 1 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { d } . 0 , $ { p } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` ;if(d===a&&f==null)return t? `
float $ { s } ( int row , int col , int depth ) {
float texR = dot ( vec2 ( row , col ) , vec2 ( $ { r } Shape [ 1 ] , 1 ) ) ;
2021-09-11 16:29:31 +02:00
float texC = float ( depth ) ;
2021-10-04 22:29:15 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth ) {
2021-09-11 16:29:31 +02:00
float texR = dot ( vec2 ( row , col ) , vec2 ( $ { n [ 1 ] } , 1 ) ) ;
float texC = float ( depth ) ;
2021-10-04 22:29:15 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { d } . 0 , $ { p } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let h=gh(r);return t? `
float $ { s } ( int row , int col , int depth ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int stride0 = $ { r } Shape [ 1 ] * $ { r } Shape [ 2 ] ;
int stride1 = $ { r } Shape [ 2 ] ;
int index = row * $ { o } + col * $ { a } + depth + $ { h } ;
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int index = row * $ { o } + col * $ { a } + depth + $ { h } ;
vec2 uv = uvFromFlat ( $ { p } , $ { d } , index ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function jSe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=Us();if(t)return `
vec4 $ { r } ( int b2 , int b , int row , int col ) {
2021-09-11 16:29:31 +02:00
int valuesPerRow = int ( ceil ( float ( $ { n } Shape [ 3 ] ) / 2.0 ) ) ;
int texelsInBatch = valuesPerRow * int ( ceil ( float ( $ { n } Shape [ 2 ] ) / 2.0 ) ) ;
int index = b * texelsInBatch + ( row / 2 ) * valuesPerRow + ( col / 2 ) ;
texelsInBatch *= $ { n } Shape [ 1 ] ;
index = b2 * texelsInBatch + index ;
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { n } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { n } TexShape [ 1 ] ) / 2.0 ) ) ;
int texR = index / packedTexShape [ 1 ] ;
int texC = index - texR * packedTexShape [ 1 ] ;
2021-10-04 22:29:15 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( packedTexShape [ 1 ] , packedTexShape [ 0 ] ) ; return $ { s . texture2D } ( $ { n } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let o=e.shapeInfo.logicalShape,a=o.length,i=e.shapeInfo.texShape,l=[Math.ceil(i[0]/2),Math.ceil(i[1]/2)],u=l[0],c=l[1],p=Math.ceil(o[a-1]/2),d=p*Math.ceil(o[a-2]/2),f="int b, int row, int col",h= ` b * $ { d } + ( row / 2 ) * $ { p } + ( col / 2 ) ` ;for(let m=2;m<a-1;m++)f= ` int b$ { m } , ` +f,d*=o[a-m-1],h= ` b$ { m } * $ { d } + ` +h;return `
vec4 $ { r } ( $ { f } ) {
int index = $ { h } ;
2021-09-11 17:17:13 +02:00
int texR = index / $ { c } ;
int texC = index - texR * $ { c } ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { c } , $ { u } ) ;
2021-10-04 22:29:15 +02:00
return $ { s . texture2D } ( $ { n } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function HSe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=n[3],a=n[2]*o,i=n[1]*a,{newShape:l,keptDims:u}=E.squeezeShape(n);if(l.length<n.length){let y=K0(e,l),b=["row","col","depth","depth2"];return `
$ { H0 ( y , t ) }
float $ { s } ( int row , int col , int depth , int depth2 ) {
return $ { s } ( $ { X0 ( b , u ) } ) ;
2021-09-11 16:29:31 +02:00
}
` }if(e.shapeInfo.isUniform)return `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
int index = round ( dot ( vec4 ( row , col , depth , depth2 ) ,
2021-10-04 22:29:15 +02:00
vec4 ( $ { i } , $ { a } , $ { o } , 1 ) ) ) ;
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let c=e.shapeInfo.flatOffset,p=e.shapeInfo.texShape,d=p[0],f=p[1],h= ` int stride2 = $ { r } Shape [ 3 ] ; ` ,m= ` int stride1 = $ { r } Shape [ 2 ] * stride2 ; ` ,g= ` int stride0 = $ { r } Shape [ 1 ] * stride1 ; ` ;if(f===i&&c==null)return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
$ { h }
2021-09-11 17:17:13 +02:00
$ { m }
2021-09-11 16:29:31 +02:00
float texR = float ( row ) ;
float texC =
dot ( vec3 ( col , depth , depth2 ) ,
vec3 ( stride1 , stride2 , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
float texR = float ( row ) ;
float texC =
dot ( vec3 ( col , depth , depth2 ) ,
2021-10-04 22:29:15 +02:00
vec3 ( $ { a } , $ { o } , 1 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;if(f===o&&c==null)return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
float texR = dot ( vec3 ( row , col , depth ) ,
2021-10-04 22:29:15 +02:00
vec3 ( $ { r } Shape [ 1 ] * $ { r } Shape [ 2 ] , $ { r } Shape [ 2 ] , 1 ) ) ;
2021-09-11 16:29:31 +02:00
float texC = float ( depth2 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
float texR = dot ( vec3 ( row , col , depth ) ,
vec3 ( $ { n [ 1 ] * n [ 2 ] } , $ { n [ 2 ] } , 1 ) ) ;
float texC = float ( depth2 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let x=gh(r);return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
$ { h }
2021-09-11 17:17:13 +02:00
$ { m }
2021-09-11 16:29:31 +02:00
$ { g }
int index = row * stride0 + col * stride1 +
depth * stride2 + depth2 ;
2021-10-04 22:29:15 +02:00
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index + $ { x } ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
` : `
2021-10-04 22:29:15 +02:00
float $ { s } ( int row , int col , int depth , int depth2 ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int index = row * $ { i } + col * $ { a } +
depth * $ { o } + depth2 ;
vec2 uv = uvFromFlat ( $ { d } , $ { f } , index + $ { x } ) ;
return sampleTexture ( $ { r } , uv ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function qSe(e){let t=e.shapeInfo.logicalShape,n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=t[4],o=t[3]*s,a=t[2]*o,i=t[1]*a,{newShape:l,keptDims:u}=E.squeezeShape(t);if(l.length<t.length){let m=K0(e,l),g=["row","col","depth","depth2","depth3"];return `
$ { H0 ( m ) }
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
return $ { r } ( $ { X0 ( g , u ) } ) ;
2021-09-11 16:29:31 +02:00
}
` }if(e.shapeInfo.isUniform)return `
2021-10-04 22:29:15 +02:00
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
2021-09-11 16:29:31 +02:00
float index = dot (
vec4 ( row , col , depth , depth2 ) ,
2021-10-04 22:29:15 +02:00
vec4 ( $ { i } , $ { a } , $ { o } , $ { s } ) ) +
2021-09-11 16:29:31 +02:00
depth3 ;
2021-10-04 22:29:15 +02:00
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let c=e.shapeInfo.flatOffset,p=e.shapeInfo.texShape,d=p[0],f=p[1];if(f===i&&c==null)return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
2021-09-11 16:29:31 +02:00
int texR = row ;
float texC = dot ( vec4 ( col , depth , depth2 , depth3 ) ,
2021-10-04 22:29:15 +02:00
vec4 ( $ { a } , $ { o } , $ { s } , 1 ) ) ;
2021-09-11 16:29:31 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` ;if(f===s&&c==null)return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
2021-09-11 16:29:31 +02:00
float texR = dot (
vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { t [ 1 ] * t [ 2 ] * t [ 3 ] } ,
$ { t [ 2 ] * t [ 3 ] } , $ { t [ 3 ] } , 1 ) ) ;
int texC = depth3 ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` ;let h=gh(n);return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
2021-09-11 16:29:31 +02:00
// Explicitly use integer operations as dot() only works on floats.
2021-10-04 22:29:15 +02:00
int index = row * $ { i } + col * $ { a } + depth * $ { o } +
depth2 * $ { s } + depth3 + $ { h } ;
vec2 uv = uvFromFlat ( $ { d } , $ { f } , index ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` }function KSe(e){let t=e.shapeInfo.logicalShape,n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),{newShape:s,keptDims:o}=E.squeezeShape(t);if(s.length<t.length){let g=K0(e,s),x=["row","col","depth","depth2","depth3","depth4"];return `
$ { H0 ( g ) }
float $ { r } ( int row , int col , int depth ,
2021-09-11 16:29:31 +02:00
int depth2 , int depth3 , int depth4 ) {
2021-10-04 22:29:15 +02:00
return $ { r } ( $ { X0 ( x , o ) } ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }let a=t[5],i=t[4]*a,l=t[3]*i,u=t[2]*l,c=t[1]*u;if(e.shapeInfo.isUniform)return `
float $ { r } ( int row , int col , int depth ,
2021-09-11 16:29:31 +02:00
int depth2 , int depth3 , int depth4 ) {
int index = round ( dot (
vec4 ( row , col , depth , depth2 ) ,
2021-09-11 17:17:13 +02:00
vec4 ( $ { c } , $ { u } , $ { l } , $ { i } ) ) +
2021-09-11 16:29:31 +02:00
dot (
vec2 ( depth3 , depth4 ) ,
2021-10-04 22:29:15 +02:00
vec2 ( $ { a } , 1 ) ) ) ;
$ { q0 ( e ) }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;let p=e.shapeInfo.flatOffset,d=e.shapeInfo.texShape,f=d[0],h=d[1];if(h===c&&p==null)return `
float $ { r } ( int row , int col , int depth ,
2021-09-11 16:29:31 +02:00
int depth2 , int depth3 , int depth4 ) {
int texR = row ;
float texC = dot ( vec4 ( col , depth , depth2 , depth3 ) ,
2021-10-04 22:29:15 +02:00
vec4 ( $ { u } , $ { l } , $ { i } , $ { a } ) ) +
2021-09-11 16:29:31 +02:00
float ( depth4 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { h } . 0 , $ { f } . 0 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` ;if(h===a&&p==null)return `
float $ { r } ( int row , int col , int depth ,
2021-09-11 16:29:31 +02:00
int depth2 , int depth3 , int depth4 ) {
float texR = dot ( vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { t [ 1 ] * t [ 2 ] * t [ 3 ] * t [ 4 ] } ,
$ { t [ 2 ] * t [ 3 ] * t [ 4 ] } ,
$ { t [ 3 ] * t [ 4 ] } ,
$ { t [ 4 ] } ) ) + float ( depth3 ) ;
int texC = depth4 ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { h } . 0 , $ { f } . 0 ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` ;let m=gh(n);return `
float $ { r } ( int row , int col , int depth ,
2021-09-11 16:29:31 +02:00
int depth2 , int depth3 , int depth4 ) {
// Explicitly use integer operations as dot() only works on floats.
2021-09-11 17:17:13 +02:00
int index = row * $ { c } + col * $ { u } + depth * $ { l } +
2021-10-04 22:29:15 +02:00
depth2 * $ { i } + depth3 * $ { a } + depth4 + $ { m } ;
vec2 uv = uvFromFlat ( $ { f } , $ { h } , index ) ;
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , uv ) ;
}
2021-10-04 22:29:15 +02:00
` }function q0(e){let t=e.name,n=E.sizeFromShape(e.shapeInfo.logicalShape);return n<2? ` return $ { t } ; ` : `
2021-09-11 16:29:31 +02:00
for ( int i = 0 ; i < $ { n } ; i ++ ) {
if ( i == index ) {
return $ { t } [ i ] ;
}
}
2021-10-04 22:29:15 +02:00
` }function XSe(e,t){let n=e.name,r=n.charAt(0).toUpperCase()+n.slice(1),s="get"+r+"AtOutCoords",o=e.shapeInfo.logicalShape.length,a=t.logicalShape.length,i=HV(e.shapeInfo.logicalShape,t.logicalShape),l=Dn(a),u=a-o,c,p=["x","y","z","w","u","v"];o===0?c="":a<2&&i.length>=1?c="coords = 0;":c=i.map(y=> ` coords . $ { p [ y + u ] } = 0 ; ` ).join( `
` );let d="";a<2&&o>0?d="coords":d=e.shapeInfo.logicalShape.map((y,b)=> ` coords . $ { p [ b + u ] } ` ).join(", ");let f="return outputValue;",m=E.sizeFromShape(e.shapeInfo.logicalShape)===1,x=E.sizeFromShape(t.logicalShape)===1;if(o===1&&!m&&!x)f= `
2021-09-11 16:29:31 +02:00
return vec4 ( outputValue . xy , outputValue . xy ) ;
2021-10-04 22:29:15 +02:00
` ;else if(m&&!x)a===1?f= `
2021-09-11 16:29:31 +02:00
return vec4 ( outputValue . x , outputValue . x , 0. , 0. ) ;
2021-10-04 22:29:15 +02:00
` :f= `
2021-09-11 16:29:31 +02:00
return vec4 ( outputValue . x ) ;
2021-10-04 22:29:15 +02:00
` ;else if(i.length){let y=o-2,b=o-1;i.indexOf(y)>-1&&i.indexOf(b)>-1?f="return vec4(outputValue.x);":i.indexOf(y)>-1?f="return vec4(outputValue.x, outputValue.y, outputValue.x, outputValue.y);":i.indexOf(b)>-1&&(f="return vec4(outputValue.xx, outputValue.zz);")}return `
vec4 $ { s } ( ) {
2021-09-11 16:29:31 +02:00
$ { l } coords = getOutputCoords ( ) ;
2021-09-11 17:17:13 +02:00
$ { c }
2021-10-04 22:29:15 +02:00
vec4 outputValue = get$ { r } ( $ { d } ) ;
$ { f }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function ZSe(e,t){let n=e.name,r=n.charAt(0).toUpperCase()+n.slice(1),s="get"+r+"AtOutCoords",o=t.texShape,a=e.shapeInfo.texShape,i=e.shapeInfo.logicalShape.length,l=t.logicalShape.length;if(!e.shapeInfo.isUniform&&i===l&&e.shapeInfo.flatOffset==null&&E.arraysEqual(a,o))return `
float $ { s } ( ) {
2021-09-11 16:29:31 +02:00
return sampleTexture ( $ { n } , resultUV ) ;
}
2021-10-04 22:29:15 +02:00
` ;let u=Dn(l),c=HV(e.shapeInfo.logicalShape,t.logicalShape),p=l-i,d,f=["x","y","z","w","u","v"];i===0?d="":l<2&&c.length>=1?d="coords = 0;":d=c.map(m=> ` coords . $ { f [ m + p ] } = 0 ; ` ).join( `
` );let h="";return l<2&&i>0?h="coords":h=e.shapeInfo.logicalShape.map((m,g)=> ` coords . $ { f [ g + p ] } ` ).join(", "), `
float $ { s } ( ) {
2021-09-11 16:29:31 +02:00
$ { u } coords = getOutputCoords ( ) ;
2021-10-04 22:29:15 +02:00
$ { d }
return get$ { r } ( $ { h } ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }function Dn(e){if(e<=1)return"int";if(e===2)return"ivec2";if(e===3)return"ivec3";if(e===4)return"ivec4";if(e===5)return"ivec5";if(e===6)return"ivec6";throw Error( ` GPU for rank $ { e } is not yet supported ` )}function _R(e,t,n){let{newShape:r,keptDims:s}=E.squeezeShape(t),o=t.length,a=e&&o===3&&t[0]===1,i=a?t.slice(1):r,l=!e&&o>1&&!E.arraysEqual(t,n)&&r.length<o||a;return{useSqueezeShape:l,uniformShape:l?i:t,keptDims:s}}function K0(e,t){let n=JSON.parse(JSON.stringify(e));return n.shapeInfo.logicalShape=t,n}function X0(e,t){return t.map(n=>e[n]).join(", ")}function YSe(e,t,n,r){let s=n.map((b,A)=>{let v={logicalShape:b.shape,texShape:b.isUniform?null:b.texData.texShape,isUniform:b.isUniform,isPacked:b.isUniform?!1:b.texData.isPacked,flatOffset:null};return b.texData!=null&&b.texData.slice!=null&&b.texData.slice.flatOffset>0&&(v.flatOffset=b.texData.slice.flatOffset),{name:t.variableNames[A],shapeInfo:v}}),o=s.map(b=>b.shapeInfo),a={logicalShape:r.shape,texShape:r.texData.texShape,isUniform:!1,isPacked:r.texData.isPacked,flatOffset:null},i=mSe(s,a,t),l=e.createProgram(i),u=null,c=e.getUniformLocation(l,"NAN",!1);be().getNumber("WEBGL_VERSION")===1&&(u=e.getUniformLocation(l,"INFINITY",!1));let p=!1,d={},f={},h={};for(let b=0;b<t.variableNames.length;b++){let A=t.variableNames[b];d[A]=e.getUniformLocation(l,A,p),d[ ` offset$ { A } ` ]=e.getUniformLocation(l, ` offset$ { A } ` ,p),t.enableShapeUniforms&&(f[ ` $ { A } Shape ` ]=e.getUniformLocation(l, ` $ { A } Shape ` ,p),h[ ` $ { A } TexShape ` ]=e.getUniformLocation(l, ` $ { A } TexShape ` ,p))}let m,g,x;t.enableShapeUniforms&&(m=e.getUniformLocation(l,"outShape",p),x=e.getUniformLocation(l,"outShapeStrides",p),g=e.getUniformLocation(l,"outTexShape",p));let y=[];return t.customUniforms&&t.customUniforms.forEach((b,A)=>{y[A]=e.getUniformLocation(l,b.name,p)}),{program:t,source:i,webGLProgram:l,uniformLocations:d,customUniformLocations:y,inShapeInfos:o,outShapeInfo:a,infLoc:u,nanLoc:c,inShapesLocations:f,inTexShapesLocations:h,outShapeLocation:m,outShapeStridesLocation:x,outTexShapeLocation:g}}function XV(e,t){if(e.length!==t.length)throw Error( ` Binary was compiled with $ { e . length } inputs , but was executed with $ { t . length } inputs ` );e.forEach((n,r)=>{let s=n.logicalShape,o=t[r],a=o.shape;if(!E.arraysEqual(s,a))throw Error( ` Binary was compiled with different shapes than the current args . Shapes $ { s } and $ { a } must match ` );if(n.isUniform&&o.isUniform)return;let i=n.texShape,l=o.isUniform?null:o.texData.texShape;if(!E.arraysEqual(i,l))throw Error( ` Binary was compiled with different texture shapes than the current args . Shape $ { i } and $ { l } must match ` )})}function JSe(e,t,n,r,s){t.program.enableShapeUniforms||(XV(t.inShapeInfos,n),XV([t.outShapeInfo],[r]));let o=r.texData.texture,a=r.texData.texShape;r.texData.isPacked?e.setOutputPackedMatrixTexture(o,a[0],a[1]):e.setOutputMatrixTexture(o,a[0],a[1]),e.setProgram(t.webGLProgram),be().getNumber("WEBGL_VERSION")===1&&t.infLoc!==null&&e.gl.uniform1f(t.infLoc,1/0),t.nanLoc!==null&&e.gl.uniform1f(t.nanLoc,NaN),n.forEach((l,u)=>{let c=t.program.variableNames[u],p=t.uniformLocations[c],d=t.uniformLocations[ ` offset$ { c } ` ],f=t.inShapesLocations[ ` $ { c } Shape ` ],h=t.inTexShapesLocations[ ` $ { c } TexShape ` ];if(f){let{uniformShape:m}=_R(t.program.packedInputs,l.shape,l.texData.texShape);switch(m.length){case 1:e.gl.uniform1iv(f,new Int32Array(m));break;case 2:e.gl.uniform2iv(f,new Int32Array(m));break;case 3:e.gl.uniform3iv(f,new Int32Array(m));break;case 4:e.gl.uniform4iv(f,new Int32Array(m));break;default:break}}if(h&&e.gl.uniform2i(h,l.texData.texShape[0],l.texData.texShape[1]),p!=null){if(l.isUniform){if(E.sizeFromShape(l.shape)<2)e.gl.uniform1f(p,l.uniformValues[0]);else{let m=l.uniformValues;m instanceof Float32Array||(m=new Float32Array(m)),e.gl.uniform1fv(p,m)}return}l.texData.slice!=null&&d!=null&&e.gl.uniform1i(d,l.texData.slice.flatOffset),e.setInputMatrixTexture(l.texData.texture,p,u)}});let i=t.outShapeLocation;if(i)switch(r.shape.length){case 1:e.gl.uniform1iv(i,new Int32Array(r.shape));break;case 2:e.gl.uniform2iv(i,new Int32Array(r.shape));break;case 3:e.gl.uniform3iv(i,new Int32Ar
2021-09-11 16:29:31 +02:00
ivec3 outCoordsFromFlatIndex ( int index ) {
2021-10-04 22:29:15 +02:00
$ { this . enableShapeUniforms ? D5 ( [ "r" , "c" , "d" ] , e ) : mh ( [ "r" , "c" , "d" ] , e ) }
2021-09-11 16:29:31 +02:00
return ivec3 ( r , c , d ) ;
}
void main ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( texShape [ 0 ] , texShape [ 1 ] ) ) ;
int index = 4 * ( resTexRC . x * texShape [ 1 ] + resTexRC . y ) ;
vec4 result = vec4 ( 0. ) ;
for ( int i = 0 ; i < 4 ; i ++ ) {
int flatIndex = index + i ;
ivec3 rc = outCoordsFromFlatIndex ( flatIndex ) ;
result [ i ] = getA ( rc . x , rc . y , rc . z ) ;
2021-08-31 19:00:06 +02:00
}
2021-09-11 16:29:31 +02:00
$ { t . output } = result ;
}
2021-10-04 22:29:15 +02:00
` }},tNe=class{constructor(e){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outPackingScheme=Yb.DENSE,this.customUniforms=[{name:"texShape",type:"ivec2"}];let t=Us();this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length),this.userCode= `
2021-09-11 16:29:31 +02:00
ivec3 outCoordsFromFlatIndex ( int index ) {
2021-10-04 22:29:15 +02:00
$ { this . enableShapeUniforms ? D5 ( [ "r" , "c" , "d" ] , e ) : mh ( [ "r" , "c" , "d" ] , e ) }
2021-09-11 16:29:31 +02:00
return ivec3 ( r , c , d ) ;
}
void main ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( texShape [ 0 ] , texShape [ 1 ] ) ) ;
int index = 4 * ( resTexRC . x * texShape [ 1 ] + resTexRC . y ) ;
vec4 result = vec4 ( 0. ) ;
for ( int i = 0 ; i < 4 ; i ++ ) {
int flatIndex = index + i ;
ivec3 rc = outCoordsFromFlatIndex ( flatIndex ) ;
result [ i ] = getChannel ( getA ( rc . x , rc . y , rc . z ) , vec2 ( rc . y , rc . z ) ) ;
2021-08-31 19:00:06 +02:00
}
2021-09-11 16:29:31 +02:00
$ { t . output } = result ;
}
2021-10-04 22:29:15 +02:00
` }},nNe=class{constructor(e){this.variableNames=["A"],this.outTexUsage=ma.DOWNLOAD;let t=Us();this.outputShape=e,this.userCode= `
$ { jV }
2021-09-11 16:29:31 +02:00
void main ( ) {
float x = getAAtOutCoords ( ) ;
$ { t . output } = encode _float ( x ) ;
}
2021-10-04 22:29:15 +02:00
` }},rNe=class{constructor(e){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!1,this.outTexUsage=ma.DOWNLOAD;let t=Us();this.outputShape=e,this.userCode= `
$ { jV }
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
float x = getChannel ( getAAtOutCoords ( ) , vec2 ( coords . y , coords . z ) ) ;
$ { t . output } = encode _float ( x ) ;
}
2021-10-04 22:29:15 +02:00
` }},sNe=class{constructor(e,t=!1){this.variableNames=["A"],this.customUniforms=[{name:"texShape",type:"ivec2"}];let n=Us();this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length);let r="result";t&&(r="floor(result * 255. + 0.5)"),this.userCode= `
$ { this . enableShapeUniforms ? TR ( ) : NR ( e ) }
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int flatIndex = getFlatIndex ( coords ) ;
int offset = imod ( flatIndex , 4 ) ;
flatIndex = idiv ( flatIndex , 4 , 1. ) ;
int r = flatIndex / texShape [ 1 ] ;
int c = imod ( flatIndex , texShape [ 1 ] ) ;
vec2 uv = ( vec2 ( c , r ) + halfCR ) / vec2 ( texShape [ 1 ] , texShape [ 0 ] ) ;
vec4 values = $ { n . texture2D } ( A , uv ) ;
float result ;
if ( offset == 0 ) {
result = values [ 0 ] ;
} else if ( offset == 1 ) {
result = values [ 1 ] ;
} else if ( offset == 2 ) {
result = values [ 2 ] ;
} else {
result = values [ 3 ] ;
2021-08-31 19:00:06 +02:00
}
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
$ { n . output } = vec4 ( $ { r } , 0. , 0. , 0. ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }},oNe=class{constructor(e,t=!1){this.variableNames=["A"],this.packedInputs=!1,this.packedOutput=!0,this.customUniforms=[{name:"texShape",type:"ivec2"}];let n=Us();this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length);let r="",s="result";t&&(s="floor(result * 255. + 0.5)");for(let o=0;o<=1;o++)for(let a=0;a<=1;a++){let i=o*2+a;r+= `
2021-09-11 16:29:31 +02:00
localCoords = coords ;
2021-10-04 22:29:15 +02:00
if ( localCoords [ 2 ] + $ { a } < $ { this . enableShapeUniforms ? "outShape[2]" : ` ${ e [ 2 ] } ` } ) {
localCoords [ 2 ] += $ { a } ;
if ( localCoords [ 1 ] + $ { o } < $ { this . enableShapeUniforms ? "outShape[1]" : ` ${ e [ 1 ] } ` } ) {
localCoords [ 1 ] += $ { o } ;
2021-09-11 16:29:31 +02:00
flatIndex = getFlatIndex ( localCoords ) ;
offset = imod ( flatIndex , 4 ) ;
flatIndex = idiv ( flatIndex , 4 , 1. ) ;
int r = flatIndex / texShape [ 1 ] ;
int c = imod ( flatIndex , texShape [ 1 ] ) ;
vec2 uv = ( vec2 ( c , r ) + halfCR ) / vec2 ( texShape [ 1 ] , texShape [ 0 ] ) ;
values = $ { n . texture2D } ( A , uv ) ;
if ( offset == 0 ) {
2021-09-11 17:17:13 +02:00
result [ $ { i } ] = values [ 0 ] ;
2021-09-11 16:29:31 +02:00
} else if ( offset == 1 ) {
2021-09-11 17:17:13 +02:00
result [ $ { i } ] = values [ 1 ] ;
2021-09-11 16:29:31 +02:00
} else if ( offset == 2 ) {
2021-09-11 17:17:13 +02:00
result [ $ { i } ] = values [ 2 ] ;
2021-08-31 19:00:06 +02:00
} else {
2021-09-11 17:17:13 +02:00
result [ $ { i } ] = values [ 3 ] ;
2021-08-31 19:00:06 +02:00
}
}
}
2021-09-11 16:29:31 +02:00
` }this.userCode= `
2021-10-04 22:29:15 +02:00
$ { this . enableShapeUniforms ? TR ( ) : NR ( e ) }
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
int flatIndex , r , c , offset ;
ivec3 localCoords ;
vec2 uv ;
vec4 values ;
2021-10-04 22:29:15 +02:00
$ { r }
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
$ { n . output } = $ { s } ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` }},ZV={};Rt(ZV,{bindVertexProgramAttributeStreams:()=>oW,createBufferFromOutputTexture:()=>lW,createFloat16MatrixTexture:()=>tW,createFloat16PackedMatrixTexture:()=>sW,createFloat32MatrixTexture:()=>eW,createIndexBuffer:()=>QV,createPackedMatrixTexture:()=>rW,createUnsignedBytesMatrixTexture:()=>nW,createVertexBuffer:()=>JV,createVertexShader:()=>YV,downloadByteEncodedFloatMatrixFromOutputTexture:()=>cW,downloadFloat32MatrixFromBuffer:()=>uW,downloadMatrixFromPackedOutputTexture:()=>dW,downloadPackedMatrixFromBuffer:()=>pW,getInternalFormatForFloat16MatrixTexture:()=>RR,getInternalFormatForFloat16PackedMatrixTexture:()=>FR,getInternalFormatForFloat32MatrixTexture:()=>ER,getInternalFormatForPackedMatrixTexture:()=> $ R,getInternalFormatForUnsignedBytesMatrixTexture:()=>DR,uploadDenseMatrixToTexture:()=>aW,uploadPixelDataToTexture:()=>iW});function YV(e){let t=Us(),n= ` $ { t . version }
2021-09-11 16:29:31 +02:00
precision highp float ;
$ { t . attribute } vec3 clipSpacePos ;
$ { t . attribute } vec2 uv ;
$ { t . varyingVs } vec2 resultUV ;
void main ( ) {
gl _Position = vec4 ( clipSpacePos , 1 ) ;
resultUV = uv ;
2021-10-04 22:29:15 +02:00
} ` ;return kV(e,n)}function JV(e){let t=new Float32Array([-1,1,0,0,1,-1,-1,0,0,0,1,1,0,1,1,1,-1,0,1,0]);return NV(e,t)}function QV(e){let t=new Uint16Array([0,1,2,2,1,3]);return TV(e,t)}function nA(e,t,n,r,s,o){EV(t,n);let a=_V(e),i=e.TEXTURE_2D;return Qe(e,()=>e.bindTexture(i,a)),Qe(e,()=>e.texParameteri(i,e.TEXTURE_WRAP_S,e.CLAMP_TO_EDGE)),Qe(e,()=>e.texParameteri(i,e.TEXTURE_WRAP_T,e.CLAMP_TO_EDGE)),Qe(e,()=>e.texParameteri(i,e.TEXTURE_MIN_FILTER,e.NEAREST)),Qe(e,()=>e.texParameteri(i,e.TEXTURE_MAG_FILTER,e.NEAREST)),Qe(e,()=>e.texImage2D(i,0,r,t,n,0,s,o,null)),Qe(e,()=>e.bindTexture(e.TEXTURE_2D,null)),a}function ER(e){return e.internalFormatFloat}function eW(e,t,n,r){let[s,o]=Jb(t,n);return nA(e,s,o,ER(r),r.textureFormatFloat,e.FLOAT)}function RR(e){return e.internalFormatHalfFloat}function tW(e,t,n,r){let[s,o]=Jb(t,n);return nA(e,s,o,RR(r),r.textureFormatFloat,r.textureTypeHalfFloat)}function DR(e){return e.downloadTextureFormat}function nW(e,t,n,r){let[s,o]=Jb(t,n);return nA(e,s,o,DR(r),e.RGBA,e.UNSIGNED_BYTE)}function $ R(e){return e.internalFormatPackedFloat}function rW(e,t,n,r){let[s,o]=U0(t,n);return nA(e,s,o, $ R(r),e.RGBA,e.FLOAT)}function FR(e){return e.internalFormatPackedHalfFloat}function sW(e,t,n,r){let[s,o]=U0(t,n);return nA(e,s,o,FR(r),e.RGBA,r.textureTypeHalfFloat)}function oW(e,t,n){let r=0,s=3*4,o=3*4+2*4;return Qe(e,()=>e.bindBuffer(e.ARRAY_BUFFER,n)),kR(e,t,"clipSpacePos",n,3,o,r)&&kR(e,t,"uv",n,2,o,s)}function aW(e,t,n,r,s,o){Qe(e,()=>e.bindTexture(e.TEXTURE_2D,t));let a,i,l;s instanceof Uint8Array?(a=new Uint8Array(n*r*4),i=e.UNSIGNED_BYTE,l=e.RGBA):(a=new Float32Array(n*r*4),i=e.FLOAT,l=o.internalFormatPackedFloat),a.set(s),Qe(e,()=>e.texImage2D(e.TEXTURE_2D,0,l,n,r,0,e.RGBA,i,a)),Qe(e,()=>e.bindTexture(e.TEXTURE_2D,null))}function iW(e,t,n){Qe(e,()=>e.bindTexture(e.TEXTURE_2D,t)),n.data instanceof Uint8Array?Qe(e,()=>e.texImage2D(e.TEXTURE_2D,0,e.RGBA,n.width,n.height,0,e.RGBA,e.UNSIGNED_BYTE,n.data)):Qe(e,()=>e.texImage2D(e.TEXTURE_2D,0,e.RGBA,e.RGBA,e.UNSIGNED_BYTE,n)),Qe(e,()=>e.bindTexture(e.TEXTURE_2D,null))}function lW(e,t,n,r){let s=e.createBuffer();Qe(e,()=>e.bindBuffer(e.PIXEL_PACK_BUFFER,s));let i=4*4*t*n;return Qe(e,()=>e.bufferData(e.PIXEL_PACK_BUFFER,i,e.STREAM_READ)),Qe(e,()=>e.readPixels(0,0,n,t,e.RGBA,e.FLOAT,0)),Qe(e,()=>e.bindBuffer(e.PIXEL_PACK_BUFFER,null)),s}function uW(e,t,n){let r=e,s=new Float32Array(n);return r.bindBuffer(r.PIXEL_PACK_BUFFER,t),r.getBufferSubData(r.PIXEL_PACK_BUFFER,0,s),r.bindBuffer(r.PIXEL_PACK_BUFFER,null),s}function cW(e,t,n,r){let[s,o]=Jb(t,n),a=4,i=new Uint8Array(eSe(t*n,a));return Qe(e,()=>e.readPixels(0,0,s,o,r.downloadTextureFormat,e.UNSIGNED_BYTE,i)),new Float32Array(i.buffer)}function pW(e,t,n,r,s,o,a,i){let l=e,u=new Float32Array(tSe(o,a));return l.bindBuffer(l.PIXEL_PACK_BUFFER,t),l.getBufferSubData(l.PIXEL_PACK_BUFFER,0,u),l.bindBuffer(l.PIXEL_PACK_BUFFER,null),u}function dW(e,t,n){let r=new Float32Array(t*n*4);return Qe(e,()=>e.readPixels(0,0,n,t,e.RGBA,e.FLOAT,r)),r}var $ 5=class{constructor(e){this.outputTexture=null,this.program=null,this.disposed=!1,this.vertexAttrsAreBound=!1,this.itemsToPoll=[];let t=be().getNumber("WEBGL_VERSION");e!=null?(this.gl=e,I5(t,e)):this.gl=ml(t);let n="WEBGL_color_buffer_float",r="EXT_color_buffer_half_float";if(be().getNumber("WEBGL_VERSION")===1){let s="OES_texture_float",o="OES_texture_half_float";if(this.textureFloatExtension=Qb(this.gl,s),ga(this.gl,o))this.textureHalfFloatExtension=Qb(this.gl,o);else if(be().get("WEBGL_FORCE_F16_TEXTURES"))throw new Error("GL context does not support half float textures, yet the environment flag WEBGL_FORCE_F16_TEXTURES is set to true.");if(this.colorBufferFloatExtension=this.gl.getExtension(n),ga(this.gl,r))this.colorBufferHalfFloatExtension=Qb(this.gl,r);else if(be().get("WEBGL_FORCE_F16_TEXTURES"))throw new Error("GL context does not support color renderable half floats, yet the environment flag WEBGL_FORCE_F16_TEXTURES is set to true.")}else if(n="EXT_color_buffer_float",ga(this.gl,n))this.colorBufferFloatExtension=this.gl.getExtension(n);else if(ga(this.gl,r)
2021-09-11 16:29:31 +02:00
void main ( ) {
setOutput ( vec4 ( getA ( ) , 0. , 0. , 0. ) ) ;
2021-07-29 22:35:16 +02:00
}
2021-10-04 22:29:15 +02:00
` ;else{let n=js("rc",t),r=Dn(t),s=KNe(t,e,n),o=XNe(t,e[e.length-1],e[e.length-2],n),a=ZNe(e,n);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r } rc = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { s } ) {
2021-09-11 16:29:31 +02:00
setOutput ( vec4 ( 0 ) ) ;
2021-09-05 22:42:11 +02:00
} else {
2021-10-04 22:29:15 +02:00
$ { o }
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
setOutput ( vec4 ( $ { a } ) ) ;
2021-09-01 00:24:30 +02:00
}
}
2021-10-04 22:29:15 +02:00
` }}};function qNe(e,t){let n=[];for(let r=0;r<=1;r++)for(let s=0;s<=1;s++){let o= ` $ { r === 0 ? "r" : "rp1" } , $ { s === 0 ? "c" : "cp1" } ` ;for(let a=2;a<e;a++)o= ` $ { t [ t . length - 1 - a ] } , ` +o;n.push(o)}return n}function KNe(e,t,n){if(e===1)return ` rc > $ { t [ 0 ] } ` ;let r="";for(let s=e-2;s<e;s++)r+= ` $ { n [ s ] } >= $ { t [ s ] } ` ,s<e-1&&(r+="||");return r}function XNe(e,t,n,r){if(e===1)return"";let s=r.slice(-2);return `
int r = $ { s [ 0 ] } ;
int c = $ { s [ 1 ] } ;
2021-09-11 16:29:31 +02:00
int rp1 = r + 1 ;
int cp1 = c + 1 ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
bool cEdge = cp1 >= $ { t } ;
bool rEdge = rp1 >= $ { n } ;
2021-10-04 22:29:15 +02:00
` }function ZNe(e,t){let n=e.length,r=qNe(n,t);return n===1? ` getA ( rc ) ,
2021-09-11 16:29:31 +02:00
rc + 1 >= $ { e [ 0 ] } ? 0. : getA ( rc + 1 ) ,
2021-10-04 22:29:15 +02:00
0 , 0 ` : ` getA ( $ { r [ 0 ] } ) ,
cEdge ? 0. : getA ( $ { r [ 1 ] } ) ,
rEdge ? 0. : getA ( $ { r [ 2 ] } ) ,
rEdge || cEdge ? 0. : getA ( $ { r [ 3 ] } ) ` }var xW=class{constructor(e,t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"inputShape",type:"ivec3"}],this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length);let n="";for(let r=0;r<4;r++){let s="thisRC = rc;";r%2==1&&(s+="thisRC.z += 1;"),r>1&&(s+="thisRC.y += 1;"),n+= `
$ { s }
$ { r > 0 ? "if(thisRC.y < rows && thisRC.z < cols){" : "" }
2021-09-11 16:29:31 +02:00
int flatIndex = getFlatIndex ( thisRC ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
ivec3 inputRC = inputCoordsFromReshapedOutCoords ( flatIndex ) ;
vec2 inputRCInnerDims = vec2 ( float ( inputRC . y ) , float ( inputRC . z ) ) ;
2021-06-07 02:34:29 +02:00
2021-10-04 22:29:15 +02:00
result [ $ { r } ] =
2021-09-11 16:29:31 +02:00
getChannel ( getA ( inputRC . x , inputRC . y , inputRC . z ) , inputRCInnerDims ) ;
2021-10-04 22:29:15 +02:00
$ { r > 0 ? "}" : "" }
2021-09-11 16:29:31 +02:00
` }this.userCode= `
2021-10-04 22:29:15 +02:00
$ { YNe ( t , this . enableShapeUniforms ) }
$ { this . enableShapeUniforms ? TR ( ) : NR ( e ) }
2021-06-07 02:34:29 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 rc = getOutputCoords ( ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0. ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
ivec3 thisRC ;
int rows = $ { this . enableShapeUniforms ? "outShape[1]" : e [ 1 ] } ;
int cols = $ { this . enableShapeUniforms ? "outShape[2]" : e [ 2 ] } ;
$ { n }
2021-08-14 17:17:51 +02:00
2021-09-01 00:24:30 +02:00
setOutput ( result ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function YNe(e,t){return `
2021-09-11 16:29:31 +02:00
ivec3 inputCoordsFromReshapedOutCoords ( int index ) {
2021-10-04 22:29:15 +02:00
$ { t ? hSe ( [ "r" , "c" , "d" ] , "inputShape" ) : mh ( [ "r" , "c" , "d" ] , e ) }
2021-09-11 16:29:31 +02:00
return ivec3 ( r , c , d ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }var JNe=class{constructor(e){this.gpgpu=e,this.numUsedTextures=0,this.numFreeTextures=0,this._numBytesAllocated=0,this._numBytesFree=0,this.freeTextures={},this.logEnabled=!1,this.usedTextures={}}acquireTexture(e,t,n){let r=bW(t,n),s=AW(e,r,n);s in this.freeTextures||(this.freeTextures[s]=[]),s in this.usedTextures||(this.usedTextures[s]=[]);let o=yW(e,r,this.gpgpu.gl,this.gpgpu.textureConfig,n);if(this.freeTextures[s].length>0){this.numFreeTextures--,this.numUsedTextures++,this._numBytesFree-=o,this.log();let i=this.freeTextures[s].shift();return this.usedTextures[s].push(i),i}let a;return r===ws.PACKED_2X2_FLOAT32?a=this.gpgpu.createPackedMatrixTexture(e[0],e[1]):r===ws.PACKED_2X2_FLOAT16?a=this.gpgpu.createFloat16PackedMatrixTexture(e[0],e[1]):r===ws.UNPACKED_FLOAT32?a=this.gpgpu.createFloat32MatrixTexture(e[0],e[1]):r===ws.UNPACKED_FLOAT16?a=this.gpgpu.createFloat16MatrixTexture(e[0],e[1]):r===ws.PACKED_4X1_UNSIGNED_BYTE&&(a=this.gpgpu.createUnsignedBytesMatrixTexture(e[0],e[1])),this.usedTextures[s].push(a),this.numUsedTextures++,this._numBytesAllocated+=o,this.log(),a}releaseTexture(e,t,n,r){if(this.freeTextures==null)return;let s=bW(n,r),o=AW(t,s,r);o in this.freeTextures||(this.freeTextures[o]=[]);let a=yW(t,s,this.gpgpu.gl,this.gpgpu.textureConfig,r),i=be().get("WEBGL_DELETE_TEXTURE_THRESHOLD");i!==-1&&this._numBytesAllocated>i?(this.gpgpu.deleteMatrixTexture(e),this._numBytesAllocated-=a):(this.freeTextures[o].push(e),this.numFreeTextures++,this._numBytesFree+=a),this.numUsedTextures--;let l=this.usedTextures[o],u=l.indexOf(e);if(u<0)throw new Error("Cannot release a texture that was never provided by this texture manager");l.splice(u,1),this.log()}log(){if(!this.logEnabled)return;let e=this.numFreeTextures+this.numUsedTextures;console.log("Free/Used", ` $ { this . numFreeTextures } / $ { this . numUsedTextures } ` , ` ( $ { e } ) ` );let t=this._numBytesFree/this._numBytesAllocated;console.log( ` Bytes allocated : $ { this . _numBytesAllocated } ` ),console.log( ` Bytes unused : $ { this . _numBytesFree } ( $ { Math . round ( 100 * t ) } % ) ` )}get numBytesAllocated(){return this._numBytesAllocated}get numBytesFree(){return this._numBytesFree}getNumUsedTextures(){return this.numUsedTextures}getNumFreeTextures(){return this.numFreeTextures}dispose(){if(this.freeTextures!=null){for(let e in this.freeTextures)this.freeTextures[e].forEach(t=>{this.gpgpu.deleteMatrixTexture(t)});for(let e in this.usedTextures)this.usedTextures[e].forEach(t=>{this.gpgpu.deleteMatrixTexture(t)});this.freeTextures=null,this.usedTextures=null,this.numUsedTextures=0,this.numFreeTextures=0,this._numBytesAllocated=0,this._numBytesFree=0}}};function QNe(e,t){let n=e;if(t===n.R32F)return 4;if(t===n.R16F)return 2;if(t===n.RGBA32F)return 16;if(t===e.RGBA)return 16;if(t===n.RGBA16F)return 8;throw new Error( ` Unknown internal format $ { t } ` )}function yW(e,t,n,r,s){let o=eTe(t,r),a;if(s){let[l,u]=U0(e[0],e[1]);a=l*u}else{let[l,u]=Jb(e[0],e[1]);a=l*u}let i=QNe(n,o);return a*i}function eTe(e,t){switch(e){case ws.PACKED_2X2_FLOAT32:return $ R(t);case ws.PACKED_2X2_FLOAT16:return FR(t);case ws.UNPACKED_FLOAT32:return ER(t);case ws.UNPACKED_FLOAT16:return RR(t);case ws.PACKED_4X1_UNSIGNED_BYTE:return DR(t);default:throw new Error( ` Unknown physical texture type $ { e } ` )}}function tTe(e){return be().getBool("WEBGL_RENDER_FLOAT32_ENABLED")?e?ws.PACKED_2X2_FLOAT32:ws.UNPACKED_FLOAT32:e?ws.PACKED_2X2_FLOAT16:ws.UNPACKED_FLOAT16}function bW(e,t){if(e===ma.UPLOAD)return ws.PACKED_2X2_FLOAT32;if(e===ma.RENDER||e==null)return tTe(t);if(e===ma.DOWNLOAD||e===ma.PIXELS)return ws.PACKED_4X1_UNSIGNED_BYTE;throw new Error( ` Unknown logical texture type $ { e } ` )}function AW(e,t,n){return ` $ { e [ 0 ] } _$ { e [ 1 ] } _$ { t } _$ { n } ` }var rp=class{constructor(e,t){this.variableNames=["A"],this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length),this.userCode= `
2021-09-11 16:29:31 +02:00
float unaryOperation ( float x ) {
$ { t }
}
2021-09-01 00:22:16 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
float x = getAAtOutCoords ( ) ;
float y = unaryOperation ( x ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( y ) ;
}
2021-10-04 22:29:15 +02:00
` }},Ti="if (isnan(x)) return x;",nTe="return x;",vW="return abs(x);",rTe="return (x >= 0.0) ? x : (exp(x) - 1.0);",sTe=Ti+ `
2021-09-11 16:29:31 +02:00
return ( x < 0.0 ) ? 0.0 : x ;
2021-10-04 22:29:15 +02:00
` ,oTe=Ti+ `
2021-09-11 16:29:31 +02:00
return ( x < 0.0 ) ? 0.0 : min ( 6.0 , x ) ;
2021-10-04 22:29:15 +02:00
` ,F5="return x;",aTe="return 1.0 / (1.0 + exp(-1.0 * x));",iTe="return x;",lTe= `
2021-09-11 16:29:31 +02:00
vec4 result ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
result . r = ( x . r >= 0.0 ) ? x . r : ( exp ( x . r ) - 1.0 ) ;
result . g = ( x . g >= 0.0 ) ? x . g : ( exp ( x . g ) - 1.0 ) ;
result . b = ( x . b >= 0.0 ) ? x . b : ( exp ( x . b ) - 1.0 ) ;
result . a = ( x . a >= 0.0 ) ? x . a : ( exp ( x . a ) - 1.0 ) ;
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,uTe= `
2021-09-11 16:29:31 +02:00
vec4 result = x * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
2021-08-14 17:17:51 +02:00
2021-09-05 22:42:11 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,cTe= `
2021-09-11 16:29:31 +02:00
vec4 result = min ( x , vec4 ( 6. ) ) * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
2021-09-01 00:22:16 +02:00
2021-09-11 16:29:31 +02:00
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,pTe="return 1.0 / (1.0 + exp(-1.0 * x));",Z0=class{constructor(e,t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length),this.userCode= `
2021-09-11 16:29:31 +02:00
vec4 unaryOperation ( vec4 x ) {
$ { t }
}
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
vec4 x = getAAtOutCoords ( ) ;
vec4 y = unaryOperation ( x ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( y ) ;
}
2021-10-04 22:29:15 +02:00
` }},dTe=class{constructor(e){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!1,this.outputShape=e;let t=e.length,n=js("rc",t),r=Dn(t),s=jNe(t,n),o=n.slice(-2),a=t<=1?"rc": ` vec2 ( $ { o . join ( "," ) } ) ` ;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r } rc = getOutputCoords ( ) ;
vec4 packedInput = getA ( $ { s } ) ;
2021-08-14 17:17:51 +02:00
2021-10-04 22:29:15 +02:00
setOutput ( getChannel ( packedInput , $ { a } ) ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }},fTe=ll.whereImpl,hTe=1e-7,mTe=1e-4,O5={};function gTe(e){return e in O5||(O5[e]={}),O5[e]}var xTe=be().getNumber("CPU_HANDOFF_SIZE_THRESHOLD"),yTe=600;function bTe(){return be().global.screen==null?1024:be().global.screen.height*be().global.screen.width*window.devicePixelRatio*yTe/1024/1024}var Y0=class extends $ 1{constructor(e){super();if(this.pendingRead=new WeakMap,this.pendingDisposal=new WeakSet,this.dataRefCount=new WeakMap,this.numBytesInGPU=0,this.uploadWaitMs=0,this.downloadWaitMs=0,this.lastGlFlushTime=0,this.warnedAboutMemory=!1,this.pendingDeletes=0,this.disposed=!1,!be().getBool("HAS_WEBGL"))throw new Error("WebGL is not supported on this device");if(e==null){let t=ml(be().getNumber("WEBGL_VERSION"));this.binaryCache=gTe(be().getNumber("WEBGL_VERSION")),this.gpgpu=new $ 5(t),this.canvas=t.canvas,this.gpgpuCreatedLocally=!0}else this.gpgpu=e,this.binaryCache={},this.gpgpuCreatedLocally=!1,this.canvas=e.gl.canvas;this.textureManager=new JNe(this.gpgpu),this.numMBBeforeWarning=bTe(),this.texData=new Xv(this,zo())}nextDataId(){return Y0.nextDataId++}numDataIds(){return this.texData.numDataIds()-this.pendingDeletes}write(e,t,n){if((be().getBool("WEBGL_CHECK_NUMERICAL_PROBLEMS")||be().getBool("DEBUG"))&&this.checkNumericalProblems(e),n==="complex64"&&e!=null)throw new Error("Cannot write to a complex64 dtype. Please use tf.complex(real, imag).");let r={id:this.nextDataId()};return this.texData.set(r,{shape:t,dtype:n,values:e,usage:ma.UPLOAD,refCount:1}),r}refCount(e){return this.texData.has(e)?this.texData.get(e).refCount:0}incRef(e){let t=this.texData.get(e);t.refCount++}decRef(e){if(this.texData.has(e)){let t=this.texData.get(e);t.refCount--}}move(e,t,n,r,s){if(be().getBool("DEBUG")&&this.checkNumericalProblems(t),r==="complex64")throw new Error("Cannot write to a complex64 dtype. Please use tf.complex(real, imag).");this.texData.set(e,{shape:n,dtype:r,values:t,usage:ma.UPLOAD,refCount:s})}disposeIntermediateTensorInfo(e){this.disposeData(e.dataId)}readSync(e){let t=this.texData.get(e),{values:n,dtype:r,complexTensorInfos:s,slice:o,shape:a,isPacked:i}=t;if(o!=null){let p;i?p=new Z0(a,F5):p=new rp(a,F5);let d=this.runWebGLProgram(p,[{dataId:e,shape:a,dtype:r}],r),f=this.readSync(d.dataId);return this.disposeIntermediateTensorInfo(d),f}if(n!=null)return this.convertAndCacheOnCPU(e);if(r==="string")return n;let l=this.activeTimers!=null,u;l&&(u=E.now());let c;if(r==="complex64"){let p=this.readSync(s.real.dataId),d=this.readSync(s.imag.dataId);c=G.mergeRealAndImagArrays(p,d)}else c=this.getValuesFromTexture(e);return l&&(this.downloadWaitMs+=E.now()-u),this.convertAndCacheOnCPU(e,c)}async read(e){if(this.pendingRead.has(e)){let f=this.pendingRead.get(e);return new Promise(h=>f.push(h))}let t=this.texData.get(e),{values:n,shape:r,slice:s,dtype:o,complexTensorInfos:a,isPacked:i}=t;if(s!=null){let f;i?f=new Z0(r,F5):f=new rp(r,F5);let h=this.runWebGLProgram(f,[{dataId:e,shape:r,dtype:o}],o),m=this.read(h.dataId);return this.disposeIntermediateTensorInfo(h),m}if(n!=null)return this.convertAndCacheOnCPU(e);if(!be().getBool("WEBGL_DOWNLOAD_FLOAT_ENABLED")&&be().getNumber("WEBGL_VERSION")===2)throw new Error("tensor.data() with WEBGL_DOWNLOAD_FLOAT_ENABLED=false and WEBGL_VERSION=2 not yet supported.");let l=null,u;if(o!=="complex64"&&be().get("WEBGL_BUFFER_SUPPORTED")){u=this.decode(e);let f=this.texData.get(u.dataId);l=this.gpgpu.createBufferFromTexture(f.texture,...C5(r))}this.pendingRead.set(e,[]),o!=="complex64"&&await this.gpgpu.createAndWaitForFence();let c;if(o==="complex64"){let f=await Promise.all([this.read(a.real.dataId),this.read(a.imag.dataId)]),h=f[0],m=f[1];c=G.mergeRealAndImagArrays(h,m)}else if(l==null)c=this.getValuesFromTexture(e);else{let f=E.sizeFromShape(r);c=this.gpgpu.downloadFloat32MatrixFromBuffer(l,f)}if(u!=null&&this.disposeIntermediateTensorInfo(u),l!=null){let f=this.gpgpu.gl;Qe(f,()=>f.deleteBuffer(l))}let p=this.convertAndCacheOnCPU(e,c),d=this.pendingRead.get(e);return this.pendingRead.delete(e),d.forEach(f=>f(p)),this.pendingDisposal.has(e)&&(this.pendingDisposal.delete(e),this.d
2021-09-11 16:29:31 +02:00
if ( isnan ( a ) ) return a ;
if ( isnan ( b ) ) return b ;
2021-10-04 22:29:15 +02:00
` ,J0=class{constructor(e,t,n){this.variableNames=["A","B"],this.outputShape=G.assertAndGetBroadcastShape(t,n),this.enableShapeUniforms=xa(this.outputShape.length),this.userCode= `
2021-09-11 16:29:31 +02:00
float binaryOperation ( float a , float b ) {
$ { e }
}
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
float a = getAAtOutCoords ( ) ;
float b = getBAtOutCoords ( ) ;
setOutput ( binaryOperation ( a , b ) ) ;
}
2021-10-04 22:29:15 +02:00
` }},P5= `
2021-09-11 16:29:31 +02:00
result . r = isNaN . r > 0. ? NAN : result . r ;
result . g = isNaN . g > 0. ? NAN : result . g ;
result . b = isNaN . b > 0. ? NAN : result . b ;
result . a = isNaN . a > 0. ? NAN : result . a ;
2021-10-04 22:29:15 +02:00
` ,rA=class{constructor(e,t,n,r=!1){this.variableNames=["A","B"],this.supportsBroadcasting=!0,this.packedInputs=!0,this.packedOutput=!0,this.outputShape=G.assertAndGetBroadcastShape(t,n);let s=this.outputShape.length;this.enableShapeUniforms=xa(s);let o="";if(r)if(s===0||E.sizeFromShape(this.outputShape)===1)o= `
2021-09-11 16:29:31 +02:00
result . y = 0. ;
result . z = 0. ;
result . w = 0. ;
2021-10-04 22:29:15 +02:00
` ;else if(o= `
$ { Dn ( s ) } coords = getOutputCoords ( ) ;
` ,s===1)this.enableShapeUniforms?o+= `
2021-09-11 16:29:31 +02:00
result . y = ( coords + 1 ) >= outShape ? 0. : result . y ;
result . z = 0. ;
result . w = 0. ;
2021-10-04 22:29:15 +02:00
` :o+= `
2021-09-11 16:29:31 +02:00
result . y = ( coords + 1 ) >= $ { this . outputShape [ 0 ] } ? 0. : result . y ;
result . z = 0. ;
result . w = 0. ;
2021-10-04 22:29:15 +02:00
` ;else{let i=js("coords",s);this.enableShapeUniforms?o+= `
2021-09-11 16:29:31 +02:00
bool nextRowOutOfBounds =
2021-10-04 22:29:15 +02:00
( $ { i [ s - 2 ] } + 1 ) >= outShape [ $ { s } - 2 ] ;
2021-09-11 16:29:31 +02:00
bool nextColOutOfBounds =
2021-10-04 22:29:15 +02:00
( $ { i [ s - 1 ] } + 1 ) >= outShape [ $ { s } - 1 ] ;
2021-09-11 16:29:31 +02:00
result . y = nextColOutOfBounds ? 0. : result . y ;
result . z = nextRowOutOfBounds ? 0. : result . z ;
result . w = nextColOutOfBounds || nextRowOutOfBounds ? 0. : result . w ;
2021-10-04 22:29:15 +02:00
` :o+= `
2021-09-11 16:29:31 +02:00
bool nextRowOutOfBounds =
2021-10-04 22:29:15 +02:00
( $ { i [ s - 2 ] } + 1 ) >= $ { this . outputShape [ s - 2 ] } ;
2021-09-11 16:29:31 +02:00
bool nextColOutOfBounds =
2021-10-04 22:29:15 +02:00
( $ { i [ s - 1 ] } + 1 ) >= $ { this . outputShape [ s - 1 ] } ;
2021-09-11 16:29:31 +02:00
result . y = nextColOutOfBounds ? 0. : result . y ;
result . z = nextRowOutOfBounds ? 0. : result . z ;
result . w = nextColOutOfBounds || nextRowOutOfBounds ? 0. : result . w ;
` }this.userCode= `
vec4 binaryOperation ( vec4 a , vec4 b ) {
$ { e }
}
void main ( ) {
vec4 a = getAAtOutCoords ( ) ;
vec4 b = getBAtOutCoords ( ) ;
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
vec4 result = binaryOperation ( a , b ) ;
2021-10-04 22:29:15 +02:00
$ { o }
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }};function Xo(e){let{inputs:t,backend:n}=e,{x:r}=t;return n.incRef(r.dataId),{dataId:r.dataId,shape:r.shape,dtype:r.dtype}}var kTe={kernelName:sf,backendName:"webgl",kernelFunc:Xo};function sp(e){let{inputs:t,backend:n}=e,{real:r,imag:s}=t,o=n.makeTensorInfo(r.shape,"complex64"),a=n.texData.get(o.dataId),i=Xo({inputs:{x:r},backend:n}),l=Xo({inputs:{x:s},backend:n});return a.complexTensorInfos={real:i,imag:l},o}var ITe={kernelName:rw,backendName:"webgl",kernelFunc:sp},IW="return (a < 0.) ? b * a : a;",CW= `
2021-09-11 16:29:31 +02:00
vec4 aLessThanZero = vec4 ( lessThan ( a , vec4 ( 0. ) ) ) ;
return ( aLessThanZero * ( b * a ) ) + ( ( vec4 ( 1.0 ) - aLessThanZero ) * a ) ;
2021-10-04 22:29:15 +02:00
` ;function CTe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{alpha:o}=r,a=n.makeTensorInfo([],"float32",E.createScalarValue(o,"float32")),i=be().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new rA(CW,s.shape,a.shape):new J0(IW,s.shape,a.shape),l=n.runWebGLProgram(i,[s,a],s.dtype);return n.disposeIntermediateTensorInfo(a),l}var STe={kernelName:of,backendName:"webgl",kernelFunc:CTe},SW="return (a < 0.) ? b * a : a;",NW= `
2021-09-11 16:29:31 +02:00
vec4 aLessThanZero = vec4 ( lessThan ( a , vec4 ( 0. ) ) ) ;
return ( aLessThanZero * ( b * a ) ) + ( ( vec4 ( 1.0 ) - aLessThanZero ) * a ) ;
2021-10-04 22:29:15 +02:00
` ;function NTe(e){let{inputs:t,backend:n}=e,{x:r,alpha:s}=t,o=be().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new rA(NW,r.shape,s.shape):new J0(SW,r.shape,s.shape);return n.runWebGLProgram(o,[r,s],r.dtype)}var TTe={kernelName:bf,backendName:"webgl",kernelFunc:NTe},TW="if (isnan(x)) return x;",_Te= `
2021-09-11 16:29:31 +02:00
if ( isnan ( a ) ) return a ;
if ( isnan ( b ) ) return b ;
2021-10-04 22:29:15 +02:00
` ,ETe= `
2021-09-11 16:29:31 +02:00
result . r = isNaN . r > 0. ? NAN : result . r ;
result . g = isNaN . g > 0. ? NAN : result . g ;
result . b = isNaN . b > 0. ? NAN : result . b ;
result . a = isNaN . a > 0. ? NAN : result . a ;
2021-10-04 22:29:15 +02:00
` ;function tn({opSnippet:e,packedOpSnippet:t,cpuKernelImpl:n,dtype:r}){return({inputs:s,backend:o})=>{let{x:a}=s,i=o,l=r||a.dtype;if(i.shouldExecuteOnCPU([a])&&n!=null){let p=i.texData.get(a.dataId),d=n(p.values,l);return i.makeTensorInfo(a.shape,l,d)}let u=be().getBool("WEBGL_PACK_UNARY_OPERATIONS")&&t!=null,c;return u?c=new Z0(a.shape,t):c=new rp(a.shape,e),i.runWebGLProgram(c,[a],l)}}function ks({opSnippet:e,packedOpSnippet:t,checkOutOfBounds:n=!1,supportsComplex:r=!1,cpuKernelImpl:s,dtype:o}){return({inputs:a,backend:i})=>{let{a:l,b:u}=a,c=i;if(r&&l.dtype==="complex64"){let h=c.texData.get(l.dataId),m=c.texData.get(u.dataId),[g,x]=[[h.complexTensorInfos.real,m.complexTensorInfos.real],[h.complexTensorInfos.imag,m.complexTensorInfos.imag]].map(b=>{let[A,v]=b,w={dataId:A.dataId,dtype:A.dtype,shape:l.shape},k={dataId:v.dataId,dtype:v.dtype,shape:u.shape},I=new J0(e,l.shape,u.shape);return c.runWebGLProgram(I,[w,k], $ a(A.dtype,v.dtype))}),y=sp({inputs:{real:g,imag:x},backend:c});return c.disposeIntermediateTensorInfo(g),c.disposeIntermediateTensorInfo(x),y}let p=o|| $ a(l.dtype,u.dtype);if((l.dtype==="string"||u.dtype==="string"||c.shouldExecuteOnCPU([l,u]))&&s!=null){let h=c.texData.get(l.dataId).values,m=c.texData.get(u.dataId).values,g=l.dtype==="string"?G.fromUint8ToStringArray(h):h,x=l.dtype==="string"?G.fromUint8ToStringArray(m):m,[y,b]=s(l.shape,u.shape,g,x,p),A=c.makeTensorInfo(b,p),v=c.texData.get(A.dataId);return v.values=y,A}let d=be().getBool("WEBGL_PACK_BINARY_OPERATIONS")&&t!=null,f;return d?f=new rA(t,l.shape,u.shape,n):f=new J0(e,l.shape,u.shape),c.runWebGLProgram(f,[l,u],p)}}function M5(e,t=!1){if(e==="linear")return t?iTe:nTe;if(e==="relu")return t?uTe:sTe;if(e==="elu")return t?lTe:rTe;if(e==="relu6")return t?cTe:oTe;if(e==="prelu")return t?NW:SW;if(e==="leakyrelu")return t?CW:IW;if(e==="sigmoid")return t?pTe:aTe;throw new Error( ` Activation $ { e } has not been implemented for the WebGL backend . ` )}var _W=class{constructor(e,t,n,r=!1,s=!1,o=!1,a=null,i=!1,l=!1){this.variableNames=["matrixA","matrixB"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=n,this.enableShapeUniforms=xa(this.outputShape.length);let u=r?e[1]:e[2],c=Math.ceil(u/2),p=r?"i * 2, rc.y":"rc.y, i * 2",d=s?"rc.z, i * 2":"i * 2, rc.z",f=r?["a.xxyy","a.zzww"]:["a.xxzz","a.yyww"],h=s?["b.xzxz","b.ywyw"]:["b.xyxy","b.zwzw"],m="",g="";a&&(i?m= ` vec4 activation ( vec4 a ) {
2021-09-11 16:29:31 +02:00
vec4 b = getPreluActivationWeightsAtOutCoords ( ) ;
2021-10-04 22:29:15 +02:00
$ { a }
} ` :l?m= ` vec4 activation ( vec4 a ) {
2021-09-11 16:29:31 +02:00
vec4 b = getLeakyreluAlphaAtOutCoords ( ) ;
2021-10-04 22:29:15 +02:00
$ { a }
} ` :m= ` vec4 activation ( vec4 x ) {
$ { a }
} ` ,g="result = activation(result);");let x=o?"result += getBiasAtOutCoords();":"";o&&this.variableNames.push("bias"),i&&this.variableNames.push("preluActivationWeights"),l&&this.variableNames.push("leakyreluAlpha");let y="rc.x",b="rc.x";e[0]<t[0]?y= ` int ( min ( float ( rc . x ) , $ { e [ 0 ] - 1 } . ) ) ` :t[0]<e[0]&&(b= ` int ( min ( float ( rc . x ) , $ { t [ 0 ] - 1 } . ) ) ` ),this.userCode= `
$ { m }
2021-09-11 16:29:31 +02:00
// Don't use uniform for sharedDimensionPacked for performance.
2021-09-11 17:17:13 +02:00
const float sharedDimension = $ { c } . 0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
vec4 dot2x2ARowBCol ( ivec3 rc ) {
vec4 result = vec4 ( 0 ) ;
2021-09-11 17:17:13 +02:00
for ( int i = 0 ; i < $ { c } ; i ++ ) {
2021-09-11 16:29:31 +02:00
int batchA = $ { y } ;
2021-10-04 22:29:15 +02:00
int batchB = $ { b } ;
vec4 a = getMatrixA ( batchA , $ { p } ) ;
vec4 b = getMatrixB ( batchB , $ { d } ) ;
2021-09-11 16:29:31 +02:00
// These swizzled products need to be separately added.
// See: https://github.com/tensorflow/tfjs/issues/1735
2021-10-04 22:29:15 +02:00
result += ( $ { f [ 0 ] } * $ { h [ 0 ] } ) ;
result += ( $ { f [ 1 ] } * $ { h [ 1 ] } ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
return result ;
2021-09-01 00:22:16 +02:00
}
2021-09-01 00:24:30 +02:00
2021-08-14 17:17:51 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
ivec3 rc = getOutputCoords ( ) ;
vec4 result = dot2x2ARowBCol ( rc ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { x }
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { g }
2021-09-01 00:22:16 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }},EW={REAL:"return areal * breal - aimag * bimag;",IMAG:"return areal * bimag + aimag * breal;"},RW=class{constructor(e,t,n){this.variableNames=["AReal","AImag","BReal","BImag"],this.outputShape=G.assertAndGetBroadcastShape(t,n),this.userCode= `
2021-09-11 16:29:31 +02:00
float binaryOpComplex (
float areal , float aimag , float breal , float bimag ) {
$ { e }
}
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
float areal = getARealAtOutCoords ( ) ;
float aimag = getAImagAtOutCoords ( ) ;
float breal = getBRealAtOutCoords ( ) ;
float bimag = getBImagAtOutCoords ( ) ;
setOutput ( binaryOpComplex ( areal , aimag , breal , bimag ) ) ;
}
2021-10-04 22:29:15 +02:00
` }},DW="return a * b;";function PR(e){let{inputs:t,backend:n}=e,{a:r,b:s}=t,o=G.upcastType(r.dtype,s.dtype);if(r.dtype==="complex64"){let i=n.texData.get(r.dataId),l=n.texData.get(s.dataId),u=new RW(EW.REAL,r.shape,s.shape),c=new RW(EW.IMAG,r.shape,s.shape),p=[{dataId:i.complexTensorInfos.real.dataId,dtype:i.complexTensorInfos.real.dtype,shape:r.shape},{dataId:i.complexTensorInfos.imag.dataId,dtype:i.complexTensorInfos.imag.dtype,shape:r.shape},{dataId:l.complexTensorInfos.real.dataId,dtype:l.complexTensorInfos.real.dtype,shape:s.shape},{dataId:l.complexTensorInfos.imag.dataId,dtype:l.complexTensorInfos.imag.dtype,shape:s.shape}],d=n.runWebGLProgram(u,p,"float32"),f=n.runWebGLProgram(c,p,"float32"),h=sp({inputs:{real:d,imag:f},backend:n});return n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}if(n.shouldExecuteOnCPU([r,s])){let i=n.texData.get(r.dataId),l=n.texData.get(s.dataId),[u,c]=SNe(r.shape,s.shape,i.values,l.values,o),p=n.makeTensorInfo(c,o),d=n.texData.get(p.dataId);return d.values=u,p}let a;return be().getBool("WEBGL_PACK_BINARY_OPERATIONS")?a=new rA(DW,r.shape,s.shape):a=new J0(DW,r.shape,s.shape),n.runWebGLProgram(a,[r,s],o)}var RTe={kernelName:mf,backendName:"webgl",kernelFunc:PR};function DTe(e,t,n){let r=[fh(e.shape),...hh(e.shape)],s={dtype:e.dtype,shape:r,dataId:e.dataId},o=[fh(t),...hh(t)],a=new xW(o,r),i=!0,l=[r],u=n.runWebGLProgram(a,[s],e.dtype,l,i);return{dataId:u.dataId,shape:t,dtype:u.dtype}}function qe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{shape:o}=r,a=n,i=E.sizeFromShape(s.shape),l=E.inferFromImplicitShape(o,i),u=E.sizeFromShape(l);E.assert(i===u,()=> ` The new shape ( $ { l } ) has $ { u } elements and the old shape ( $ { s . shape } ) has $ { i } elements . The new shape and old shape must have the same number of elements . ` );let c=a.texData.get(s.dataId);return c.isPacked&&!tA(s.shape,l)&&!(c.texture!==null&&tA(c.shape,l))?DTe(s,l,a):(a.incRef(s.dataId),{dataId:s.dataId,shape:l,dtype:s.dtype})}var $ Te={kernelName:Xg,backendName:"webgl",kernelFunc:qe}, $ W=class{constructor(e,t){this.variableNames=["x"];let{windowSize:n,batchSize:r,inSize:s,outSize:o}=e;this.outputShape=[r,o];let a=Math.floor(n/4)*4,i=n%4,l="sumValue += dot(values, ones);";if(t!=null){let c=1/t;l= ` sumValue += dot ( values * $ { E . isInt ( c ) ? c . toPrecision ( 2 ) : c } , ones ) ; ` }let u="";s%n>0&&(u= `
if ( inIdx < 0 || inIdx >= $ { s } ) {
2021-09-11 16:29:31 +02:00
return 0.0 ;
}
` ),this.userCode= `
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
2021-09-01 00:22:16 +02:00
2021-09-11 16:29:31 +02:00
float getValue ( int batch , int inIdx ) {
$ { u }
return getX ( batch , inIdx ) ;
}
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = outIdx * $ { n } ;
2021-09-01 00:22:16 +02:00
2021-09-11 16:29:31 +02:00
float sumValue = 0.0 ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int i = 0 ; i < $ { a } ; i += 4 ) {
2021-09-11 16:29:31 +02:00
int inIdx = inOffset + i ;
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { l }
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
int inIdx = inOffset + $ { a } ;
2021-09-11 17:17:13 +02:00
if ( $ { i === 1 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 ( getValue ( batch , inIdx ) , 0.0 , 0.0 , 0.0 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { l }
2021-09-11 17:17:13 +02:00
} else if ( $ { i === 2 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) , 0.0 , 0.0 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { l }
2021-09-11 17:17:13 +02:00
} else if ( $ { i === 3 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) , 0.0 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { l }
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( sumValue ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},FTe=class{constructor(e,t){this.variableNames=["x"];let{windowSize:n,batchSize:r,inSize:s,outSize:o}=e;this.outputShape=[r,o];let a="0.0",i="";t==="prod"?a="1.0":t==="min"?(a="1.0 / 1e-20",i="min"):t==="max"&&(a="-1.0 / 1e-20",i="max");let l= ` $ { t } ( $ { t } ( $ { t } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;t==="sum"?l="sumValue":t==="prod"?l="prodValue":t==="all"?l="allValue":t==="any"&&(l="anyValue");let u=Math.floor(n/4)*4,c=n%4,p= `
2021-09-11 16:29:31 +02:00
if ( $ { t === "sum" } ) {
sumValue += dot ( values , ones ) ;
} else if ( $ { t === "prod" } ) {
vec2 tmp = vec2 ( values [ 0 ] , values [ 1 ] ) * vec2 ( values [ 2 ] , values [ 3 ] ) ;
prodValue *= tmp [ 0 ] * tmp [ 1 ] ;
2021-09-05 22:42:11 +02:00
} else {
2021-09-11 17:17:13 +02:00
minMaxValue = $ { i } ( values , minMaxValue ) ;
2021-09-11 16:29:31 +02:00
if ( $ { t === "min" } || $ { t === "max" } ) {
2021-09-11 17:17:13 +02:00
minMaxValue = $ { i } ( values , minMaxValue ) ;
2021-09-11 16:29:31 +02:00
bvec4 isNaN = isnan ( values ) ;
if ( isNaN . r || isNaN . g || isNaN . b || isNaN . a ) {
minMaxValue = vec4 ( NAN ) ;
2021-09-05 22:42:11 +02:00
}
}
2021-09-01 00:24:30 +02:00
}
2021-10-04 22:29:15 +02:00
` ,d="vec4";t==="all"?(a="1.0",p= `
2021-09-11 16:29:31 +02:00
bool reducedAllValue = all ( values ) ;
float floatedReducedAllValue = float ( reducedAllValue ) ;
allValue = float ( allValue >= 1.0 && floatedReducedAllValue >= 1.0 ) ;
2021-10-04 22:29:15 +02:00
` ,d="bvec4"):t==="any"&&(a="0.0",p= `
2021-09-11 16:29:31 +02:00
bool reducedAnyValue = any ( values ) ;
float floatedReducedAnyValue = float ( reducedAnyValue ) ;
anyValue = float ( anyValue >= 1.0 || floatedReducedAnyValue >= 1.0 ) ;
2021-10-04 22:29:15 +02:00
` ,d="bvec4");let f="";s%n>0&&(f= `
if ( inIdx < 0 || inIdx >= $ { s } ) {
2021-09-11 16:29:31 +02:00
return initializationValue ;
}
` ),this.userCode= `
2021-10-04 22:29:15 +02:00
const float initializationValue = $ { a } ;
2021-09-11 16:29:31 +02:00
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
float getValue ( int batch , int inIdx ) {
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 16:29:31 +02:00
return getX ( batch , inIdx ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
2021-09-05 22:42:11 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
2021-09-01 00:24:30 +02:00
int batch = coords [ 0 ] ;
2021-09-11 16:29:31 +02:00
int outIdx = coords [ 1 ] ;
int inOffset = outIdx * $ { n } ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
vec4 minMaxValue = vec4 ( $ { a } ) ;
2021-09-11 16:29:31 +02:00
float prodValue = 1.0 ;
float sumValue = 0.0 ;
float allValue = 1.0 ;
float anyValue = 0.0 ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
for ( int i = 0 ; i < $ { u } ; i += 4 ) {
int inIdx = inOffset + i ;
2021-10-04 22:29:15 +02:00
$ { d } values = $ { d } (
2021-09-11 16:29:31 +02:00
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-05 22:42:11 +02:00
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int inIdx = inOffset + $ { u } ;
2021-09-11 17:17:13 +02:00
if ( $ { c === 1 } ) {
2021-10-04 22:29:15 +02:00
$ { d } values = $ { d } (
2021-09-11 16:29:31 +02:00
getValue ( batch , inIdx ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-11 17:17:13 +02:00
} else if ( $ { c === 2 } ) {
2021-10-04 22:29:15 +02:00
$ { d } values = $ { d } (
2021-09-11 16:29:31 +02:00
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
initializationValue ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-11 17:17:13 +02:00
} else if ( $ { c === 3 } ) {
2021-10-04 22:29:15 +02:00
$ { d } values = $ { d } (
2021-09-11 16:29:31 +02:00
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-11 16:29:31 +02:00
}
setOutput ( $ { l } ) ;
}
2021-10-04 22:29:15 +02:00
` }};function OTe(e){let t=[];for(;t.length===0||t[t.length-1].outSize!==1;){let n=t.length?t[t.length-1].outSize:e[1],r=G.computeOptimalWindowSize(n);t.push({inSize:n,windowSize:r,outSize:Math.ceil(n/r)})}return t}function xh(e,t,n,r){let s=OTe(e.shape),o=e;for(let a=0;a<s.length;a++){let{inSize:i,windowSize:l,outSize:u}=s[a],c,p;n==="mean"?c=a===0?new $ W({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u},i):new $ W({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u}):c=new FTe({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u},n),p=o,o=r.runWebGLProgram(c,[o],t),p.dataId!==e.dataId&&r.disposeIntermediateTensorInfo(p)}return o}var PTe=class{constructor(e,t){this.variableNames=["A"];let n=new Array(e.length);for(let o=0;o<n.length;o++)n[o]=e[t[o]];this.outputShape=n,this.rank=n.length;let r=Dn(this.rank),s=MTe(t);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r } resRC = getOutputCoords ( ) ;
setOutput ( getA ( $ { s } ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function MTe(e){let t=e.length;if(t>6)throw Error( ` Transpose for rank $ { t } is not yet supported ` );let n=["resRC.x","resRC.y","resRC.z","resRC.w","resRC.u","resRC.v"],r=new Array(t);for(let s=0;s<e.length;s++)r[e[s]]=n[s];return r.join()}var zTe=class{constructor(e,t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0;let n=new Array(e.length);for(let u=0;u<n.length;u++)n[u]=e[t[u]];if(this.outputShape=n,this.rank=n.length,this.rank>6)throw Error( ` Packed transpose for rank $ { this . rank } is not yet supported . ` );let r=Dn(this.rank),s=gW("rc",this.rank),o=new Array(this.rank);for(let u=0;u<t.length;u++)o[t[u]]=s[u];let a= ` vec2 ( $ { o . slice ( - 2 ) . join ( ) } ) ` ,i= ` ++ $ { s [ this . rank - 1 ] } < $ { n [ this . rank - 1 ] } ` ,l= ` getChannel ( getA ( $ { o . join ( ) } ) , $ { a } ) ` ;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r } rc = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0. ) ;
result [ 0 ] = $ { l } ;
2021-09-11 17:17:13 +02:00
if ( $ { i } ) {
2021-09-11 16:29:31 +02:00
result [ 1 ] = $ { l } ;
}
2021-10-04 22:29:15 +02:00
-- $ { s [ this . rank - 1 ] } ;
if ( ++ $ { s [ this . rank - 2 ] } < $ { n [ this . rank - 2 ] } ) {
2021-09-11 16:29:31 +02:00
result [ 2 ] = $ { l } ;
2021-09-11 17:17:13 +02:00
if ( $ { i } ) {
2021-09-11 16:29:31 +02:00
result [ 3 ] = $ { l } ;
}
}
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function z5(e,t,n){let r=be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new zTe(e.shape,t):new PTe(e.shape,t);return n.runWebGLProgram(r,[e],e.dtype)}function LTe(e,t,n,r){let s=t,o=e.shape.length,a=E.parseAxisParam(s,e.shape),i=a,l=G.getAxesPermutation(i,o),u=l!=null,c=e;u&&(c=z5(e,l,r),i=G.getInnerMostAxes(i.length,o)),G.assertAxesAreInnerMostDims("sum",i,o);let[p,d]=G.computeOutAndReduceShapes(c.shape,i),f=p;n&&(f=G.expandShapeToKeepDim(p,a));let h=E.sizeFromShape(d),g=E.sizeFromShape(e.shape)/h,x=qe({inputs:{x:c},attrs:{shape:[g,h]},backend:r}),y=Ww(e.dtype),b=xh(x,y,"sum",r),A=qe({inputs:{x:b},attrs:{shape:f},backend:r});return r.disposeIntermediateTensorInfo(x),r.disposeIntermediateTensorInfo(b),u&&r.disposeIntermediateTensorInfo(c),A}function L5(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r;return LTe(s,o,a,n)}var BTe={kernelName:_f,backendName:"webgl",kernelFunc:L5};function Hs(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{perm:o}=r,a=n,i=s.shape.length,l=new Array(i);for(let c=0;c<l.length;c++)l[c]=s.shape[o[c]];let u;if(a.shouldExecuteOnCPU([s])){let p=a.texData.get(s.dataId).values,d=OR(p,s.shape,s.dtype,o,l);u=a.makeTensorInfo(l,s.dtype);let f=a.texData.get(u.dataId);f.values=d}else u=z5(s,o,a);return u}var VTe={kernelName:Of,backendName:"webgl",kernelFunc:Hs},FW=1e3;function B5({a:e,b:t,transposeA:n,transposeB:r,backend:s,bias:o=null,preluActivationWeights:a=null,leakyreluAlpha:i=0,activation:l=null}){let u=e.shape.length,c=t.shape.length,p=n?e.shape[u-2]:e.shape[u-1],d=r?t.shape[c-1]:t.shape[c-2],f=n?e.shape[u-1]:e.shape[u-2],h=r?t.shape[c-2]:t.shape[c-1],m=e.shape.slice(0,-2),g=t.shape.slice(0,-2),x=E.sizeFromShape(m),y=E.sizeFromShape(g),b=x===y||x===1||y===1;E.assert(u>=2&&c>=2&&b,()=> ` Error in matMul : the input batch dimensions must either be the same or at least one input batch dimension must be 1. Got input batch dimensions of ( $ { m } ) and ( $ { g } ) . ` );let v=(x>y?e.shape.slice(0,-2):t.shape.slice(0,-2)).concat([f,h]);E.assert(p===d,()=> ` Error in matMul : inner shapes ( $ { p } ) and ( $ { d } ) of Tensors with shapes $ { e . shape } and $ { t . shape } and transposeA = $ { n } and transposeB = $ { r } must match . ` );let w=n?[x,p,f]:[x,f,p],k=r?[y,h,d]:[y,d,h],I=qe({inputs:{x:e},backend:s,attrs:{shape:w}}),N=qe({inputs:{x:t},backend:s,attrs:{shape:k}}),T=[I,N],_=Math.max(x,y),R=n?I.shape[1]:I.shape[2],S=o!=null,F=a!=null,B=l==="leakyrelu",W=l!=null?M5(l,!0):null,q=S||F||B||W!=null,K;if((f===1||h===1)&&R>FW&&q===!1){let Q=I,ee=N;n&&(Q=Hs({inputs:{x:I},backend:s,attrs:{perm:[0,2,1]}}),T.push(Q)),r&&(ee=Hs({inputs:{x:N},backend:s,attrs:{perm:[0,2,1]}}),T.push(ee));let ue=h!==1,re=h===1,de=Q;ue&&(de=qe({inputs:{x:Q},backend:s,attrs:{shape:[_,R,1]}}),T.push(de));let fe=h===1?2:1,me=ee;re&&(me=qe({inputs:{x:ee},backend:s,attrs:{shape:[_,1,R]}}),T.push(me));let xe=PR({inputs:{a:de,b:me},backend:s});K=L5({inputs:{x:xe},backend:s,attrs:{axis:fe,keepDims:!0}}),T.push(xe)}else{let Q= $ a(e.dtype,t.dtype),ee=new _W(w,k,[_,f,h],n,r,S,W,F,B),ue=[I,N];if(o!=null&&ue.push(o),F&&ue.push(a),B){let re=s.makeTensorInfo([],"float32",E.createScalarValue(i,"float32"));ue.push(re),T.push(re)}K=s.runWebGLProgram(ee,ue,Q)}let J=qe({inputs:{x:K},backend:s,attrs:{shape:v}});T.push(K);for(let Q of T)s.disposeIntermediateTensorInfo(Q);return J}function WTe(e){let{inputs:t,backend:n,attrs:r}=e,{a:s,b:o,bias:a,preluActivationWeights:i}=t,{transposeA:l,transposeB:u,activation:c,leakyreluAlpha:p}=r;return B5({a:s,b:o,transposeA:l,transposeB:u,backend:n,bias:a,preluActivationWeights:i,leakyreluAlpha:p,activation:c})}var GTe={kernelName:Pf,backendName:"webgl",kernelFunc:WTe},OW="return abs(x);";function UTe(e){let{inputs:t,backend:n}=e,{x:r}=t;if(n.shouldExecuteOnCPU([r])&&r.dtype!=="complex64"){let o=n.texData.get(r.dataId),a=hW(o.values);return n.makeTensorInfo(r.shape,r.dtype,a)}let s;return be().getBool("WEBGL_PACK_UNARY_OPERATIONS")?s=new Z0(r.shape,OW):s=new rp(r.shape,OW),n.runWebGLProgram(s,[r],r.dtype)}var jTe={kernelName:cg,backendName:"webgl",kernelFunc:UTe},HTe=Ti+ `
2021-09-11 16:29:31 +02:00
if ( abs ( x ) > 1. ) {
return NAN ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
return acos ( x ) ;
2021-10-04 22:29:15 +02:00
` ,qTe=tn({opSnippet:HTe}),KTe={kernelName:pg,backendName:"webgl",kernelFunc:qTe},XTe=Ti+ `
2021-09-11 16:29:31 +02:00
if ( x < 1.0 ) return NAN ;
2021-10-04 22:29:15 +02:00
return log ( x + sqrt ( x * x - 1.0 ) ) ; ` ,ZTe=tn({opSnippet:XTe}),YTe={kernelName:dg,backendName:"webgl",kernelFunc:ZTe},PW="return a + b;",JTe=ks({opSnippet:PW,packedOpSnippet:PW,supportsComplex:!0,cpuKernelImpl:iNe}),QTe={kernelName: $ c,backendName:"webgl",kernelFunc:JTe},e_e=class{constructor(e,t){this.outputShape=[],this.outputShape=e,this.variableNames=t.map((s,o)=> ` T$ { o } ` );let n=[];this.variableNames.forEach(s=>{n.push( ` float v$ { s } = get$ { s } AtOutCoords ( ) ; ` )});let r=this.variableNames.map(s=> ` v$ { s } ` ).join(" + ");this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
$ { n . join ( `
` )}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
float result = $ { r } ;
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-01 00:24:30 +02:00
}
2021-10-04 22:29:15 +02:00
` }},t_e=class{constructor(e,t){this.outputShape=[],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=e,this.variableNames=t.map((s,o)=> ` T$ { o } ` );let n=[];this.variableNames.forEach(s=>{n.push( ` vec4 v$ { s } = get$ { s } AtOutCoords ( ) ; ` )});let r=this.variableNames.map(s=> ` v$ { s } ` ).join(" + ");this.userCode= `
2021-09-05 22:42:11 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
$ { n . join ( `
` )}
2021-10-04 22:29:15 +02:00
vec4 result = $ { r } ;
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function V5(e){let{inputs:t,backend:n}=e,r=t;if(r.length===1)return Xo({inputs:{x:r[0]},backend:n});if(r.length>be().get("WEBGL_MAX_TEXTURES_IN_SHADER")){let l=Math.floor(r.length/2),u=V5({inputs:r.slice(0,l),backend:n}),c=V5({inputs:r.slice(l),backend:n});return V5({inputs:[u,c],backend:n})}let s=r.map(l=>l.dtype).reduce((l,u)=> $ a(l,u)),o=r.map(l=>l.shape),i=be().getBool("WEBGL_PACK")?new t_e(r[0].shape,o):new e_e(r[0].shape,o);return n.runWebGLProgram(i,r,s)}var n_e={kernelName:Ld,backendName:"webgl",kernelFunc:V5};function r_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=E.parseAxisParam(o,s.shape),u=l,c=G.getAxesPermutation(u,i),p=s;c!=null&&(p=Hs({inputs:{x:s},backend:n,attrs:{perm:c}}),u=G.getInnerMostAxes(u.length,i)),G.assertAxesAreInnerMostDims("all",u,i);let[d,f]=G.computeOutAndReduceShapes(p.shape,u),h=E.sizeFromShape(f),m=qe({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=xh(m,m.dtype,"all",n),x;if(a){let y=G.expandShapeToKeepDim(d,l);x=qe({inputs:{x:g},backend:n,attrs:{shape:y}})}else x=qe({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var s_e={kernelName:fg,backendName:"webgl",kernelFunc:r_e};function o_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=E.parseAxisParam(o,s.shape),u=l,c=G.getAxesPermutation(u,i),p=s;c!=null&&(p=Hs({inputs:{x:s},backend:n,attrs:{perm:c}}),u=G.getInnerMostAxes(u.length,i)),G.assertAxesAreInnerMostDims("any",u,i);let[d,f]=G.computeOutAndReduceShapes(p.shape,u),h=E.sizeFromShape(f),m=qe({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=xh(m,m.dtype,"any",n),x;if(a){let y=G.expandShapeToKeepDim(d,l);x=qe({inputs:{x:g},backend:n,attrs:{shape:y}})}else x=qe({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var a_e={kernelName:hg,backendName:"webgl",kernelFunc:o_e},i_e=class{constructor(e,t,n){this.variableNames=["A"];let{windowSize:r,batchSize:s,outSize:o}=e;n||this.variableNames.push("bestIndicesA"),this.outputShape=[s,o];let a=t==="max"?">":"<",i=n?"inOffset + i;":"round(getBestIndicesA(batch, inOffset + i));";this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
2021-10-04 22:29:15 +02:00
int inOffset = outIdx * $ { r } ;
2021-09-11 16:29:31 +02:00
int bestIndex = inOffset ;
float bestValue = getA ( batch , bestIndex ) ;
2021-10-04 22:29:15 +02:00
for ( int i = 0 ; i < $ { r } ; i ++ ) {
2021-09-11 17:17:13 +02:00
int inIdx = $ { i } ;
2021-09-11 16:29:31 +02:00
float candidate = getA ( batch , inIdx ) ;
2021-10-04 22:29:15 +02:00
if ( candidate $ { a } bestValue ) {
2021-09-11 16:29:31 +02:00
bestValue = candidate ;
bestIndex = inIdx ;
}
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( float ( bestIndex ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},l_e=class{constructor(e,t,n,r){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,E.assert(e.length>2,()=> ` Packed arg$ { n . charAt ( 0 ) . toUpperCase ( ) + n . slice ( 1 ) } supports only inputs with rank above 2. ` );let s=e[e.length-1],o=Math.ceil(s/t);this.outputShape=e.slice(0,-1),o>1&&this.outputShape.push(o),r||this.variableNames.push("bestIndicesA");let a=this.outputShape,i=a.length,l=Dn(i),u=js("coords",i),c,p;if(o===1){p=i+1;let k=Dn(p);c= `
$ { k } sourceLocR = $ { k } ( $ { u . join ( ) } , 0 ) ;
2021-09-11 17:17:13 +02:00
++ $ { u [ i - 1 ] } ;
2021-10-04 22:29:15 +02:00
$ { k } sourceLocG = $ { k } ( $ { u . join ( ) } , 0 ) ;
2021-09-11 17:17:13 +02:00
++ $ { u [ i - 2 ] } ;
2021-10-04 22:29:15 +02:00
$ { k } sourceLocA = $ { k } ( $ { u . join ( ) } , 0 ) ;
2021-09-11 17:17:13 +02:00
-- $ { u [ i - 1 ] } ;
2021-10-04 22:29:15 +02:00
$ { k } sourceLocB = $ { k } ( $ { u . join ( ) } , 0 ) ;
-- $ { u [ i - 2 ] } ; ` }else p=i,c= `
2021-09-11 16:29:31 +02:00
$ { l } sourceLocR = coords ;
2021-09-11 17:17:13 +02:00
++ $ { u [ i - 1 ] } ;
2021-09-11 16:29:31 +02:00
$ { l } sourceLocG = coords ;
2021-09-11 17:17:13 +02:00
++ $ { u [ i - 2 ] } ;
2021-09-11 16:29:31 +02:00
$ { l } sourceLocA = coords ;
2021-09-11 17:17:13 +02:00
-- $ { u [ i - 1 ] } ;
2021-09-11 16:29:31 +02:00
$ { l } sourceLocB = coords ;
2021-10-04 22:29:15 +02:00
-- $ { u [ i - 2 ] } ; ` ;let d=["x","y","z","w","u","v"].slice(0,p),f="."+d[p-1],h=d.map(k=>"int "+k),m=js("sourceLocR",p-1).concat("inIdx.r"),g=js("sourceLocG",p-1).concat("inIdx.g"),x=js("sourceLocB",p-1).concat("inIdx.b"),y=js("sourceLocA",p-1).concat("inIdx.a"),b=n==="max"?"greaterThan":"lessThan",A=r?"": `
inIdx = round ( vec4 ( getBestIndicesAChannel ( $ { m . join ( ) } ) ,
2021-09-11 16:29:31 +02:00
getBestIndicesAChannel ( $ { g . join ( ) } ) ,
2021-10-04 22:29:15 +02:00
getBestIndicesAChannel ( $ { x . join ( ) } ) ,
2021-09-11 17:17:13 +02:00
getBestIndicesAChannel ( $ { y . join ( ) } ) ) ) ; ` ,v= ` vec4 (
2021-10-04 22:29:15 +02:00
getAChannel ( $ { m . join ( ) } ) ,
2021-09-11 16:29:31 +02:00
hasNextCol ? getAChannel ( $ { g . join ( ) } ) : 0. ,
2021-10-04 22:29:15 +02:00
hasNextRow ? getAChannel ( $ { x . join ( ) } ) : 0. ,
hasNextRow && hasNextCol ? getAChannel ( $ { y . join ( ) } ) : 0. ) ` ,w=r?"": `
float getBestIndicesAChannel ( $ { h . join ( ) } ) {
return getChannel ( getBestIndicesA ( $ { d . join ( ) } ) ,
vec2 ( $ { d . slice ( - 2 ) . join ( ) } ) ) ;
2021-09-11 16:29:31 +02:00
} ` ;this.userCode= `
2021-10-04 22:29:15 +02:00
float getAChannel ( $ { h . join ( ) } ) {
return getChannel ( getA ( $ { d . join ( ) } ) ,
vec2 ( $ { d . slice ( - 2 ) . join ( ) } ) ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
$ { w }
2021-09-05 22:42:11 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
$ { l } coords = getOutputCoords ( ) ;
2021-10-04 22:29:15 +02:00
bool hasNextCol = $ { u [ i - 1 ] } < $ { a [ i - 1 ] - 1 } ;
bool hasNextRow = $ { u [ i - 2 ] } < $ { a [ i - 2 ] - 1 } ;
2021-09-11 17:17:13 +02:00
$ { c }
2021-10-04 22:29:15 +02:00
ivec4 srcIdx = ivec4 ( sourceLocR$ { f } , sourceLocG$ { f } ,
sourceLocB$ { f } , sourceLocA$ { f } ) * $ { t } ;
2021-09-11 16:29:31 +02:00
ivec4 inIdx = srcIdx ;
vec4 bestIndex = vec4 ( inIdx ) ;
2021-09-11 17:17:13 +02:00
vec4 bestValue = $ { v } ;
2021-09-11 16:29:31 +02:00
for ( int i = 0 ; i < $ { t } ; i ++ ) {
inIdx = srcIdx ;
2021-10-04 22:29:15 +02:00
$ { A }
2021-09-11 17:17:13 +02:00
vec4 candidate = $ { v } ;
2021-09-11 16:29:31 +02:00
bvec4 nan = isnan ( candidate ) ;
bvec4 replace = bvec4 (
2021-10-04 22:29:15 +02:00
vec4 ( $ { b } ( candidate , bestValue ) ) * ( vec4 ( 1.0 ) - vec4 ( nan ) ) ) ;
2021-09-11 16:29:31 +02:00
bestValue = vec4 ( replace . x ? candidate . x : bestValue . x ,
replace . y ? candidate . y : bestValue . y ,
replace . z ? candidate . z : bestValue . z ,
replace . w ? candidate . w : bestValue . w ) ;
bestIndex = mix ( bestIndex , vec4 ( inIdx ) , vec4 ( replace ) ) ;
srcIdx ++ ;
}
setOutput ( bestIndex ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function MW(e,t,n,r=null){let s=t.shape[0],o=t.shape[1];r!=null&&(s=r.shape[0],o=r.shape[1]);let a=G.computeOptimalWindowSize(o),i={windowSize:a,inSize:o,batchSize:s,outSize:Math.ceil(o/a)},l=new i_e(i,n,r==null),u=[t];r!=null&&u.push(r);let c=e.runWebGLProgram(l,u,"int32");if(c.shape[1]===1)return c;let p=MW(e,t,n,c);return e.disposeIntermediateTensorInfo(c),p}function zW(e,t,n,r=null){let s=r!=null?r.shape:t.shape,o=s[s.length-1],a=G.computeOptimalWindowSize(o),i=new l_e(s,a,n,r==null),l=r==null?[t]:[t,r],u=e.runWebGLProgram(i,l,"int32");if(u.shape.length===t.shape.length){let c=zW(e,t,n,u);return e.disposeIntermediateTensorInfo(u),c}return u}function LW(e,t,n,r){let s=[n];if(G.assertAxesAreInnerMostDims("arg"+r.charAt(0).toUpperCase()+r.slice(1),s,t.shape.length),!be().getBool("WEBGL_PACK_REDUCE")||t.shape.length<=2){let o=[],a=e.texData.get(t.dataId),i=a!==null&&a.isPacked,l=t;i&&(l=e.unpackTensor(t),o.push(l));let[u,c]=G.computeOutAndReduceShapes(l.shape,s),p=E.sizeFromShape(c),d=qe({inputs:{x:l},backend:e,attrs:{shape:[-1,p]}});o.push(d);let f=MW(e,d,r);o.push(f);let h=qe({inputs:{x:f},backend:e,attrs:{shape:u}});return o.forEach(m=>e.disposeIntermediateTensorInfo(m)),h}return zW(e,t,r)}function u_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o}=r,a=E.parseAxisParam(o,s.shape),i=G.getAxesPermutation(a,s.shape.length),l=s,u=[];i!=null&&(l=Hs({inputs:{x:s},backend:n,attrs:{perm:i}}),u.push(l),a=G.getInnerMostAxes(a.length,l.shape.length)),G.assertAxesAreInnerMostDims("argMax",[a[0]],l.shape.length);let c=LW(n,l,a[0],"max");return u.forEach(p=>n.disposeIntermediateTensorInfo(p)),c}var c_e={kernelName:Bd,backendName:"webgl",kernelFunc:u_e};function p_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o}=r,a=E.parseAxisParam(o,s.shape),i=G.getAxesPermutation(a,s.shape.length),l=s,u=[];i!=null&&(l=Hs({inputs:{x:s},backend:n,attrs:{perm:i}}),u.push(l),a=G.getInnerMostAxes(a.length,l.shape.length)),G.assertAxesAreInnerMostDims("argMin",[a[0]],l.shape.length);let c=LW(n,l,a[0],"min");return u.forEach(p=>n.disposeIntermediateTensorInfo(p)),c}var d_e={kernelName:P1,backendName:"webgl",kernelFunc:p_e},f_e=Ti+ `
2021-09-11 16:29:31 +02:00
if ( abs ( x ) > 1. ) {
2021-09-05 22:42:11 +02:00
return NAN ;
}
2021-09-11 16:29:31 +02:00
return asin ( x ) ;
2021-10-04 22:29:15 +02:00
` ,h_e=tn({opSnippet:f_e}),m_e={kernelName:mg,backendName:"webgl",kernelFunc:h_e},g_e=Ti+"return log(x + sqrt(x * x + 1.0));",x_e=tn({opSnippet:g_e}),y_e={kernelName:gg,backendName:"webgl",kernelFunc:x_e},b_e=Ti+ `
2021-09-11 16:29:31 +02:00
return atan ( x ) ;
2021-10-04 22:29:15 +02:00
` ,A_e=tn({opSnippet:b_e}),v_e={kernelName:xg,backendName:"webgl",kernelFunc:A_e},w_e=_Te+ `
2021-09-11 16:29:31 +02:00
return atan ( a , b ) ;
2021-10-04 22:29:15 +02:00
` ,k_e= `
2021-09-11 16:29:31 +02:00
vec4 result = atan ( a , b ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
2021-10-04 22:29:15 +02:00
` +ETe+ `
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,I_e=ks({opSnippet:w_e,packedOpSnippet:k_e}),C_e={kernelName:bg,backendName:"webgl",kernelFunc:I_e},S_e=Ti+ `
2021-09-11 16:29:31 +02:00
if ( ( x < - 1.0 ) || ( x > 1.0 ) ) return NAN ;
2021-10-04 22:29:15 +02:00
return ( log ( 1.0 + x ) - log ( 1.0 - x ) ) / 2.0 ; ` ,N_e=tn({opSnippet:S_e}),T_e={kernelName:yg,backendName:"webgl",kernelFunc:N_e},sA=class{constructor(e,t,n,r=!1,s=!1){if(this.variableNames=["x"],t==="avg"&&n)throw new Error("Cannot compute positions for average pool.");let o=e.filterWidth,a=e.strideHeight,i=e.strideWidth,l=e.dilationHeight,u=e.dilationWidth,c=e.effectiveFilterHeight,p=e.effectiveFilterWidth,d=e.padInfo.top,f=e.padInfo.left;this.outputShape=e.outShape;let h=t==="avg",m= ` ( ( batch * $ { e . inHeight } + xR ) * $ { e . inWidth } + xC ) * $ { e . inChannels } + d ` ,g= ` ( xR * $ { e . inWidth } + xC ) * $ { e . inChannels } + d ` ,x="0.0";if(h||(x="-1.0 / 1e-20"),n){let k=">=";this.userCode= `
const ivec2 strides = ivec2 ( $ { a } , $ { i } ) ;
const ivec2 pads = ivec2 ( $ { d } , $ { f } ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d = coords [ 3 ] ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
ivec2 xRCCorner = coords . yz * strides - pads ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
// max/min x(?, ?, d) to get y(yR, yC, d).
// ? = to be determined
float minMaxValue = 0.0 ;
float minMaxValueFound = 0.0 ;
int minMaxPosition = 0 ;
float avgValue = 0.0 ;
2021-09-01 00:24:30 +02:00
2021-09-11 17:17:13 +02:00
for ( int wR = 0 ; wR < $ { c } ;
2021-09-11 16:29:31 +02:00
wR += $ { l } ) {
int xR = xRCorner + wR ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { p } ;
2021-09-11 16:29:31 +02:00
wC += $ { u } ) {
int xC = xCCorner + wC ;
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
}
float value = getX ( batch , xR , xC , d ) ;
// If a min / max value has already been found, use it. If not,
// use the current value.
float currMinMaxValue = mix (
value , minMaxValue , minMaxValueFound ) ;
2021-10-04 22:29:15 +02:00
if ( value $ { k } currMinMaxValue ) {
2021-09-11 16:29:31 +02:00
minMaxValue = value ;
minMaxValueFound = 1.0 ;
2021-10-04 22:29:15 +02:00
minMaxPosition = $ { r ? s ? m : g : ` wR * ${ p } + wC ` } ;
2021-09-11 16:29:31 +02:00
}
}
}
setOutput ( float ( minMaxPosition ) ) ;
}
2021-10-04 22:29:15 +02:00
` ;return}let y="max",b= ` $ { t } ( $ { t } ( $ { t } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;t==="avg"&&(b="avgValue / count");let A=Math.floor(o/4)*4,v=o%4,w= `
if ( $ { h } ) {
2021-09-11 16:29:31 +02:00
avgValue += dot ( values , ones ) ;
2021-09-05 22:42:11 +02:00
} else {
2021-09-11 16:29:31 +02:00
minMaxValue = $ { y } ( values , minMaxValue ) ;
}
` ;this.userCode= `
2021-10-04 22:29:15 +02:00
const ivec2 strides = ivec2 ( $ { a } , $ { i } ) ;
const ivec2 pads = ivec2 ( $ { d } , $ { f } ) ;
const float initializationValue = $ { x } ;
2021-09-11 16:29:31 +02:00
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
float count = 0.0 ;
float getValue ( int batch , int xR , int xC , int d ) {
if ( xC < 0 || xC >= $ { e . inWidth } ) {
return initializationValue ;
}
count += 1.0 ;
return getX ( batch , xR , xC , d ) ;
2021-08-14 17:16:26 +02:00
}
2021-09-01 00:24:30 +02:00
void main ( ) {
2021-09-05 22:42:11 +02:00
ivec4 coords = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
int batch = coords [ 0 ] ;
2021-09-05 22:42:11 +02:00
int d = coords [ 3 ] ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
ivec2 xRCCorner = coords . yz * strides - pads ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// max/min x(?, ?, d) to get y(yR, yC, d).
// ? = to be determined
2021-10-04 22:29:15 +02:00
vec4 minMaxValue = vec4 ( $ { x } ) ;
2021-09-11 16:29:31 +02:00
float avgValue = 0.0 ;
count = 0.0 ;
2021-09-01 00:24:30 +02:00
2021-09-11 17:17:13 +02:00
for ( int wR = 0 ; wR < $ { c } ;
2021-09-11 16:29:31 +02:00
wR += $ { l } ) {
int xR = xRCorner + wR ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { A } ; wC += 4 ) {
2021-09-11 16:29:31 +02:00
int xC = xCCorner + wC * $ { u } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { u } , d ) ,
getValue ( batch , xR , xC + 2 * $ { u } , d ) ,
getValue ( batch , xR , xC + 3 * $ { u } , d )
) ;
2021-10-04 22:29:15 +02:00
$ { w }
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
int xC = xCCorner + $ { A } ;
2021-09-11 17:17:13 +02:00
if ( $ { v === 1 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { w }
2021-09-11 17:17:13 +02:00
} else if ( $ { v === 2 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { u } , d ) ,
initializationValue ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { w }
2021-09-11 17:17:13 +02:00
} else if ( $ { v === 3 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { u } , d ) ,
getValue ( batch , xR , xC + 2 * $ { u } , d ) ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { w }
2021-09-11 16:29:31 +02:00
}
}
2021-10-04 22:29:15 +02:00
setOutput ( $ { b } ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},MR=class{constructor(e,t,n,r=!1,s=!1){if(this.variableNames=["x"],t==="avg"&&n)throw new Error("Cannot compute positions for average pool.");let o=e.filterWidth,a=e.strideDepth,i=e.strideHeight,l=e.strideWidth,u=e.dilationDepth,c=e.dilationHeight,p=e.dilationWidth,d=e.effectiveFilterDepth,f=e.effectiveFilterHeight,h=e.effectiveFilterWidth,m=e.padInfo.front,g=e.padInfo.top,x=e.padInfo.left;this.outputShape=e.outShape;let y=t==="avg",b="0.0";if(y||(b="-1.0 / 1e-20"),n){let N=">=";this.userCode= `
2021-09-11 16:29:31 +02:00
const ivec3 strides =
2021-10-04 22:29:15 +02:00
ivec3 ( $ { a } , $ { i } , $ { l } ) ;
const ivec3 pads = ivec3 ( $ { m } , $ { g } , $ { x } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 xCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xDCorner = xCorner . x ;
int xRCorner = xCorner . y ;
int xCCorner = xCorner . z ;
// max/min x(?, ?, ?, ch) to get y(yD, yR, yC, ch).
// ? = to be determined
float minMaxValue = 0.0 ;
float minMaxValueFound = 0.0 ;
int minMaxPosition = 0 ;
2021-10-04 22:29:15 +02:00
for ( int wD = 0 ; wD < $ { d } ;
2021-09-11 16:29:31 +02:00
wD += $ { u } ) {
int xD = xDCorner + wD ;
if ( xD < 0 || xD >= $ { e . inDepth } ) {
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { f } ;
2021-09-11 17:17:13 +02:00
wR += $ { c } ) {
2021-09-11 16:29:31 +02:00
int xR = xRCorner + wR ;
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { h } ;
wC += $ { p } ) {
2021-09-11 16:29:31 +02:00
int xC = xCCorner + wC ;
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
}
float value = getX ( batch , xD , xR , xC , ch ) ;
// If a min / max value has already been found, use it. If not,
// use the current value.
float currMinMaxValue = mix (
value , minMaxValue , minMaxValueFound ) ;
2021-10-04 22:29:15 +02:00
if ( value $ { N } currMinMaxValue ) {
2021-09-11 16:29:31 +02:00
minMaxValue = value ;
minMaxValueFound = 1.0 ;
2021-10-04 22:29:15 +02:00
minMaxPosition = $ { r ? s ? ` (((batch * ${ e . inDepth } + xD) * ${ e . inHeight } + xR) * ${ e . inWidth } + xC) * ${ e . inChannels } + ch ` : ` ((xD * ${ e . inHeight } + xR) * ${ e . inWidth } + xC) * ${ e . inChannels } + ch ` : ` wD * ${ f } * ${ h } +
wR * $ { h } + wC ` };
2021-09-11 16:29:31 +02:00
}
}
}
}
setOutput ( float ( minMaxPosition ) ) ;
}
2021-10-04 22:29:15 +02:00
` ;return}let A="max",v= ` $ { t } ( $ { t } ( $ { t } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;t==="avg"&&(v="avgValue / count");let w=Math.floor(o/4)*4,k=o%4,I= `
2021-09-11 16:29:31 +02:00
if ( $ { y } ) {
avgValue += dot ( values , ones ) ;
2021-09-05 22:42:11 +02:00
} else {
2021-10-04 22:29:15 +02:00
minMaxValue = $ { A } ( values , minMaxValue ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
` ;this.userCode= `
const ivec3 strides =
2021-10-04 22:29:15 +02:00
ivec3 ( $ { a } , $ { i } , $ { l } ) ;
const ivec3 pads = ivec3 ( $ { m } , $ { g } , $ { x } ) ;
const float initializationValue = $ { b } ;
2021-09-11 16:29:31 +02:00
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
float count = 0.0 ;
float getValue ( int batch , int xD , int xR , int xC , int ch ) {
if ( xC < 0 || xC >= $ { e . inWidth } ) {
return initializationValue ;
}
count += 1.0 ;
return getX ( batch , xD , xR , xC , ch ) ;
2021-09-01 00:22:16 +02:00
}
2021-09-05 22:42:11 +02:00
2021-09-01 00:24:30 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
ivec3 xCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xDCorner = xCorner . x ;
int xRCorner = xCorner . y ;
int xCCorner = xCorner . z ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
// max/min x(?, ?, ?, d) to get y(yD, yR, yC, ch).
// ? = to be determined
2021-10-04 22:29:15 +02:00
vec4 minMaxValue = vec4 ( $ { b } ) ;
2021-09-11 16:29:31 +02:00
float avgValue = 0.0 ;
count = 0.0 ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wD = 0 ; wD < $ { d } ;
2021-09-11 16:29:31 +02:00
wD += $ { u } ) {
int xD = xDCorner + wD ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xD < 0 || xD >= $ { e . inDepth } ) {
continue ;
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { f } ;
2021-09-11 17:17:13 +02:00
wR += $ { c } ) {
2021-09-11 16:29:31 +02:00
int xR = xRCorner + wR ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { w } ; wC += 4 ) {
int xC = xCCorner + wC * $ { p } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
2021-10-04 22:29:15 +02:00
getValue ( batch , xD , xR , xC + $ { p } , ch ) ,
getValue ( batch , xD , xR , xC + 2 * $ { p } , ch ) ,
getValue ( batch , xD , xR , xC + 3 * $ { p } , ch )
2021-09-11 16:29:31 +02:00
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { I }
2021-09-11 16:29:31 +02:00
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
int xC = xCCorner + $ { w } ;
if ( $ { k === 1 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { I }
} else if ( $ { k === 2 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
2021-10-04 22:29:15 +02:00
getValue ( batch , xD , xR , xC + $ { p } , ch ) ,
2021-09-11 16:29:31 +02:00
initializationValue ,
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { I }
} else if ( $ { k === 3 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
2021-10-04 22:29:15 +02:00
getValue ( batch , xD , xR , xC + $ { p } , ch ) ,
getValue ( batch , xD , xR , xC + 2 * $ { p } , ch ) ,
2021-09-11 16:29:31 +02:00
initializationValue
) ;
2021-10-04 22:29:15 +02:00
$ { I }
2021-09-11 16:29:31 +02:00
}
}
2021-09-11 17:17:13 +02:00
setOutput ( $ { v } ) ;
2021-09-11 16:29:31 +02:00
}
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function __e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t;j0(s,"avgPool");let{filterSize:o,strides:a,pad:i,dimRoundingMode:l}=r,u=1;E.assert(G.eitherStridesOrDilationsAreOne(a,u),()=> ` Error in avgPool : Either strides or dilations must be 1. Got strides $ { a } and dilations '${u}' ` );let c=G.computePool2DInfo(s.shape,o,a,u,i,l);if(c.filterWidth===1&&c.filterHeight===1&&E.arraysEqual(c.inShape,c.outShape))return Xo({inputs:{x:s},backend:n});let p=new sA(c,"avg",!1);return n.runWebGLProgram(p,[s],"float32")}var E_e={kernelName:Vd,backendName:"webgl",kernelFunc:__e};function R_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{filterSize:o,strides:a,pad:i,dimRoundingMode:l,dataFormat:u}=r,c=[1,1,1],p=G.computePool3DInfo(s.shape,o,a,c,i,l,u),d=new MR(p,"avg",!1);return n.runWebGLProgram(d,[s],"float32")}var D_e={kernelName:M1,backendName:"webgl",kernelFunc:R_e}, $ _e=class{constructor(e){this.variableNames=["dy"],this.outputShape=e.inShape;let t=e.filterHeight,n=e.filterWidth,r=e.strideHeight,s=e.strideWidth,o=e.dilationHeight,a=e.dilationWidth,i=e.effectiveFilterHeight,l=e.effectiveFilterWidth,u=i-1-e.padInfo.top,c=l-1-e.padInfo.left,p=1/(t*n);this.userCode= `
2021-09-11 17:17:13 +02:00
const ivec2 pads = ivec2 ( $ { u } , $ { c } ) ;
2021-10-04 22:29:15 +02:00
const float avgMultiplier = float ( $ { p } ) ;
2021-09-11 16:29:31 +02:00
2021-08-14 17:17:51 +02:00
void main ( ) {
2021-09-01 00:24:30 +02:00
ivec4 coords = getOutputCoords ( ) ;
2021-09-05 22:42:11 +02:00
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
ivec2 dyRCCorner = coords . yz - pads ;
int dyRCorner = dyRCCorner . x ;
int dyCCorner = dyRCCorner . y ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
// Convolve dy(?, ?, d) with pos mask(:, :, d) to get dx(xR, xC, d).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
2021-09-11 17:17:13 +02:00
for ( int wR = 0 ; wR < $ { i } ;
2021-10-04 22:29:15 +02:00
wR += $ { o } ) {
float dyR = float ( dyRCorner + wR ) / $ { r } . 0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
for ( int wC = 0 ; wC < $ { l } ;
2021-10-04 22:29:15 +02:00
wC += $ { a } ) {
float dyC = float ( dyCCorner + wC ) / $ { s } . 0 ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
float dyValue = getDy ( b , idyR , idyC , d ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
dotProd += dyValue * avgMultiplier ;
}
}
setOutput ( dotProd ) ;
}
2021-10-04 22:29:15 +02:00
` }},F_e=class{constructor(e){this.variableNames=["dy"],this.outputShape=e.inShape;let t=e.filterDepth,n=e.filterHeight,r=e.filterWidth,s=e.strideDepth,o=e.strideHeight,a=e.strideWidth,i=e.dilationDepth,l=e.dilationHeight,u=e.dilationWidth,c=e.effectiveFilterDepth,p=e.effectiveFilterHeight,d=e.effectiveFilterWidth,f=c-1-e.padInfo.front,h=p-1-e.padInfo.top,m=d-1-e.padInfo.left,g=1/(t*n*r);this.userCode= `
const ivec3 pads = ivec3 ( $ { f } , $ { h } , $ { m } ) ;
2021-09-11 16:29:31 +02:00
const float avgMultiplier = float ( $ { g } ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyDCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
// Convolve dy(?, ?, ?, d) with pos mask(:, :, :, ch) to get
// dx(xD, xR, xC, ch).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
2021-09-11 17:17:13 +02:00
for ( int wD = 0 ; wD < $ { c } ;
wD += $ { i } ) {
2021-10-04 22:29:15 +02:00
float dyD = float ( dyDCorner + wD ) / $ { s } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyD < 0.0 || dyD >= $ { e . outDepth } . 0 || fract ( dyD ) > 0.0 ) {
2021-09-01 00:24:30 +02:00
continue ;
}
2021-09-11 16:29:31 +02:00
int idyD = int ( dyD ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { p } ;
2021-09-11 16:29:31 +02:00
wR += $ { l } ) {
2021-10-04 22:29:15 +02:00
float dyR = float ( dyRCorner + wR ) / $ { o } . 0 ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
2021-09-01 00:24:30 +02:00
continue ;
}
2021-09-11 16:29:31 +02:00
int idyR = int ( dyR ) ;
2021-08-14 17:17:51 +02:00
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { d } ;
2021-09-11 16:29:31 +02:00
wC += $ { u } ) {
2021-10-04 22:29:15 +02:00
float dyC = float ( dyCCorner + wC ) / $ { a } . 0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
float dyValue = getDy ( batch , idyD , idyR , idyC , ch ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
dotProd += dyValue * avgMultiplier ;
2021-09-05 22:42:11 +02:00
}
}
}
2021-09-11 16:29:31 +02:00
setOutput ( dotProd ) ;
}
2021-10-04 22:29:15 +02:00
` }};function O_e(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o,{filterSize:i,strides:l,pad:u,dimRoundingMode:c}=r,p=[1,1,1],d=G.computePool3DInfo(a.shape,i,l,p,u,c),f=new F_e(d);return n.runWebGLProgram(f,[s],a.dtype)}var P_e={kernelName:tw,backendName:"webgl",kernelFunc:O_e};function M_e(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o;j0([s,o],"avgPoolGrad");let{filterSize:i,strides:l,pad:u}=r,c=G.computePool2DInfo(a.shape,i,l,1,u),p=new $ _e(c);return n.runWebGLProgram(p,[s],a.dtype)}var z_e={kernelName:ew,backendName:"webgl",kernelFunc:M_e};function L_e(e){let{inputs:t,backend:n,attrs:r}=e,{a:s,b:o}=t,{transposeA:a,transposeB:i}=r;return B5({a:s,b:o,transposeA:a,transposeB:i,backend:n})}var B_e={kernelName:Wd,backendName:"webgl",kernelFunc:L_e},V_e=class{constructor(e,t,n,r,s,o){this.outputShape=[],this.variableNames=["x","mean","variance"],G.assertAndGetBroadcastShape(e,t),G.assertAndGetBroadcastShape(e,n);let a="0.0";r!=null&&(G.assertAndGetBroadcastShape(e,r),this.variableNames.push("offset"),a="getOffsetAtOutCoords()");let i="1.0";s!=null&&(G.assertAndGetBroadcastShape(e,s),this.variableNames.push("scale"),i="getScaleAtOutCoords()"),this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
float x = getXAtOutCoords ( ) ;
float mean = getMeanAtOutCoords ( ) ;
float variance = getVarianceAtOutCoords ( ) ;
2021-10-04 22:29:15 +02:00
float offset = $ { a } ;
2021-09-11 17:17:13 +02:00
float scale = $ { i } ;
2021-10-04 22:29:15 +02:00
float inv = scale * inversesqrt ( variance + float ( $ { o } ) ) ;
2021-09-11 16:29:31 +02:00
setOutput ( dot ( vec3 ( x , - mean , offset ) , vec3 ( inv , inv , 1 ) ) ) ;
}
2021-10-04 22:29:15 +02:00
` }},W_e=class{constructor(e,t,n,r,s,o){this.packedInputs=!0,this.packedOutput=!0,this.variableNames=["x","mean","variance"],G.assertAndGetBroadcastShape(e,t),G.assertAndGetBroadcastShape(e,n);let a="vec4(0.0)";r!=null&&(G.assertAndGetBroadcastShape(e,r),this.variableNames.push("offset"),a="getOffsetAtOutCoords()");let i="vec4(1.0)";s!=null&&(G.assertAndGetBroadcastShape(e,s),this.variableNames.push("scale"),i="getScaleAtOutCoords()"),this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
vec4 offset = $ { a } ;
2021-09-11 17:17:13 +02:00
vec4 scale = $ { i } ;
2021-09-01 00:22:16 +02:00
2021-09-11 16:29:31 +02:00
vec4 x = getXAtOutCoords ( ) ;
vec4 mean = getMeanAtOutCoords ( ) ;
vec4 variance = getVarianceAtOutCoords ( ) ;
2021-10-04 22:29:15 +02:00
vec4 inv = scale * inversesqrt ( variance + vec4 ( $ { o } ) ) ;
2021-09-11 16:29:31 +02:00
setOutput ( ( x - mean ) * inv + offset ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},G_e=({inputs:e,backend:t,attrs:n})=>{let{x:r,mean:s,variance:o,offset:a,scale:i}=e;E.assert(s.shape.length===o.shape.length,()=>"Batch normalization gradient requires mean and variance to have equal ranks."),E.assert(a==null||s.shape.length===a.shape.length,()=>"Batch normalization gradient requires mean and offset to have equal ranks."),E.assert(i==null||s.shape.length===i.shape.length,()=>"Batch normalization gradient requires mean and scale to have equal ranks.");let{varianceEpsilon:l}=n;l==null&&(l=.001);let u=[r,s,o],c=null;a!=null&&(c=a.shape,u.push(a));let p=null;i!=null&&(p=i.shape,u.push(i));let d=be().getBool("WEBGL_PACK_NORMALIZATION")?new W_e(r.shape,s.shape,o.shape,c,p,l):new V_e(r.shape,s.shape,o.shape,c,p,l);return t.runWebGLProgram(d,u,u[0].dtype)},U_e={kernelName:nf,backendName:"webgl",kernelFunc:G_e},j_e=class{constructor(e){this.variableNames=["source"],this.outputShape=e,this.rank=e.length;let t=Dn(this.rank);this.customUniforms=[{name:"start",arrayIndex:this.rank,type:"int"}];let n=H_e(this.rank),r,s=e.map((o,a)=> ` sourceLoc . $ { zR [ a ] } = start [ $ { a } ] + coords . $ { zR [ a ] } ; ` );r= `
2021-09-11 16:29:31 +02:00
$ { t } sourceLoc ;
$ { t } coords = getOutputCoords ( ) ;
2021-10-04 22:29:15 +02:00
$ { s . join ( `
2021-09-11 16:29:31 +02:00
` )}
` ,this.userCode= `
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r }
2021-09-11 16:29:31 +02:00
setOutput ( getSource ( $ { n } ) ) ;
}
2021-10-04 22:29:15 +02:00
` }},zR=["x","y","z","w","u","v"];function H_e(e){if(e===1)return"sourceLoc";if(e<=6)return zR.slice(0,e).map(t=>"sourceLoc."+t).join(",");throw Error( ` Slicing for rank $ { e } is not yet supported ` )}var q_e=class{constructor(e){this.variableNames=["source"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=e,this.rank=e.length,this.customUniforms=[{name:"start",arrayIndex:this.rank,type:"int"}];let t=Dn(this.rank),n=js("coords",this.rank),r=js("sourceLoc",this.rank),s=this.rank===1?"sourceLoc": ` vec2 ( $ { r . slice ( - 2 ) . join ( ) } ) ` ,o= ` getChannel ( getSource ( $ { r . join ( ) } ) , $ { s } ) ` ,a= `
result . x = $ { o } ;
2021-09-11 16:29:31 +02:00
if ( ++ $ { n [ this . rank - 1 ] } < $ { e [ this . rank - 1 ] } ) {
2021-10-04 22:29:15 +02:00
++ $ { r [ this . rank - 1 ] } ;
result . y = $ { o } ;
-- $ { r [ this . rank - 1 ] } ;
2021-09-11 16:29:31 +02:00
}
2021-09-11 17:17:13 +02:00
` ,i=this.rank===1?"": `
2021-09-11 16:29:31 +02:00
-- $ { n [ this . rank - 1 ] } ;
if ( ++ $ { n [ this . rank - 2 ] } < $ { e [ this . rank - 2 ] } ) {
2021-10-04 22:29:15 +02:00
++ $ { r [ this . rank - 2 ] } ;
result . z = $ { o } ;
2021-09-11 16:29:31 +02:00
if ( ++ $ { n [ this . rank - 1 ] } < $ { e [ this . rank - 1 ] } ) {
2021-10-04 22:29:15 +02:00
++ $ { r [ this . rank - 1 ] } ;
result . w = $ { o } ;
2021-09-11 16:29:31 +02:00
}
}
` ,l=this.rank<=4? ` sourceLoc = coords +
2021-10-04 22:29:15 +02:00
$ { t } ( $ { e . map ( ( u , c ) => ` start[ ${ c } ] ` ) . join ( ) } ) ; ` :e.map((u,c)=> ` $ { r [ c ] } = $ { n [ c ] } + start [ $ { c } ] ; ` ).join( `
2021-09-11 16:29:31 +02:00
` );this.userCode= `
void main ( ) {
$ { t } coords = getOutputCoords ( ) ;
$ { t } sourceLoc ;
$ { l }
vec4 result = vec4 ( 0. ) ;
2021-10-04 22:29:15 +02:00
$ { a }
2021-09-11 17:17:13 +02:00
$ { i }
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function K_e(e,t,n,r){let s=r.texData.get(e.dataId),o=r.makeTensorInfo(n,e.dtype),a=r.texData.get(o.dataId);Object.assign(a,s),a.refCount=1,a.shape=n,a.dtype=e.dtype;let i=Bs.computeFlatOffset(t,E.computeStrides(e.shape));s.slice&&(i+=s.slice.flatOffset),a.slice={flatOffset:i,origDataId:s.slice&&s.slice.origDataId||e.dataId};let l=r.dataRefCount.get(a.slice.origDataId)||1;return r.dataRefCount.set(a.slice.origDataId,l+1),o}function Q0(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,size:a}=r,[i,l]=Bs.parseSliceParams(s,o,a);if(Bs.assertParamsValid(s,i,l),E.sizeFromShape(l)===0)return n.makeTensorInfo(l,s.dtype,[]);if(n.shouldExecuteOnCPU([s])||s.dtype==="string"){let p=n.texData.get(s.dataId),d= $ Ne(p.values,i,l,s.shape,s.dtype);return n.makeTensorInfo(l,s.dtype,d)}let{isPacked:u}=n.texData.get(s.dataId),c=Bs.isSliceContinous(s.shape,i,l);if(u||!c){let p=be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new q_e(l):new j_e(l),d=[i];return n.runWebGLProgram(p,[s],s.dtype,d)}return n.uploadToGPU(s.dataId),K_e(s,i,l,n)}var X_e={kernelName:Qg,backendName:"webgl",kernelFunc:Q0},Z_e=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockShape:o,crops:a}=r;E.assert(s.shape.length<=4,()=>"batchToSpaceND for rank > 4 with a WebGL backend not implemented yet");let i=o.reduce((y,b)=>y*b),l=G.getReshaped(s.shape,o,i),u=G.getPermuted(l.length,o.length),c=G.getReshapedPermuted(s.shape,o,i),p=G.getSliceBeginCoords(a,o.length),d=G.getSliceSize(c,a,o.length),f=[],h=qe({inputs:{x:s},backend:n,attrs:{shape:l}}),m=Hs({inputs:{x:h},backend:n,attrs:{perm:u}}),g=qe({inputs:{x:m},backend:n,attrs:{shape:c}}),x=Q0({inputs:{x:g},backend:n,attrs:{begin:p,size:d}});return f.push(h),f.push(m),f.push(g),f.forEach(y=>n.disposeIntermediateTensorInfo(y)),x},Y_e={kernelName:Ag,backendName:"webgl",kernelFunc:Z_e};function J_e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,weights:o}=t,{size:a}=r,i=n.readSync(s.dataId),l=n.readSync(o.dataId),u=fW(i,l,o.dtype,o.shape,a);return n.makeTensorInfo([a],o.dtype,u)}var Q_e={kernelName:nw,backendName:"webgl",kernelFunc:J_e},eEe="return float(a != b);",BW=ks({opSnippet:eEe,cpuKernelImpl:TNe,dtype:"bool"}),tEe={kernelName:Vg,backendName:"webgl",kernelFunc:BW};function oA(e){let{inputs:t,backend:n}=e,{input:r}=t,s=n.texData.get(r.dataId);return Xo({inputs:{x:s.complexTensorInfos.real},backend:n})}var nEe={kernelName:Iw,backendName:"webgl",kernelFunc:oA},rEe="return float(int(x));";function sEe(e,t){let n=new rp(e.shape,rEe),r=t.runWebGLProgram(n,[e],"int32");return{dataId:r.dataId,shape:r.shape,dtype:r.dtype}}function LR(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{dtype:o}=r;if(o==="complex64"){if(s.dtype==="complex64")return Xo({inputs:{x:s},backend:n});let a=xr(s.shape),i=LR({inputs:{x:s},backend:n,attrs:{dtype:"float32"}}),l=sp({inputs:{real:i,imag:a},backend:n});return a.dispose(),n.disposeIntermediateTensorInfo(i),l}if(s.dtype==="complex64"){let a=oA({inputs:{input:s},backend:n}),i=LR({inputs:{x:a},backend:n,attrs:{dtype:o}});return n.disposeIntermediateTensorInfo(a),i}if(!E.hasEncodingLoss(s.dtype,o)){let a=Xo({inputs:{x:s},backend:n});return{dataId:a.dataId,shape:a.shape,dtype:o}}if(o==="int32")return sEe(s,n);if(o==="bool"){let a=n.makeTensorInfo([],"bool",E.getTypedArrayFromDType("bool",1)),l=BW({inputs:{a:s,b:a},backend:n});return n.disposeIntermediateTensorInfo(a),l}throw new Error( ` Error in Cast : failed to cast $ { s . dtype } to $ { o } ` )}var oEe={kernelName:Gd,backendName:"webgl",kernelFunc:LR},VW="return ceil(x);",aEe=tn({opSnippet:VW,packedOpSnippet:VW,cpuKernelImpl:uNe}),iEe={kernelName:Ud,backendName:"webgl",kernelFunc:aEe},lEe=class{constructor(e){this.variableNames=["A"],this.customUniforms=[{name:"minVal",type:"float"},{name:"maxVal",type:"float"}],this.outputShape=e,this.userCode= `
2021-09-01 00:24:30 +02:00
2021-09-05 22:42:11 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
float value = getAAtOutCoords ( ) ;
if ( isnan ( value ) ) {
setOutput ( value ) ;
return ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( clamp ( value , minVal , maxVal ) ) ;
}
2021-10-04 22:29:15 +02:00
` }},uEe=class{constructor(e){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"minVal",type:"float"},{name:"maxVal",type:"float"}],this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
vec4 value = getAAtOutCoords ( ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( any ( isnan ( value ) ) ) {
setOutput ( value ) ;
return ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( clamp ( value , vec4 ( minVal ) , vec4 ( maxVal ) ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function cEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{clipValueMin:o,clipValueMax:a}=r,i;be().getBool("WEBGL_PACK_CLIP")?i=new uEe(s.shape):i=new lEe(s.shape);let l=[[o],[a]];return n.runWebGLProgram(i,[s],s.dtype,l)}var pEe={kernelName:Fc,backendName:"webgl",kernelFunc:cEe},dEe=class{constructor(e){this.variableNames=["real","imag"],this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
float re = abs ( getRealAtOutCoords ( ) ) ;
float im = abs ( getImagAtOutCoords ( ) ) ;
float mx = max ( re , im ) ;
// sadly the length function in glsl is not underflow-safe
// (at least not on Intel GPUs). So the safe solution is
// to ensure underflow-safety in all cases.
setOutput (
mx == 0.0 ? 0.0 : mx * length ( vec2 ( 1 , min ( re , im ) / mx ) )
) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function WW(e,t){return{dataId:t.dataId,dtype:t.dtype,shape:e.shape}}function fEe(e){let{inputs:t,backend:n}=e,{x:r}=t,s=n.texData.get(r.dataId),o=new dEe(r.shape),a=[WW(r,s.complexTensorInfos.real),WW(r,s.complexTensorInfos.imag)];return n.runWebGLProgram(o,a,a[0].dtype)}var hEe={kernelName:z1,backendName:"webgl",kernelFunc:fEe},mEe=class{constructor(e){this.outputShape=[],this.outputShape=G.computeOutShape(e,1),this.variableNames=e.map((o,a)=> ` T$ { a } ` );let t=new Array(e.length-1);t[0]=e[0][1];for(let o=1;o<t.length;o++)t[o]=t[o-1]+e[o][1];let n=[ ` if ( yC < $ { t [ 0 ] } ) setOutput ( getT0 ( yR , yC ) ) ; ` ];for(let o=1;o<t.length;o++){let a=t[o-1];n.push( ` else if ( yC < $ { t [ o ] } ) setOutput ( getT$ { o } ( yR , yC - $ { a } ) ) ; ` )}let r=t.length,s=t[t.length-1];n.push( ` else setOutput ( getT$ { r } ( yR , yC - $ { s } ) ) ; ` ),this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int yR = coords . x ;
int yC = coords . y ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { n . join ( `
` )}
}
2021-10-04 22:29:15 +02:00
` }},gEe=class{constructor(e,t){this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[],this.outputShape=G.computeOutShape(e,t);let n=this.outputShape,r=n.length,s=Dn(r),o=js("coords",r),a=["x","y","z","w","u","v"].slice(0,r);this.variableNames=e.map((h,m)=> ` T$ { m } ` );let i=new Array(e.length-1);i[0]=e[0][t];for(let h=1;h<i.length;h++)i[h]=i[h-1]+e[h][t];let l=a[t],u=a.slice(-2),c=a.join(),p= ` if ( $ { l } < $ { i [ 0 ] } ) {
2021-09-11 16:29:31 +02:00
return getChannel (
2021-09-11 17:17:13 +02:00
getT0 ( $ { c } ) , vec2 ( $ { u . join ( ) } ) ) ;
2021-10-04 22:29:15 +02:00
} ` ;for(let h=1;h<i.length;h++){let m=i[h-1];p+= `
if ( $ { l } < $ { i [ h ] } && $ { l } >= $ { i [ h - 1 ] } ) {
2021-09-11 16:29:31 +02:00
return getChannel (
2021-10-04 22:29:15 +02:00
getT$ { h } ( $ { W5 ( a , l , m ) } ) ,
vec2 ( $ { W5 ( u , l , m ) } ) ) ;
} ` }let d=i.length,f=i[i.length-1];p+= `
2021-09-11 16:29:31 +02:00
return getChannel (
2021-10-04 22:29:15 +02:00
getT$ { d } ( $ { W5 ( a , l , f ) } ) ,
vec2 ( $ { W5 ( u , l , f ) } ) ) ; ` ,this.userCode= `
float getValue ( $ { a . map ( h => "int " + h ) } ) {
$ { p }
2021-09-05 22:42:11 +02:00
}
2021-09-01 00:24:30 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } coords = getOutputCoords ( ) ;
vec4 result = vec4 ( getValue ( $ { o } ) , 0. , 0. , 0. ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { o [ r - 1 ] } = $ { o [ r - 1 ] } + 1 ;
if ( $ { o [ r - 1 ] } < $ { n [ r - 1 ] } ) {
result . g = getValue ( $ { o } ) ;
2021-09-11 16:29:31 +02:00
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { o [ r - 2 ] } = $ { o [ r - 2 ] } + 1 ;
if ( $ { o [ r - 2 ] } < $ { n [ r - 2 ] } ) {
result . a = getValue ( $ { o } ) ;
2021-09-11 16:29:31 +02:00
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { o [ r - 1 ] } = $ { o [ r - 1 ] } - 1 ;
if ( $ { o [ r - 2 ] } < $ { n [ r - 2 ] } &&
$ { o [ r - 1 ] } < $ { n [ r - 1 ] } ) {
result . b = getValue ( $ { o } ) ;
2021-09-11 16:29:31 +02:00
}
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }};function W5(e,t,n){let r=e.indexOf(t);return e.map((o,a)=>a===r? ` $ { o } - $ { n } ` :o).join()}function G5(e){let{inputs:t,backend:n}=e,{input:r}=t,s=n.texData.get(r.dataId);return Xo({inputs:{x:s.complexTensorInfos.imag},backend:n})}var xEe={kernelName:xw,backendName:"webgl",kernelFunc:G5};function ex(e,t,n){let r=e[0].dtype;if(r==="complex64"){let c=e.map(m=>oA({inputs:{input:m},backend:n})),p=e.map(m=>G5({inputs:{input:m},backend:n})),d=ex(c,t,n),f=ex(p,t,n),h=sp({inputs:{real:d,imag:f},backend:n});return c.forEach(m=>n.disposeIntermediateTensorInfo(m)),p.forEach(m=>n.disposeIntermediateTensorInfo(m)),n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}let s=n.shouldExecuteOnCPU(e);if(r==="string"&&(s=!0),s){let c=e.map(x=>{let y=E.sizeFromShape(x.shape.slice(t));return qe({inputs:{x},backend:n,attrs:{shape:[-1,y]}})}),p=c.map(x=>({vals:n.readSync(x.dataId),shape:x.shape})),d=G.computeOutShape(c.map(x=>x.shape),1),f=c[0].shape[0]===1,h=cNe(p,d,r,f),m=G.computeOutShape(e.map(x=>x.shape),t),g=n.makeTensorInfo(m,r,h);return c.forEach(x=>n.disposeIntermediateTensorInfo(x)),g}if(e.length>be().getNumber("WEBGL_MAX_TEXTURES_IN_SHADER")){let c=Math.floor(e.length/2),p=ex(e.slice(0,c),t,n),d=ex(e.slice(c),t,n),f=ex([p,d],t,n);return n.disposeIntermediateTensorInfo(p),n.disposeIntermediateTensorInfo(d),f}if(be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")&&e[0].shape.length>1){let c=new gEe(e.map(p=>p.shape),t);return n.runWebGLProgram(c,e,r)}let{tensors2D:o,outShape:a}=yEe(e,t,n),i=new mEe(o.map(c=>c.shape)),l=n.runWebGLProgram(i,o,r);o.forEach(c=>n.disposeIntermediateTensorInfo(c));let u=qe({inputs:{x:l},attrs:{shape:a},backend:n});return n.disposeIntermediateTensorInfo(l),u}function yEe(e,t,n){let r=G.computeOutShape(e.map(o=>o.shape),t);return{tensors2D:e.map(o=>qe({inputs:{x:o},attrs:{shape:[-1,E.sizeFromShape(o.shape.slice(t))]},backend:n})),outShape:r}}function GW(e){let{inputs:t,backend:n,attrs:r}=e,{axis:s}=r,o=E.parseAxisParam(s,t[0].shape)[0],a=G.computeOutShape(t.map(u=>u.shape),o);if(E.sizeFromShape(a)===0)return n.makeTensorInfo(a,t[0].dtype,[]);let i=t.filter(u=>E.sizeFromShape(u.shape)>0);if(i.length===1)return Xo({inputs:{x:i[0]},backend:n});let l=i.map(u=>u.shape);return G.assertParamsConsistent(l,o),ex(i,o,n)}var bEe={kernelName:vg,backendName:"webgl",kernelFunc:GW},UW=class{constructor(e,t=!1,n=null,r=!1,s=!1){this.variableNames=["x","W"],this.outputShape=e.outShape;let o=e.padInfo.top,a=e.padInfo.left,i=e.strideHeight,l=e.strideWidth,u=e.dilationHeight,c=e.dilationWidth,p=e.filterHeight,d=e.filterWidth,f=Math.floor(e.inChannels/4)*4,h=e.inChannels%4,m=e.dataFormat==="channelsLast",g=m?1:2,x=m?2:3,y=m?3:1,b="",A="";n&&(r?b= ` float activation ( float a ) {
2021-09-11 16:29:31 +02:00
float b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { n }
2021-10-04 22:29:15 +02:00
} ` :s?b= ` float activation ( float a ) {
2021-09-11 16:29:31 +02:00
float b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { n }
2021-10-04 22:29:15 +02:00
} ` :b= `
2021-09-11 16:29:31 +02:00
float activation ( float x ) {
$ { n }
}
2021-10-04 22:29:15 +02:00
` ,A="result = activation(result);");let v=t?"result += getBiasAtOutCoords();":"";t&&this.variableNames.push("bias"),r&&this.variableNames.push("preluActivationWeights"),s&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
$ { b }
2021-09-05 22:42:11 +02:00
2021-09-11 17:17:13 +02:00
const ivec2 strides = ivec2 ( $ { i } , $ { l } ) ;
2021-10-04 22:29:15 +02:00
const ivec2 pads = ivec2 ( $ { o } , $ { a } ) ;
2021-09-05 22:42:11 +02:00
2021-09-01 00:24:30 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
int batch = coords [ 0 ] ;
int d2 = coords [ $ { y } ] ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
ivec2 xRCCorner =
2021-10-04 22:29:15 +02:00
ivec2 ( coords [ $ { g } ] , coords [ $ { x } ] ) * strides - pads ;
2021-09-11 16:29:31 +02:00
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Convolve x(?, ?, d1) with w(:, :, d1, d2) to get y(yR, yC, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { p } ; wR ++ ) {
2021-09-11 16:29:31 +02:00
int xR = xRCorner + wR * $ { u } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
2021-09-01 00:24:30 +02:00
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { d } ; wC ++ ) {
2021-09-11 17:17:13 +02:00
int xC = xCCorner + wC * $ { c } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xC < 0 || xC >= $ { e . inWidth } ) {
2021-09-01 00:24:30 +02:00
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int d1 = 0 ; d1 < $ { f } ; d1 += 4 ) {
2021-09-11 16:29:31 +02:00
vec4 wValues = vec4 (
getW ( wR , wC , d1 , d2 ) ,
getW ( wR , wC , d1 + 1 , d2 ) ,
getW ( wR , wC , d1 + 2 , d2 ) ,
getW ( wR , wC , d1 + 3 , d2 )
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { m } ) {
2021-09-11 16:29:31 +02:00
vec4 xValues = vec4 (
getX ( batch , xR , xC , d1 ) ,
getX ( batch , xR , xC , d1 + 1 ) ,
getX ( batch , xR , xC , d1 + 2 ) ,
getX ( batch , xR , xC , d1 + 3 )
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec4 xValues = vec4 (
getX ( batch , d1 , xR , xC ) ,
getX ( batch , d1 + 1 , xR , xC ) ,
getX ( batch , d1 + 2 , xR , xC ) ,
getX ( batch , d1 + 3 , xR , xC )
) ;
dotProd += dot ( xValues , wValues ) ;
}
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { h === 1 } ) {
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { m } ) {
2021-09-11 16:29:31 +02:00
dotProd +=
2021-10-04 22:29:15 +02:00
getX ( batch , xR , xC , $ { f } ) *
getW ( wR , wC , $ { f } , d2 ) ;
2021-09-11 16:29:31 +02:00
} else {
dotProd +=
2021-10-04 22:29:15 +02:00
getX ( batch , $ { f } , xR , xC ) *
getW ( wR , wC , $ { f } , d2 ) ;
2021-09-11 16:29:31 +02:00
}
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
} else if ( $ { h === 2 } ) {
2021-09-11 16:29:31 +02:00
vec2 wValues = vec2 (
2021-10-04 22:29:15 +02:00
getW ( wR , wC , $ { f } , d2 ) ,
getW ( wR , wC , $ { f } + 1 , d2 )
2021-09-11 16:29:31 +02:00
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { m } ) {
2021-09-11 16:29:31 +02:00
vec2 xValues = vec2 (
2021-10-04 22:29:15 +02:00
getX ( batch , xR , xC , $ { f } ) ,
getX ( batch , xR , xC , $ { f } + 1 )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec2 xValues = vec2 (
2021-10-04 22:29:15 +02:00
getX ( batch , $ { f } , xR , xC ) ,
getX ( batch , $ { f } + 1 , xR , xC )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
}
2021-10-04 22:29:15 +02:00
} else if ( $ { h === 3 } ) {
2021-09-11 16:29:31 +02:00
vec3 wValues = vec3 (
2021-10-04 22:29:15 +02:00
getW ( wR , wC , $ { f } , d2 ) ,
getW ( wR , wC , $ { f } + 1 , d2 ) ,
getW ( wR , wC , $ { f } + 2 , d2 )
2021-09-11 16:29:31 +02:00
) ;
2021-10-04 22:29:15 +02:00
if ( $ { m } ) {
2021-09-11 16:29:31 +02:00
vec3 xValues = vec3 (
2021-10-04 22:29:15 +02:00
getX ( batch , xR , xC , $ { f } ) ,
getX ( batch , xR , xC , $ { f } + 1 ) ,
getX ( batch , xR , xC , $ { f } + 2 )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec3 xValues = vec3 (
2021-10-04 22:29:15 +02:00
getX ( batch , $ { f } , xR , xC ) ,
getX ( batch , $ { f } + 1 , xR , xC ) ,
getX ( batch , $ { f } + 2 , xR , xC )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
}
2021-09-01 00:24:30 +02:00
}
}
2021-06-07 02:34:29 +02:00
}
2021-09-11 16:29:31 +02:00
float result = dotProd ;
$ { v }
2021-10-04 22:29:15 +02:00
$ { A }
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},AEe=class{constructor(e){this.variableNames=["x","W"],this.outputShape=e.outShape;let t=e.padInfo.front,n=e.padInfo.top,r=e.padInfo.left,s=e.strideDepth,o=e.strideHeight,a=e.strideWidth,i=e.dilationDepth,l=e.dilationHeight,u=e.dilationWidth,c=e.filterDepth,p=e.filterHeight,d=e.filterWidth,f=Math.floor(e.inChannels/4)*4,h=e.inChannels%4;this.userCode= `
const ivec3 strides = ivec3 ( $ { s } , $ { o } , $ { a } ) ;
const ivec3 pads = ivec3 ( $ { t } , $ { n } , $ { r } ) ;
2021-09-11 16:29:31 +02:00
2021-09-01 00:24:30 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d2 = coords . u ;
ivec3 xFRCCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xFCorner = xFRCCorner . x ;
int xRCorner = xFRCCorner . y ;
int xCCorner = xFRCCorner . z ;
// Convolve x(?, ?, ?, d1) with w(:, :, :, d1, d2) to get
// y(yF, yR, yC, d2). ? = to be determined. : = across all
// values in that axis.
float dotProd = 0.0 ;
2021-09-11 17:17:13 +02:00
for ( int wF = 0 ; wF < $ { c } ; wF ++ ) {
int xF = xFCorner + wF * $ { i } ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
if ( xF < 0 || xF >= $ { e . inDepth } ) {
continue ;
}
2021-09-05 22:42:11 +02:00
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { p } ; wR ++ ) {
2021-09-11 16:29:31 +02:00
int xR = xRCorner + wR * $ { l } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
2021-09-05 22:42:11 +02:00
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { d } ; wC ++ ) {
2021-09-11 16:29:31 +02:00
int xC = xCCorner + wC * $ { u } ;
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int d1 = 0 ; d1 < $ { f } ; d1 += 4 ) {
2021-09-11 16:29:31 +02:00
vec4 xValues = vec4 (
getX ( batch , xF , xR , xC , d1 ) ,
getX ( batch , xF , xR , xC , d1 + 1 ) ,
getX ( batch , xF , xR , xC , d1 + 2 ) ,
getX ( batch , xF , xR , xC , d1 + 3 )
) ;
vec4 wValues = vec4 (
getW ( wF , wR , wC , d1 , d2 ) ,
getW ( wF , wR , wC , d1 + 1 , d2 ) ,
getW ( wF , wR , wC , d1 + 2 , d2 ) ,
getW ( wF , wR , wC , d1 + 3 , d2 )
) ;
dotProd += dot ( xValues , wValues ) ;
}
2021-10-04 22:29:15 +02:00
if ( $ { h === 1 } ) {
2021-09-11 16:29:31 +02:00
dotProd +=
2021-10-04 22:29:15 +02:00
getX ( batch , xF , xR , xC , $ { f } ) *
getW ( wF , wR , wC , $ { f } , d2 ) ;
} else if ( $ { h === 2 } ) {
2021-09-11 16:29:31 +02:00
vec2 xValues = vec2 (
2021-10-04 22:29:15 +02:00
getX ( batch , xF , xR , xC , $ { f } ) ,
getX ( batch , xF , xR , xC , $ { f } + 1 )
2021-09-11 16:29:31 +02:00
) ;
vec2 wValues = vec2 (
2021-10-04 22:29:15 +02:00
getW ( wF , wR , wC , $ { f } , d2 ) ,
getW ( wF , wR , wC , $ { f } + 1 , d2 )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
2021-10-04 22:29:15 +02:00
} else if ( $ { h === 3 } ) {
2021-09-11 16:29:31 +02:00
vec3 xValues = vec3 (
2021-10-04 22:29:15 +02:00
getX ( batch , xF , xR , xC , $ { f } ) ,
getX ( batch , xF , xR , xC , $ { f } + 1 ) ,
getX ( batch , xF , xR , xC , $ { f } + 2 )
2021-09-11 16:29:31 +02:00
) ;
vec3 wValues = vec3 (
2021-10-04 22:29:15 +02:00
getW ( wF , wR , wC , $ { f } , d2 ) ,
getW ( wF , wR , wC , $ { f } + 1 , d2 ) ,
getW ( wF , wR , wC , $ { f } + 2 , d2 )
2021-09-11 16:29:31 +02:00
) ;
dotProd += dot ( xValues , wValues ) ;
}
}
}
}
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},vEe=class{constructor(e,t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"inputShape",type:"ivec3"},{name:"pad",type:"ivec2"},{name:"stride",type:"ivec2"},{name:"dilation",type:"ivec2"},{name:"inChannels",type:"int"},{name:"itemsPerBlockRow",type:"int"},{name:"outWidth",type:"int"}],this.outputShape=e,this.enableShapeUniforms=xa(this.outputShape.length);let{dataFormat:n}=t,r=Us(),s=n==="channelsLast",o=s?0:1,a=s?1:2,i=this.enableShapeUniforms?"if(blockIndex < outShape[1] && pos < outShape[0]) {": ` if ( blockIndex < $ { e [ 1 ] } && pos < $ { e [ 0 ] } ) { ` ,l="";for(let u=0;u<=1;u++)for(let c=0;c<=1;c++)l+= `
2021-09-11 17:17:13 +02:00
blockIndex = rc . y + $ { c } ;
2021-09-11 16:29:31 +02:00
pos = rc . x + $ { u } ;
2021-09-01 00:24:30 +02:00
2021-09-11 17:17:13 +02:00
$ { i }
2021-09-11 16:29:31 +02:00
offsetY = int ( blockIndex / outWidth ) * stride [ 0 ] - pad [ 0 ] ;
d0 = offsetY + dilation [ 0 ] * ( pos / itemsPerBlockRow ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( d0 < inputShape [ $ { o } ] && d0 >= 0 ) {
2021-09-11 16:29:31 +02:00
// Use custom imod instead mod. On Intel GPU, mod may generate
// unexpected value.
// https://github.com/tensorflow/tfjs/issues/5447
offsetX = imod ( blockIndex , outWidth ) * stride [ 1 ] - pad [ 1 ] ;
d1 = offsetX + dilation [ 1 ] * ( imod ( pos , itemsPerBlockRow ) /
inChannels ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
if ( d1 < inputShape [ $ { a } ] && d1 >= 0 ) {
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
ch = imod ( pos , inChannels ) ;
2021-09-05 22:42:11 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { s } ) {
2021-09-11 16:29:31 +02:00
innerDims = vec2 ( d1 , ch ) ;
2021-09-11 17:17:13 +02:00
result [ $ { u * 2 + c } ] = getChannel (
2021-09-11 16:29:31 +02:00
getA ( d0 , int ( innerDims . x ) ,
int ( innerDims . y ) ) , innerDims ) ;
} else {
innerDims = vec2 ( d0 , d1 ) ;
2021-09-11 17:17:13 +02:00
result [ $ { u * 2 + c } ] = getChannel (
2021-09-11 16:29:31 +02:00
getA ( ch , int ( innerDims . x ) ,
int ( innerDims . y ) ) , innerDims ) ;
}
}
}
}
` ;this.userCode= `
void main ( ) {
ivec2 rc = getOutputCoords ( ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int blockIndex , pos , offsetY , d0 , offsetX , d1 , ch ;
vec2 innerDims ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
$ { l }
2021-10-04 22:29:15 +02:00
$ { r . output } = result ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function jW({x:e,filter:t,convInfo:n,backend:r,bias:s=null,preluActivationWeights:o=null,leakyreluAlpha:a=0,activation:i=null}){let l=e.shape,u=r.texData.get(e.dataId),c=n.inChannels,p=l[0]*l[1]*l[2],d=n.outChannels,f=n.dataFormat==="channelsLast",h=!1,m=!1,g,x=[];if(!((p===1||d===1)&&c>FW)&&u.isPacked&&f&&u.texture!=null&&l[2]%2!=0&&E.arraysEqual(u.shape.slice(-3),l.slice(-3))){let A=l[0]*l[1]*(l[2]+1),v={dataId:e.dataId,shape:[1,A,n.inChannels],dtype:e.dtype},w=u.shape;u.shape=u.shape.slice(),u.shape[u.shape.length-2]++,E.assert(tA(u.shape,v.shape),()=> ` packed reshape $ { u . shape } to $ { v . shape } isn ' t free ` );let k=qe({inputs:{x:t},backend:r,attrs:{shape:[1,n.inChannels,n.outChannels]}});x.push(k);let I=B5({a:v,b:k,backend:r,transposeA:h,transposeB:m,bias:s,activation:i,preluActivationWeights:o,leakyreluAlpha:a}),N=r.texData.get(I.dataId);E.assert(N.isPacked,()=>"batchMatMul result is expected to be packed"),u.shape=w,N.shape=n.outShape,g=Xo({inputs:{x:I},backend:r}),g.shape=n.outShape,x.push(I)}else{let A=f?l[0]*l[1]*l[2]:l[0]*l[2]*l[3],v=qe({inputs:{x:e},backend:r,attrs:{shape:[1,A,n.inChannels]}}),w=qe({inputs:{x:t},backend:r,attrs:{shape:[1,n.inChannels,n.outChannels]}}),k=B5({a:v,b:w,transposeA:h,transposeB:m,backend:r,bias:s,activation:i,preluActivationWeights:o,leakyreluAlpha:a});g=qe({inputs:{x:k},backend:r,attrs:{shape:n.outShape}}),x.push(v),x.push(w),x.push(k)}for(let A of x)r.disposeIntermediateTensorInfo(A);return g}function HW({x:e,filter:t,convInfo:n,backend:r,bias:s=null,preluActivationWeights:o=null,leakyreluAlpha:a=0,activation:i=null}){let{filterWidth:l,filterHeight:u,inChannels:c,outWidth:p,outHeight:d,dataFormat:f}=n,h=f==="channelsLast",m=l*u*c,g=d*p,x=[m,g],y=!0,b=!1,A=[],v=qe({inputs:{x:e},backend:r,attrs:{shape:e.shape.slice(1)}}),w=qe({inputs:{x:t},backend:r,attrs:{shape:[1,m,E.sizeFromShape(t.shape)/m]}});A.push(v),A.push(w);let k=new vEe(x,n),I=[v.shape,[n.padInfo.top,n.padInfo.left],[n.strideHeight,n.strideWidth],[n.dilationHeight,n.dilationWidth],[n.inChannels],[n.filterWidth*n.inChannels],[n.outWidth]],N=r.runWebGLProgram(k,[v],"float32",I),T=qe({inputs:{x:N},backend:r,attrs:{shape:[1,x[0],x[1]]}});A.push(N),A.push(T);let _=s!=null,R=o!=null,S=i==="leakyrelu",F=i?M5(i,!0):null,B=new _W(T.shape,w.shape,[1,g,n.outChannels],y,b,_,F,R,S),W=[T,w];if(s&&W.push(s),R&&W.push(o),S){let Q=r.makeTensorInfo([],"float32",E.createScalarValue(a,"float32"));W.push(Q),A.push(Q)}let q=r.runWebGLProgram(B,W,"float32"),K=h?[1,d,p,n.outChannels]:[1,n.outChannels,d,p],J=qe({inputs:{x:q},backend:r,attrs:{shape:K}});A.push(q);for(let Q of A)r.disposeIntermediateTensorInfo(Q);return J}function wEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dataFormat:l,dilations:u,dimRoundingMode:c}=r,p=G.convertConv2DDataFormat(l),d=G.computeConv2DInfo(s.shape,o.shape,a,u,i,c,!1,p),f;if(d.filterHeight===1&&d.filterWidth===1&&d.dilationHeight===1&&d.dilationWidth===1&&d.strideHeight===1&&d.strideWidth===1&&(d.padInfo.type==="SAME"||d.padInfo.type==="VALID"))f=jW({x:s,filter:o,convInfo:d,backend:n});else if(be().getBool("WEBGL_CONV_IM2COL")&&s.shape[0]===1)f=HW({x:s,filter:o,convInfo:d,backend:n});else{let m=new UW(d);f=n.runWebGLProgram(m,[s,o],"float32")}let h=qe({inputs:{x:f},backend:n,attrs:{shape:d.outShape}});return n.disposeIntermediateTensorInfo(f),h}var kEe={kernelName:jd,backendName:"webgl",kernelFunc:wEe},IEe=class{constructor(e){this.variableNames=["x","dy"],this.outputShape=e.filterShape;let t=e.strideHeight,n=e.strideWidth,r=e.padInfo.top,s=e.padInfo.left,o=e.dataFormat==="channelsLast";this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int wR = coords . x ;
int wC = coords . y ;
int d1 = coords . z ;
int d2 = coords . w ;
// Convolve x(?, ?, d1) with dy(:, :, d2) to get dw(wR, wC, d1, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int b = 0 ; b < $ { e . batchSize } ; b ++ ) {
for ( int yR = 0 ; yR < $ { e . outHeight } ; yR ++ ) {
2021-10-04 22:29:15 +02:00
int xR = wR + yR * $ { t } - $ { r } ;
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { e . outWidth } ; yC ++ ) {
2021-10-04 22:29:15 +02:00
int xC = wC + yC * $ { n } - $ { s } ;
2021-09-11 16:29:31 +02:00
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
}
2021-10-04 22:29:15 +02:00
if ( $ { o } ) {
2021-09-11 16:29:31 +02:00
float dyValue = getDy ( b , yR , yC , d2 ) ;
float xValue = getX ( b , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
2021-09-05 22:42:11 +02:00
} else {
2021-09-11 16:29:31 +02:00
float dyValue = getDy ( b , d2 , yR , yC ) ;
float xValue = getX ( b , d1 , xR , xC ) ;
dotProd += ( xValue * dyValue ) ;
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
}
}
setOutput ( dotProd ) ;
}
2021-10-04 22:29:15 +02:00
` }},CEe=class{constructor(e){this.variableNames=["dy","W"],this.outputShape=e.inShape;let t=e.filterHeight,n=e.filterWidth,r=e.strideHeight,s=e.strideWidth,o=e.dataFormat==="channelsLast",a=t-1-e.padInfo.top,i=n-1-e.padInfo.left,l=o?1:2,u=o?2:3,c=o?3:1;this.userCode= `
const ivec2 pads = ivec2 ( $ { a } , $ { i } ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
2021-09-11 17:17:13 +02:00
int d1 = coords [ $ { c } ] ;
2021-09-11 16:29:31 +02:00
ivec2 dyCorner = ivec2 ( coords [ $ { l } ] , coords [ $ { u } ] ) - pads ;
int dyRCorner = dyCorner . x ;
int dyCCorner = dyCorner . y ;
// Convolve dy(?, ?, d2) with w(:, :, d1, d2) to compute dx(xR, xC, d1).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { t } ; wR ++ ) {
2021-10-04 22:29:15 +02:00
float dyR = float ( dyRCorner + wR ) / $ { r } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { t } - 1 - wR ;
for ( int wC = 0 ; wC < $ { n } ; wC ++ ) {
2021-10-04 22:29:15 +02:00
float dyC = float ( dyCCorner + wC ) / $ { s } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
int wCPerm = $ { n } - 1 - wC ;
for ( int d2 = 0 ; d2 < $ { e . outChannels } ; d2 ++ ) {
2021-10-04 22:29:15 +02:00
if ( $ { o } ) {
2021-09-11 16:29:31 +02:00
float xValue = getDy ( batch , idyR , idyC , d2 ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
2021-09-05 22:42:11 +02:00
} else {
2021-09-11 16:29:31 +02:00
float xValue = getDy ( batch , d2 , idyR , idyC ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
}
}
}
setOutput ( dotProd ) ;
}
2021-10-04 22:29:15 +02:00
` }},SEe=class{constructor(e){this.variableNames=["x","dy"],this.outputShape=e.filterShape;let t=e.strideDepth,n=e.strideHeight,r=e.strideWidth,s=e.padInfo.front,o=e.padInfo.top,a=e.padInfo.left;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int wF = coords . x ;
int wR = coords . y ;
int wC = coords . z ;
int d1 = coords . w ;
int d2 = coords . u ;
float dotProd = 0.0 ;
for ( int b = 0 ; b < $ { e . batchSize } ; b ++ ) {
for ( int yF = 0 ; yF < $ { e . outDepth } ; yF ++ ) {
2021-10-04 22:29:15 +02:00
int xF = wF + yF * $ { t } - $ { s } ;
2021-09-11 16:29:31 +02:00
if ( xF < 0 || xF >= $ { e . inDepth } ) {
continue ;
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
for ( int yR = 0 ; yR < $ { e . outHeight } ; yR ++ ) {
2021-10-04 22:29:15 +02:00
int xR = wR + yR * $ { n } - $ { o } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { e . outWidth } ; yC ++ ) {
2021-10-04 22:29:15 +02:00
int xC = wC + yC * $ { r } - $ { a } ;
2021-09-11 16:29:31 +02:00
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
float dyValue = getDy ( b , yF , yR , yC , d2 ) ;
float xValue = getX ( b , xF , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
2021-09-01 00:24:30 +02:00
}
}
2021-09-11 16:29:31 +02:00
}
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( dotProd ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }},NEe=class{constructor(e){this.variableNames=["dy","W"],this.outputShape=e.inShape;let t=e.filterDepth,n=e.filterHeight,r=e.filterWidth,s=e.strideDepth,o=e.strideHeight,a=e.strideWidth,i=t-1-e.padInfo.front,l=n-1-e.padInfo.top,u=r-1-e.padInfo.left;this.userCode= `
2021-09-11 17:17:13 +02:00
const ivec3 pads = ivec3 ( $ { i } , $ { l } , $ { u } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d1 = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyFCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
float dotProd = 0.0 ;
for ( int wF = 0 ; wF < $ { t } ; wF ++ ) {
2021-10-04 22:29:15 +02:00
float dyF = float ( dyFCorner + wF ) / $ { s } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyF < 0.0 || dyF >= $ { e . outDepth } . 0 || fract ( dyF ) > 0.0 ) {
continue ;
}
int idyF = int ( dyF ) ;
int wFPerm = $ { t } - 1 - wF ;
for ( int wR = 0 ; wR < $ { n } ; wR ++ ) {
2021-10-04 22:29:15 +02:00
float dyR = float ( dyRCorner + wR ) / $ { o } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { n } - 1 - wR ;
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { r } ; wC ++ ) {
float dyC = float ( dyCCorner + wC ) / $ { a } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
2021-10-04 22:29:15 +02:00
int wCPerm = $ { r } - 1 - wC ;
2021-09-11 16:29:31 +02:00
for ( int d2 = 0 ; d2 < $ { e . outChannels } ; d2 ++ ) {
float xValue = getDy ( batch , idyF , idyR , idyC , d2 ) ;
float wValue = getW ( wFPerm , wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
}
}
}
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function TEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,pad:i,dataFormat:l,dimRoundingMode:u,filterShape:c}=r,p=G.convertConv2DDataFormat(l),d=G.computeConv2DInfo(s.shape,c,a,1,i,u,!1,p),f=new IEe(d);return n.runWebGLProgram(f,[s,o],"float32")}var _Ee={kernelName:sw,backendName:"webgl",kernelFunc:TEe};function EEe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{inputShape:a,strides:i,pad:l,dataFormat:u,dimRoundingMode:c}=r,p=G.convertConv2DDataFormat(u),d=G.computeConv2DInfo(a,o.shape,i,1,l,c,!1,p),f=new CEe(d);return n.runWebGLProgram(f,[s,o],"float32")}var REe={kernelName:Hd,backendName:"webgl",kernelFunc:EEe};function DEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l}=r,u=G.computeConv3DInfo(s.shape,o.shape,a,l,i),c=new AEe(u);return n.runWebGLProgram(c,[s,o],"float32")}var $ Ee={kernelName:L1,backendName:"webgl",kernelFunc:DEe};function FEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,pad:i,filterShape:l}=r,u=G.computeConv3DInfo(s.shape,l,a,1,i),c=new SEe(u);return n.runWebGLProgram(c,[s,o],"float32")}var OEe={kernelName:ow,backendName:"webgl",kernelFunc:FEe};function PEe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{pad:a,strides:i,inputShape:l}=r,u=G.computeConv3DInfo(l,o.shape,i,1,a),c=new NEe(u);return n.runWebGLProgram(c,[s,o],"float32")}var MEe={kernelName:aw,backendName:"webgl",kernelFunc:PEe},zEe=TW+ `
2021-09-11 16:29:31 +02:00
return cos ( x ) ;
2021-10-04 22:29:15 +02:00
` ,LEe=tn({opSnippet:zEe}),BEe={kernelName:qd,backendName:"webgl",kernelFunc:LEe},VEe= `
2021-09-11 16:29:31 +02:00
float e2x = exp ( - x ) ;
return ( e2x + 1.0 / e2x ) / 2.0 ;
2021-10-04 22:29:15 +02:00
` ,WEe=tn({opSnippet:VEe}),GEe={kernelName:Kd,backendName:"webgl",kernelFunc:WEe},UEe=class{constructor(e,t,n,r,s){this.variableNames=["Image","Boxes","BoxInd"],this.outputShape=[];let[o,a,i,l]=e,[u]=t,[c,p]=n;this.outputShape=[u,c,p,l];let d=r==="bilinear"?1:0,[f,h]=[ ` $ { a - 1 } . 0 ` , ` $ { i - 1 } . 0 ` ],[m,g,x]=c>1?[ ` $ { ( a - 1 ) / ( c - 1 ) } ` ,"(y2-y1) * height_ratio", ` y1 * $ { f } + float ( y ) * ( height _scale ) ` ]:["0.0","0.0", ` 0.5 * ( y1 + y2 ) * $ { f } ` ],[y,b,A]=p>1?[ ` $ { ( i - 1 ) / ( p - 1 ) } ` ,"(x2-x1) * width_ratio", ` x1 * $ { h } + float ( x ) * ( width _scale ) ` ]:["0.0","0.0", ` 0.5 * ( x1 + x2 ) * $ { h } ` ];this.userCode= `
const float height _ratio = float ( $ { m } ) ;
2021-09-11 16:29:31 +02:00
const float width _ratio = float ( $ { y } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int y = coords [ 1 ] ;
int x = coords [ 2 ] ;
int d = coords [ 3 ] ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// get box vals
float y1 = getBoxes ( b , 0 ) ;
float x1 = getBoxes ( b , 1 ) ;
float y2 = getBoxes ( b , 2 ) ;
float x2 = getBoxes ( b , 3 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// get image in batch index
int bInd = round ( getBoxInd ( b ) ) ;
2021-10-04 22:29:15 +02:00
if ( bInd < 0 || bInd >= $ { o } ) {
2021-09-11 16:29:31 +02:00
return ;
}
float height _scale = $ { g } ;
2021-10-04 22:29:15 +02:00
float width _scale = $ { b } ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
float in _y = $ { x } ;
if ( in _y < 0.0 || in _y > $ { f } ) {
setOutput ( float ( $ { s } ) ) ;
2021-09-11 16:29:31 +02:00
return ;
}
2021-10-04 22:29:15 +02:00
float in _x = $ { A } ;
if ( in _x < 0.0 || in _x > $ { h } ) {
setOutput ( float ( $ { s } ) ) ;
2021-09-11 16:29:31 +02:00
return ;
}
vec2 sourceFracIndexCR = vec2 ( in _x , in _y ) ;
2021-10-04 22:29:15 +02:00
if ( $ { d } == 1 ) {
2021-09-11 16:29:31 +02:00
// Compute the four integer indices.
ivec2 sourceFloorCR = ivec2 ( sourceFracIndexCR ) ;
ivec2 sourceCeilCR = ivec2 ( ceil ( sourceFracIndexCR ) ) ;
float topLeft = getImage ( b , sourceFloorCR . y , sourceFloorCR . x , d ) ;
float bottomLeft = getImage ( b , sourceCeilCR . y , sourceFloorCR . x , d ) ;
float topRight = getImage ( b , sourceFloorCR . y , sourceCeilCR . x , d ) ;
float bottomRight = getImage ( b , sourceCeilCR . y , sourceCeilCR . x , d ) ;
vec2 fracCR = sourceFracIndexCR - vec2 ( sourceFloorCR ) ;
float top = topLeft + ( topRight - topLeft ) * fracCR . x ;
float bottom = bottomLeft + ( bottomRight - bottomLeft ) * fracCR . x ;
float newValue = top + ( bottom - top ) * fracCR . y ;
setOutput ( newValue ) ;
} else {
// Compute the coordinators of nearest neighbor point.
ivec2 sourceNearestCR = ivec2 ( floor (
sourceFracIndexCR + vec2 ( 0.5 , 0.5 ) ) ) ;
float newValue = getImage ( b , sourceNearestCR . y , sourceNearestCR . x , d ) ;
setOutput ( newValue ) ;
}
}
2021-10-04 22:29:15 +02:00
` }},jEe=e=>{let{inputs:t,backend:n,attrs:r}=e,{image:s,boxes:o,boxInd:a}=t,{cropSize:i,method:l,extrapolationValue:u}=r,c=new UEe(s.shape,o.shape,i,l,u);return n.runWebGLProgram(c,[s,o,a],"float32")},HEe={kernelName:wg,backendName:"webgl",kernelFunc:jEe},qW=class{constructor(e,t,n){this.variableNames=["x"],this.customUniforms=[{name:"index",type:"float"}],this.outputShape=e;let r=e.length,s=t?"0.0": ` getX ( $ { KW ( r , "coords" ) } ) ` ,o=e[e.length-1],a="",i="";t?(a=n? ` end != $ { o - 1 } ` :"end != 0",i=n?"end + 1":"end - 1"):(a=n? ` end + pow2 < $ { o } ` :"end >= pow2",i=n?"end + pow2":"end - pow2"),this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { Dn ( r ) } coords = getOutputCoords ( ) ;
int end = $ { XW ( r , "coords" ) } ;
float val = $ { s } ;
2021-09-11 16:29:31 +02:00
int pow2 = int ( pow ( 2.0 , index ) ) ;
2021-10-04 22:29:15 +02:00
if ( $ { a } ) {
2021-09-11 17:17:13 +02:00
int idx = $ { i } ;
2021-10-04 22:29:15 +02:00
$ { XW ( r , "coords" ) } = idx ;
val += getX ( $ { KW ( r , "coords" ) } ) ;
2021-09-11 16:29:31 +02:00
}
setOutput ( val ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function KW(e,t){if(e===1)return ` $ { t } ` ;if(e===2)return ` $ { t } . x , $ { t } . y ` ;if(e===3)return ` $ { t } . x , $ { t } . y , $ { t } . z ` ;if(e===4)return ` $ { t } . x , $ { t } . y , $ { t } . z , $ { t } . w ` ;throw Error( ` Cumulative sum for rank $ { e } is not yet supported ` )}function XW(e,t){if(e===1)return ` $ { t } ` ;if(e===2)return ` $ { t } . y ` ;if(e===3)return ` $ { t } . z ` ;if(e===4)return ` $ { t } . w ` ;throw Error( ` Cumulative sum for rank $ { e } is not yet supported ` )}function qEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,exclusive:a,reverse:i}=r,l=s.shape.length,u=G.getAxesPermutation([o],l),c=s;u!=null&&(c=Hs({inputs:{x:s},backend:n,attrs:{perm:u}}));let p=G.getInnerMostAxes(1,l)[0];if(p!==l-1)throw new Error( ` WebGL cumsum shader expects an inner - most axis = $ { s . shape . length - 1 } but got axis = $ { o } ` );let d=c.shape[p],f=Xo({inputs:{x:c},backend:n});for(let h=0;h<=Math.ceil(Math.log2(d))-1;h++){let m=new qW(c.shape,!1,i),g=[[h]],x=f;f=n.runWebGLProgram(m,[f],f.dtype,g),n.disposeIntermediateTensorInfo(x)}if(a){let h=new qW(c.shape,a,i),m=f;f=n.runWebGLProgram(h,[f],f.dtype),n.disposeIntermediateTensorInfo(m)}if(u!=null){let h=G.getUndoAxesPermutation(u),m=Hs({inputs:{x:f},backend:n,attrs:{perm:h}});return n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(c),m}return f}var KEe={kernelName:Xd,backendName:"webgl",kernelFunc:qEe};function XEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,weights:o}=t,{size:a,binaryOutput:i}=r;if(s.shape.length===1){let l=n.readSync(s.dataId),u=n.readSync(o.dataId),c=fW(l,u,o.dtype,o.shape,a);return n.makeTensorInfo([a],o.dtype,c)}else if(s.shape.length===2){let l=n.bufferSync(s),u=n.bufferSync(o),c=lNe(l,u,a,i);return n.makeTensorInfo(c.shape,o.dtype,c.values)}throw new Error( ` Error in denseBincount : input must be at most rank 2 , but got rank$ { s . shape . length } . ` )}var ZEe={kernelName:iw,backendName:"webgl",kernelFunc:XEe},YEe=class{constructor(e,t,n){this.variableNames=["x"],this.outputShape=[],this.outputShape=e,this.blockSize=t,this.dataFormat=n,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int h = $ { this . getHeightCoordString ( ) } ;
int w = $ { this . getWidthCoordString ( ) } ;
int d = $ { this . getDepthCoordString ( ) } ;
int in _h = h / $ { t } ;
int offset _h = imod ( h , $ { t } ) ;
int in _w = w / $ { t } ;
int offset _w = imod ( w , $ { t } ) ;
int offset _d = ( offset _h * $ { t } + offset _w ) *
$ { this . getOutputDepthSize ( ) } ;
int in _d = d + offset _d ;
float result = $ { this . getInputSamplingString ( ) } ;
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }getHeightCoordString(){return this.dataFormat==="NHWC"?"coords[1]":"coords[2]"}getWidthCoordString(){return this.dataFormat==="NHWC"?"coords[2]":"coords[3]"}getDepthCoordString(){return this.dataFormat==="NHWC"?"coords[3]":"coords[1]"}getOutputDepthSize(){return this.dataFormat==="NHWC"?this.outputShape[3]:this.outputShape[1]}getInputSamplingString(){return this.dataFormat==="NHWC"?"getX(b, in_h, in_w, in_d)":"getX(b, in_d, in_h, in_w)"}};function JEe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockSize:o,dataFormat:a}=r;E.assert(o>1,()=> ` blockSize should be > 1 for depthToSpace , but was : $ { o } ` );let i=s.shape[0],l=a==="NHWC"?s.shape[1]:s.shape[2],u=a==="NHWC"?s.shape[2]:s.shape[3],c=a==="NHWC"?s.shape[3]:s.shape[1],p=l*o,d=u*o,f=c/(o*o),h=a==="NHWC"?[i,p,d,f]:[i,f,p,d],m=new YEe(h,o,a);return n.runWebGLProgram(m,[s],s.dtype)}var QEe={kernelName:kg,backendName:"webgl",kernelFunc:JEe},ZW=class{constructor(e,t=!1,n=null,r=!1,s=!1){this.variableNames=["x","W"],this.customUniforms=[{name:"pads",type:"ivec2"},{name:"strides",type:"ivec2"},{name:"dilations",type:"ivec2"},{name:"inDims",type:"ivec2"}],this.outputShape=e.outShape,this.enableShapeUniforms=xa(this.outputShape.length);let o=e.filterHeight,a=e.filterWidth,i=e.outChannels/e.inChannels,l="",u="";n&&(r?l= ` float activation ( float a ) {
2021-09-11 16:29:31 +02:00
float b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { n }
2021-10-04 22:29:15 +02:00
} ` :s?l= ` float activation ( float a ) {
2021-09-11 16:29:31 +02:00
float b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { n }
} ` :l= `
float activation ( float x ) {
$ { n }
}
2021-10-04 22:29:15 +02:00
` ,u="result = activation(result);");let c=t?"result += getBiasAtOutCoords();":"";t&&this.variableNames.push("bias"),r&&this.variableNames.push("preluActivationWeights"),s&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
2021-09-11 16:29:31 +02:00
$ { l }
2021-09-01 00:24:30 +02:00
void main ( ) {
2021-09-11 16:29:31 +02:00
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int d2 = coords . w ;
2021-09-11 17:17:13 +02:00
int d1 = d2 / $ { i } ;
int q = d2 - d1 * $ { i } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Convolve x(?, ?, d1) with w(:, :, d1, q) to get y(yR, yC, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
// TO DO(dsmilkov): Flatten the two for loops and vec4 the operations.
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { o } ; wR ++ ) {
2021-09-11 16:29:31 +02:00
int xR = xRCorner + wR * dilations [ 0 ] ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= inDims [ 0 ] ) {
continue ;
}
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { a } ; wC ++ ) {
2021-09-11 16:29:31 +02:00
int xC = xCCorner + wC * dilations [ 1 ] ;
if ( xC < 0 || xC >= inDims [ 1 ] ) {
continue ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
float xVal = getX ( batch , xR , xC , d1 ) ;
float wVal = getW ( wR , wC , d1 , q ) ;
dotProd += xVal * wVal ;
}
2021-09-05 22:42:11 +02:00
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
float result = dotProd ;
2021-09-11 17:17:13 +02:00
$ { c }
2021-09-11 16:29:31 +02:00
$ { u }
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }},YW=class{constructor(e,t=!1,n=null,r=!1,s=!1){this.variableNames=["x","W"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"pads",type:"ivec2"},{name:"strides",type:"ivec2"},{name:"dilations",type:"ivec2"},{name:"inDims",type:"ivec2"}],this.outputShape=e.outShape,this.enableShapeUniforms=xa(this.outputShape.length);let o=e.outChannels/e.inChannels,a=e.padInfo.left,i=e.strideWidth,l=e.dilationWidth,u=e.filterHeight,c=e.filterWidth,p=c,d= `
2021-09-11 16:29:31 +02:00
int xR ; int xC ; int xCOffset ;
2021-10-04 22:29:15 +02:00
vec4 wTexel ; vec4 previous ; vec4 final ; ` ;for(let g=0;g<c;g++)d+= `
2021-09-11 16:29:31 +02:00
vec4 xTexelC$ { g * 2 } ;
int xTexelC$ { g * 2 } Ready ;
vec4 xTexelC$ { g * 2 + 1 } ;
int xTexelC$ { g * 2 + 1 } Ready ;
2021-10-04 22:29:15 +02:00
vec4 xC$ { g } ; ` ;for(let g=0;g<u;g++){for(let x=0;x<c;x++)d+= `
xTexelC$ { x * 2 } = vec4 ( 0.0 ) ;
xTexelC$ { x * 2 } Ready = 0 ;
xTexelC$ { x * 2 + 1 } = vec4 ( 0.0 ) ;
xTexelC$ { x * 2 + 1 } Ready = 0 ;
xC$ { x } = vec4 ( 0.0 ) ; ` ;d+= `
2021-09-11 16:29:31 +02:00
xR = xRCorner + $ { g } * dilations [ 0 ] ;
if ( xR >= 0 && xR < inDims [ 0 ] ) {
2021-10-04 22:29:15 +02:00
` ;for(let x=0;x<(p+1)/2;x++){let y=x*2;if(d+= `
2021-09-11 16:29:31 +02:00
xC = xCCorner + $ { y * l } ;
2021-10-04 22:29:15 +02:00
` ,i===1){if(y<c&&(a%2==1?(d+= `
2021-09-11 16:29:31 +02:00
xCOffset = xC + 1 ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y } Ready == 0 ) {
xTexelC$ { y } = getX ( batch , xR , xCOffset , d1 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { y } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y } Ready = 1 ;
}
2021-10-04 22:29:15 +02:00
` ,l===1&&y>0?d+= `
2021-09-11 16:29:31 +02:00
xC$ { y } = vec4 ( xTexelC$ { y - 2 } . zw , xTexelC$ { y } . xy ) ;
2021-10-04 22:29:15 +02:00
` :d+= `
2021-09-11 16:29:31 +02:00
xCOffset = xC + 1 - 2 ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] ) {
previous = getX ( batch , xR , xCOffset , d1 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
previous . zw = vec2 ( 0.0 ) ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
xC$ { y } = vec4 ( previous . zw , xTexelC$ { y } . xy ) ;
} else {
xC$ { y } = vec4 ( 0.0 , 0.0 , xTexelC$ { y } . xy ) ;
}
2021-10-04 22:29:15 +02:00
` ):d+= `
2021-09-11 16:29:31 +02:00
if ( xC >= 0 && xC < inDims [ 1 ] && xTexelC$ { y } Ready == 0 ) {
xTexelC$ { y } = getX ( batch , xR , xC , d1 ) ;
if ( xC + 1 >= inDims [ 1 ] ) {
xTexelC$ { y } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y } Ready = 1 ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
xC$ { y } = xTexelC$ { y } ;
2021-10-04 22:29:15 +02:00
` ,y+1<c)){let b=a%2==0?E.nearestLargerEven(l):l;l%2==0&&a%2==1||l%2!=0&&a%2!=1?(d+= `
xCOffset = xC + imod ( pads [ 1 ] , 2 ) + $ { b } ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y + 1 } Ready == 0 ) {
xTexelC$ { y + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { y + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y + 1 } Ready = 1 ;
}
2021-10-04 22:29:15 +02:00
` ,l>1&&(d+= `
2021-09-11 16:29:31 +02:00
xCOffset -= 2 ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y } Ready == 0 ) {
xTexelC$ { y } = getX ( batch , xR , xCOffset , d1 ) ;
xTexelC$ { y } Ready = 1 ;
}
2021-10-04 22:29:15 +02:00
` ),d+= `
2021-09-11 16:29:31 +02:00
xC$ { y + 1 } = vec4 ( xTexelC$ { y } . zw , xTexelC$ { y + 1 } . xy ) ;
2021-10-04 22:29:15 +02:00
` ):b===1?d+= `
2021-09-11 16:29:31 +02:00
xC$ { y + 1 } = xTexelC$ { y } ;
2021-10-04 22:29:15 +02:00
` :d+= `
xCOffset = xC + $ { b } ;
2021-09-11 16:29:31 +02:00
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y + 1 } Ready == 0 ) {
xTexelC$ { y + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { y + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y + 1 } Ready = 1 ;
}
xC$ { y + 1 } = xTexelC$ { y + 1 } ;
2021-10-04 22:29:15 +02:00
` }}else y<c&&(a%2==1?(d+= `
2021-09-11 16:29:31 +02:00
xCOffset = xC + 1 - strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y } Ready == 0 ) {
xTexelC$ { y } = getX ( batch , xR , xCOffset , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { y } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y } Ready = 1 ;
}
if ( xC + 1 >= 0 && xC + 1 < inDims [ 1 ] && xTexelC$ { y + 1 } Ready == 0 ) {
xTexelC$ { y + 1 } = getX ( batch , xR , xC + 1 , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xC + 2 >= inDims [ 1 ] ) {
xTexelC$ { y + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y + 1 } Ready = 1 ;
}
xC$ { y } = vec4 ( xTexelC$ { y } . zw , xTexelC$ { y + 1 } . zw ) ;
2021-10-04 22:29:15 +02:00
` ,y+1<c&&(d+= `
2021-09-11 16:29:31 +02:00
final = vec4 ( 0.0 ) ;
xCOffset = xC + 1 + strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] ) {
final = getX ( batch , xR , xCOffset , d1 ) ;
}
xC$ { y + 1 } = vec4 ( xTexelC$ { y + 1 } . xy , final . xy ) ;
2021-10-04 22:29:15 +02:00
` )):(d+= `
2021-09-11 16:29:31 +02:00
if ( xC >= 0 && xC < inDims [ 1 ] && xTexelC$ { y } Ready == 0 ) {
xTexelC$ { y } = getX ( batch , xR , xC , d1 ) ;
if ( xC + 1 >= inDims [ 1 ] ) {
xTexelC$ { y } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { y } Ready = 1 ;
}
xCOffset = xC + strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { y + 1 } Ready == 0 ) {
xTexelC$ { y + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { y + 1 } . zw = vec2 ( 0. ) ;
}
xTexelC$ { y + 1 } Ready = 1 ;
}
xC$ { y } = vec4 (
xTexelC$ { y } . xy , xTexelC$ { y + 1 } . xy ) ;
2021-10-04 22:29:15 +02:00
` ,y+1<c&&(d+= `
2021-09-11 16:29:31 +02:00
xC$ { y + 1 } = vec4 ( xTexelC$ { y } . zw , xTexelC$ { y + 1 } . zw ) ;
2021-10-04 22:29:15 +02:00
` )));y<c&&(d+= `
2021-09-11 16:29:31 +02:00
wTexel = getW ( $ { g } , $ { y } , d1 , q ) ;
dotProd += xC$ { y } * vec4 ( wTexel . xz , wTexel . xz ) ;
2021-10-04 22:29:15 +02:00
` ,y+1<c&&(d+= `
2021-09-11 16:29:31 +02:00
wTexel = getW ( $ { g } , $ { y + 1 } , d1 , q ) ;
dotProd += xC$ { y + 1 } * vec4 ( wTexel . xz , wTexel . xz ) ;
2021-10-04 22:29:15 +02:00
` ))}d+= `
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }let f="",h="";n&&(r?f= ` vec4 activation ( vec4 a ) {
2021-09-11 16:29:31 +02:00
vec4 b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { n }
2021-10-04 22:29:15 +02:00
} ` :s?f= ` vec4 activation ( vec4 a ) {
2021-09-11 16:29:31 +02:00
vec4 b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { n }
2021-10-04 22:29:15 +02:00
} ` :f= ` vec4 activation ( vec4 x ) {
2021-09-11 16:29:31 +02:00
$ { n }
2021-10-04 22:29:15 +02:00
} ` ,h="result = activation(result);");let m=t?"result += getBiasAtOutCoords();":"";t&&this.variableNames.push("bias"),r&&this.variableNames.push("preluActivationWeights"),s&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
$ { f }
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int d2 = coords . w ;
2021-10-04 22:29:15 +02:00
int d1 = d2 / $ { o } ;
int q = d2 - d1 * $ { o } ;
2021-09-11 16:29:31 +02:00
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
//intialize dotProd with a small epsilon seems to reduce GPU accuracy loss.
vec4 dotProd = vec4 ( 0.000000000000001 ) ;
2021-10-04 22:29:15 +02:00
$ { d }
2021-09-11 16:29:31 +02:00
vec4 result = dotProd - vec4 ( 0.000000000000001 ) ;
2021-09-28 23:07:34 +02:00
$ { m }
2021-10-04 22:29:15 +02:00
$ { h }
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function e4e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l,dimRoundingMode:u}=r,c=l;c==null&&(c=[1,1]),E.assert(G.eitherStridesOrDilationsAreOne(a,c),()=> ` Error in depthwiseConv2d : Either strides or dilations must be 1. Got strides $ { a } and dilations '${c}' ` );let p=G.computeConv2DInfo(s.shape,o.shape,a,c,i,u,!0),d;be().getBool("WEBGL_PACK_DEPTHWISECONV")&&p.strideWidth<=2&&p.outChannels/p.inChannels==1?d=new YW(p):d=new ZW(p);let f=[[p.padInfo.top,p.padInfo.left],[p.strideHeight,p.strideWidth],[p.dilationHeight,p.dilationWidth],[p.inHeight,p.inWidth]];return n.runWebGLProgram(d,[s,o],"float32",f)}var t4e={kernelName:Zd,backendName:"webgl",kernelFunc:e4e},n4e=class{constructor(e){this.variableNames=["x","dy"],this.outputShape=e.filterShape;let t=e.strideHeight,n=e.strideWidth,r=e.padInfo.top,s=e.padInfo.left,o=e.outChannels/e.inChannels;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int wR = coords . x ;
int wC = coords . y ;
int d1 = coords . z ;
int dm = coords . w ;
2021-10-04 22:29:15 +02:00
int d2 = d1 * $ { o } + dm ;
2021-09-11 16:29:31 +02:00
float dotProd = 0.0 ;
// TO DO: Vec4 over the batch size
for ( int b = 0 ; b < $ { e . batchSize } ; b ++ ) {
for ( int yR = 0 ; yR < $ { e . outHeight } ; yR ++ ) {
2021-10-04 22:29:15 +02:00
int xR = wR + yR * $ { t } - $ { r } ;
2021-09-11 16:29:31 +02:00
if ( xR < 0 || xR >= $ { e . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { e . outWidth } ; yC ++ ) {
2021-10-04 22:29:15 +02:00
int xC = wC + yC * $ { n } - $ { s } ;
2021-09-11 16:29:31 +02:00
if ( xC < 0 || xC >= $ { e . inWidth } ) {
continue ;
}
float dyValue = getDy ( b , yR , yC , d2 ) ;
float xValue = getX ( b , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
}
}
}
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},r4e=class{constructor(e){this.variableNames=["dy","W"],this.outputShape=e.inShape;let t=e.filterHeight,n=e.filterWidth,r=e.strideHeight,s=e.strideWidth,o=t-1-e.padInfo.top,a=n-1-e.padInfo.left,i=e.outChannels/e.inChannels;this.userCode= `
const ivec2 pads = ivec2 ( $ { o } , $ { a } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d1 = coords [ 3 ] ;
ivec2 dyCorner = coords . yz - pads ;
int dyRCorner = dyCorner . x ;
int dyCCorner = dyCorner . y ;
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { t } ; wR ++ ) {
2021-10-04 22:29:15 +02:00
float dyR = float ( dyRCorner + wR ) / $ { r } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { t } - 1 - wR ;
for ( int wC = 0 ; wC < $ { n } ; wC ++ ) {
2021-10-04 22:29:15 +02:00
float dyC = float ( dyCCorner + wC ) / $ { s } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
int wCPerm = $ { n } - 1 - wC ;
// TO DO: Vec4 over the channelMul
2021-09-11 17:17:13 +02:00
for ( int dm = 0 ; dm < $ { i } ; dm ++ ) {
int d2 = d1 * $ { i } + dm ;
2021-09-11 16:29:31 +02:00
float xValue = getDy ( batch , idyR , idyC , d2 ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , dm ) ;
dotProd += xValue * wValue ;
}
}
}
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function s4e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,dilations:i,pad:l,dimRoundingMode:u,filterShape:c}=r,p=G.computeConv2DInfo(s.shape,c,a,i,l,u,!0),d=new n4e(p);return n.runWebGLProgram(d,[s,o],"float32")}var o4e={kernelName:lw,backendName:"webgl",kernelFunc:s4e};function a4e(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{strides:a,dilations:i,pad:l,dimRoundingMode:u,inputShape:c}=r,p=G.computeConv2DInfo(c,o.shape,a,i,l,u,!0),d=new r4e(p);return n.runWebGLProgram(d,[s,o],"float32")}var i4e={kernelName:uw,backendName:"webgl",kernelFunc:a4e},l4e=class{constructor(e){this.variableNames=["X"],this.outputShape=[e,e],this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
float val = coords [ 0 ] == coords [ 1 ] ? getX ( coords [ 0 ] ) : 0.0 ;
setOutput ( val ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function u4e(e){let{inputs:t,backend:n}=e,{x:r}=t,s=[...r.shape,...r.shape],o=E.sizeFromShape(r.shape),a=qe({inputs:{x:r},backend:n,attrs:{shape:[o]}}),i=new l4e(o),l=n.runWebGLProgram(i,[a],a.dtype),u=qe({inputs:{x:l},backend:n,attrs:{shape:s}});return n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(l),u}var c4e={kernelName:cw,backendName:"webgl",kernelFunc:u4e},p4e=class{constructor(e){this.variableNames=["x","W"],this.outputShape=e.outShape;let{inHeight:t,inWidth:n,padInfo:r,strideHeight:s,strideWidth:o,filterHeight:a,filterWidth:i,dilationHeight:l,dilationWidth:u}=e,{top:c,left:p}=r;this.userCode= `
const ivec2 strides = ivec2 ( $ { s } , $ { o } ) ;
const ivec2 pads = ivec2 ( $ { c } , $ { p } ) ;
2021-09-11 16:29:31 +02:00
const float neg _infinity = - 3.4 e38 ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d1 = coords . w ;
ivec2 outTopLeftCorner =
coords . yz * strides - pads ;
int hBeg = outTopLeftCorner . x ;
int wBeg = outTopLeftCorner . y ;
float curVal = neg _infinity ;
2021-10-04 22:29:15 +02:00
for ( int h = 0 ; h < $ { a } ; h ++ ) {
2021-09-11 16:29:31 +02:00
int hIn = hBeg + h * $ { l } ;
if ( hIn >= 0 && hIn < $ { t } ) {
2021-09-11 17:17:13 +02:00
for ( int w = 0 ; w < $ { i } ; w ++ ) {
2021-09-11 16:29:31 +02:00
int wIn = wBeg + w * $ { u } ;
if ( wIn >= 0 && wIn < $ { n } ) {
float xVal = getX ( batch , hIn , wIn , d1 ) ;
float wVal = getW ( h , w , d1 ) ;
float val = xVal + wVal ;
if ( val > curVal ) {
curVal = val ;
}
}
}
}
}
float result = curVal ;
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function d4e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l}=r,u=G.computeDilation2DInfo(s.shape,o.shape,a,i,"NHWC",l),c,p=new p4e(u);c=n.runWebGLProgram(p,[s,o],"float32");let d=qe({inputs:{x:c},backend:n,attrs:{shape:u.outShape}});return n.disposeIntermediateTensorInfo(c),d}var f4e={kernelName:B1,backendName:"webgl",kernelFunc:d4e};function h4e(e){let{inputs:t,backend:n,attrs:r}=e,{equation:s}=r,o=t,{allDims:a,summedDims:i,idDims:l}=G.decodeEinsumEquation(s,o.length);G.checkEinsumDimSizes(a.length,l,o);let{path:u,steps:c}=G.getEinsumComputePath(i,l),p=c.length,d=null,f=a.length,h=[];for(let m=0;m<p;++m){for(let g of c[m]){let{permutationIndices:x,expandDims:y}=G.getEinsumPermutation(f,l[g]),b;G.isIdentityPermutation(x)?b=o[g]:(b=Hs({inputs:{x:o[g]},backend:n,attrs:{perm:x}}),h.push(b));let A=b.shape.slice();for(let v=0;v<y.length;++v)A.splice(y[v],0,1);E.arraysEqual(b.shape,A)||(b=qe({inputs:{x:b},backend:n,attrs:{shape:A}}),h.push(b)),d===null?d=b:(d=PR({inputs:{a:b,b:d},backend:n}),h.push(d))}m<p-1&&(u[m]>=0&&(d=L5({inputs:{x:d},backend:n,attrs:{axis:u[m]-(a.length-f),keepDims:!1}}),h.push(d)),f--)}for(let m of h)m!==d&&n.disposeIntermediateTensorInfo(m);return d}var m4e={kernelName:fw,backendName:"webgl",kernelFunc:h4e},g4e="return (x >= 0.0) ? x : (exp(x) - 1.0);",x4e= `
2021-09-11 16:29:31 +02:00
vec4 result ;
result . r = ( x . r >= 0.0 ) ? x . r : ( exp ( x . r ) - 1.0 ) ;
result . g = ( x . g >= 0.0 ) ? x . g : ( exp ( x . g ) - 1.0 ) ;
result . b = ( x . b >= 0.0 ) ? x . b : ( exp ( x . b ) - 1.0 ) ;
result . a = ( x . a >= 0.0 ) ? x . a : ( exp ( x . a ) - 1.0 ) ;
return result ;
2021-10-04 22:29:15 +02:00
` ,y4e=tn({opSnippet:g4e,packedOpSnippet:x4e}),b4e={kernelName:Jd,backendName:"webgl",kernelFunc:y4e},A4e="return (b >= 1.0) ? a : a * (b + 1.0);",v4e= `
2021-09-11 16:29:31 +02:00
vec4 bGTEZero = vec4 ( greaterThanEqual ( b , vec4 ( 0. ) ) ) ;
return ( bGTEZero * a ) + ( ( vec4 ( 1.0 ) - bGTEZero ) * ( a * ( b + vec4 ( 1.0 ) ) ) ) ;
2021-10-04 22:29:15 +02:00
` ,w4e=e=>{let{inputs:t,backend:n}=e,{dy:r,y:s}=t,o=be().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new rA(v4e,r.shape,s.shape):new J0(A4e,r.shape,s.shape);return n.runWebGLProgram(o,[r,s],r.dtype)},k4e={kernelName:hw,backendName:"webgl",kernelFunc:w4e},I4e= `
2021-09-11 16:29:31 +02:00
return vec4 ( equal ( a , b ) ) ;
2021-10-04 22:29:15 +02:00
` ,C4e="return float(a == b);",S4e=ks({opSnippet:C4e,packedOpSnippet:I4e,dtype:"bool",cpuKernelImpl:pNe}),N4e={kernelName:Cg,backendName:"webgl",kernelFunc:S4e},T4e= `
2021-09-11 16:29:31 +02:00
// Error function is calculated approximately with elementary function.
// See "Handbook of Mathematical Functions with Formulas,
// Graphs, and Mathematical Tables", Abramowitz and Stegun.
2021-10-04 22:29:15 +02:00
float p = $ { G . ERF _P } ;
float a1 = $ { G . ERF _A1 } ;
float a2 = $ { G . ERF _A2 } ;
float a3 = $ { G . ERF _A3 } ;
float a4 = $ { G . ERF _A4 } ;
float a5 = $ { G . ERF _A5 } ;
2021-09-11 16:29:31 +02:00
float sign = sign ( x ) ;
x = abs ( x ) ;
float t = 1.0 / ( 1.0 + p * x ) ;
return sign * ( 1.0 - ( ( ( ( ( a5 * t + a4 ) * t ) + a3 ) * t + a2 ) * t + a1 ) * t * exp ( - x * x ) ) ;
2021-10-04 22:29:15 +02:00
` ,_4e=tn({opSnippet:T4e}),E4e={kernelName:Ig,backendName:"webgl",kernelFunc:_4e},JW="return exp(x);",QW=tn({opSnippet:JW,packedOpSnippet:JW,cpuKernelImpl:dNe}),R4e={kernelName:Qd,backendName:"webgl",kernelFunc:QW};function BR(e){let{inputs:t,attrs:n,backend:r}=e,{dim:s}=n,{input:o}=t,a=o.shape.length,i=o.shape.slice(),l=s;return s<0&&(E.assert(-(a+1)<=s,()=> ` Axis must be in the interval [ $ { - ( a + 1 ) } , $ { a } ] ` ),l=a+s+1),i.splice(l,0,1),qe({inputs:{x:o},backend:r,attrs:{shape:i}})}var D4e={kernelName:Sg,backendName:"webgl",kernelFunc:BR},eG="return exp(x) - 1.0;", $ 4e=tn({opSnippet:eG,packedOpSnippet:eG,cpuKernelImpl:fNe}),F4e={kernelName:Ng,backendName:"webgl",kernelFunc: $ 4e},tG=class{constructor(e,t,n){this.variableNames=["real","imag"];let r=t[1];this.outputShape=t;let s=n? ` 2.0 * $ { Math . PI } ` : ` - 2.0 * $ { Math . PI } ` ,o=n? ` $ { r } . 0 ` :"1.0",a;if(e==="real")a="return real * expR - imag * expI;";else if(e==="imag")a="return real * expI + imag * expR;";else throw new Error( ` FFT component must be either "real" or "imag" , got $ { e } . ` );this.userCode= `
const float exponentMultiplier = $ { s } ;
2021-09-11 16:29:31 +02:00
float unaryOpComplex ( float real , float expR , float imag , float expI ) {
2021-10-04 22:29:15 +02:00
$ { a }
2021-09-01 00:24:30 +02:00
}
2021-09-11 16:29:31 +02:00
float mulMatDFT ( int batch , int index ) {
2021-10-04 22:29:15 +02:00
float indexRatio = float ( index ) / float ( $ { r } ) ;
2021-09-11 16:29:31 +02:00
float exponentMultiplierTimesIndexRatio =
exponentMultiplier * indexRatio ;
float result = 0.0 ;
2021-10-04 22:29:15 +02:00
for ( int i = 0 ; i < $ { r } ; i ++ ) {
2021-09-11 16:29:31 +02:00
// x = (-2|2 * PI / N) * index * i;
float x = exponentMultiplierTimesIndexRatio * float ( i ) ;
float expR = cos ( x ) ;
float expI = sin ( x ) ;
float real = getReal ( batch , i ) ;
float imag = getImag ( batch , i ) ;
result +=
2021-10-04 22:29:15 +02:00
unaryOpComplex ( real , expR , imag , expI ) / $ { o } ;
2021-09-11 16:29:31 +02:00
}
return result ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
setOutput ( mulMatDFT ( coords [ 0 ] , coords [ 1 ] ) ) ;
}
2021-10-04 22:29:15 +02:00
` }};function nG(e,t,n){let r=n.texData.get(e.dataId),s=E.sizeFromShape(e.shape),o=e.shape[e.shape.length-1],a=s/o,i=qe({inputs:{x:e},backend:n,attrs:{shape:[a,o]}}),l=i.shape,u=new tG("real",l,t),c=new tG("imag",l,t),p=[{dataId:r.complexTensorInfos.real.dataId,dtype:r.complexTensorInfos.real.dtype,shape:l},{dataId:r.complexTensorInfos.imag.dataId,dtype:r.complexTensorInfos.imag.dtype,shape:l}],d=n.runWebGLProgram(u,p,"float32"),f=n.runWebGLProgram(c,p,"float32"),h=sp({inputs:{real:d,imag:f},backend:n});n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f);let m=qe({inputs:{x:h},backend:n,attrs:{shape:e.shape}});return n.disposeIntermediateTensorInfo(i),n.disposeIntermediateTensorInfo(h),m}function O4e(e){let{inputs:t,backend:n}=e,{input:r}=t;return nG(r,!1,n)}var P4e={kernelName:mw,backendName:"webgl",kernelFunc:O4e},M4e=class{constructor(e,t){this.outputShape=[],this.customUniforms=[{name:"value",type:"float"}],this.variableNames=["x"],this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
// Input can be obtained from uniform value.
setOutput ( value ) ;
}
2021-10-04 22:29:15 +02:00
` }};function aA(e){let{backend:t,attrs:n}=e,{shape:r,value:s}=n,{dtype:o}=n;if(o=o||E.inferDtype(s),o==="string"){let a=E.getArrayFromDType(o,E.sizeFromShape(r));return a.fill(s),t.makeTensorInfo(r,o,a)}else{let a=new M4e(r,s),i=[[s]];return t.runWebGLProgram(a,[],o,i)}}var z4e={kernelName:V1,backendName:"webgl",kernelFunc:aA},L4e=class{constructor(e){this.variableNames=["Image"],this.outputShape=[];let t=e[2];this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int x = coords [ 2 ] ;
int coordX = $ { t } - x - 1 ;
float outputValue ;
if ( coordX >= 0 && coordX < $ { t } ) {
outputValue = getImage ( coords [ 0 ] , coords [ 1 ] , coordX , coords [ 3 ] ) ;
} else {
outputValue = getImage ( coords [ 0 ] , coords [ 1 ] , coords [ 2 ] , coords [ 3 ] ) ;
}
setOutput ( outputValue ) ;
}
2021-10-04 22:29:15 +02:00
` }},B4e={kernelName:Tg,backendName:"webgl",kernelFunc:({inputs:e,backend:t})=>{let{image:n}=e,r=t,s=new L4e(n.shape);return r.runWebGLProgram(s,[n],n.dtype)}},rG="return floor(x);",V4e=tn({opSnippet:rG,packedOpSnippet:rG,cpuKernelImpl:hNe}),W4e={kernelName:ef,backendName:"webgl",kernelFunc:V4e},G4e= `
2021-09-11 16:29:31 +02:00
float s = sign ( a ) * sign ( b ) ;
int ia = round ( a ) ;
int ib = round ( b ) ;
if ( ib != 0 ) {
// Windows (D3D) wants guaranteed non-zero int division at compile-time.
return float ( idiv ( ia , ib , s ) ) ;
} else {
return NAN ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` ,U4e= `
2021-09-11 16:29:31 +02:00
ivec4 ia = round ( a ) ;
ivec4 ib = round ( b ) ;
bvec4 cond = notEqual ( ib , ivec4 ( 0 ) ) ;
ivec4 result = ivec4 ( 0 ) ;
vec4 s = sign ( a ) * sign ( b ) ;
// Windows (D3D) wants guaranteed non-zero int division at compile-time.
if ( cond [ 0 ] ) {
result [ 0 ] = idiv ( ia [ 0 ] , ib [ 0 ] , s [ 0 ] ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( cond [ 1 ] ) {
result [ 1 ] = idiv ( ia [ 1 ] , ib [ 1 ] , s [ 1 ] ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( cond [ 2 ] ) {
result [ 2 ] = idiv ( ia [ 2 ] , ib [ 2 ] , s [ 2 ] ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( cond [ 3 ] ) {
result [ 3 ] = idiv ( ia [ 3 ] , ib [ 3 ] , s [ 3 ] ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
return vec4 ( result ) ;
2021-10-04 22:29:15 +02:00
` ,j4e=ks({opSnippet:G4e,packedOpSnippet:U4e,dtype:"int32"}),H4e={kernelName:tf,backendName:"webgl",kernelFunc:j4e},q4e=class{constructor(e){this.variableNames=["A"];let t=Us(),[n,r]=e;this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int texR = coords [ 0 ] ;
int texC = coords [ 1 ] ;
int depth = coords [ 2 ] ;
2021-10-04 22:29:15 +02:00
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { r } . 0 , $ { n } . 0 ) ;
2021-09-11 16:29:31 +02:00
vec4 values = $ { t . texture2D } ( A , uv ) ;
float value ;
if ( depth == 0 ) {
value = values . r ;
} else if ( depth == 1 ) {
value = values . g ;
} else if ( depth == 2 ) {
value = values . b ;
} else if ( depth == 3 ) {
value = values . a ;
}
setOutput ( floor ( value * 255.0 + 0.5 ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},K4e=class{constructor(e){this.variableNames=["A"],this.packedInputs=!1,this.packedOutput=!0;let t=Us(),[n,r]=e;this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int texR = coords [ 0 ] ;
int texC = coords [ 1 ] ;
int depth = coords [ 2 ] ;
vec4 result = vec4 ( 0. ) ;
for ( int row = 0 ; row <= 1 ; row ++ ) {
for ( int col = 0 ; col <= 1 ; col ++ ) {
texC = coords [ 1 ] + row ;
depth = coords [ 2 ] + col ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
2021-10-04 22:29:15 +02:00
vec2 ( $ { r } . 0 , $ { n } . 0 ) ;
2021-09-11 16:29:31 +02:00
vec4 values = $ { t . texture2D } ( A , uv ) ;
float value ;
if ( depth == 0 ) {
value = values . r ;
} else if ( depth == 1 ) {
value = values . g ;
} else if ( depth == 2 ) {
value = values . b ;
} else if ( depth == 3 ) {
value = values . a ;
}
result [ row * 2 + col ] = floor ( value * 255.0 + 0.5 ) ;
}
}
$ { t . output } = result ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},X4e={kernelName:Pw,backendName:"webgl",kernelFunc:Z4e},tx;function Z4e(e){let{inputs:t,backend:n,attrs:r}=e,{pixels:s}=t,{numChannels:o}=r,a=typeof HTMLVideoElement!="undefined"&&s instanceof HTMLVideoElement,i=typeof HTMLImageElement!="undefined"&&s instanceof HTMLImageElement,[l,u]=a?[s.videoWidth,s.videoHeight]:[s.width,s.height],c=[u,l],p=[u,l,o];(i||a)&&(tx==null&&(tx=document.createElement("canvas").getContext("2d")),tx.canvas.width=l,tx.canvas.height=u,tx.drawImage(s,0,0,l,u),s=tx.canvas);let d=n.makeTensorInfo(c,"int32");n.texData.get(d.dataId).usage=ma.PIXELS,n.gpgpu.uploadPixelDataToTexture(n.getTexture(d.dataId),s);let f=be().getBool("WEBGL_PACK")?new K4e(p):new q4e(p),h=n.runWebGLProgram(f,[d],"int32");return n.disposeData(d.dataId),h}function Y4e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o,bias:a,preluActivationWeights:i}=t,{strides:l,pad:u,dataFormat:c,dilations:p,dimRoundingMode:d,activation:f,leakyreluAlpha:h}=r,m=G.convertConv2DDataFormat(c),g=G.computeConv2DInfo(s.shape,o.shape,l,p,u,d,!1,m),x,y=[];if(g.filterHeight===1&&g.filterWidth===1&&g.dilationHeight===1&&g.dilationWidth===1&&g.strideHeight===1&&g.strideWidth===1&&(g.padInfo.type==="SAME"||g.padInfo.type==="VALID"))x=jW({x:s,filter:o,convInfo:g,backend:n,bias:a,activation:f,preluActivationWeights:i,leakyreluAlpha:h});else if(be().getBool("WEBGL_CONV_IM2COL")&&s.shape[0]===1)x=HW({x:s,filter:o,convInfo:g,backend:n,bias:a,activation:f,preluActivationWeights:i,leakyreluAlpha:h});else{let A=a!=null,v=i!=null,w=f==="leakyrelu",k=f?M5(f,!1):null,I=new UW(g,A,k,v,w),N=[s,o];if(a&&N.push(a),i&&N.push(i),w){let T=n.makeTensorInfo([],"float32",E.createScalarValue(h,"float32"));N.push(T),y.push(T)}x=n.runWebGLProgram(I,N,"float32")}let b=qe({inputs:{x},backend:n,attrs:{shape:g.outShape}});return y.push(x),y.forEach(A=>n.disposeIntermediateTensorInfo(A)),b}var J4e={kernelName:Mf,backendName:"webgl",kernelFunc:Y4e};function Q4e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o,bias:a,preluActivationWeights:i}=t,{strides:l,pad:u,dilations:c,dimRoundingMode:p,activation:d,leakyreluAlpha:f}=r,h=[],m=c;m==null&&(m=[1,1]),E.assert(G.eitherStridesOrDilationsAreOne(l,m),()=> ` Error in depthwiseConv2d : Either strides or dilations must be 1. Got strides $ { l } and dilations '${m}' ` );let g=G.computeConv2DInfo(s.shape,o.shape,l,m,u,p,!0),x=be().getBool("WEBGL_PACK_DEPTHWISECONV")&&g.strideWidth<=2&&g.outChannels/g.inChannels==1,y=d?M5(d,x):null,b=[s,o],A=a!=null,v=i!=null,w=d==="leakyrelu";if(A&&b.push(a),v&&b.push(i),w){let T=n.makeTensorInfo([],"float32",E.createScalarValue(f,"float32"));b.push(T),h.push(T)}let k;x?k=new YW(g,A,y,v,w):k=new ZW(g,A,y,v,w);let I=[[g.padInfo.top,g.padInfo.left],[g.strideHeight,g.strideWidth],[g.dilationHeight,g.dilationWidth],[g.inHeight,g.inWidth]],N=n.runWebGLProgram(k,b,"float32",I);return h.forEach(T=>n.disposeIntermediateTensorInfo(T)),N}var eRe={kernelName:zf,backendName:"webgl",kernelFunc:Q4e},tRe=class{constructor(e,t,n){this.sliceDim=e,this.strides=t,this.variableNames=["x","indices"],this.outputShape=n;let r=Dn(t.length),s=Dn(n.length),o=this.sliceDim>1?"strides[j]":"strides";this.userCode= `
$ { r } strides = $ { r } ( $ { this . strides } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } coords = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
int flattenIndex = 0 ;
for ( int j = 0 ; j < $ { this . sliceDim } ; j ++ ) {
int index = round ( getIndices ( coords [ 0 ] , j ) ) ;
2021-10-04 22:29:15 +02:00
flattenIndex += index * $ { o } ;
2021-09-11 16:29:31 +02:00
}
setOutput ( getX ( flattenIndex , coords [ 1 ] ) ) ;
2021-09-01 00:24:30 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function nRe(e){let{inputs:t,backend:n}=e,{params:r,indices:s}=t,o=s.shape,a=o[o.length-1],i=E.sizeFromShape(r.shape),[l,u,c,p]=G.prepareAndValidate(r,s),d=qe({inputs:{x:s},backend:n,attrs:{shape:[u,a]}}),f=qe({inputs:{x:r},backend:n,attrs:{shape:[E.sizeFromShape(r.shape)/c,c]}});if(n.shouldExecuteOnCPU([r,s])||r.dtype==="string"){let x=n.readSync(s.dataId),y=n.bufferSync(r),b=mNe(x,y,r.dtype,u,a,c,p,r.shape,i);return n.makeTensorInfo(l,r.dtype,b.values)}let h=new tRe(a,p,[u,c]),m=n.runWebGLProgram(h,[f,d],f.dtype),g=qe({inputs:{x:m},backend:n,attrs:{shape:l}});return n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(m),g}var rRe={kernelName:Eg,backendName:"webgl",kernelFunc:nRe},sRe=class{constructor(e,t){this.variableNames=["A","indices"],this.outputShape=t,this.rank=t.length;let n=Dn(this.rank),r=oRe(e,2);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
$ { n } resRC = getOutputCoords ( ) ;
2021-10-04 22:29:15 +02:00
setOutput ( getA ( $ { r } ) ) ;
2021-09-01 00:24:30 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function oRe(e,t){let n=["resRC.x","resRC.y","resRC.z","resRC.w"],r=[];for(let s=0;s<e.length;s++)s===2?r.push("int(getIndices(resRC.x, resRC.z))"):r.push( ` $ { n [ s ] } ` );return r.join()}function sG(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,indices:o}=t,{axis:a,batchDims:i}=r,l=E.parseAxisParam(a,s.shape)[0],u=G.segment_util.collectGatherOpShapeInfo(s,o,l,i),c=E.sizeFromShape(o.shape),p=[],d=qe({inputs:{x:s},backend:n,attrs:{shape:[u.batchSize,u.outerSize,u.dimSize,u.sliceSize]}}),f=qe({inputs:{x:o},backend:n,attrs:{shape:[u.batchSize,c/u.batchSize]}});p.push(d),p.push(f);let h=[u.batchSize,u.outerSize,c/u.batchSize,u.sliceSize];if(n.shouldExecuteOnCPU([s,o])||s.dtype==="string"){let y=n.bufferSync(f),b=n.bufferSync(d),A=gNe(b,y,h);return p.forEach(v=>n.disposeIntermediateTensorInfo(v)),n.makeTensorInfo(u.outputShape,A.dtype,A.values)}let m=new sRe(d.shape,h),g=n.runWebGLProgram(m,[d,f],d.dtype);p.push(g);let x=qe({inputs:{x:g},backend:n,attrs:{shape:u.outputShape}});return p.forEach(y=>n.disposeIntermediateTensorInfo(y)),x}var aRe={kernelName:_g,backendName:"webgl",kernelFunc:sG},iRe="return float(a > b);",lRe= `
2021-09-11 16:29:31 +02:00
return vec4 ( greaterThan ( a , b ) ) ;
2021-10-04 22:29:15 +02:00
` ,uRe=ks({opSnippet:iRe,packedOpSnippet:lRe,cpuKernelImpl:xNe,dtype:"bool"}),cRe={kernelName:Rg,backendName:"webgl",kernelFunc:uRe},pRe="return float(a >= b);",dRe= `
2021-09-11 16:29:31 +02:00
return vec4 ( greaterThanEqual ( a , b ) ) ;
2021-10-04 22:29:15 +02:00
` ,fRe=ks({opSnippet:pRe,packedOpSnippet:dRe,dtype:"bool",cpuKernelImpl:yNe}),hRe={kernelName:rf,backendName:"webgl",kernelFunc:fRe};function mRe(e){let{inputs:t,backend:n}=e,{input:r}=t;return nG(r,!0,n)}var gRe={kernelName:gw,backendName:"webgl",kernelFunc:mRe},xRe="return float(!isnan(x) && !isinf(x));",yRe=tn({opSnippet:xRe,dtype:"bool"}),bRe={kernelName:Dg,backendName:"webgl",kernelFunc:yRe},ARe="return float(isinf(x));",vRe=tn({opSnippet:ARe,dtype:"bool"}),wRe={kernelName: $ g,backendName:"webgl",kernelFunc:vRe},kRe="return float(isnan(x));",IRe=tn({opSnippet:kRe,dtype:"bool"}),CRe={kernelName:Fg,backendName:"webgl",kernelFunc:IRe},SRe="return float(a < b);",NRe= `
2021-09-11 16:29:31 +02:00
return vec4 ( lessThan ( a , b ) ) ;
2021-10-04 22:29:15 +02:00
` ,TRe=ks({opSnippet:SRe,packedOpSnippet:NRe,cpuKernelImpl:bNe,dtype:"bool"}),_Re={kernelName:Og,backendName:"webgl",kernelFunc:TRe},ERe="return float(a <= b);",RRe= `
2021-09-11 16:29:31 +02:00
return vec4 ( lessThanEqual ( a , b ) ) ;
2021-10-04 22:29:15 +02:00
` ,DRe=ks({opSnippet:ERe,packedOpSnippet:RRe,cpuKernelImpl:ANe,dtype:"bool"}), $ Re={kernelName:Pg,backendName:"webgl",kernelFunc:DRe};function FRe(e){let{backend:t,attrs:n}=e,{start:r,stop:s,num:o}=n,a=vNe(r,s,o);return t.makeTensorInfo([a.length],"float32",a)}var ORe={kernelName:yw,backendName:"webgl",kernelFunc:FRe},PRe= ` if ( x < 0.0 ) return NAN ;
return log ( x ) ; ` ,MRe= `
2021-09-11 16:29:31 +02:00
vec4 result = log ( x ) ;
vec4 isNaN = vec4 ( lessThan ( x , vec4 ( 0.0 ) ) ) ;
result . r = isNaN . r == 1.0 ? NAN : result . r ;
result . g = isNaN . g == 1.0 ? NAN : result . g ;
result . b = isNaN . b == 1.0 ? NAN : result . b ;
result . a = isNaN . a == 1.0 ? NAN : result . a ;
return result ;
2021-10-04 22:29:15 +02:00
` ,zRe=tn({opSnippet:PRe,packedOpSnippet:MRe,cpuKernelImpl:wNe}),LRe={kernelName:af,backendName:"webgl",kernelFunc:zRe},BRe="return log(1.0 + x);",VRe=tn({opSnippet:BRe}),WRe={kernelName:Mg,backendName:"webgl",kernelFunc:VRe},GRe="return float(a >= 1.0 && b >= 1.0);",URe= `
2021-09-11 16:29:31 +02:00
return vec4 (
vec4 ( greaterThanEqual ( a , vec4 ( 1.0 ) ) ) *
vec4 ( greaterThanEqual ( b , vec4 ( 1.0 ) ) ) ) ;
2021-10-04 22:29:15 +02:00
` ,jRe=ks({opSnippet:GRe,packedOpSnippet:URe,dtype:"bool"}),HRe={kernelName:zg,backendName:"webgl",kernelFunc:jRe},qRe="return float(!(x >= 1.0));",KRe=tn({opSnippet:qRe}),XRe={kernelName:W1,backendName:"webgl",kernelFunc:KRe},ZRe="return float(a >= 1.0 || b >= 1.0);",YRe= `
2021-09-11 16:29:31 +02:00
return min (
vec4 ( greaterThanEqual ( a , vec4 ( 1.0 ) ) ) +
vec4 ( greaterThanEqual ( b , vec4 ( 1.0 ) ) ) ,
vec4 ( 1.0 ) ) ;
2021-10-04 22:29:15 +02:00
` ,JRe=ks({opSnippet:ZRe,packedOpSnippet:YRe,dtype:"bool"}),QRe={kernelName:G1,backendName:"webgl",kernelFunc:JRe},e7e=class{constructor(e,t,n,r,s){this.variableNames=["x"],this.outputShape=[];let o=t,a=e[3]-1;this.outputShape=e;let i,l= ` float ( $ { n } ) + float ( $ { r } ) * sum ` ;s===.5?i= ` inversesqrt ( $ { l } ) ` :s===1?i= ` 1.0 / ( $ { l } ) ` :i= ` exp ( log ( $ { l } ) * float ( - $ { s } ) ) ; ` ,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
int d = coords [ 3 ] ;
float x = getX ( b , r , c , d ) ;
float sum = 0.0 ;
2021-10-04 22:29:15 +02:00
for ( int j = - $ { o } ; j <= $ { o } ; j ++ ) {
2021-09-11 16:29:31 +02:00
int idx = d + j ;
2021-10-04 22:29:15 +02:00
if ( idx >= 0 && idx <= $ { a } ) {
2021-09-11 16:29:31 +02:00
float z = getX ( b , r , c , idx ) ;
sum += z * z ;
}
}
2021-09-11 17:17:13 +02:00
float val = x * $ { i } ;
2021-09-11 16:29:31 +02:00
setOutput ( val ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},t7e=class{constructor(e,t,n,r,s){this.variableNames=["x"],this.outputShape=[],this.packedInputs=!0,this.packedOutput=!0;let o=t,a=e[3]-1;this.outputShape=e;let i,l= ` float ( $ { n } ) + float ( $ { r } ) * sum ` ;s===.5?i= ` inversesqrt ( $ { l } ) ` :s===1?i= ` 1.0 / ( $ { l } ) ` :i= ` exp ( log ( $ { l } ) * float ( - $ { s } ) ) ; ` ,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords . x ;
int r = coords . y ;
int c = coords . z ;
int d = coords . w ;
bool hasNextCol = d < $ { this . outputShape [ 3 ] } ;
bool hasNextRow = c < $ { this . outputShape [ 2 ] } ;
vec4 sum = vec4 ( 0. ) ;
vec4 xFragAtOutputCoords = getX ( b , r , c , d ) ;
vec4 xAtOutputCoords = vec4 (
getChannel ( xFragAtOutputCoords , vec2 ( c , d ) ) ,
hasNextCol ?
getChannel ( xFragAtOutputCoords , vec2 ( c , d + 1 ) ) : 0.0 ,
hasNextRow ?
getChannel ( xFragAtOutputCoords , vec2 ( c + 1 , d ) ) : 0.0 ,
( hasNextRow && hasNextCol ) ?
getChannel ( xFragAtOutputCoords , vec2 ( c + 1 , d + 1 ) ) : 0.0
) ;
2021-10-04 22:29:15 +02:00
int firstChannel = d - $ { o } ;
2021-09-11 16:29:31 +02:00
vec2 cache = vec2 ( 0. ) ;
if ( firstChannel >= 0 ) {
vec4 firstChannelFrag = getX ( b , r , c , firstChannel ) ;
cache . x = getChannel ( firstChannelFrag , vec2 ( c , firstChannel ) ) ;
if ( hasNextRow ) {
cache . y = getChannel ( firstChannelFrag , vec2 ( c + 1 , firstChannel ) ) ;
}
}
ivec2 depth = ivec2 ( d , d + 1 ) ;
2021-10-04 22:29:15 +02:00
for ( int j = - $ { o } ; j <= $ { o } ; j ++ ) {
2021-09-11 16:29:31 +02:00
ivec2 idx = depth + j ;
bvec2 aboveLowerBound = greaterThanEqual ( idx , ivec2 ( 0 ) ) ;
2021-10-04 22:29:15 +02:00
bvec2 belowUpperBound = lessThanEqual ( idx , ivec2 ( $ { a } ) ) ;
2021-09-11 16:29:31 +02:00
bool depthInRange = aboveLowerBound . x && belowUpperBound . x ;
bool depthPlusOneInRange = aboveLowerBound . y && belowUpperBound . y ;
if ( depthInRange || depthPlusOneInRange ) {
vec4 z = vec4 ( 0. ) ;
vec4 xFragAtCurrentDepth ;
z . xz = cache . xy ;
if ( depthPlusOneInRange && hasNextCol ) {
xFragAtCurrentDepth = idx . y != d ?
getX ( b , r , c , idx . y ) : xFragAtOutputCoords ;
z . y = getChannel ( xFragAtCurrentDepth , vec2 ( c , idx . y ) ) ;
if ( hasNextRow ) {
z . w = getChannel ( xFragAtCurrentDepth , vec2 ( c + 1 , idx . y ) ) ;
}
}
cache . xy = z . yw ;
sum += z * z ;
}
}
2021-09-11 17:17:13 +02:00
vec4 result = xAtOutputCoords * $ { i } ;
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},n7e=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{depthRadius:o,bias:a,alpha:i,beta:l}=r,u=be().getBool("WEBGL_PACK_NORMALIZATION")?new t7e(s.shape,o,a,i,l):new e7e(s.shape,o,a,i,l);return n.runWebGLProgram(u,[s],s.dtype)},r7e={kernelName:U1,backendName:"webgl",kernelFunc:n7e},s7e=class{constructor(e,t,n,r,s){this.variableNames=["inputImage","outputImage","dy"],this.outputShape=[],this.outputShape=e,this.depth=e[3],this.depthRadius=t,this.bias=n,this.alpha=r,this.beta=s,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
float result = 0.0 ;
for ( int d = 0 ; d < $ { this . depth } ; ++ d ) {
int depthBegin = int ( max ( 0.0 , float ( d - $ { t } ) ) ) ;
int depthEnd = int ( min ( float ( $ { this . depth } ) ,
float ( d + $ { t } + 1 ) ) ) ;
const int MIN _DEPTH _BEGIN = 0 ;
const int MAX _DEPTH _END = $ { this . depth } ;
float norm = 0.0 ;
for ( int k = MIN _DEPTH _BEGIN ; k < MAX _DEPTH _END ; ++ k ) {
if ( k < depthBegin ) {
continue ;
}
else if ( k >= depthBegin && k < depthEnd ) {
norm += getInputImage ( b , r , c , k ) * getInputImage ( b , r , c , k ) ;
}
else {
break ;
}
}
2021-10-04 22:29:15 +02:00
norm = float ( $ { r } ) * norm + float ( $ { n } ) ;
2021-09-11 16:29:31 +02:00
for ( int k = MIN _DEPTH _BEGIN ; k < MAX _DEPTH _END ; ++ k ) {
if ( k < depthBegin ) {
continue ;
}
else if ( k >= depthBegin && k < depthEnd ) {
2021-10-04 22:29:15 +02:00
float dyi = - 2.0 * float ( $ { r } )
* float ( $ { s } )
2021-09-11 16:29:31 +02:00
* getInputImage ( b , r , c , k ) * getOutputImage ( b , r , c , d )
/ n o r m ;
if ( k == d ) {
2021-10-04 22:29:15 +02:00
dyi += pow ( norm , - 1.0 * $ { s } ) ;
2021-09-11 16:29:31 +02:00
}
if ( k == coords [ 3 ] ) {
dyi *= getDy ( b , r , c , d ) ;
result += dyi ;
}
}
else {
break ;
}
}
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},o7e=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s,y:o,dy:a}=t,{depthRadius:i,bias:l,alpha:u,beta:c}=r,p=new s7e(s.shape,i,l,u,c);return n.runWebGLProgram(p,[s,o,a],s.dtype)},a7e={kernelName:bw,backendName:"webgl",kernelFunc:o7e};function i7e(e,t,n,r){let s=E.sizeFromShape(t),a=E.sizeFromShape(e.shape)/s,i=qe({inputs:{x:e},attrs:{shape:[a,s]},backend:r}),l=xh(i,e.dtype,"max",r),u=qe({inputs:{x:l},attrs:{shape:n},backend:r});return r.disposeIntermediateTensorInfo(i),r.disposeIntermediateTensorInfo(l),u}function oG(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reductionIndices:o,keepDims:a}=r,i=s.shape.length,l=E.parseAxisParam(o,s.shape),u=l,c=G.getAxesPermutation(u,i),p=c!=null,d=n.shouldExecuteOnCPU([s]),f=s;if(p){if(d){let b=n.texData.get(f.dataId).values,A=new Array(i);for(let k=0;k<A.length;k++)A[k]=s.shape[c[k]];let v=OR(b,s.shape,s.dtype,c,A);f=n.makeTensorInfo(A,s.dtype);let w=n.texData.get(f.dataId);w.values=v}else f=z5(s,c,n);u=G.getInnerMostAxes(u.length,i)}G.assertAxesAreInnerMostDims("max",u,i);let[h,m]=G.computeOutAndReduceShapes(f.shape,u),g=h;a&&(g=G.expandShapeToKeepDim(h,l));let x;if(d){let b=n.texData.get(f.dataId).values,A=kNe(b,E.sizeFromShape(m),g,s.dtype);x=n.makeTensorInfo(g,s.dtype);let v=n.texData.get(x.dataId);v.values=A}else x=i7e(f,m,g,n);return p&&n.disposeIntermediateTensorInfo(f),x}var l7e={kernelName:lf,backendName:"webgl",kernelFunc:oG},u7e=kW+ `
2021-09-11 16:29:31 +02:00
return max ( a , b ) ;
2021-10-04 22:29:15 +02:00
` ,c7e= `
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( max ( a , b ) ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
2021-10-04 22:29:15 +02:00
` +P5+ `
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,p7e=ks({opSnippet:u7e,packedOpSnippet:c7e,cpuKernelImpl:INe}),d7e={kernelName:uf,backendName:"webgl",kernelFunc:p7e};function f7e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t;j0(s,"maxPool");let{filterSize:o,strides:a,pad:i,dimRoundingMode:l}=r,u=1;E.assert(G.eitherStridesOrDilationsAreOne(a,u),()=> ` Error in maxPool : Either strides or dilations must be 1. Got strides $ { a } and dilations '${u}' ` );let c=G.computePool2DInfo(s.shape,o,a,u,i,l);if(c.filterWidth===1&&c.filterHeight===1&&E.arraysEqual(c.inShape,c.outShape))return Xo({inputs:{x:s},backend:n});let p=new sA(c,"max",!1);return n.runWebGLProgram(p,[s],s.dtype)}var h7e={kernelName:cf,backendName:"webgl",kernelFunc:f7e};function m7e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{filterSize:o,strides:a,pad:i,dataFormat:l,dimRoundingMode:u}=r,c=[1,1,1],p=G.computePool3DInfo(s.shape,o,a,c,i,u,l),d=new MR(p,"max",!1);return n.runWebGLProgram(d,[s],s.dtype)}var g7e={kernelName:j1,backendName:"webgl",kernelFunc:m7e},x7e=class{constructor(e){this.variableNames=["dy","maxPos"],this.outputShape=e.inShape;let t=e.strideHeight,n=e.strideWidth,r=e.dilationHeight,s=e.effectiveFilterHeight,o=e.effectiveFilterWidth,a=s-1-e.padInfo.top,i=o-1-e.padInfo.left,l=s*o-1;this.userCode= `
const ivec2 pads = ivec2 ( $ { a } , $ { i } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 dyRCCorner = coords . yz - pads ;
int dyRCorner = dyRCCorner . x ;
int dyCCorner = dyRCCorner . y ;
// Convolve dy(?, ?, d) with pos mask(:, :, d) to get dx(xR, xC, d).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
2021-10-04 22:29:15 +02:00
for ( int wR = 0 ; wR < $ { s } ;
wR += $ { r } ) {
2021-09-11 16:29:31 +02:00
float dyR = float ( dyRCorner + wR ) / $ { t } . 0 ;
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
2021-10-04 22:29:15 +02:00
for ( int wC = 0 ; wC < $ { o } ; wC ++ ) {
2021-09-11 16:29:31 +02:00
float dyC = float ( dyCCorner + wC ) / $ { n } . 0 ;
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( b , idyR , idyC , d ) ;
int maxPosValue = $ { l } - int ( getMaxPos ( b , idyR , idyC , d ) ) ;
// Get the current value, check it against the value from the
// position matrix.
2021-10-04 22:29:15 +02:00
int curPosValue = wR * $ { o } + wC ;
2021-09-11 16:29:31 +02:00
float mask = float ( maxPosValue == curPosValue ? 1.0 : 0.0 ) ;
dotProd += dyValue * mask ;
}
}
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},y7e=class{constructor(e){this.variableNames=["dy","maxPos"],this.outputShape=e.inShape;let t=e.strideDepth,n=e.strideHeight,r=e.strideWidth,s=e.dilationDepth,o=e.dilationHeight,a=e.dilationWidth,i=e.effectiveFilterDepth,l=e.effectiveFilterHeight,u=e.effectiveFilterWidth,c=i-1-e.padInfo.front,p=l-1-e.padInfo.top,d=u-1-e.padInfo.left,f=i*l*u-1;this.userCode= `
const ivec3 pads = ivec3 ( $ { c } , $ { p } , $ { d } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyDCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
// Convolve dy(?, ?, ?, ch) with pos mask(:, :, :, d) to get
// dx(xD, xR, xC, ch).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
2021-09-11 17:17:13 +02:00
for ( int wD = 0 ; wD < $ { i } ;
2021-10-04 22:29:15 +02:00
wD += $ { s } ) {
2021-09-11 16:29:31 +02:00
float dyD = float ( dyDCorner + wD ) / $ { t } . 0 ;
if ( dyD < 0.0 || dyD >= $ { e . outDepth } . 0 || fract ( dyD ) > 0.0 ) {
continue ;
}
int idyD = int ( dyD ) ;
for ( int wR = 0 ; wR < $ { l } ;
2021-10-04 22:29:15 +02:00
wR += $ { o } ) {
2021-09-11 16:29:31 +02:00
float dyR = float ( dyRCorner + wR ) / $ { n } . 0 ;
if ( dyR < 0.0 || dyR >= $ { e . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
for ( int wC = 0 ; wC < $ { u } ;
2021-10-04 22:29:15 +02:00
wC += $ { a } ) {
float dyC = float ( dyCCorner + wC ) / $ { r } . 0 ;
2021-09-11 16:29:31 +02:00
if ( dyC < 0.0 || dyC >= $ { e . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( batch , idyD , idyR , idyC , ch ) ;
2021-10-04 22:29:15 +02:00
int maxPosValue = $ { f } -
2021-09-11 16:29:31 +02:00
int ( getMaxPos ( batch , idyD , idyR , idyC , ch ) ) ;
// Get the current value, check it against the value from the
// position matrix.
int curPosValue =
wD * $ { l } * $ { u } +
wR * $ { u } + wC ;
float mask = float ( maxPosValue == curPosValue ? 1.0 : 0.0 ) ;
dotProd += dyValue * mask ;
}
}
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( dotProd ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function b7e(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o,{filterSize:i,strides:l,pad:u,dimRoundingMode:c}=r,p=[1,1,1],d=G.computePool3DInfo(a.shape,i,l,p,u,c),f=new MR(d,"max",!0),h=n.runWebGLProgram(f,[a],a.dtype),m=new y7e(d),g=n.runWebGLProgram(m,[s,h],a.dtype);return n.disposeIntermediateTensorInfo(h),g}var A7e={kernelName:vw,backendName:"webgl",kernelFunc:b7e};function v7e(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o,output:a}=t,i=o;j0([o,a],"maxPoolGrad");let{filterSize:l,strides:u,pad:c,dimRoundingMode:p}=r,d=G.computePool2DInfo(i.shape,l,u,1,c,p),f=!0,h=new sA(d,"max",f),m=n.runWebGLProgram(h,[i],i.dtype),g=new x7e(d),x=n.runWebGLProgram(g,[s,m],i.dtype);return n.disposeIntermediateTensorInfo(m),x}var w7e={kernelName:Aw,backendName:"webgl",kernelFunc:v7e};function k7e(e,t,n,r){let s=new sA(n,"max",!1),o=r.runWebGLProgram(s,[e],"float32");s=new sA(n,"max",!0,!0,t);let a=r.runWebGLProgram(s,[e],"float32");return[o,a]}var I7e={kernelName:ww,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{x:r}=e,{filterSize:s,strides:o,pad:a,includeBatchInIndex:i}=t,l=n;E.assert(r.shape.length===4,()=> ` Error in maxPool : input must be rank 4 but got rank $ { r . shape . length } . ` );let u=[1,1];E.assert(G.eitherStridesOrDilationsAreOne(o,u),()=> ` Error in maxPool : Either strides or dilations must be 1. Got strides $ { o } and dilations '${u}' ` );let c=G.computePool2DInfo(r.shape,s,o,u,a),[p,d]=k7e(r,i,c,l);return[p,d]}};function C7e(e,t,n,r){let s=E.sizeFromShape(t),a=E.sizeFromShape(e.shape)/s,i=qe({inputs:{x:e},attrs:{shape:[a,s]},backend:r}),l=xh(i,"float32","mean",r),u=qe({inputs:{x:l},attrs:{shape:n},backend:r});return r.disposeIntermediateTensorInfo(i),r.disposeIntermediateTensorInfo(l),u}var S7e={kernelName:pf,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{x:r}=e,{keepDims:s,axis:o}=t,a=n,i=r.shape.length,l=E.parseAxisParam(o,r.shape),u=l,c=G.getAxesPermutation(u,i),p=c!=null,d=a.shouldExecuteOnCPU([r]),f=[],h=r;if(p){if(d){let A=a.texData.get(h.dataId).values,v=new Array(i);for(let I=0;I<v.length;I++)v[I]=r.shape[c[I]];let w=OR(A,r.shape,r.dtype,c,v);h=a.makeTensorInfo(v,r.dtype);let k=a.texData.get(h.dataId);k.values=w}else h=z5(r,c,a);f.push(h),u=G.getInnerMostAxes(u.length,i)}G.assertAxesAreInnerMostDims("sum",u,i);let[m,g]=G.computeOutAndReduceShapes(h.shape,u),x=m;s&&(x=G.expandShapeToKeepDim(m,l));let y=C7e(h,g,x,a);for(let b of f)a.disposeIntermediateTensorInfo(b);return y}};function N7e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=E.parseAxisParam(o,s.shape),u=l,c=G.getAxesPermutation(u,i),p=s;c!=null&&(p=Hs({inputs:{x:s},backend:n,attrs:{perm:c}}),u=G.getInnerMostAxes(u.length,s.shape.length)),G.assertAxesAreInnerMostDims("min",u,i);let[d,f]=G.computeOutAndReduceShapes(p.shape,u),h=E.sizeFromShape(f),m=qe({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=xh(m,m.dtype,"min",n),x;if(a){let y=G.expandShapeToKeepDim(d,l);x=qe({inputs:{x:g},backend:n,attrs:{shape:y}})}else x=qe({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var T7e={kernelName:df,backendName:"webgl",kernelFunc:N7e},_7e=kW+ `
2021-09-11 16:29:31 +02:00
return min ( a , b ) ;
2021-10-04 22:29:15 +02:00
` ,E7e= `
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( min ( a , b ) ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
2021-10-04 22:29:15 +02:00
` +P5+ `
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,R7e=ks({opSnippet:_7e,packedOpSnippet:E7e,cpuKernelImpl:CNe}),D7e={kernelName:ff,backendName:"webgl",kernelFunc:R7e}, $ 7e=class{constructor(e,t,n){this.variableNames=["x"],this.outputShape=t.map((u,c)=>u[0]+e[c]+u[1]);let r=e.length,s=Dn(r),o=t.map(u=>u[0]).join(","),a=t.map((u,c)=>u[0]+e[c]).join(","),i=["coords[0]","coords[1]","coords[2]","coords[3]"].slice(0,r),l=n==="reflect"?0:1;if(r===1){this.userCode= `
int start = $ { o } ;
int end = $ { a } ;
2021-09-11 16:29:31 +02:00
void main ( ) {
int outC = getOutputCoords ( ) ;
if ( outC < start ) {
outC = start * 2 - outC - $ { l } ;
} else if ( outC >= end ) {
outC = ( end - 1 ) * 2 - outC + $ { l } ;
}
setOutput ( getX ( outC - start ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
` ;return}this.userCode= `
2021-10-04 22:29:15 +02:00
$ { s } start = $ { s } ( $ { o } ) ;
$ { s } end = $ { s } ( $ { a } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } outC = getOutputCoords ( ) ;
for ( int i = 0 ; i < $ { r } ; i ++ ) {
2021-09-11 16:29:31 +02:00
if ( outC [ i ] < start [ i ] ) {
outC [ i ] = start [ i ] * 2 - outC [ i ] - $ { l } ;
} else if ( outC [ i ] >= end [ i ] ) {
outC [ i ] = ( end [ i ] - 1 ) * 2 - outC [ i ] + $ { l } ;
}
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
$ { s } coords = outC - start ;
2021-09-11 17:17:13 +02:00
setOutput ( getX ( $ { i } ) ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }},F7e=class{constructor(e,t,n){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=t.map((f,h)=>f[0]+e[h]+f[1]);let r=e.length,s=Dn(r),o=t.map(f=>f[0]).join(","),a=t.map((f,h)=>f[0]+e[h]).join(","),i=js("rc",r),l=js("source",r),u= ` $ { i [ r - 1 ] } < $ { this . outputShape [ r - 1 ] } ` ,c=r===1?"source": ` vec2 ( $ { l . slice ( - 2 ) . join ( ) } ) ` ,p=n==="reflect"?0:1,d="";if(r===1){let f= `
$ { s } source = rc ;
2021-09-11 16:29:31 +02:00
if ( source < start ) {
2021-10-04 22:29:15 +02:00
source = start * 2 - source - $ { p } ;
2021-09-11 16:29:31 +02:00
} else if ( source >= end ) {
2021-10-04 22:29:15 +02:00
source = ( end - 1 ) * 2 - source + $ { p } ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
source -= start ;
2021-10-04 22:29:15 +02:00
` ;d= `
$ { s } rc = outputLoc ;
$ { f }
2021-09-11 17:17:13 +02:00
result [ 0 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-10-04 22:29:15 +02:00
$ { i [ r - 1 ] } += 1 ;
2021-09-11 16:29:31 +02:00
if ( $ { u } ) {
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 17:17:13 +02:00
result [ 1 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }else{let f= `
$ { s } source = rc ;
$ { s } lt = $ { s } ( lessThan ( source , start ) ) ;
$ { s } gte = $ { s } ( greaterThanEqual ( source , end ) ) ;
$ { s } orig = 1 - ( lt + gte ) ;
2021-09-11 16:29:31 +02:00
source = orig * source +
2021-10-04 22:29:15 +02:00
lt * ( start * 2 - source - $ { p } ) +
gte * ( ( end - 1 ) * 2 - source + $ { p } ) ;
2021-09-11 16:29:31 +02:00
source -= start ;
2021-10-04 22:29:15 +02:00
` ;d= `
$ { s } rc = outputLoc ;
$ { f }
2021-09-11 17:17:13 +02:00
result [ 0 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-10-04 22:29:15 +02:00
$ { i [ r - 1 ] } += 1 ;
2021-09-11 16:29:31 +02:00
if ( $ { u } ) {
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 17:17:13 +02:00
result [ 1 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
rc = outputLoc ;
2021-10-04 22:29:15 +02:00
$ { i [ r - 2 ] } += 1 ;
if ( $ { i [ r - 2 ] } < $ { this . outputShape [ r - 2 ] } ) {
$ { f }
2021-09-11 17:17:13 +02:00
result [ 2 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-10-04 22:29:15 +02:00
$ { i [ r - 1 ] } += 1 ;
2021-09-11 16:29:31 +02:00
if ( $ { u } ) {
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 17:17:13 +02:00
result [ 3 ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-09-11 16:29:31 +02:00
}
}
` }this.userCode= `
2021-10-04 22:29:15 +02:00
const $ { s } start = $ { s } ( $ { o } ) ;
const $ { s } end = $ { s } ( $ { a } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } outputLoc = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0. ) ;
2021-10-04 22:29:15 +02:00
$ { d }
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }},O7e=({inputs:e,backend:t,attrs:n})=>{let{x:r}=e,{paddings:s,mode:o}=n,a=be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new F7e(r.shape,s,o):new $ 7e(r.shape,s,o);return t.runWebGLProgram(a,[r],r.dtype)},P7e={kernelName:hf,backendName:"webgl",kernelFunc:O7e},M7e= ` if ( b == 0.0 ) return NAN ;
return mod ( a , b ) ; ` ,z7e= `
2021-09-11 16:29:31 +02:00
vec4 result = mod ( a , b ) ;
vec4 isNaN = vec4 ( equal ( b , vec4 ( 0.0 ) ) ) ;
2021-10-04 22:29:15 +02:00
` +P5+ `
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,L7e=ks({opSnippet:M7e,packedOpSnippet:z7e}),B7e={kernelName:Lg,backendName:"webgl",kernelFunc:L7e},V7e=class{constructor(e,t,n){this.variableNames=["probs"],this.customUniforms=[{name:"seed",type:"float"}],this.outputShape=[e,n],this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
float r = random ( seed ) ;
float cdf = 0.0 ;
for ( int i = 0 ; i < $ { t - 1 } ; i ++ ) {
cdf += getProbs ( batch , i ) ;
if ( r < cdf ) {
setOutput ( float ( i ) ) ;
return ;
}
}
// If no other event happened, last event happened.
setOutput ( float ( $ { t - 1 } ) ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }},W7e= `
2021-09-11 16:29:31 +02:00
if ( a == b ) {
return 1.0 ;
} ;
2021-10-04 22:29:15 +02:00
return a / b ; ` ,G7e= `
2021-09-11 16:29:31 +02:00
// vec4 one = vec4(equal(a, b));
// return one + (vec4(1.0) - one) * a / b;
vec4 result = a / b ;
if ( a . x == b . x ) {
result . x = 1. ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( a . y == b . y ) {
result . y = 1. ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( a . z == b . z ) {
result . z = 1. ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
if ( a . w == b . w ) {
result . w = 1. ;
}
return result ;
2021-10-04 22:29:15 +02:00
` ,aG=ks({opSnippet:W7e,packedOpSnippet:G7e,checkOutOfBounds:!0}),U7e={kernelName:Yd,backendName:"webgl",kernelFunc:aG},iG="return a - b;",lG=ks({opSnippet:iG,packedOpSnippet:iG,supportsComplex:!0,cpuKernelImpl:VNe}),j7e={kernelName:Df,backendName:"webgl",kernelFunc:lG};function uG(e){let{inputs:t,backend:n,attrs:r}=e,{logits:s}=t,{dim:o}=r,a=E.parseAxisParam([o],s.shape),i=oG({inputs:{x:s},backend:n,attrs:{reductionIndices:a,keepDims:!1}}),l=G.expandShapeToKeepDim(i.shape,a),u=qe({inputs:{x:i},backend:n,attrs:{shape:l}}),c=lG({inputs:{a:s,b:u},backend:n}),p=QW({inputs:{x:c},backend:n}),d=L5({inputs:{x:p},backend:n,attrs:{axis:a,keepDims:!1}}),f=qe({inputs:{x:d},backend:n,attrs:{shape:l}}),h=aG({inputs:{a:p,b:f},backend:n});return n.disposeIntermediateTensorInfo(i),n.disposeIntermediateTensorInfo(u),n.disposeIntermediateTensorInfo(c),n.disposeIntermediateTensorInfo(p),n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}var H7e={kernelName:Ef,backendName:"webgl",kernelFunc:uG};function q7e(e){let{inputs:t,backend:n,attrs:r}=e,{logits:s}=t,{numSamples:o,seed:a,normalized:i}=r,l=i?s:uG({inputs:{logits:s},backend:n,attrs:{dim:s.shape.length-1}}),u=l.shape[0],c=l.shape[1],p=new V7e(u,c,o),d=[[a]],f=n.runWebGLProgram(p,[l],"int32",d);return i||n.disposeIntermediateTensorInfo(l),f}var K7e={kernelName:kw,backendName:"webgl",kernelFunc:q7e},cG="return -x;";function X7e(e){let{inputs:t,backend:n}=e,{x:r}=t;if(n.shouldExecuteOnCPU([r])){let o=n.texData.get(r.dataId),[a,i]=NNe(o.values,r.shape,r.dtype);return n.makeTensorInfo(i,r.dtype,a)}let s;return be().getBool("WEBGL_PACK_UNARY_OPERATIONS")?s=new Z0(r.shape,cG):s=new rp(r.shape,cG),n.runWebGLProgram(s,[r],r.dtype)}var Z7e={kernelName:Bg,backendName:"webgl",kernelFunc:X7e},Y7e=ll.nonMaxSuppressionV3Impl;function J7e(e){G.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l}=r,u=n.readSync(s.dataId),c=n.readSync(o.dataId),{selectedIndices:p}=Y7e(u,c,a,i,l);return n.makeTensorInfo([p.length],"int32",new Int32Array(p))}var Q7e={kernelName:Wg,backendName:"webgl",kernelFunc:J7e},eDe=ll.nonMaxSuppressionV4Impl;function tDe(e){G.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l,padToMaxOutputSize:u}=r,c=n.readSync(s.dataId),p=n.readSync(o.dataId),{selectedIndices:d,validOutputs:f}=eDe(c,p,a,i,l,u);return[n.makeTensorInfo([d.length],"int32",new Int32Array(d)),n.makeTensorInfo([],"int32",new Int32Array([f]))]}var nDe={kernelName:Gg,backendName:"webgl",kernelFunc:tDe},rDe=ll.nonMaxSuppressionV5Impl;function sDe(e){G.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l,softNmsSigma:u}=r,c=n.readSync(s.dataId),p=n.readSync(o.dataId),d=a,f=i,h=l,m=u,{selectedIndices:g,selectedScores:x}=rDe(c,p,d,f,h,m);return[n.makeTensorInfo([g.length],"int32",new Int32Array(g)),n.makeTensorInfo([x.length],"float32",new Float32Array(x))]}var oDe={kernelName:Ug,backendName:"webgl",kernelFunc:sDe},aDe=class{constructor(e,t,n,r){this.variableNames=["indices"],this.outputShape=[e,t],this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int index = round ( getIndices ( coords . x ) ) ;
2021-10-04 22:29:15 +02:00
setOutput ( mix ( float ( $ { r } ) , float ( $ { n } ) ,
2021-09-11 16:29:31 +02:00
float ( index == coords . y ) ) ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }},iDe=e=>{let{inputs:t,backend:n,attrs:r}=e,{indices:s}=t,{depth:o,onValue:a,offValue:i}=r,l=E.sizeFromShape(s.shape),u=new aDe(l,o,a,i),c=qe({inputs:{x:s},backend:n,attrs:{shape:[l]}}),p=n.runWebGLProgram(u,[c],s.dtype);n.disposeIntermediateTensorInfo(c);let d=[...s.shape,o],f=qe({inputs:{x:p},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(p),f},lDe={kernelName:gf,backendName:"webgl",kernelFunc:iDe};function U5(e){let{inputs:t,backend:n}=e,{x:r}=t;if(r.dtype==="complex64"){let s=oA({inputs:{input:r},backend:n}),o=U5({inputs:{x:s},backend:n}),a=G5({inputs:{input:r},backend:n}),i=U5({inputs:{x:a},backend:n}),l=sp({inputs:{real:o,imag:i},backend:n});return n.disposeIntermediateTensorInfo(s),n.disposeIntermediateTensorInfo(o),n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(i),l}else return aA({attrs:{shape:r.shape,dtype:r.dtype,value:r.dtype==="string"?"":0},backend:n})}var uDe={kernelName:u0,backendName:"webgl",kernelFunc:U5};function pG(e){let{inputs:t,backend:n}=e,{x:r}=t;if(r.dtype==="string")throw new Error("onesLike is not supported under string dtype");if(r.dtype==="complex64"){let s=oA({inputs:{input:r},backend:n}),o=pG({inputs:{x:s},backend:n}),a=G5({inputs:{input:r},backend:n}),i=U5({inputs:{x:a},backend:n}),l=sp({inputs:{real:o,imag:i},backend:n});return n.disposeIntermediateTensorInfo(s),n.disposeIntermediateTensorInfo(o),n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(i),l}else return aA({attrs:{shape:r.shape,dtype:r.dtype,value:1},backend:n})}var cDe={kernelName:jg,backendName:"webgl",kernelFunc:pG};function pDe(e){let{inputs:t,backend:n,attrs:r}=e,{axis:s}=r;if(t.length===1)return BR({inputs:{input:t[0]},backend:n,attrs:{dim:s}});let o=t[0].shape,a=t[0].dtype;t.forEach(c=>{E.assertShapesMatch(o,c.shape,"All tensors passed to stack must have matching shapes"),E.assert(a===c.dtype,()=>"All tensors passed to stack must have matching dtypes")});let i=[],l=t.map(c=>{let p=BR({inputs:{input:c},backend:n,attrs:{dim:s}});return i.push(p),p}),u=GW({inputs:l,backend:n,attrs:{axis:s}});return i.forEach(c=>n.disposeIntermediateTensorInfo(c)),u}var dDe={kernelName:Hg,backendName:"webgl",kernelFunc:pDe},fDe=class{constructor(e,t,n){this.variableNames=["x"],this.customUniforms=[{name:"value",type:"float"}],this.outputShape=t.map((l,u)=>l[0]+e[u]+l[1]);let r=e.length,s=Dn(r),o=t.map(l=>l[0]).join(","),a=t.map((l,u)=>l[0]+e[u]).join(","),i=["coords[0]","coords[1]","coords[2]","coords[3]"].slice(0,r);if(r===1){this.userCode= `
int start = $ { o } ;
int end = $ { a } ;
2021-09-11 16:29:31 +02:00
void main ( ) {
int outC = getOutputCoords ( ) ;
if ( outC < start || outC >= end ) {
setOutput ( value ) ;
} else {
setOutput ( getX ( outC - start ) ) ;
}
}
` ;return}this.userCode= `
2021-10-04 22:29:15 +02:00
$ { s } start = $ { s } ( $ { o } ) ;
$ { s } end = $ { s } ( $ { a } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } outC = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
if ( any ( lessThan ( outC , start ) ) || any ( greaterThanEqual ( outC , end ) ) ) {
setOutput ( value ) ;
} else {
2021-10-04 22:29:15 +02:00
$ { s } coords = outC - start ;
2021-09-11 17:17:13 +02:00
setOutput ( getX ( $ { i } ) ) ;
2021-09-11 16:29:31 +02:00
}
}
2021-10-04 22:29:15 +02:00
` }},hDe=class{constructor(e,t,n){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"value",type:"float"}],this.outputShape=t.map((h,m)=>h[0]+e[m]+h[1]);let r=e.length,s=Dn(r),o=t.map(h=>h[0]).join(","),a=t.map((h,m)=>h[0]+e[m]).join(","),i=js("rc",r),l=js("source",r),u= ` $ { i [ r - 1 ] } < $ { this . outputShape [ r - 1 ] } ` ,c=r===1?"source": ` vec2 ( $ { l . slice ( - 2 ) . join ( ) } ) ` ,p=[ ` $ { s } rc = outputLoc ; ` , ` $ { i [ r - 1 ] } += 1 ;
2021-09-11 16:29:31 +02:00
if ( $ { u } ) {
2021-10-04 22:29:15 +02:00
` ,r===1?"": ` }
2021-09-11 16:29:31 +02:00
rc = outputLoc ;
2021-10-04 22:29:15 +02:00
$ { i [ r - 2 ] } += 1 ;
if ( $ { i [ r - 2 ] } < $ { this . outputShape [ r - 2 ] } ) { ` ,r===1?"": ` $ { i [ r - 1 ] } += 1 ;
if ( $ { u } ) { ` ],d=r===1?"rc < start || rc >= end":"any(lessThan(rc, start)) || any(greaterThanEqual(rc, end))",f="";for(let h=0,m=r===1?2:4;h<m;h++)f+= `
$ { p [ h ] }
if ( $ { d } ) {
result [ $ { h } ] = float ( value ) ;
2021-09-11 16:29:31 +02:00
} else {
2021-10-04 22:29:15 +02:00
$ { s } source = rc - start ;
result [ $ { h } ] = getChannel ( getX ( $ { l . join ( ) } ) , $ { c } ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` ;f+=r===1?"} ":"}}",this.userCode= `
const $ { s } start = $ { s } ( $ { o } ) ;
const $ { s } end = $ { s } ( $ { a } ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { s } outputLoc = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0. ) ;
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
}
2021-10-04 22:29:15 +02:00
` }},dG=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{paddings:o,constantValue:a}=r;if(E.sizeFromShape(s.shape)===0){let u=o.map((c,p)=>c[0]+s.shape[p]+c[1]);return aA({backend:n,attrs:{shape:u,value:a,dtype:s.dtype}})}let i=be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new hDe(s.shape,o,a):new fDe(s.shape,o,a),l=[[a]];return n.runWebGLProgram(i,[s],s.dtype,l)},mDe={kernelName:xf,backendName:"webgl",kernelFunc:dG},gDe= `
2021-09-11 16:29:31 +02:00
if ( a < 0.0 && floor ( b ) < b ) {
return NAN ;
}
if ( b == 0.0 ) {
return 1.0 ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
return ( round ( mod ( b , 2.0 ) ) != 1 ) ?
pow ( abs ( a ) , b ) : sign ( a ) * pow ( abs ( a ) , b ) ;
2021-10-04 22:29:15 +02:00
` ,xDe= `
2021-09-11 16:29:31 +02:00
// isModRound1 has 1 for components with round(mod(b, 2.0)) == 1, 0 otherwise.
vec4 isModRound1 = vec4 ( equal ( round ( mod ( b , 2.0 ) ) , ivec4 ( 1 ) ) ) ;
vec4 multiplier = sign ( a ) * isModRound1 + ( vec4 ( 1.0 ) - isModRound1 ) ;
vec4 result = multiplier * pow ( abs ( a ) , b ) ;
// Ensure that a^0 = 1, including 0^0 = 1 as this correspond to TF and JS
bvec4 isExpZero = equal ( b , vec4 ( 0.0 ) ) ;
result . r = isExpZero . r ? 1.0 : result . r ;
result . g = isExpZero . g ? 1.0 : result . g ;
result . b = isExpZero . b ? 1.0 : result . b ;
result . a = isExpZero . a ? 1.0 : result . a ;
vec4 isNaN = vec4 ( lessThan ( a , vec4 ( 0.0 ) ) ) * vec4 ( lessThan ( floor ( b ) , b ) ) ;
2021-10-04 22:29:15 +02:00
` +P5+ `
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,yDe=ks({opSnippet:gDe,packedOpSnippet:xDe}),bDe={kernelName:yf,backendName:"webgl",kernelFunc:yDe};function ADe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=[],u=E.parseAxisParam(o,s.shape),c=u,p=G.getAxesPermutation(c,i),d=s;p!=null&&(d=Hs({inputs:{x:s},backend:n,attrs:{perm:p}}),c=G.getInnerMostAxes(c.length,i),l.push(d)),G.assertAxesAreInnerMostDims("prod",c,i);let f;if(n.shouldExecuteOnCPU([d])){let h=n.texData.get(d.dataId).values,{outVals:m,outShape:g,outDtype:x}=_Ne(d.shape,d.dtype,h,c);f=n.makeTensorInfo(g,x,m)}else{let[h,m]=G.computeOutAndReduceShapes(d.shape,c),g=E.sizeFromShape(m),x=qe({inputs:{x:d},backend:n,attrs:{shape:[-1,g]}}),y=Ww(s.dtype),b=xh(x,y,"prod",n);f=qe({inputs:{x:b},backend:n,attrs:{shape:h}}),l.push(x),l.push(b)}if(a){l.push(f);let h=G.expandShapeToKeepDim(f.shape,u);f=qe({inputs:{x:f},backend:n,attrs:{shape:h}})}return l.forEach(h=>n.disposeIntermediateTensorInfo(h)),f}var vDe={kernelName:qg,backendName:"webgl",kernelFunc:ADe},fG=e=>{let{backend:t,attrs:n}=e,{start:r,stop:s,step:o,dtype:a}=n,i=ENe(r,s,o,a);return t.makeTensorInfo([i.length],a,i)},wDe={kernelName:H1,backendName:"webgl",kernelFunc:fG},kDe="return 1.0 / x;",IDe=tn({opSnippet:kDe}),CDe={kernelName:Kg,backendName:"webgl",kernelFunc:IDe},SDe=Ti+ `
2021-09-11 16:29:31 +02:00
return ( x < 0.0 ) ? 0.0 : x ;
2021-10-04 22:29:15 +02:00
` ,NDe= `
2021-09-11 16:29:31 +02:00
vec4 result = x * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
2021-10-04 22:29:15 +02:00
` ,TDe=tn({opSnippet:SDe,packedOpSnippet:NDe}),_De={kernelName:Af,backendName:"webgl",kernelFunc:TDe},EDe=Ti+ `
2021-09-11 16:29:31 +02:00
return ( x < 0.0 ) ? 0.0 : min ( 6.0 , x ) ;
2021-10-04 22:29:15 +02:00
` ,RDe= `
2021-09-11 16:29:31 +02:00
vec4 result = min ( x , vec4 ( 6. ) ) * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
2021-10-04 22:29:15 +02:00
` ,DDe=tn({opSnippet:EDe,packedOpSnippet:RDe}), $ De={kernelName:wf,backendName:"webgl",kernelFunc:DDe},FDe=class{constructor(e,t,n,r,s){this.variableNames=["A"],this.outputShape=[];let[o,a,i,l]=e;this.outputShape=[o,t,n,l];let u=[r&&t>1?a-1:a,r&&n>1?i-1:i],c=[r&&t>1?t-1:t,r&&n>1?n-1:n],p;s?p="(vec2(yRC) + vec2(0.5)) * effectiveInputOverOutputRatioRC - vec2(0.5)":p="vec2(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
2021-09-11 16:29:31 +02:00
const vec2 effectiveInputOverOutputRatioRC = vec2 (
2021-09-11 17:17:13 +02:00
$ { u [ 0 ] / c [ 0 ] } ,
$ { u [ 1 ] / c [ 1 ] } ) ;
2021-10-04 22:29:15 +02:00
const vec2 inputShapeRC = vec2 ( $ { a } . 0 , $ { i } . 0 ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 yRC = coords . yz ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Fractional source index.
2021-10-04 22:29:15 +02:00
vec2 sourceFracIndexRC = $ { p } ;
2021-09-11 16:29:31 +02:00
// Compute the four integer indices.
ivec2 sourceFloorRC = ivec2 ( max ( sourceFracIndexRC , vec2 ( 0.0 ) ) ) ;
ivec2 sourceCeilRC = ivec2 (
min ( inputShapeRC - 1.0 , ceil ( sourceFracIndexRC ) ) ) ;
float topLeft = getA ( b , sourceFloorRC . x , sourceFloorRC . y , d ) ;
float bottomLeft = getA ( b , sourceCeilRC . x , sourceFloorRC . y , d ) ;
float topRight = getA ( b , sourceFloorRC . x , sourceCeilRC . y , d ) ;
float bottomRight = getA ( b , sourceCeilRC . x , sourceCeilRC . y , d ) ;
vec2 fracRC = sourceFracIndexRC - vec2 ( sourceFloorRC ) ;
float top = topLeft + ( topRight - topLeft ) * fracRC . y ;
float bottom = bottomLeft + ( bottomRight - bottomLeft ) * fracRC . y ;
float newValue = top + ( bottom - top ) * fracRC . x ;
setOutput ( newValue ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }},ODe=class{constructor(e,t,n,r,s){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[];let[o,a,i,l]=e;this.outputShape=[o,t,n,l];let u=[r&&t>1?a-1:a,r&&n>1?i-1:i],c=[r&&t>1?t-1:t,r&&n>1?n-1:n],p;s?p="(vec3(yRC) + vec3(0.5)) * effectiveInputOverOutputRatioRC - vec3(0.5)":p="vec3(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
2021-09-11 16:29:31 +02:00
const vec3 effectiveInputOverOutputRatioRC = vec3 (
2021-09-11 17:17:13 +02:00
$ { u [ 0 ] / c [ 0 ] } ,
$ { u [ 1 ] / c [ 1 ] } ,
$ { u [ 1 ] / c [ 1 ] } ) ;
2021-10-04 22:29:15 +02:00
const vec3 inputShapeRC = vec3 ( $ { a } . 0 , $ { i } . 0 ,
2021-09-11 17:17:13 +02:00
$ { i } . 0 ) ;
2021-09-11 16:29:31 +02:00
float getAValue ( int b , int r , int c , int d ) {
return getChannel ( getA ( b , r , c , d ) , vec2 ( c , d ) ) ;
2021-09-05 22:42:11 +02:00
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
// Calculate values for next column in yRC.z.
ivec3 yRC = coords . yzz + ivec3 ( 0 , 0 , 1 ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Fractional source index.
2021-10-04 22:29:15 +02:00
vec3 sourceFracIndexRC = $ { p } ;
2021-09-11 16:29:31 +02:00
// Compute the four integer indices.
ivec3 sourceFloorRC = ivec3 ( max ( sourceFracIndexRC , vec3 ( 0.0 ) ) ) ;
ivec3 sourceCeilRC = ivec3 (
min ( inputShapeRC - 1.0 , ceil ( sourceFracIndexRC ) ) ) ;
// Should we calculate next column and row elements in 2x2 packed cell.
bool hasNextCol = d < $ { l - 1 } ;
bool hasNextRow = coords . z < $ { n - 1 } ;
// In parallel, construct four corners for all four components in
// packed 2x2 cell.
vec4 topLeft = vec4 (
getAValue ( b , sourceFloorRC . x , sourceFloorRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceFloorRC . x , sourceFloorRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceFloorRC . x , sourceFloorRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceFloorRC . x , sourceFloorRC . z , d + 1 ) : 0.0 ) ;
vec4 bottomLeft = vec4 (
getAValue ( b , sourceCeilRC . x , sourceFloorRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceCeilRC . x , sourceFloorRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceCeilRC . x , sourceFloorRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceCeilRC . x , sourceFloorRC . z , d + 1 ) : 0.0 ) ;
vec4 topRight = vec4 (
getAValue ( b , sourceFloorRC . x , sourceCeilRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceFloorRC . x , sourceCeilRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceFloorRC . x , sourceCeilRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceFloorRC . x , sourceCeilRC . z , d + 1 ) : 0.0 ) ;
vec4 bottomRight = vec4 (
getAValue ( b , sourceCeilRC . x , sourceCeilRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceCeilRC . x , sourceCeilRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceCeilRC . x , sourceCeilRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceCeilRC . x , sourceCeilRC . z , d + 1 ) : 0.0 ) ;
vec3 fracRC = sourceFracIndexRC - vec3 ( sourceFloorRC ) ;
vec4 top = mix ( topLeft , topRight , fracRC . yyzz ) ;
vec4 bottom = mix ( bottomLeft , bottomRight , fracRC . yyzz ) ;
vec4 newValue = mix ( top , bottom , fracRC . x ) ;
setOutput ( newValue ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function PDe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s}=t,{alignCorners:o,halfPixelCenters:a,size:i}=r,[l,u]=i,c=be().getBool("WEBGL_PACK_IMAGE_OPERATIONS")?new ODe(s.shape,l,u,o,a):new FDe(s.shape,l,u,o,a);return n.runWebGLProgram(c,[s],"float32")}var MDe={kernelName:vf,backendName:"webgl",kernelFunc:PDe},zDe=class{constructor(e,t,n){this.variableNames=["dy"],this.outputShape=[],this.outputShape=t;let[,r,s]=t,[,o,a]=e,i=[n&&o>1?r-1:r,n&&a>1?s-1:s],l=[n&&o>1?o-1:o,n&&a>1?a-1:a],u=i[0]/l[0],c=i[1]/l[1],p=1/u,d=1/c,f=Math.ceil(p)*2+2,h=Math.ceil(d)*2+2;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
float accumulator = 0.0 ;
const float heightScale = float ( $ { u } ) ;
2021-09-11 17:17:13 +02:00
const float widthScale = float ( $ { c } ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
const float invHeightScale = float ( $ { p } ) ;
const float invWidthScale = float ( $ { d } ) ;
2021-09-11 16:29:31 +02:00
2021-10-04 22:29:15 +02:00
const int winHeight = int ( $ { f } ) ;
const int winWidth = int ( $ { h } ) ;
2021-09-11 16:29:31 +02:00
// Compute bounds for where in dy we will look
float startRLerp = floor ( float ( r ) * invHeightScale ) ;
int startDyR = int ( startRLerp - float ( winHeight / 2 ) ) ;
float startCLerp = floor ( float ( c ) * invWidthScale ) ;
int startDyC = int ( startCLerp - float ( winWidth / 2 ) ) ;
// Loop over dy
for ( int dyROffset = 0 ; dyROffset < winHeight ; dyROffset ++ ) {
int dyR = dyROffset + startDyR ;
// Guard against the window exceeding the bounds of dy
2021-10-04 22:29:15 +02:00
if ( dyR < 0 || dyR >= $ { o } ) {
2021-09-11 16:29:31 +02:00
continue ;
2021-09-01 00:22:16 +02:00
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
for ( int dyCOffset = 0 ; dyCOffset < winWidth ; dyCOffset ++ ) {
int dyC = dyCOffset + startDyC ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Guard against the window exceeding the bounds of dy
2021-10-04 22:29:15 +02:00
if ( dyC < 0 || dyC >= $ { a } ) {
2021-09-11 16:29:31 +02:00
continue ;
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
float dxR = float ( dyR ) * heightScale ;
int topDxRIndex = int ( floor ( dxR ) ) ;
2021-10-04 22:29:15 +02:00
int bottomDxRIndex = int ( min ( ceil ( dxR ) , $ { r - 1 } . 0 ) ) ;
2021-09-11 16:29:31 +02:00
float dxRLerp = dxR - float ( topDxRIndex ) ;
float inverseDxRLerp = 1.0 - dxRLerp ;
float dxC = float ( dyC ) * widthScale ;
int leftDxCIndex = int ( floor ( dxC ) ) ;
2021-10-04 22:29:15 +02:00
int rightDxCIndex = int ( min ( ceil ( dxC ) , $ { s - 1 } . 0 ) ) ;
2021-09-11 16:29:31 +02:00
float dxCLerp = dxC - float ( leftDxCIndex ) ;
float inverseDxCLerp = 1.0 - dxCLerp ;
if ( r == topDxRIndex && c == leftDxCIndex ) {
// topLeft
accumulator +=
getDy ( b , dyR , dyC , d ) * inverseDxRLerp * inverseDxCLerp ;
}
if ( r == topDxRIndex && c == rightDxCIndex ) {
// topRight
accumulator += getDy ( b , dyR , dyC , d ) * inverseDxRLerp * dxCLerp ;
}
if ( r == bottomDxRIndex && c == leftDxCIndex ) {
// bottomLeft
accumulator += getDy ( b , dyR , dyC , d ) * dxRLerp * inverseDxCLerp ;
}
if ( r == bottomDxRIndex && c == rightDxCIndex ) {
// bottomRight
accumulator += getDy ( b , dyR , dyC , d ) * dxRLerp * dxCLerp ;
2021-09-05 22:42:11 +02:00
}
}
}
2021-09-11 16:29:31 +02:00
// End loop over dy
setOutput ( accumulator ) ;
2021-09-01 00:22:16 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function LDe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s,dy:o}=t,{alignCorners:a}=r,i=new zDe(o.shape,s.shape,a);return n.runWebGLProgram(i,[o],o.dtype)}var BDe={kernelName:Sw,backendName:"webgl",kernelFunc:LDe},VDe=class{constructor(e,t,n,r,s){this.variableNames=["A"],this.outputShape=[];let[o,a,i,l]=e;this.outputShape=[o,t,n,l];let u=[r&&t>1?a-1:a,r&&n>1?i-1:i],c=[r&&t>1?t-1:t,r&&n>1?n-1:n],p=r?"0.5":"0.0",d;s?d="max((vec2(yRC) + vec2(0.5)) * effectiveInputOverOutputRatioRC, vec2(0.0))":d="vec2(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
2021-09-11 16:29:31 +02:00
const vec2 effectiveInputOverOutputRatioRC = vec2 (
2021-09-11 17:17:13 +02:00
$ { u [ 0 ] / c [ 0 ] } ,
$ { u [ 1 ] / c [ 1 ] } ) ;
2021-10-04 22:29:15 +02:00
const vec2 inputShapeRC = vec2 ( $ { a } . 0 , $ { i } . 0 ) ;
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 yRC = coords . yz ;
// Fractional source index.
2021-10-04 22:29:15 +02:00
vec2 sourceFracIndexRC = $ { d } ;
2021-09-11 16:29:31 +02:00
// Compute the coordinators of nearest neighbor point.
ivec2 sourceNearestRC = ivec2 (
2021-10-04 22:29:15 +02:00
min ( inputShapeRC - 1.0 , floor ( sourceFracIndexRC + $ { p } ) ) ) ;
2021-09-11 16:29:31 +02:00
float newValue = getA ( b , sourceNearestRC . x , sourceNearestRC . y , d ) ;
setOutput ( newValue ) ;
}
2021-10-04 22:29:15 +02:00
` }},WDe=class{constructor(e,t,n,r,s){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[];let[o,a,i,l]=e;this.outputShape=[o,t,n,l];let u=[r&&t>1?a-1:a,r&&n>1?i-1:i],c=[r&&t>1?t-1:t,r&&n>1?n-1:n],p=r?"0.5":"0.0",d;s?d="max((vec3(yRC) + vec3(0.5)) * effectiveInputOverOutputRatioRC, vec3(0.0))":d="vec3(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
2021-09-11 16:29:31 +02:00
const vec3 effectiveInputOverOutputRatioRC = vec3 (
2021-09-11 17:17:13 +02:00
$ { u [ 0 ] / c [ 0 ] } ,
$ { u [ 1 ] / c [ 1 ] } ,
$ { u [ 1 ] / c [ 1 ] } ) ;
2021-10-04 22:29:15 +02:00
const vec3 inputShapeRC = vec3 ( $ { a } . 0 , $ { i } . 0 ,
2021-09-11 17:17:13 +02:00
$ { i } . 0 ) ;
2021-09-11 16:29:31 +02:00
float getAValue ( int b , int r , int c , int d ) {
return getChannel ( getA ( b , r , c , d ) , vec2 ( c , d ) ) ;
}
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
// Calculate values for next column in yRC.z.
ivec3 yRC = coords . yzz + ivec3 ( 0 , 0 , 1 ) ;
// Fractional source index.
2021-10-04 22:29:15 +02:00
vec3 sourceFracIndexRC = $ { d } ;
2021-09-11 16:29:31 +02:00
// Compute the coordinators of nearest neighbor point.
ivec3 sourceNearestRC = ivec3 (
2021-10-04 22:29:15 +02:00
min ( inputShapeRC - 1.0 , floor ( sourceFracIndexRC + $ { p } ) ) ) ;
2021-09-11 16:29:31 +02:00
// Should we calculate next column and row elements in 2x2 packed cell.
bool hasNextCol = d < $ { l - 1 } ;
bool hasNextRow = coords . z < $ { n - 1 } ;
vec4 newValue = vec4 (
getAValue ( b , sourceNearestRC . x , sourceNearestRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceNearestRC . x , sourceNearestRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceNearestRC . x , sourceNearestRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceNearestRC . x , sourceNearestRC . z , d + 1 ) : 0.0 ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( newValue ) ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function GDe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s}=t,{alignCorners:o,halfPixelCenters:a,size:i}=r,[l,u]=i,c=be().getBool("WEBGL_PACK_IMAGE_OPERATIONS")?new WDe(s.shape,l,u,o,a):new VDe(s.shape,l,u,o,a);return n.runWebGLProgram(c,[s],s.dtype)}var UDe={kernelName:q1,backendName:"webgl",kernelFunc:GDe},jDe=class{constructor(e,t,n){this.variableNames=["dy"],this.outputShape=[],this.outputShape=t;let[,r,s]=t,[,o,a]=e,i=[n&&o>1?r-1:r,n&&a>1?s-1:s],l=[n&&o>1?o-1:o,n&&a>1?a-1:a],u=i[0]/l[0],c=i[1]/l[1],p=1/u,d=1/c,f=Math.ceil(p)*2+2,h=Math.ceil(d)*2+2;this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
float accumulator = 0.0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
const float heightScale = float ( $ { u } ) ;
2021-09-11 17:17:13 +02:00
const float widthScale = float ( $ { c } ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
const float invHeightScale = float ( $ { p } ) ;
const float invWidthScale = float ( $ { d } ) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
const int winHeight = int ( $ { f } ) ;
const int winWidth = int ( $ { h } ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
// Compute bounds for where in dy we will look
float startRLerp = floor ( float ( r ) * invHeightScale ) ;
int startDyR = int ( floor ( startRLerp - float ( winHeight / 2 ) ) ) ;
float startCLerp = floor ( float ( c ) * invWidthScale ) ;
int startDyC = int ( floor ( startCLerp - float ( winWidth / 2 ) ) ) ;
// Loop over dy
for ( int dyROffset = 0 ; dyROffset < winHeight ; dyROffset ++ ) {
int dyR = dyROffset + startDyR ;
// Guard against the window exceeding the bounds of dy
2021-10-04 22:29:15 +02:00
if ( dyR < 0 || dyR >= $ { o } ) {
2021-07-29 22:06:03 +02:00
continue ;
2021-09-11 16:29:31 +02:00
}
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
for ( int dyCOffset = 0 ; dyCOffset < winWidth ; dyCOffset ++ ) {
int dyC = dyCOffset + startDyC ;
2021-08-14 17:16:26 +02:00
2021-09-11 16:29:31 +02:00
// Guard against the window exceeding the bounds of dy
2021-10-04 22:29:15 +02:00
if ( dyC < 0 || dyC >= $ { a } ) {
2021-09-11 16:29:31 +02:00
continue ;
}
2021-07-29 15:53:13 +02:00
2021-09-11 16:29:31 +02:00
float sourceFracRow =
2021-09-11 17:17:13 +02:00
float ( $ { i [ 0 ] } ) *
2021-09-11 16:29:31 +02:00
( float ( dyR ) / float ( $ { l [ 0 ] } ) ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
float sourceFracCol =
2021-09-11 17:17:13 +02:00
float ( $ { i [ 1 ] } ) *
2021-09-11 16:29:31 +02:00
( float ( dyC ) / float ( $ { l [ 1 ] } ) ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
int sourceNearestRow = int ( min (
2021-10-04 22:29:15 +02:00
float ( int ( $ { r } ) - 1 ) ,
2021-09-11 16:29:31 +02:00
$ { n } ? float ( round ( sourceFracRow ) ) :
float ( floor ( sourceFracRow ) ) ) ) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int sourceNearestCol = int ( min (
2021-10-04 22:29:15 +02:00
float ( int ( $ { s } ) - 1 ) ,
2021-09-11 16:29:31 +02:00
$ { n } ? float ( round ( sourceFracCol ) ) :
float ( floor ( sourceFracCol ) ) ) ) ;
if ( r == sourceNearestRow && c == sourceNearestCol ) {
accumulator += getDy ( b , dyR , dyC , d ) ;
}
2021-07-29 22:06:03 +02:00
}
}
2021-09-11 16:29:31 +02:00
// End loop over dy
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
setOutput ( accumulator ) ;
2021-07-29 22:06:03 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function HDe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s,dy:o}=t,{alignCorners:a}=r,i=new jDe(o.shape,s.shape,a);return n.runWebGLProgram(i,[o],o.dtype)}var qDe={kernelName:Cw,backendName:"webgl",kernelFunc:HDe},KDe=class{constructor(e,t){this.variableNames=["x"];let n=e.length;if(n>4)throw new Error( ` WebGL backend : Reverse of rank - $ { n } tensor is not yet supported ` );if(this.outputShape=e,n===1){this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
int coord = getOutputCoords ( ) ;
setOutput ( getX ( $ { e [ 0 ] } - coord - 1 ) ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` ;return}let r=a=>t.indexOf(a)!==-1&&e[a]!==1? ` $ { e [ a ] } - coords [ $ { a } ] - 1 ` : ` coords [ $ { a } ] ` ,s=e.map((a,i)=>r(i)).join(","),o=Dn(n);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { o } coords = getOutputCoords ( ) ;
setOutput ( getX ( $ { s } ) ) ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
` }},XDe=class{constructor(e,t){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0;let n=e.length;if(n>4)throw new Error( ` WebGL backend : Reverse of rank - $ { n } tensor is not yet supported ` );this.outputShape=e;let r=js("rc",n),s= ` $ { r [ n - 1 ] } + 1 < $ { this . outputShape [ n - 1 ] } ` ,o= ` $ { r [ n - 2 ] } + 1 < $ { this . outputShape [ n - 2 ] } ` ,a=Dn(n);n===1?this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
int rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
result . r = getChannel ( getX ( $ { e [ 0 ] } - rc - 1 ) ,
$ { e [ 0 ] } - rc - 1 ) ;
2021-10-04 22:29:15 +02:00
if ( $ { s } ) {
2021-09-11 16:29:31 +02:00
result . g = getChannel ( getX ( $ { e [ 0 ] } - ( rc + 1 ) - 1 ) ,
$ { e [ 0 ] } - ( rc + 1 ) - 1 ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( result ) ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
` :this.userCode= `
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { a } rc = getOutputCoords ( ) ;
2021-09-11 16:29:31 +02:00
vec4 result = vec4 ( 0. ) ;
2021-10-04 22:29:15 +02:00
result . r = $ { i ( r . slice ( ) ) } ;
if ( $ { s } ) {
result . g = $ { l ( r . slice ( ) ) } ;
2021-09-11 16:29:31 +02:00
}
2021-10-04 22:29:15 +02:00
if ( $ { o } ) {
result . b = $ { u ( r . slice ( ) ) } ;
if ( $ { s } ) {
result . a = $ { c ( r . slice ( ) ) } ;
2021-09-11 16:29:31 +02:00
}
}
setOutput ( result ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` ;function i(f){return p(f)}function l(f){return f[n-1]="("+f[n-1]+" + 1)",p(f)}function u(f){return f[n-2]="("+f[n-2]+" + 1)",p(f)}function c(f){return f[n-1]="("+f[n-1]+" + 1)",f[n-2]="("+f[n-2]+" + 1)",p(f)}function p(f){let h=e.map((x,y)=>d(y,f)),m=h.join(","),g=h.slice(-2).join(",");return ` getChannel ( getX ( $ { m } ) , vec2 ( $ { g } ) ) ` }function d(f,h){return t.indexOf(f)!==-1&&e[f]!==1? ` $ { e [ f ] } - $ { h [ f ] } - 1 ` : ` $ { h [ f ] } ` }}};function ZDe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{dims:o}=r,a=s.shape.length,i=E.parseAxisParam(o,s.shape);if(a===0)return Xo({inputs:{x:s},backend:n});let l=be().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new XDe(s.shape,i):new KDe(s.shape,i);return n.runWebGLProgram(l,[s],s.dtype)}var YDe={kernelName:kf,backendName:"webgl",kernelFunc:ZDe},JDe=class{constructor(e,t){this.variableNames=["Image"],this.outputShape=[],this.customUniforms=[{name:"params",type:"vec4"}];let n=e[1],r=e[2];this.outputShape=e;let s="";typeof t=="number"?s= ` float outputValue = $ { t . toFixed ( 2 ) } ; ` :s= `
2021-09-11 16:29:31 +02:00
vec3 fill = vec3 ( $ { t . join ( "," ) } ) ;
float outputValue = fill [ coords [ 3 ] ] ; ` ,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int x = coords [ 2 ] ;
int y = coords [ 1 ] ;
float coordXFloat = ( float ( x ) - params [ 0 ] ) * params [ 3 ] -
( float ( y ) - params [ 1 ] ) * params [ 2 ] ;
float coordYFloat = ( float ( x ) - params [ 0 ] ) * params [ 2 ] +
( float ( y ) - params [ 1 ] ) * params [ 3 ] ;
int coordX = int ( round ( coordXFloat + params [ 0 ] ) ) ;
int coordY = int ( round ( coordYFloat + params [ 1 ] ) ) ;
2021-10-04 22:29:15 +02:00
$ { s }
if ( coordX >= 0 && coordX < $ { r } && coordY >= 0 && coordY < $ { n } ) {
2021-09-11 16:29:31 +02:00
outputValue = getImage ( coords [ 0 ] , coordY , coordX , coords [ 3 ] ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( outputValue ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` }},QDe={kernelName:c0,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{image:r}=e,{radians:s,fillValue:o,center:a}=t,i=n,l=new JDe(r.shape,o),[u,c]=G.getImageCenter(a,r.shape[1],r.shape[2]),p=[[u,c,Math.sin(s),Math.cos(s)]];return i.runWebGLProgram(l,[r],r.dtype,p)}},e $ e= `
2021-09-11 16:29:31 +02:00
// OpenGL ES does not support round function.
// The algorithm is based on banker's rounding.
float base = floor ( x ) ;
if ( ( x - base ) < 0.5 ) {
return floor ( x ) ;
} else if ( ( x - base ) > 0.5 ) {
return ceil ( x ) ;
} else {
if ( mod ( base , 2.0 ) == 0.0 ) {
return base ;
} else {
return base + 1.0 ;
2021-09-05 22:42:11 +02:00
}
}
2021-10-04 22:29:15 +02:00
` ,t $ e=tn({opSnippet:e $ e}),n $ e={kernelName:If,backendName:"webgl",kernelFunc:t $ e},r $ e="return inversesqrt(x);",s $ e=tn({opSnippet:r $ e,cpuKernelImpl:RNe}),o $ e={kernelName:Cf,backendName:"webgl",kernelFunc:s $ e},hG=class{constructor(e,t,n,r,s,o,a=!0){this.variableNames=["updates","indices","defaultValue"],this.outputShape=o;let i=Dn(s.length),l=Dn(o.length),u="";n===1?u="i":n===2&&(u="i, j");let c= ` getIndices ( $ { u } ) ` ,p="";r===1?p="i":r===2&&(p="i, coords[1]");let d= ` getUpdates ( $ { p } ) ` ,f=t>1?"strides[j]":"strides";this.userCode= `
$ { i } strides = $ { i } ( $ { s } ) ;
2021-09-05 22:42:11 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
$ { l } coords = getOutputCoords ( ) ;
float sum = 0.0 ;
bool found = false ;
for ( int i = 0 ; i < $ { e } ; i ++ ) {
int flattenedIndex = 0 ;
for ( int j = 0 ; j < $ { t } ; j ++ ) {
2021-09-11 17:17:13 +02:00
int index = round ( $ { c } ) ;
2021-10-04 22:29:15 +02:00
flattenedIndex += index * $ { f } ;
2021-09-11 16:29:31 +02:00
}
if ( flattenedIndex == coords [ 0 ] ) {
2021-10-04 22:29:15 +02:00
sum += $ { d } ;
2021-09-11 16:29:31 +02:00
found = true ;
}
}
setOutput ( mix ( getDefaultValue ( ) , sum , float ( found ) ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function a $ e(e){let{inputs:t,backend:n,attrs:r}=e,{indices:s,updates:o}=t,{shape:a}=r,{sliceRank:i,numUpdates:l,sliceSize:u,strides:c,outputSize:p}=G.calculateShapes(o,s,a),d=[p/u,u];if(p===0)return n.makeTensorInfo(a,s.dtype);let f=qe({inputs:{x:s},backend:n,attrs:{shape:[l,i]}}),h=qe({inputs:{x:o},backend:n,attrs:{shape:[l,u]}}),m=n.makeTensorInfo([],"float32",new Float32Array([0])),g=new hG(l,i,f.shape.length,h.shape.length,c,d),x=n.runWebGLProgram(g,[h,f,m],h.dtype),y=qe({inputs:{x},backend:n,attrs:{shape:a}});return n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(h),n.disposeIntermediateTensorInfo(x),n.disposeIntermediateTensorInfo(m),y}var i $ e={kernelName:Zg,backendName:"webgl",kernelFunc:a $ e},l $ e=class{constructor(e,t,n){this.variableNames=["c","a","b"],this.outputShape=t;let r,s;if(n>4)throw Error( ` Where for rank $ { n } is not yet supported ` );if(n===1)s="resRC",r="resRC";else{let a=["resRC.x","resRC.y","resRC.z","resRC.w"],i=[],l=[];for(let u=0;u<t.length;u++)l.push( ` $ { a [ u ] } ` ),u<e&&i.push( ` $ { a [ u ] } ` );r=i.join(),s=l.join()}let o=Dn(n);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { o } resRC = getOutputCoords ( ) ;
float cVal = getC ( $ { r } ) ;
2021-09-11 16:29:31 +02:00
if ( cVal >= 1.0 ) {
2021-10-04 22:29:15 +02:00
setOutput ( getA ( $ { s } ) ) ;
2021-09-11 16:29:31 +02:00
} else {
2021-10-04 22:29:15 +02:00
setOutput ( getB ( $ { s } ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function u $ e(e){let{inputs:t,backend:n}=e,{condition:r,t:s,e:o}=t,a=new l $ e(r.shape.length,s.shape,s.shape.length);return n.runWebGLProgram(a,[r,s,o], $ a(s.dtype,o.dtype))}var c $ e={kernelName:Yg,backendName:"webgl",kernelFunc:u $ e},p $ e= `
2021-09-11 16:29:31 +02:00
// Stable and Attracting Fixed Point (0, 1) for Normalized Weights.
// see: https://arxiv.org/abs/1706.02515
2021-10-04 22:29:15 +02:00
float scaleAlpha = $ { G . SELU _SCALEALPHA } ;
float scale = $ { G . SELU _SCALE } ;
2021-09-11 16:29:31 +02:00
return ( x >= 0.0 ) ? scale * x : scaleAlpha * ( exp ( x ) - 1.0 ) ;
2021-10-04 22:29:15 +02:00
` ,d $ e=tn({opSnippet:p $ e}),f $ e={kernelName:Jg,backendName:"webgl",kernelFunc:d $ e},mG="return 1.0 / (1.0 + exp(-1.0 * x));",h $ e=tn({opSnippet:mG,packedOpSnippet:mG,cpuKernelImpl:DNe}),m $ e={kernelName:Nf,backendName:"webgl",kernelFunc:h $ e},g $ e= `
2021-09-11 16:29:31 +02:00
if ( isnan ( x ) ) { return 0.0 ; }
return sign ( x ) ;
2021-10-04 22:29:15 +02:00
` ,x $ e=tn({opSnippet:g $ e}),y $ e={kernelName:t0,backendName:"webgl",kernelFunc:x $ e},b $ e=TW+ `
2021-09-11 16:29:31 +02:00
return sin ( x ) ;
2021-10-04 22:29:15 +02:00
` ,A $ e=tn({opSnippet:b $ e}),v $ e={kernelName:Sf,backendName:"webgl",kernelFunc:A $ e},w $ e= `
2021-09-11 16:29:31 +02:00
float e2x = exp ( x ) ;
return ( e2x - 1.0 / e2x ) / 2.0 ;
2021-10-04 22:29:15 +02:00
` ,k $ e=tn({opSnippet:w $ e}),I $ e={kernelName:e0,backendName:"webgl",kernelFunc:k $ e},C $ e= `
2021-09-11 16:29:31 +02:00
float epsilon = 1.1920928955078125 e - 7 ;
float threshold = log ( epsilon ) + 2.0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
bool too _large = x > - threshold ;
bool too _small = x < threshold ;
2021-08-14 17:16:26 +02:00
2021-09-11 16:29:31 +02:00
float result ;
float exp _x = exp ( x ) ;
2021-06-07 02:34:29 +02:00
2021-09-11 16:29:31 +02:00
if ( too _large ) {
result = x ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
else if ( too _small ) {
result = exp _x ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
else {
result = log ( exp _x + 1.0 ) ;
2021-08-14 17:16:26 +02:00
}
2021-09-11 16:29:31 +02:00
return result ;
2021-10-04 22:29:15 +02:00
` ,S $ e=tn({opSnippet:C $ e}),N $ e={kernelName:n0,backendName:"webgl",kernelFunc:S $ e},T $ e=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockShape:o,paddings:a}=r;E.assert(s.shape.length<=4,()=>"spaceToBatchND for rank > 4 with a WebGL backend not implemented yet");let i=o.reduce((x,y)=>x*y),l=[[0,0]];l.push(...a);for(let x=1+o.length;x<s.shape.length;++x)l.push([0,0]);let u=[],c=dG({inputs:{x:s},backend:n,attrs:{paddings:l,constantValue:0}}),p=G.getReshaped(c.shape,o,i,!1),d=G.getPermuted(p.length,o.length,!1),f=G.getReshapedPermuted(c.shape,o,i,!1),h=qe({inputs:{x:c},backend:n,attrs:{shape:p}}),m=Hs({inputs:{x:h},backend:n,attrs:{perm:d}}),g=qe({inputs:{x:m},backend:n,attrs:{shape:f}});return u.push(c),u.push(h),u.push(m),u.forEach(x=>n.disposeIntermediateTensorInfo(x)),g},_ $ e={kernelName:r0,backendName:"webgl",kernelFunc:T $ e};function E $ e(e){let{inputs:t,backend:n}=e,{indices:r,values:s,denseShape:o,defaultValue:a}=t;if(o.shape.length!==1)throw new Error( ` Dense shape must be a vector , saw :
$ { o . shape } ` );if(r.shape.length!==2)throw new Error( ` Indices must be a matrix , saw :
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Values must be a vector , saw :
$ { s . shape } ` );if(a.shape.length!==0)throw new Error( ` Default value must be a scalar , saw :
$ { a . shape } ` );let i=n.readSync(r.dataId),l=n.readSync(s.dataId),u=n.readSync(o.dataId),c=n.readSync(a.dataId)[0],[p,d,f,h,m]=FNe(i,r.shape,r.dtype,l,s.dtype,u,c);return[n.makeTensorInfo(d,r.dtype,p),n.makeTensorInfo([d[0]],s.dtype,f),n.makeTensorInfo([h.length],"bool",new Uint8Array(h.map(g=>Number(g)))),n.makeTensorInfo([m.length],r.dtype,new Int32Array(m))]}var R $ e={kernelName:Nw,backendName:"webgl",kernelFunc:E $ e};function D $ e(e){let{inputs:t,backend:n}=e,{inputIndices:r,inputShape:s,newShape:o}=t;if(r.shape.length!==2)throw new Error( ` Input indices should be a matrix but received shape $ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Input shape should be a vector but received shape $ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Target shape should be a vector but received shape $ { o . shape } ` );let a=Array.from(n.readSync(s.dataId)),i=n.readSync(r.dataId),l=Array.from(n.readSync(o.dataId)),[u,c,p]=ONe(i,r.shape,r.dtype,a,l);return[n.makeTensorInfo(c,r.dtype,u),n.makeTensorInfo([p.length],o.dtype,new Int32Array(p))]}var $ $ e={kernelName:Tw,backendName:"webgl",kernelFunc:D $ e};function F $ e(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.readSync(r.dataId),i=n.readSync(s.dataId),l=n.readSync(o.dataId),[u,c]=mW(a,r.shape,r.dtype,i,l,!0);return n.makeTensorInfo(c,r.dtype,u)}var O $ e={kernelName:_w,backendName:"webgl",kernelFunc:F $ e};function P $ e(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.readSync(r.dataId),i=n.readSync(s.dataId),l=n.readSync(o.dataId),[u,c]=mW(a,r.shape,r.dtype,i,l);return n.makeTensorInfo(c,r.dtype,u)}var M $ e={kernelName:Ew,backendName:"webgl",kernelFunc:P $ e};function z $ e(e){let{inputs:t,backend:n,attrs:r}=e,{sparseIndices:s,sparseValues:o,defaultValue:a}=t,{outputShape:i}=r,{sliceRank:l,numUpdates:u,strides:c,outputSize:p}=G.calculateShapes(o,s,i),d=!1,f=new hG(u,l,s.shape.length,o.shape.length,c,[p,1],d),h=n.runWebGLProgram(f,[o,s,a],o.dtype),m=qe({inputs:{x:h},backend:n,attrs:{shape:i}});return n.disposeIntermediateTensorInfo(h),m}var L $ e={kernelName:Rw,backendName:"webgl",kernelFunc:z $ e};function B $ e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{numOrSizeSplits:o,axis:a}=r,i=E.parseAxisParam(a,s.shape)[0],l=G.prepareSplitSize(s,o,i),u=s.shape.length,c=new Array(u).fill(0),p=s.shape.slice();return l.map(d=>{let f=[...p];f[i]=d;let h=Q0({inputs:{x:s},backend:n,attrs:{begin:c,size:f}});return c[i]+=d,h})}var V $ e={kernelName:s0,backendName:"webgl",kernelFunc:B $ e},gG="return sqrt(x);",W $ e=tn({opSnippet:gG,packedOpSnippet:gG,cpuKernelImpl:PNe}),G $ e={kernelName:Tf,backendName:"webgl",kernelFunc:W $ e},U $ e="return x * x;",j $ e=tn({opSnippet:U $ e}),H $ e={kernelName:K1,backendName:"webgl",kernelFunc:j $ e},xG="return (a - b) * (a - b);",q $ e=ks({opSnippet:xG,packedOpSnippet:xG}),K $ e={kernelName:Rf,backendName:"webgl",kernelFunc:q $ e};function X $ e({inputs:e,attrs:t,backend:n}){let{x:r}=e,s=Ti+ `
2021-09-11 16:29:31 +02:00
return x > 0.0 ? 1.0 : float ( $ { t . alpha } ) ;
2021-10-04 22:29:15 +02:00
` ,o=new rp(r.shape,s);return n.runWebGLProgram(o,[r],r.dtype)}var Z $ e={kernelName:Pc,backendName:"webgl",kernelFunc:X $ e},Y $ e=class{constructor(e,t,n){this.variableNames=["x"],this.outputShape=n;let r=n.length,s=Dn(n.length),o=Dn(n.length),a="";if(r===1)a="coords * strides + begin";else{let i=0;a=n.map((l,u)=>(i++,n.length===1? ` coords * strides [ $ { u } ] + begin [ $ { u } ] ` : ` coords [ $ { i - 1 } ] * strides [ $ { u } ] + begin [ $ { u } ] ` )).join(",")}this.userCode= `
$ { s } begin = $ { s } ( $ { e } ) ;
$ { s } strides = $ { s } ( $ { t } ) ;
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { o } coords = getOutputCoords ( ) ;
setOutput ( getX ( $ { a } ) ) ;
2021-08-14 17:16:26 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function J $ e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,end:a,strides:i,beginMask:l,endMask:u,ellipsisMask:c,newAxisMask:p,shrinkAxisMask:d}=r,{nonStrided:f, $ begin:h, $ strides:m,size:g,newShape:x,outShape:y}=Bs.sliceInfo(s.shape,o,a,i,l,u,c,p,d),b=qe({inputs:{x:s},backend:n,attrs:{shape:x}}),A;if(f){let w=Q0({inputs:{x:b},backend:n,attrs:{begin:h,size:g}});A=qe({inputs:{x:w},backend:n,attrs:{shape:y}}),n.disposeIntermediateTensorInfo(w)}else if(y.some(w=>w===0))A=n.makeTensorInfo(y,s.dtype,[]);else if(n.shouldExecuteOnCPU([b])){let I=n.texData.get(b.dataId).values,N=Wt(b.shape,b.dtype,I),T=MNe(y,N,m,h);A=n.makeTensorInfo(y,b.dtype,T.values)}else{let k=new Y $ e(h,m,y);A=n.runWebGLProgram(k,[b],b.dtype)}let v=qe({inputs:{x:A},backend:n,attrs:{shape:y}});return n.disposeIntermediateTensorInfo(b),n.disposeIntermediateTensorInfo(A),v}var Q $ e={kernelName:o0,backendName:"webgl",kernelFunc:J $ e};function e6e(e){let{inputs:t,backend:n,attrs:r}=e,{separator:s,nGramWidths:o,leftPad:a,rightPad:i,padWidth:l,preserveShortSequences:u}=r,{data:c,dataSplits:p}=t,d=n.readSync(c.dataId),f=n.readSync(p.dataId),[h,m]=zNe(d,f,s,o,a,i,l,u);return[n.makeTensorInfo([h.length],"string",h),n.makeTensorInfo(p.shape,"int32",m)]}var t6e={kernelName:Dw,backendName:"webgl",kernelFunc:e6e};function n6e(e){let{inputs:t,backend:n,attrs:r}=e,{skipEmpty:s}=r,{input:o,delimiter:a}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(o.shape.length!==1)throw new Error( ` Input must be a vector , got shape : $ { o . shape } ` );if(a.shape.length!==0)throw new Error( ` Delimiter must be a scalar , got shape : $ { a . shape } ` );let i=n.readSync(o.dataId),l=n.readSync(a.dataId)[0],[u,c,p]=LNe(i,l,s),d=c.length;return[n.makeTensorInfo([d,2],"int32",u),n.makeTensorInfo([d],"string",c),n.makeTensorInfo([2],"int32",new Int32Array(p))]}var r6e={kernelName: $ w,backendName:"webgl",kernelFunc:n6e};function s6e(e){let{inputs:t,backend:n,attrs:r}=e,{numBuckets:s}=r,{input:o}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(s<=0)throw new Error("Number of buckets must be at least 1");let a=n.readSync(o.dataId),i=BNe(a,s);return n.makeTensorInfo(o.shape,"int32",i)}var o6e={kernelName:Fw,backendName:"webgl",kernelFunc:s6e},a6e="return tan(x);",i6e=tn({opSnippet:a6e}),l6e={kernelName: $ f,backendName:"webgl",kernelFunc:i6e},u6e= `
2021-09-11 16:29:31 +02:00
float e2x = exp ( - 2.0 * abs ( x ) ) ;
return sign ( x ) * ( 1.0 - e2x ) / ( 1.0 + e2x ) ;
2021-10-04 22:29:15 +02:00
` ,c6e=tn({opSnippet:u6e}),p6e={kernelName:Ff,backendName:"webgl",kernelFunc:c6e},d6e=class{constructor(e,t){this.variableNames=["A"];let n=new Array(e.length);for(let o=0;o<n.length;o++)n[o]=e[o]*t[o];this.outputShape=n,this.rank=n.length;let r=Dn(this.rank),s=f6e(e);this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
2021-10-04 22:29:15 +02:00
$ { r } resRC = getOutputCoords ( ) ;
setOutput ( getA ( $ { s } ) ) ;
2021-08-14 17:17:51 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function f6e(e){let t=e.length;if(t>5)throw Error( ` Tile for rank $ { t } is not yet supported ` );if(t===1)return ` imod ( resRC , $ { e [ 0 ] } ) ` ;let n=["resRC.x","resRC.y","resRC.z","resRC.w","resRC.u"],r=[];for(let s=0;s<e.length;s++)r.push( ` imod ( $ { n [ s ] } , $ { e [ s ] } ) ` );return r.join()}function yG(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reps:o}=r;if(s.dtype==="string"||s.shape.length>5){let l=n.readSync(s.dataId),u=s.dtype==="string"?l.map(d=>E.decodeString(d)):l,c=Wt(s.shape,s.dtype,u),p=WNe(c,o);return n.makeTensorInfo(p.shape,p.dtype,p.values)}let a=new d6e(s.shape,o);return n.runWebGLProgram(a,[s],s.dtype)}var h6e={kernelName:Oc,backendName:"webgl",kernelFunc:yG},m6e=class{constructor(e){this.variableNames=["x","indices"],this.customUniforms=[{name:"n",type:"int"},{name:"firstPass",type:"int"},{name:"negativeInf",type:"float"},{name:"dir",type:"int"},{name:"inc",type:"int"}],this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int elemIdx = coords [ 1 ] ;
// We compare elements pair-wise within a group of size 2 * inc.
// The comparing rule for each group alternates between ascending
// and descending. Within each group, we compare each pair at
// positions i and i+inc. To decide whether an element at position i
// is x0 or x1, we mod it by 2 * inc, if the result is smaller than
// inc, it is in the first half of the group, we denote it as x0,
// otherwise we denote it as x1.
// For example, as shown in the Bitonic top K paper referenced above,
// Figure5(a) shows that element[1] is in the
// second half of the group when group size is 2, but it is in the
// first half of the group when group size is 4.
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
bool isFirstInPair = imod ( elemIdx , 2 * inc ) < inc ;
int i = isFirstInPair ? elemIdx : elemIdx - inc ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int i0 = firstPass == 1 ? i : int ( getIndices ( batch , i ) ) ;
int i1 = firstPass == 1 ? i + inc : int ( getIndices ( batch , i + inc ) ) ;
float x0 = i0 < n ? getX ( batch , i0 ) : negativeInf ;
float x1 = i1 < n ? getX ( batch , i1 ) : negativeInf ;
// Denotes which direction indices are in (ascending or descending).
bool reverse = imod ( elemIdx , 2 * dir ) >= dir ;
bool isGreater = x0 > x1 || ( x0 == x1 && i1 > i0 ) ;
if ( reverse == isGreater ) { // Elements in opposite order of direction
int iTemp = i0 ;
i0 = i1 ;
i1 = iTemp ;
}
if ( isFirstInPair ) {
setOutput ( float ( i0 ) ) ;
} else {
setOutput ( float ( i1 ) ) ;
}
}
2021-10-04 22:29:15 +02:00
` }},g6e=class{constructor(e){this.variableNames=["x","indices"],this.customUniforms=[{name:"n",type:"int"},{name:"firstPass",type:"int"},{name:"k",type:"int"}],this.outputShape=e,this.userCode= `
2021-09-11 16:29:31 +02:00
void main ( ) {
// Takes max of indices (0, k), (1, k + 1), (2, k + 2) ...
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int elemIdx = coords [ 1 ] ;
// The output size is half of the previous size.
// If the previous sequence is | | | | _ _ _ _ | | | | _ _ _ _ (k=4),
// we only need to output the indices at positions |, the indices at
// positions _ can be thrown away, see Figure5(b) After Phase 2
// (Merge phase) in the Bitonic Top K paper referenced above.
// For example, the paper shows we only need to output the orange bars.
// The output sequence should look like this | | | | | | | |.
// Because the sequence is halved, to map the output index back
// to the previous sequence to find the corresponding value,
// we need to double the index. When we double the index,
// we basically interpolate a position, so 2i looks like
// | _ | _ | _ | _ | _ | _ | _. We move the | to the first k position
// of each 2k positions by - elemIdx % k. E.g. for output at
// index 4,5,6,7, we want to get the corresponding element at
// original index 8,9,10,11, for output at index 8,9,10,11,
// we want to get the corresponding element at original index
// 16,17,18,19, so on and so forth.
int i = elemIdx < k ? elemIdx : ( elemIdx * 2 - imod ( elemIdx , k ) ) ;
int i0 = firstPass == 1 ? i : int ( getIndices ( batch , i ) ) ;
int i1 = firstPass == 1 ? i + k : int ( getIndices ( batch , i + k ) ) ;
float x0 = getX ( batch , i0 ) ;
float x1 = i1 < n ? getX ( batch , i1 ) : x0 ;
setOutput ( x0 >= x1 ? float ( i0 ) : float ( i1 ) ) ;
}
2021-10-04 22:29:15 +02:00
` }};function yh(e,t){t!==null&&e.disposeIntermediateTensorInfo(t)}function bG(e){let t=1;for(;t<e;)t*=2;return t}function x6e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{k:o,sorted:a}=r,i=be().getNumber("TOPK_LAST_DIM_CPU_HANDOFF_SIZE_THRESHOLD"),l=be().getNumber("TOPK_K_CPU_HANDOFF_THRESHOLD"),u=s.shape,c=u[u.length-1];if(n.shouldExecuteOnCPU([s])||c<i||o>l){let T=n.readSync(s.dataId),[_,R]=GNe(T,u,s.dtype,o,a);return[n.makeTensorInfo(_.shape,_.dtype,_.values),n.makeTensorInfo(R.shape,R.dtype,R.values)]}if(o===0)return u[u.length-1]=0,[n.makeTensorInfo(u,s.dtype,[]),n.makeTensorInfo(u,"int32",[])];if(c===1)return[s,aA({attrs:{shape:u,dtype:"int32",value:0},backend:n})];let p=n.texData.get(s.dataId),d=p!==null&&p.isPacked,f=d?n.unpackTensor(s):s,m=E.sizeFromShape(u)/c,g=qe({inputs:{x:f},attrs:{shape:[m,c]},backend:n});d&&yh(n,f);let x=bG(o),y=bG(c),b=null,A=()=>b===null?[g,g]:[g,b],v=(T,_,R)=>{let S=A(),F=new m6e(R),W=[[c],[b===null?1:0],[Number.NEGATIVE_INFINITY],[T],[_]],q=b;b=n.runWebGLProgram(F,S,"int32",W),yh(n,q)};for(let T=1;T<x;T*=2){let _=T*2;for(let R=T;R>=1;R/=2)v(_,R,[m,y])}for(let T=y;T>x;T/=2){let _=A(),R=new g6e([m,T/2]),F=[[c],[b===null?1:0],[x]],B=b;b=n.runWebGLProgram(R,_,"int32",F),yh(n,B);let W=x/2,q=W*2;for(let K=W;K>=1;K/=2)v(q,K,b.shape)}let w=b;b=Q0({inputs:{x:b},backend:n,attrs:{begin:0,size:[m,o]}}),yh(n,w);let k=sG({inputs:{x:g,indices:b},backend:n,attrs:{axis:1,batchDims:1}});yh(n,g);let I=u.slice(0,-1);I.push(o),w=b,b=qe({inputs:{x:b},attrs:{shape:I},backend:n}),yh(n,w);let N=k;return k=qe({inputs:{x:k},attrs:{shape:I},backend:n}),yh(n,N),[k,b]}var y6e={kernelName:a0,backendName:"webgl",kernelFunc:x6e},b6e=class{constructor(e,t,n,r,s,o){this.variableNames=["Image","Transforms"],this.outputShape=o;let a=n==="nearest"?1:2,i;switch(r){case"constant":i=1;break;case"reflect":i=2;break;case"wrap":i=3;break;case"nearest":i=4;break;default:i=1;break}this.userCode= `
2021-09-11 16:29:31 +02:00
float mapCoord ( float outCoord , float len ) {
float inCoord = outCoord ;
2021-09-11 17:17:13 +02:00
if ( $ { i } == 2 ) {
2021-09-11 16:29:31 +02:00
if ( inCoord < 0.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz2 = 2.0 * len ;
if ( inCoord < sz2 ) {
inCoord = sz2 * float ( int ( float ( - inCoord / sz2 ) ) ) +
inCoord ;
}
inCoord = inCoord < - len ? inCoord + sz2 : - inCoord - 1.0 ;
}
} else if ( inCoord > len - 1.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz2 = 2.0 * len ;
inCoord -= sz2 * float ( int ( float ( inCoord / sz2 ) ) ) ;
if ( inCoord >= len ) {
inCoord = sz2 - inCoord - 1.0 ;
}
}
}
return clamp ( inCoord , 0.0 , len - 1.0 ) ;
2021-09-11 17:17:13 +02:00
} else if ( $ { i } == 3 ) {
2021-09-11 16:29:31 +02:00
if ( inCoord < 0.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz = len - 1.0 ;
inCoord += len * ( float ( int ( float ( - inCoord / sz ) ) ) + 1.0 ) ;
}
} else if ( inCoord > len - 1.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz = len - 1.0 ;
inCoord -= len * float ( int ( float ( inCoord / sz ) ) ) ;
}
}
return clamp ( inCoord , 0.0 , len - 1.0 ) ;
2021-09-11 17:17:13 +02:00
} else if ( $ { i } == 4 ) {
2021-09-11 16:29:31 +02:00
return clamp ( outCoord , 0.0 , len - 1.0 ) ;
} else {
return outCoord ;
}
2021-07-29 22:35:16 +02:00
}
2021-09-11 16:29:31 +02:00
float readWithFillValue ( int batch , int coordY , int coordX ,
int channel ) {
float outputValue ;
if ( 0 <= coordY && coordY < $ { e } && 0 <= coordX && coordX < $ { t } ) {
outputValue = getImage ( batch , coordY , coordX , channel ) ;
} else {
2021-10-04 22:29:15 +02:00
outputValue = float ( $ { s } ) ;
2021-09-11 16:29:31 +02:00
}
return outputValue ;
}
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
float outputValue ;
int batch = coords [ 0 ] ;
int x = coords [ 2 ] ;
int y = coords [ 1 ] ;
int channel = coords [ 3 ] ;
float xf = float ( x ) ;
float yf = float ( y ) ;
float a1 = getTransforms ( batch , 0 ) ;
float a2 = getTransforms ( batch , 1 ) ;
float a3 = getTransforms ( batch , 2 ) ;
float b1 = getTransforms ( batch , 3 ) ;
float b2 = getTransforms ( batch , 4 ) ;
float b3 = getTransforms ( batch , 5 ) ;
float c1 = getTransforms ( batch , 6 ) ;
float c2 = getTransforms ( batch , 7 ) ;
float projection = c1 * xf + c2 * yf + 1.0 ;
if ( projection == 0.0 ) {
2021-10-04 22:29:15 +02:00
outputValue = float ( $ { s } ) ;
2021-09-11 16:29:31 +02:00
} else {
float inX = ( a1 * xf + a2 * yf + a3 ) / projection ;
float inY = ( b1 * xf + b2 * yf + b3 ) / projection ;
float mapX = mapCoord ( inX , float ( $ { t } ) ) ;
float mapY = mapCoord ( inY , float ( $ { e } ) ) ;
2021-08-14 17:16:26 +02:00
2021-10-04 22:29:15 +02:00
if ( $ { a } == 1 ) {
2021-09-11 16:29:31 +02:00
int coordY = int ( round ( mapY ) ) ;
int coordX = int ( round ( mapX ) ) ;
outputValue = readWithFillValue ( batch , coordY , coordX ,
channel ) ;
} else {
float yFloor = floor ( mapY ) ;
float xFloor = floor ( mapX ) ;
float yCeil = yFloor + 1.0 ;
float xCeil = xFloor + 1.0 ;
float valueYFloor = ( xCeil - mapX ) *
readWithFillValue ( batch , int ( yFloor ) , int ( xFloor ) , channel ) +
( mapX - xFloor ) *
readWithFillValue ( batch , int ( yFloor ) , int ( xCeil ) , channel ) ;
float valueYCeil = ( xCeil - mapX ) *
readWithFillValue ( batch , int ( yCeil ) , int ( xFloor ) , channel ) +
( mapX - xFloor ) *
readWithFillValue ( batch , int ( yCeil ) , int ( xCeil ) , channel ) ;
outputValue = ( yCeil - mapY ) * valueYFloor +
( mapY - yFloor ) * valueYCeil ;
}
}
setOutput ( outputValue ) ;
}
2021-10-04 22:29:15 +02:00
` }};function A6e(e){let{inputs:t,backend:n,attrs:r}=e,{image:s,transforms:o}=t,{interpolation:a,fillMode:i,fillValue:l,outputShape:u}=r,[c,p,d,f]=s.shape,[h,m]=u!=null?u:[p,d],g=[c,h,m,f],x=new b6e(p,d,a,i,l,g);return n.runWebGLProgram(x,[s,o],"float32")}var v6e={kernelName:i0,backendName:"webgl",kernelFunc:A6e};function w6e(e){let{inputs:t,attrs:n,backend:r}=e,{axis:s}=n,{x:o}=t;j0(o,"unique"),console.warn("WARNING: ","UI might be locked temporarily as data is being downloaded");let a=r.readSync(o.dataId),{outputValues:i,outputShape:l,indices:u}=UNe(a,s,o.shape,o.dtype);return[r.makeTensorInfo(l,o.dtype,i),r.makeTensorInfo([u.length],"int32",u)]}var k6e={kernelName:Ow,backendName:"webgl",kernelFunc:w6e};function I6e(e){let{inputs:t,backend:n,attrs:r}=e,{value:s}=t,{axis:o}=r;o<0&&(o+=s.shape.length);let a=s,i=a.shape.length,l=s.shape[o],u=new Array(i-1),c=0;for(let m=0;m<i;m++)m!==o&&(u[c++]=a.shape[m]);let p=[],d=new Array(i).fill(0),f=a.shape.slice();f[o]=1;let h=new Array(l);for(let m=0;m<h.length;m++){d[o]=m;let g=Q0({inputs:{x:a},backend:n,attrs:{begin:d,size:f}}),x=qe({inputs:{x:g},backend:n,attrs:{shape:u}});h[m]=x,p.push(g)}return p.forEach(m=>n.disposeIntermediateTensorInfo(m)),h}var C6e={kernelName:l0,backendName:"webgl",kernelFunc:I6e},S6e=class{constructor(e,t){this.variableNames=["x","segmentIds"];let n=e.windowSize,r=e.batchSize,s=e.inSize,o=e.numSegments,a=o*Math.ceil(s/n);this.outputShape=[r,a];let i="0.0",l="sumValue",u=Math.floor(n/4)*4,c=n%4,p= `
2021-09-11 16:29:31 +02:00
sumValue += dot ( values , segFilter ) ;
2021-10-04 22:29:15 +02:00
` ,d="";s%n>0&&(d= `
if ( inIdx < 0 || inIdx >= $ { s } ) {
2021-09-11 16:29:31 +02:00
return initializationValue ;
2021-09-05 22:42:11 +02:00
}
2021-10-04 22:29:15 +02:00
` );let f="";s%n>0&&(f= `
if ( inIdx < 0 || inIdx >= $ { s } ) {
2021-09-11 16:29:31 +02:00
return - 1.0 ;
2021-09-05 22:42:11 +02:00
}
2021-09-11 16:29:31 +02:00
` ),this.userCode= `
2021-09-11 17:17:13 +02:00
const float initializationValue = $ { i } ;
2021-09-11 16:29:31 +02:00
float getValue ( int batch , int inIdx ) {
2021-10-04 22:29:15 +02:00
$ { d }
2021-09-11 16:29:31 +02:00
return getX ( batch , inIdx ) ;
2021-09-01 00:22:16 +02:00
}
2021-07-29 22:35:16 +02:00
2021-09-11 16:29:31 +02:00
float getSegmentIdAtIndex ( int inIdx ) {
2021-10-04 22:29:15 +02:00
$ { f }
2021-09-11 16:29:31 +02:00
return getSegmentIds ( inIdx ) ;
2021-09-05 22:42:11 +02:00
}
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = int ( floor ( float ( outIdx ) / float (
2021-10-04 22:29:15 +02:00
$ { o } ) ) * float ( $ { n } ) ) ;
int currentSeg = int ( mod ( float ( outIdx ) , float ( $ { o } ) ) ) ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
float sumValue = 0.0 ;
2021-08-14 17:17:51 +02:00
2021-09-11 16:29:31 +02:00
for ( int i = 0 ; i < $ { u } ; i += 4 ) {
int inIdx = inOffset + i ;
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 2 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 3 ) ) == currentSeg ? 1 : 0
) ;
2021-09-01 00:24:30 +02:00
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-05 22:42:11 +02:00
}
2021-09-01 00:24:30 +02:00
2021-09-11 16:29:31 +02:00
int inIdx = inOffset + $ { u } ;
2021-09-11 17:17:13 +02:00
if ( $ { c === 1 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
int inIdxSeg = int ( getSegmentIdAtIndex ( inIdx ) ) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
0 ,
0 ,
0
) ;
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-11 17:17:13 +02:00
} else if ( $ { c === 2 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
initializationValue ,
initializationValue
) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
0 ,
0
) ;
2021-10-04 22:29:15 +02:00
$ { p }
2021-09-11 17:17:13 +02:00
} else if ( $ { c === 3 } ) {
2021-09-11 16:29:31 +02:00
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
initializationValue
) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 2 ) ) == currentSeg ? 1 : 0 ,
0
) ;
2021-10-04 22:29:15 +02:00
$ { p }
2021-06-07 02:34:29 +02:00
}
2021-09-11 16:29:31 +02:00
setOutput ( $ { l } ) ;
2021-06-07 02:34:29 +02:00
}
2021-10-04 22:29:15 +02:00
` }};function N6e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,segmentIds:o}=t,{numSegments:a}=r,i=s.shape.length,l=[],u=0,c=G.getAxesPermutation([u],i),p=s;c!=null&&(p=Hs({inputs:{x:s},backend:n,attrs:{perm:c}}),l.push(p),u=G.getInnerMostAxes(1,i)[0]);let d=G.segment_util.computeOutShape(p.shape,u,a),f=E.sizeFromShape([p.shape[u]]),h=qe({inputs:{x:p},backend:n,attrs:{shape:[-1,f]}});l.push(h);let m=Ww(s.dtype),g=(A,v,w,k,I)=>{let N=A.shape[0],T=A.shape[1],_=G.segment_util.segOpComputeOptimalWindowSize(T,I),R={windowSize:_,inSize:T,batchSize:N,numSegments:I},S=new S6e(R,v),F=n.compileAndRun(S,[A,w],k);if(l.push(F),F.shape[1]===I)return F;let B=fG({backend:n,attrs:{start:0,stop:I,step:1,dtype:"float32"}}),W=yG({inputs:{x:B},backend:n,attrs:{reps:[T/_]}});return l.push(B),l.push(W),g(F,v,W,k,I)},x=g(h,"unsortedSegmentSum",o,m,a),y=qe({inputs:{x},backend:n,attrs:{shape:d}}),b=y;if(c!=null){l.push(y);let A=G.getUndoAxesPermutation(c);b=Hs({inputs:{x:b},backend:n,attrs:{perm:A}})}return l.forEach(A=>n.disposeIntermediateTensorInfo(A)),b}var T6e={kernelName:X1,backendName:"webgl",kernelFunc:N6e},_6e=[r7e,a7e,GTe,jTe,KTe,YTe,QTe,n_e,s_e,a_e,c_e,d_e,m_e,y_e,C_e,v_e,T_e,D_e,E_e,P_e,z_e,B_e,U_e,Y_e,Q_e,oEe,iEe,pEe,hEe,ITe,bEe,_Ee,REe,kEe,OEe,MEe, $ Ee,BEe,GEe,HEe,KEe,ZEe,QEe,o4e,i4e,t4e,c4e,f4e,m4e,b4e,k4e,N4e,E4e,R4e,D4e,F4e,P4e,z4e,B4e,W4e,H4e,X4e,J4e,eRe,rRe,aRe,cRe,hRe,kTe,gRe,xEe,bRe,wRe,CRe,STe,_Re, $ Re,ORe,WRe,LRe,HRe,XRe,QRe,l7e,g7e,h7e,A7e,w7e,I7e,d7e,S7e,T7e,D7e,P7e,B7e,K7e,RTe,Z7e,Q7e,nDe,oDe,tEe,lDe,cDe,dDe,mDe,bDe,TTe,vDe,wDe,nEe,U7e,CDe, $ De,_De, $ Te,MDe,BDe,UDe,qDe,YDe,QDe,n $ e,o $ e,i $ e,c $ e,f $ e,m $ e,y $ e,v $ e,I $ e,X_e,H7e,N $ e,_ $ e,R $ e, $ $ e,O $ e,M $ e,L $ e,V $ e,G $ e,H $ e,K $ e,Z $ e,Q $ e,t6e,r6e,o6e,j7e,BTe,l6e,p6e,h6e,y6e,v6e,VTe,k6e,C6e,T6e,uDe];for(let e of _6e)Mc(e);var ko;(function(e){e[e.float32=0]="float32",e[e.int32=1]="int32",e[e.bool=2]="bool",e[e.string=3]="string",e[e.complex64=4]="complex64"})(ko||(ko={}));var iA;(function(e){e[e.linear=0]="linear",e[e.relu=1]="relu",e[e.relu6=2]="relu6",e[e.prelu=3]="prelu",e[e.leakyrelu=4]="leakyrelu",e[e.sigmoid=5]="sigmoid",e[e.elu=6]="elu"})(iA||(iA={}));var AG;function E6e(e){AG=e.wasm.cwrap(Pf,null,["number","array","number","number","array","number","number","number","number","number","number","number","number"])}function R6e(e){let{inputs:t,backend:n,attrs:r}=e,{a:s,b:o,bias:a,preluActivationWeights:i}=t;if(s.dtype!=="float32"||o.dtype!=="float32")throw new Error("_FusedMatMul for non non-float32 tensors not yet supported.");let{transposeA:l,transposeB:u,activation:c,leakyreluAlpha:p}=r,d=n.dataIdMap.get(s.dataId).id,f=n.dataIdMap.get(o.dataId).id,h=0;if(a!=null){let I=n.dataIdMap.get(a.dataId);if(I.shape.length!==1)throw new Error( ` _FusedMatMul only supports rank - 1 bias but got rank $ { I . shape . length } . ` );h=I.id}let m=i==null?0:n.dataIdMap.get(i.dataId).id,g=iA[c];if(g==null)throw new Error( ` $ { c } activation not yet supported for FusedConv2D in the wasm backend . ` );let x=l?s.shape[2]:s.shape[1],y=u?o.shape[1]:o.shape[2],b=s.shape[0],A=n.makeOutput([b,x,y],s.dtype),v=n.dataIdMap.get(A.dataId).id,w=new Uint8Array(new Int32Array(s.shape).buffer),k=new Uint8Array(new Int32Array(o.shape).buffer);return AG(d,w,s.shape.length,f,k,o.shape.length,l,u,g,h,m,p||0,v),A}var D6e={kernelName:Pf,backendName:"wasm",setupFunc:E6e,kernelFunc:R6e};function Is(e){let t;function n(s){t=s.wasm.cwrap(e,null,["number","number"])}function r(s){let{backend:o,inputs:{x:a}}=s,i=o.dataIdMap.get(a.dataId).id,l=o.makeOutput(a.shape,a.dtype),u=o.dataIdMap.get(l.dataId).id;return E.sizeFromShape(l.shape)===0||t(i,u),l}return{kernelName:e,backendName:"wasm",setupFunc:n,kernelFunc:r}}var $ 6e=Is(cg);function qs(e,t,n){let r;function s(a){r=a.wasm.cwrap(e,null,["number","array","number","number","array","number","number","number"])}function o(a){let{backend:i,inputs:l}=a,{a:u,b:c}=l,p=i.dataIdMap.get(u.dataId).id,d=i.dataIdMap.get(c.dataId).id,f=n!=null?n:u.dtype,h=G.assertAndGetBroadcastShape(u.shape,c.shape),m=i.makeOutput(h,f);if(E.sizeFromShape(h)===0)return m;let g=new Uint8Array(new Int32Array(u.shape).buffer),x=new Uin
` ),w.FRAGMENT_IDENTITY=["precision highp float;","varying vec2 vUv;","uniform sampler2D texture;","void main(void) {","gl_FragColor = texture2D(texture, vUv);","}"].join( `
` ),c=new YPe(m,w.VERTEX_IDENTITY,v);let k=Float32Array.BYTES_PER_ELEMENT,I=4*k;return m.enableVertexAttribArray(c.attribute.pos),m.vertexAttribPointer(c.attribute.pos,2,m.FLOAT,!1,I,0*k),m.enableVertexAttribArray(c.attribute.uv),m.vertexAttribPointer(c.attribute.uv,2,m.FLOAT,!1,I,2*k),f[v]=c,c};p.colorMatrix=function(v){let w=new Float32Array(v);w[4]/=255,w[9]/=255,w[14]/=255,w[19]/=255;let k=w[18]===1&&w[3]===0&&w[8]===0&&w[13]===0&&w[15]===0&&w[16]===0&&w[17]===0&&w[19]===0?p.colorMatrix.SHADER.WITHOUT_ALPHA:p.colorMatrix.SHADER.WITH_ALPHA,I=A(k);m.uniform1fv(I.uniform.m,w),b()},p.colorMatrix.SHADER={},p.colorMatrix.SHADER.WITH_ALPHA=["precision highp float;","varying vec2 vUv;","uniform sampler2D texture;","uniform float m[20];","void main(void) {","vec4 c = texture2D(texture, vUv);","gl_FragColor.r = m[0] * c.r + m[1] * c.g + m[2] * c.b + m[3] * c.a + m[4];","gl_FragColor.g = m[5] * c.r + m[6] * c.g + m[7] * c.b + m[8] * c.a + m[9];","gl_FragColor.b = m[10] * c.r + m[11] * c.g + m[12] * c.b + m[13] * c.a + m[14];","gl_FragColor.a = m[15] * c.r + m[16] * c.g + m[17] * c.b + m[18] * c.a + m[19];","}"].join( `
` ),p.colorMatrix.SHADER.WITHOUT_ALPHA=["precision highp float;","varying vec2 vUv;","uniform sampler2D texture;","uniform float m[20];","void main(void) {","vec4 c = texture2D(texture, vUv);","gl_FragColor.r = m[0] * c.r + m[1] * c.g + m[2] * c.b + m[4];","gl_FragColor.g = m[5] * c.r + m[6] * c.g + m[7] * c.b + m[9];","gl_FragColor.b = m[10] * c.r + m[11] * c.g + m[12] * c.b + m[14];","gl_FragColor.a = c.a;","}"].join( `
` ),p.brightness=function(v){let w=(v||0)+1;p.colorMatrix([w,0,0,0,0,0,w,0,0,0,0,0,w,0,0,0,0,0,1,0])},p.saturation=function(v){let w=(v||0)*2/3+1,k=(w-1)*-.5;p.colorMatrix([w,k,k,0,0,k,w,k,0,0,k,k,w,0,0,0,0,0,1,0])},p.desaturate=function(){p.saturation(-1)},p.contrast=function(v){let w=(v||0)+1,k=-128*(w-1);p.colorMatrix([w,0,0,0,k,0,w,0,0,k,0,0,w,0,k,0,0,0,1,0])},p.negative=function(){p.contrast(-2)},p.hue=function(v){v=(v||0)/180*Math.PI;let w=Math.cos(v),k=Math.sin(v),I=.213,N=.715,T=.072;p.colorMatrix([I+w*(1-I)+k*-I,N+w*-N+k*-N,T+w*-T+k*(1-T),0,0,I+w*-I+k*.143,N+w*(1-N)+k*.14,T+w*-T+k*-.283,0,0,I+w*-I+k*-(1-I),N+w*-N+k*N,T+w*(1-T)+k*T,0,0,0,0,0,1,0])},p.desaturateLuminance=function(){p.colorMatrix([.2764723,.929708,.0938197,0,-37.1,.2764723,.929708,.0938197,0,-37.1,.2764723,.929708,.0938197,0,-37.1,0,0,0,1,0])},p.sepia=function(){p.colorMatrix([.393,.7689999,.18899999,0,0,.349,.6859999,.16799999,0,0,.272,.5339999,.13099999,0,0,0,0,0,1,0])},p.brownie=function(){p.colorMatrix([.5997023498159715,.34553243048391263,-.2708298674538042,0,47.43192855600873,-.037703249837783157,.8609577587992641,.15059552388459913,0,-36.96841498319127,.24113635128153335,-.07441037908422492,.44972182064877153,0,-7.562075277591283,0,0,0,1,0])},p.vintagePinhole=function(){p.colorMatrix([.6279345635605994,.3202183420819367,-.03965408211312453,0,9.651285835294123,.02578397704808868,.6441188644374771,.03259127616149294,0,7.462829176470591,.0466055556782719,-.0851232987247891,.5241648018700465,0,5.159190588235296,0,0,0,1,0])},p.kodachrome=function(){p.colorMatrix([1.1285582396593525,-.3967382283601348,-.03992559172921793,0,63.72958762196502,-.16404339962244616,1.0835251566291304,-.05498805115633132,0,24.732407896706203,-.16786010706155763,-.5603416277695248,1.6014850761964943,0,35.62982807460946,0,0,0,1,0])},p.technicolor=function(){p.colorMatrix([1.9125277891456083,-.8545344976951645,-.09155508482755585,0,11.793603434377337,-.3087833385928097,1.7658908555458428,-.10601743074722245,0,-70.35205161461398,-.231103377548616,-.7501899197440212,1.847597816108189,0,30.950940869491138,0,0,0,1,0])},p.polaroid=function(){p.colorMatrix([1.438,-.062,-.062,0,0,-.122,1.378,-.122,0,0,-.016,-.016,1.483,0,0,0,0,0,1,0])},p.shiftToBGR=function(){p.colorMatrix([0,0,1,0,0,0,1,0,0,0,1,0,0,0,0,0,0,0,1,0])},p.convolution=function(v){let w=new Float32Array(v),k=1/i,I=1/l,N=A(p.convolution.SHADER);m.uniform1fv(N.uniform.m,w),m.uniform2f(N.uniform.px,k,I),b()},p.convolution.SHADER=["precision highp float;","varying vec2 vUv;","uniform sampler2D texture;","uniform vec2 px;","uniform float m[9];","void main(void) {","vec4 c11 = texture2D(texture, vUv - px);","vec4 c12 = texture2D(texture, vec2(vUv.x, vUv.y - px.y));","vec4 c13 = texture2D(texture, vec2(vUv.x + px.x, vUv.y - px.y));","vec4 c21 = texture2D(texture, vec2(vUv.x - px.x, vUv.y) );","vec4 c22 = texture2D(texture, vUv);","vec4 c23 = texture2D(texture, vec2(vUv.x + px.x, vUv.y) );","vec4 c31 = texture2D(texture, vec2(vUv.x - px.x, vUv.y + px.y) );","vec4 c32 = texture2D(texture, vec2(vUv.x, vUv.y + px.y) );","vec4 c33 = texture2D(texture, vUv + px );","gl_FragColor = ","c11 * m[0] + c12 * m[1] + c22 * m[2] +","c21 * m[3] + c22 * m[4] + c23 * m[5] +","c31 * m[6] + c32 * m[7] + c33 * m[8];","gl_FragColor.a = c22.a;","}"].join( `
` ),p.detectEdges=function(){p.convolution.call(this,[0,1,0,1,-4,1,0,1,0])},p.sobelX=function(){p.convolution.call(this,[-1,0,1,-2,0,2,-1,0,1])},p.sobelY=function(){p.convolution.call(this,[-1,-2,-1,0,0,0,1,2,1])},p.sharpen=function(v){let w=v||1;p.convolution.call(this,[0,-1*w,0,-1*w,1+4*w,-1*w,0,-1*w,0])},p.emboss=function(v){let w=v||1;p.convolution.call(this,[-2*w,-1*w,0,-1*w,1,1*w,0,1*w,2*w])},p.blur=function(v){let w=v/7/i,k=v/7/l,I=A(p.blur.SHADER);m.uniform2f(I.uniform.px,0,k),b(h.INTERMEDIATE),m.uniform2f(I.uniform.px,w,0),b()},p.blur.SHADER=["precision highp float;","varying vec2 vUv;","uniform sampler2D texture;","uniform vec2 px;","void main(void) {","gl_FragColor = vec4(0.0);","gl_FragColor += texture2D(texture, vUv + vec2(-7.0*px.x, -7.0*px.y))*0.0044299121055113265;","gl_FragColor += texture2D(texture, vUv + vec2(-6.0*px.x, -6.0*px.y))*0.00895781211794;","gl_FragColor += texture2D(texture, vUv + vec2(-5.0*px.x, -5.0*px.y))*0.0215963866053;","gl_FragColor += texture2D(texture, vUv + vec2(-4.0*px.x, -4.0*px.y))*0.0443683338718;","gl_FragColor += texture2D(texture, vUv + vec2(-3.0*px.x, -3.0*px.y))*0.0776744219933;","gl_FragColor += texture2D(texture, vUv + vec2(-2.0*px.x, -2.0*px.y))*0.115876621105;","gl_FragColor += texture2D(texture, vUv + vec2(-1.0*px.x, -1.0*px.y))*0.147308056121;","gl_FragColor += texture2D(texture, vUv )*0.159576912161;","gl_FragColor += texture2D(texture, vUv + vec2( 1.0*px.x, 1.0*px.y))*0.147308056121;","gl_FragColor += texture2D(texture, vUv + vec2( 2.0*px.x, 2.0*px.y))*0.115876621105;","gl_FragColor += texture2D(texture, vUv + vec2( 3.0*px.x, 3.0*px.y))*0.0776744219933;","gl_FragColor += texture2D(texture, vUv + vec2( 4.0*px.x, 4.0*px.y))*0.0443683338718;","gl_FragColor += texture2D(texture, vUv + vec2( 5.0*px.x, 5.0*px.y))*0.0215963866053;","gl_FragColor += texture2D(texture, vUv + vec2( 6.0*px.x, 6.0*px.y))*0.00895781211794;","gl_FragColor += texture2D(texture, vUv + vec2( 7.0*px.x, 7.0*px.y))*0.0044299121055113265;","}"].join( `
` ),p.pixelate=function(v){let w=v/i,k=v/l,I=A(p.pixelate.SHADER);m.uniform2f(I.uniform.size,w,k),b()},p.pixelate.SHADER=["precision highp float;","varying vec2 vUv;","uniform vec2 size;","uniform sampler2D texture;","vec2 pixelate(vec2 coord, vec2 size) {","return floor( coord / size ) * size;","}","void main(void) {","gl_FragColor = vec4(0.0);","vec2 coord = pixelate(vUv, size);","gl_FragColor += texture2D(texture, coord);","}"].join( `
` )}var Ee={browser:void 0,node:void 0,worker:void 0,platform:void 0,agent:void 0,initial:!0,backends:[],offscreen:void 0,tfjs:{version:void 0},wasm:{supported:void 0,backend:void 0,simd:void 0,multithread:void 0},webgl:{supported:void 0,backend:void 0,version:void 0,renderer:void 0},webgpu:{supported:void 0,backend:void 0,adapter:void 0},kernels:[],Canvas:void 0,Image:void 0,ImageData:void 0};async function JPe(){var n;Ee.backends=Object.keys(zo().registryFactory),Ee.wasm.supported=typeof WebAssembly!="undefined",Ee.wasm.backend=Ee.backends.includes("wasm"),Ee.wasm.supported&&Ee.wasm.backend&&gi()==="wasm"&&(Ee.wasm.simd=await be().getAsync("WASM_HAS_SIMD_SUPPORT"),Ee.wasm.multithread=await be().getAsync("WASM_HAS_MULTITHREAD_SUPPORT"));let e=Zo(100,100),t=e?e.getContext("webgl2"):void 0;if(Ee.webgl.supported=typeof t!="undefined",Ee.webgl.backend=Ee.backends.includes("webgl"),Ee.webgl.supported&&Ee.webgl.backend&&(gi()==="webgl"||gi()==="humangl")){let r=sl().gpgpu!=="undefined"?await sl().getGPGPUContext().gl:null;r&&(Ee.webgl.version=r.getParameter(r.VERSION),Ee.webgl.renderer=r.getParameter(r.RENDERER))}Ee.webgpu.supported=Ee.browser&&typeof navigator.gpu!="undefined",Ee.webgpu.backend=Ee.backends.includes("webgpu"),Ee.webgpu.supported&&(Ee.webgpu.adapter=(n=await navigator.gpu.requestAdapter())==null?void 0:n.name),Ee.kernels=zl(gi()).map(r=>r.kernelName.toLowerCase())}async function K5(){if(Ee.browser=typeof navigator!="undefined",Ee.node=typeof process!="undefined",Ee.worker=Ee.browser?typeof WorkerGlobalScope!="undefined":void 0,Ee.tfjs.version=qw,Ee.offscreen=typeof Ee.offscreen=="undefined"?typeof OffscreenCanvas!="undefined":Ee.offscreen,typeof navigator!="undefined"){let e=navigator.userAgent.match(/ \( ([^()]+) \) /g);if(e&&e[0]){let t=e[0].match(/ \( ([^()]+) \) /g);Ee.platform=t&&t[0]?t[0].replace(/ \( | \) /g,""):"",Ee.agent=navigator.userAgent.replace(e[0],""),Ee.platform[1]&&(Ee.agent=Ee.agent.replace(e[1],"")),Ee.agent=Ee.agent.replace(/ /g," ")}}else typeof process!="undefined"&&(Ee.platform= ` $ { process . platform } $ { process . arch } ` ,Ee.agent= ` NodeJS $ { process . version } ` );await JPe()}async function vU(e){Ee=os(Ee,e)}var X5=2048,wt,rr, $ r;function Zo(e,t){let n;return Ee.browser?Ee.offscreen?n=new OffscreenCanvas(e,t):(n=document.createElement("canvas"),n.width=e,n.height=t):typeof Ee.Canvas!="undefined"?n=new Ee.Canvas(e,t):typeof globalThis.Canvas!="undefined"&&(n=new globalThis.Canvas(e,t)),n}function rx(e,t){let n;if(!e)return t.debug&&Ce("input is missing"),{tensor:null,canvas:null};if(!(e instanceof Vt)&&!(typeof Image!="undefined"&&e instanceof Image)&&!(typeof Ee.Canvas!="undefined"&&e instanceof Ee.Canvas)&&!(typeof globalThis.Canvas!="undefined"&&e instanceof globalThis.Canvas)&&!(typeof ImageData!="undefined"&&e instanceof ImageData)&&!(typeof ImageBitmap!="undefined"&&e instanceof ImageBitmap)&&!(typeof HTMLImageElement!="undefined"&&e instanceof HTMLImageElement)&&!(typeof HTMLMediaElement!="undefined"&&e instanceof HTMLMediaElement)&&!(typeof HTMLVideoElement!="undefined"&&e instanceof HTMLVideoElement)&&!(typeof HTMLCanvasElement!="undefined"&&e instanceof HTMLCanvasElement)&&!(typeof OffscreenCanvas!="undefined"&&e instanceof OffscreenCanvas))throw new Error("input type is not recognized");if(e instanceof Vt){if(e.isDisposedInternal)throw new Error("input tensor is disposed");if(e.shape&&e.shape.length===4&&e.shape[0]===1&&e.shape[3]===3)n=mi(e);else throw new Error( ` input tensor shape must be [ 1 , height , width , 3 ] and instead was $ { e . shape } ` )}else{if(typeof e.readyState!="undefined"&&e.readyState<=2)return t.debug&&Ce("input stream is not ready"),{tensor:null,canvas:wt};let r=e.naturalWidth||e.videoWidth||e.width||e.shape&&e.shape[1]>0,s=e.naturalHeight||e.videoHeight||e.height||e.shape&&e.shape[2]>0;if(!r||!s)return t.debug&&Ce("cannot determine input dimensions"),{tensor:null,canvas:wt};let o=r,a=s;if(o>X5&&(o=X5,a=Math.trunc(o*s/r)),a>X5&&(a=X5,o=Math.trunc(a*r/s)),(t.filter.width||0)>0?o=t.filter.width:(t.filter.height||0)>0&&(o=r*((t.filter.height||0)/s)),(t.filter.height||0)>0?a=t.filter.height:(
` )),l.join( `
` )}function pMe(e,t,n,r){let s=bn(t),o=r[r.length-1],a=new Array(o).fill(0),i=t.length,l=n==="complex64"?DA(e):e;if(i>1)for(let u=0;u<s/o;u++){let c=u*o;for(let p=0;p<o;p++)a[p]=Math.max(a[p],RA(l[c+p],0,n).length)}return a}function RA(e,t,n){let r;return Array.isArray(e)?r= ` $ { parseFloat ( e [ 0 ] . toFixed ( W7 ) ) } + $ { parseFloat ( e [ 1 ] . toFixed ( W7 ) ) } j ` :px(e)?r= ` '${e}' ` :n==="bool"?r=tH(e):r=parseFloat(e.toFixed(W7)).toString(),kh(r,t)}function tH(e){return e===0?"false":"true"}function b3(e,t,n,r,s,o=!0){let a=n==="complex64"?2:1,i=t[0],l=t.length;if(l===0){if(n==="complex64"){let m=DA(e);return[RA(m[0],0,n)]}return n==="bool"?[tH(e[0])]:[e[0].toString()]}if(l===1){if(i>Qj){let g=EA*a,x=Array.from(e.slice(0,g)),y=Array.from(e.slice((i-EA)*a,i*a));return n==="complex64"&&(x=DA(x),y=DA(y)),["["+x.map((b,A)=>RA(b,s[A],n)).join(", ")+", ..., "+y.map((b,A)=>RA(b,s[i-EA+A],n)).join(", ")+"]"]}let m=n==="complex64"?DA(e):Array.from(e);return["["+m.map((g,x)=>RA(g,s[x],n)).join(", ")+"]"]}let u=t.slice(1),c=r.slice(1),p=r[0]*a,d=[];if(i>Qj){for(let m=0;m<EA;m++){let g=m*p,x=g+p;d.push(...b3(e.slice(g,x),u,n,c,s,!1))}d.push("...");for(let m=i-EA;m<i;m++){let g=m*p,x=g+p;d.push(...b3(e.slice(g,x),u,n,c,s,m===i-1))}}else for(let m=0;m<i;m++){let g=m*p,x=g+p;d.push(...b3(e.slice(g,x),u,n,c,s,m===i-1))}let f=l===2?",":"";d[0]="["+d[0]+f;for(let m=1;m<d.length-1;m++)d[m]=" "+d[m]+f;let h= ` ,
` ;for(let m=2;m<l;m++)h+= `
` ;return d[d.length-1]=" "+d[d.length-1]+"]"+(o?"":h),d}function DA(e){let t=[];for(let n=0;n<e.length;n+=2)t.push([e[n],e[n+1]]);return t}var sn=class{constructor(t,n,r){if(this.dtype=n,this.shape=t.slice(),this.size=bn(t),r!=null){let s=r.length;V(s===this.size,()=> ` Length of values '${s}' does not match the size inferred by the shape '${this.size}' . ` )}if(n==="complex64")throw new Error("complex64 dtype TensorBuffers are not supported. Please create a TensorBuffer for the real and imaginary parts separately and call tf.complex(real, imag).");this.values=r||N7(n,this.size),this.strides=yl(t)}set(t,...n){n.length===0&&(n=[0]),V(n.length===this.rank,()=> ` The number of provided coordinates ( $ { n . length } ) must match the rank ( $ { this . rank } ) ` );let r=this.locToIndex(n);this.values[r]=t}get(...t){t.length===0&&(t=[0]);let n=0;for(let s of t){if(s<0||s>=this.shape[n]){let o= ` Requested out of range element at $ { t } . Buffer shape = $ { this . shape } ` ;throw new Error(o)}n++}let r=t[t.length-1];for(let s=0;s<t.length-1;++s)r+=this.strides[s]*t[s];return this.values[r]}locToIndex(t){if(this.rank===0)return 0;if(this.rank===1)return t[0];let n=t[t.length-1];for(let r=0;r<t.length-1;++r)n+=this.strides[r]*t[r];return n}indexToLoc(t){if(this.rank===0)return[];if(this.rank===1)return[t];let n=new Array(this.shape.length);for(let r=0;r<n.length-1;++r)n[r]=Math.floor(t/this.strides[r]),t-=n[r]*this.strides[r];return n[n.length-1]=t,n}get rank(){return this.shape.length}toTensor(){return wl().makeTensor(this.values,this.shape,this.dtype)}},wl=null,yy=null,dMe=null;function nH(e){wl=e}function rH(e){yy=e}function sH(e){dMe=e}var Ot=class{constructor(t,n,r,s){this.kept=!1,this.isDisposedInternal=!1,this.shape=t.slice(),this.dtype=n||"float32",this.size=bn(t),this.strides=yl(t),this.dataId=r,this.id=s,this.rankType=this.rank<5?this.rank.toString():"higher"}get rank(){return this.shape.length}async buffer(){let t=await this.data();return yy.buffer(this.shape,this.dtype,t)}bufferSync(){return yy.buffer(this.shape,this.dtype,this.dataSync())}async array(){let t=await this.data();return Ih(this.shape,t,this.dtype==="complex64")}arraySync(){return Ih(this.shape,this.dataSync(),this.dtype==="complex64")}async data(){this.throwIfDisposed();let t=wl().read(this.dataId);if(this.dtype==="string"){let n=await t;try{return n.map(r=>xy(r))}catch(r){throw new Error("Failed to decode the string bytes into utf-8. To get the original bytes, call tensor.bytes().")}}return t}dataSync(){this.throwIfDisposed();let t=wl().readSync(this.dataId);if(this.dtype==="string")try{return t.map(n=>xy(n))}catch(n){throw new Error("Failed to decode the string bytes into utf-8. To get the original bytes, call tensor.bytes().")}return t}async bytes(){this.throwIfDisposed();let t=await wl().read(this.dataId);return this.dtype==="string"?t:new Uint8Array(t.buffer)}dispose(){this.isDisposed||(wl().disposeTensor(this),this.isDisposedInternal=!0)}get isDisposed(){return this.isDisposedInternal}throwIfDisposed(){if(this.isDisposed)throw new Error("Tensor is disposed.")}print(t=!1){return yy.print(this,t)}clone(){return this.throwIfDisposed(),yy.clone(this)}toString(t=!1){let n=this.dataSync();return eH(n,this.shape,this.dtype,t)}cast(t){return this.throwIfDisposed(),yy.cast(this,t)}variable(t=!0,n,r){return this.throwIfDisposed(),wl().makeVariable(this,t,n,r)}};Object.defineProperty(Ot,Symbol.hasInstance,{value:e=>!!e&&e.data!=null&&e.dataSync!=null&&e.throwIfDisposed!=null});function Z(){return SA("Tensor",()=>Ot)}Z();var od=class extends Ot{constructor(t,n,r,s){super(t.shape,t.dtype,t.dataId,s);this.trainable=n,this.name=r}assign(t){if(t.dtype!==this.dtype)throw new Error( ` dtype of the new value ( $ { t . dtype } ) and previous value ( $ { this . dtype } ) must match ` );if(!Wa(t.shape,this.shape))throw new Error( ` shape of the new value ( $ { t . shape } ) and previous value ( $ { this . shape } ) must match ` );wl().disposeTensor(this),this.dataId=t.dataId,wl().incRef(this,null)}dispose(){wl().disposeVariable(this),this.isDisposedInternal=!0}};Object.defineProperty(od,Symbol.hasInstance,{value:e=>e instanceof Ot&&e.as
Manifest JSON has weights with names : $ { i . join ( ", " ) } . ` )}let l=s.reduce((f,h,m)=>(h&&f.push(m),f),[]),u=[];l.forEach(f=>{t[f].paths.forEach(h=>{let m=n+(n.endsWith("/")?"":"/")+h;u.push(m)})});let c=await e(u),p={},d=0;return l.forEach(f=>{let h=t[f].paths.length,m=0;for(let A=0;A<h;A++)m+=c[d+A].byteLength;let g=new ArrayBuffer(m),x=new Uint8Array(g),y=0;for(let A=0;A<h;A++){let v=new Uint8Array(c[d+A]);x.set(v,y),y+=v.byteLength}o[f].forEach(A=>{let v=g.slice(A.groupOffset,A.groupOffset+A.sizeBytes),w=w3(v,[A.manifestEntry]);for(let k in w)p[k]=w[k]}),d+=h}),p}}var HMe="application/octet-stream",qMe="application/json",C3=class{constructor(t,n){if(this.DEFAULT_METHOD="POST",n==null&&(n={}),this.weightPathPrefix=n.weightPathPrefix,this.onProgress=n.onProgress,this.weightUrlConverter=n.weightUrlConverter,n.fetchFunc!=null?(V(typeof n.fetchFunc=="function",()=>"Must pass a function that matches the signature of ` fetch ` (see https://developer.mozilla.org/en-US/docs/Web/API/Fetch_API)"),this.fetch=n.fetchFunc):this.fetch=he().platform.fetch,V(t!=null&&t.length>0,()=>"URL path for http must not be null, undefined or empty."),Array.isArray(t)&&V(t.length===2,()=> ` URL paths for http must have a length of 2 , ( actual length is $ { t . length } ) . ` ),this.path=t,n.requestInit!=null&&n.requestInit.body!=null)throw new Error("requestInit is expected to have no pre-existing body, but has one.");this.requestInit=n.requestInit||{}}async save(t){if(t.modelTopology instanceof ArrayBuffer)throw new Error("BrowserHTTPRequest.save() does not support saving model topology in binary formats yet.");let n=Object.assign({method:this.DEFAULT_METHOD},this.requestInit);n.body=new FormData;let r=[{paths:["./model.weights.bin"],weights:t.weightSpecs}],s=k3(t,r);n.body.append("model.json",new Blob([JSON.stringify(s)],{type:qMe}),"model.json"),t.weightData!=null&&n.body.append("model.weights.bin",new Blob([t.weightData],{type:HMe}),"model.weights.bin");let o=await this.fetch(this.path,n);if(o.ok)return{modelArtifactsInfo:kl(t),responses:[o]};throw new Error( ` BrowserHTTPRequest . save ( ) failed due to HTTP response status $ { o . status } . ` )}async load(){let t=await this.fetch(this.path,this.requestInit);if(!t.ok)throw new Error( ` Request to $ { this . path } failed with status code $ { t . status } . Please verify this URL points to the model JSON of the model to load . ` );let n;try{n=await t.json()}catch(o){let a= ` Failed to parse model JSON of response from $ { this . path } . ` ;throw this.path.endsWith(".pb")?a+=" Your path contains a .pb file extension. Support for .pb models have been removed in TensorFlow.js 1.0 in favor of .json models. You can re-convert your Python TensorFlow model using the TensorFlow.js 1.0 conversion scripts or you can convert your.pb models with the 'pb2json'NPM script in the tensorflow/tfjs-converter repository.":a+=" Please make sure the server is serving valid JSON for this request.",new Error(a)}let r=n.modelTopology,s=n.weightsManifest;if(r==null&&s==null)throw new Error( ` The JSON from HTTP path $ { this . path } contains neither model topology or manifest for weights . ` );return Ay(n,o=>this.loadWeights(o))}async loadWeights(t){let n=Array.isArray(this.path)?this.path[1]:this.path,[r,s]=KMe(n),o=this.weightPathPrefix||r,a=[];for(let c of t)a.push(...c.weights);let i=[],l=[];for(let c of t)for(let p of c.paths)this.weightUrlConverter!=null?l.push(this.weightUrlConverter(p)):i.push(o+p+s);this.weightUrlConverter&&i.push(...await Promise.all(l));let u=await iD(i,{requestInit:this.requestInit,fetchFunc:this.fetch,onProgress:this.onProgress});return[a,by(u)]}};C3.URL_SCHEME_REGEX=/^https?: \/ \/ /;function KMe(e){let t=e.lastIndexOf("/"),n=e.lastIndexOf("?"),r=e.substring(0,t),s=n>t?e.substring(n):"";return[r+"/",s]}function S3(e){return e.match(C3.URL_SCHEME_REGEX)!=null}var OH=(e,t)=>{if(typeof fetch=="undefined"&&(t==null||t.fetchFunc==null))return null;{let n=!0;if(Array.isArray(e)?n=e.every(r=>S3(r)):n=S3(e),n)return N3(e,t)}return null};Cn.registerSaveRouter(OH);Cn.registerLoadRouter(OH);function N3(e,t){return new C3(e,t)}function PH(e,t){return N3(e,t)}var T3
with dtype $ { o . dtype } . ` )}),n.length===1)return So(n[0]);let r=n,s={axis:t};return z.runKernel(Ap,r,s)}var on=L({concat_:Rze});function Dze(e){let n={x:O(e,"x","sigmoid")};return z.runKernel(zu,n)}var ra=L({sigmoid_:Dze});function $ ze(e,t,n){let r=O(e,"x","slice","string_or_numeric");if(r.rank===0)throw new Error("Slicing scalar is not possible");let s={x:r},o={begin:t,size:n};return z.runKernel(Xp,s,o)}var Tt=L({slice_: $ ze});function Fze(e){let n={x:O(e,"x","tanh")};return z.runKernel(Uu,n)}var pd=L({tanh_:Fze});function Oze(e,t,n){let r=O(e,"x","batchToSpaceND"),s=t.reduce((i,l)=>i*l);V(r.rank>=1+t.length,()=> ` input rank is $ { r . rank } but should be > than blockShape . length $ { t . length } ` ),V(n.length===t.length,()=> ` crops . length is $ { n . length } but should be equal to blockShape . length $ { t . length } ` ),V(r.shape[0]%s==0,()=> ` input tensor batch is $ { r . shape [ 0 ] } but is not divisible by the product of the elements of blockShape $ { t . join ( " * " ) } === $ { s } ` );let o={x:r},a={blockShape:t,crops:n};return z.runKernel(yp,o,a)}var Ju=L({batchToSpaceND_:Oze});function eq(e){let t;return e.rank===0||e.rank===1?t=X(e,[1,1,1,e.size]):e.rank===2?t=X(e,[1,1,e.shape[0],e.shape[1]]):e.rank===3?t=X(e,[1,e.shape[0],e.shape[1],e.shape[2]]):t=e,t}function Pze(e,t,n,r,s,o){o==null&&(o=.001);let a=O(e,"x","batchNorm"),i=O(t,"mean","batchNorm"),l=O(n,"variance","batchNorm"),u;s!=null&&(u=O(s,"scale","batchNorm"));let c;r!=null&&(c=O(r,"offset","batchNorm")),V(i.rank===l.rank,()=>"Batch normalization gradient requires mean and variance to have equal ranks."),V(c==null||i.rank===c.rank,()=>"Batch normalization gradient requires mean and offset to have equal ranks."),V(u==null||i.rank===u.rank,()=>"Batch normalization gradient requires mean and scale to have equal ranks.");let d={x:eq(a),scale:u,offset:c,mean:i,variance:l},f={varianceEpsilon:o},h=z.runKernel(Tp,d,f);return X(h,a.shape)}var qa=L({batchNorm_:Pze});function Mze(e,t,n,r,s,o){let a=O(e,"x","batchNorm"),i=O(t,"mean","batchNorm"),l=O(n,"variance","batchNorm"),u;s!=null&&(u=O(s,"scale","batchNorm"));let c;return r!=null&&(c=O(r,"offset","batchNorm")),V(a.rank===2,()=> ` Error in batchNorm2D : x must be rank 2 but got rank $ { a . rank } . ` ),V(i.rank===2||i.rank===1,()=> ` Error in batchNorm2D : mean must be rank 2 or rank 1 but got rank $ { i . rank } . ` ),V(l.rank===2||l.rank===1,()=> ` Error in batchNorm2D : variance must be rank 2 or rank 1 but got rank $ { l . rank } . ` ),u!=null&&V(u.rank===2||u.rank===1,()=> ` Error in batchNorm2D : scale must be rank 2 or rank 1 but got rank $ { u . rank } . ` ),c!=null&&V(c.rank===2||c.rank===1,()=> ` Error in batchNorm2D : offset must be rank 2 or rank 1 but got rank $ { c . rank } . ` ),qa(a,i,l,c,u,o)}var mD=L({batchNorm2d_:Mze});function zze(e,t,n,r,s,o){let a=O(e,"x","batchNorm"),i=O(t,"mean","batchNorm"),l=O(n,"variance","batchNorm"),u;s!=null&&(u=O(s,"scale","batchNorm"));let c;return r!=null&&(c=O(r,"offset","batchNorm")),V(a.rank===3,()=> ` Error in batchNorm3D : x must be rank 3 but got rank $ { a . rank } . ` ),V(i.rank===3||i.rank===1,()=> ` Error in batchNorm3D : mean must be rank 3 or rank 1 but got rank $ { i . rank } . ` ),V(l.rank===3||l.rank===1,()=> ` Error in batchNorm3D : variance must be rank 3 or rank 1 but got rank $ { l . rank } . ` ),u!=null&&V(u.rank===3||u.rank===1,()=> ` Error in batchNorm3D : scale must be rank 3 or rank 1 but got rank $ { u . rank } . ` ),c!=null&&V(c.rank===3||c.rank===1,()=> ` Error in batchNorm3D : offset must be rank 3 or rank 1 but got rank $ { c . rank } . ` ),qa(a,i,l,c,u,o)}var gD=L({batchNorm3d_:zze});function Lze(e,t,n,r,s,o){let a=O(e,"x","batchNorm"),i=O(t,"mean","batchNorm"),l=O(n,"variance","batchNorm"),u;s!=null&&(u=O(s,"scale","batchNorm"));let c;return r!=null&&(c=O(r,"offset","batchNorm")),V(a.rank===4,()=> ` Error in batchNorm4D : x must be rank 4 but got rank $ { a . rank } . ` ),V(i.rank===4||i.rank===1,()=> ` Error in batchNorm4D : mean must be rank 4 or rank 1 but got rank $ { i . rank } . ` ),V(l.rank===4||l.rank===1,()=> ` Error in batchNorm4D : variance must be rank 4 or rank 1 but got rank $ { l . rank } . ` ),u!=null&&V(u.rank===4||u.rank===1,()=> ` Error in batchNorm4D : scale must be rank 4 or rank 1 but got
$ { s } and $ { t } for depthToSpace with input shape
$ { r . shape } ` ),V(o*t>=0,()=> ` Negative dimension size caused by overflow when multiplying
$ { o } and $ { t } for depthToSpace with input shape
$ { r . shape } ` ),V(a%(t*t)==0,()=> ` Dimension size must be evenly divisible by $ { t * t } but is $ { a } for depthToSpace with input shape $ { r . shape } ` );let i={x:r},l={blockSize:t,dataFormat:n};return z.runKernel(Nx,i,l)}var KA=L({depthToSpace_:aLe});function iLe(e,t,n,r,s="NHWC",o=[1,1],a){let i=O(e,"x","depthwiseConv2d"),l=O(t,"filter","depthwiseConv2d"),u=i,c=!1;i.rank===3&&(c=!0,u=X(i,[1,i.shape[0],i.shape[1],i.shape[2]])),V(u.rank===4,()=> ` Error in depthwiseConv2d : input must be rank 4 , but got rank $ { u . rank } . ` ),V(l.rank===4,()=> ` Error in depthwiseConv2d : filter must be rank 4 , but got rank $ { l . rank } . ` ),V(u.shape[3]===l.shape[2],()=> ` Error in depthwiseConv2d : number of input channels ( $ { u . shape [ 3 ] } ) must match the inChannels dimension in filter $ { l . shape [ 2 ] } . ` ),a!=null&&V(Jt(r),()=> ` Error in depthwiseConv2d : pad must be an integer when using , dimRoundingMode $ { a } but got pad $ { r } . ` );let p={x:u,filter:l},d={strides:n,pad:r,dataFormat:s,dilations:o,dimRoundingMode:a},f=z.runKernel(Cp,p,d);return c?X(f,[f.shape[1],f.shape[2],f.shape[3]]):f}var Oi=L({depthwiseConv2d_:iLe});function lLe(e,t,n,r,s=[1,1],o="NHWC"){let a=O(e,"x","dilation2d"),i=O(t,"filter","dilation2d");V(a.rank===3||a.rank===4,()=> ` Error in dilation2d : input must be rank 3 or 4 , but got rank $ { a . rank } . ` ),V(i.rank===3,()=> ` Error in dilation2d : filter must be rank 3 , but got rank $ { i . rank } . ` ),V(o==="NHWC",()=> ` Error in dilation2d : Only NHWC is currently supported , but got dataFormat of $ { o } ` );let l=a,u=!1;a.rank===3&&(l=X(a,[1,a.shape[0],a.shape[1],a.shape[2]]),u=!0);let c={x:l,filter:i},p={strides:n,pad:r,dilations:s},d=z.runKernel(Sp,c,p);return u?X(d,[d.shape[1],d.shape[2],d.shape[3]]):d}var XA=L({dilation2d_:lLe});function uLe(e,t){let n=e.length,r=[];for(let s=0;s<n;s++){let o=n-1-s,a=e[o]||1;(t[t.length-1-s]||1)>1&&a===1&&r.unshift(o)}return r}function An(e,t){let n=[];for(let r=0;r<t.length;r++){let s=e[e.length-r-1],o=t.length-r-1,a=t[o];(s==null||s===1&&a>1)&&n.unshift(o)}return n}function kt(e,t){let n=[],r=Math.max(e.length,t.length);for(let s=0;s<r;s++){let o=e[e.length-s-1];o==null&&(o=1);let a=t[t.length-s-1];if(a==null&&(a=1),o===1)n.unshift(a);else if(a===1)n.unshift(o);else if(o!==a){let i= ` Operands could not be broadcast together with shapes $ { e } and $ { t } . ` ;throw Error(i)}else n.unshift(o)}return n}function cLe(e,t){let n=O(e,"a","equal","string_or_numeric"),r=O(t,"b","equal","string_or_numeric");[n,r]=_t(n,r),kt(n.shape,r.shape);let s={a:n,b:r};return z.runKernel(Ch,s)}var Ss=L({equal_:cLe});function pLe(e,t,n){let r=O(t,"a","where"),s=O(n,"b","where"),o=O(e,"condition","where","bool"),a=kt(kt(o.shape,r.shape),s.shape),i=Qu(o,a),l=Qu(r,a),u=Qu(s,a),c={condition:i,t:l,e:u};return z.runKernel(Kp,c)}var Sn=L({where_:pLe});function dLe(e){let n={x:O(e,"x","zerosLike")};return z.runKernel(rd,n)}var Ye=L({zerosLike_:dLe});function fLe(e,t){let n=O(e,"a","div"),r=O(t,"b","div");[n,r]=_t(n,r);let s=Pe(n,r),o=Ye(s),a=Ss(r,o);return Sn(a,o,s)}var ZA=L({divNoNan_:fLe});function hLe(e,t){let n=O(e,"t1","dot"),r=O(t,"t2","dot");V((n.rank===1||n.rank===2)&&(r.rank===1||r.rank===2),()=> ` Error in dot : inputs must all be rank 1 or 2 , but got ranks $ { n . rank } and $ { r . rank } . ` );let s=n.rank===1?n.size:n.shape[1],o=r.rank===1?r.size:r.shape[0];if(V(s===o,()=> ` Error in dot : inner dimensions of inputs must match , but got $ { s } and $ { o } . ` ),n.rank===1&&r.rank===1){let a=X(n,[1,-1]),i=X(r,[-1,1]),l=Bt(a,i);return X(l,[])}else if(n.rank===1&&r.rank===2){let a=X(n,[1,-1]),i=X(r,[r.shape[0],r.shape[1]]),l=Bt(a,i);return X(l,[l.size])}else if(n.rank===2&&r.rank===1){let a=X(r,[-1,1]),i=Bt(n,a);return X(i,[i.size])}else{let a=X(r,[r.shape[0],r.shape[1]]);return Bt(n,a)}}var tq=L({dot_:hLe});function mLe(e,...t){let n=t.map((s,o)=>O(s, ` tensors$ { o } ` ,"einsum")),r={equation:e};return z.runKernel(Ex,n,r)}var CD=L({einsum_:mLe});function gLe(e){let n={x:O(e,"x","elu")};return z.runKernel(du,n)}var Pi=L({elu_:gLe});function xLe(e){let t=O(e,"x","erf");V(t.dtype==="int32"||t.dtype==="float32",()=>"Input dtype must be ` int32 ` or ` float32 ` ."),t.dtype==="int32"&&(t=we(t,"float32"));let
rank $ { o . rank } . ` ),V(Jt(t),()=> ` Error in localResponseNormalization : depthRadius must be an integer but got depthRadius $ { t } . ` );let a=o,i=!1;o.rank===3&&(i=!0,a=X(o,[1,o.shape[0],o.shape[1],o.shape[2]]));let l={x:a},u={depthRadius:t,bias:n,alpha:r,beta:s},c=z.runKernel(Rp,l,u);return i?X(c,[c.shape[1],c.shape[2],c.shape[3]]):c}var e2=L({localResponseNormalization_:FLe});function OLe(e){let n={x:O(e,"x","log")};return z.runKernel(wu,n)}var Ns=L({log_:OLe});function PLe(e){let n={x:O(e,"x","log1p")};return z.runKernel(ku,n)}var nm=L({log1p_:PLe});function _D(e,t){V(fx(e),()=>"The f passed in variableGrads(f) must be a function"),V(t==null||Array.isArray(t)&&t.every(u=>u instanceof od),()=>"The varList passed in variableGrads(f, varList) must be an array of variables");let n=t!=null;if(!n){t=[];for(let u in z.registeredVariables)t.push(z.registeredVariables[u])}let r=n?t.filter(u=>!u.trainable):null,s=t.length;t=t.filter(u=>u.trainable),V(t.length>0,()=> ` variableGrads ( ) expects at least one of the input variables to be trainable , but none of the $ { s } variables is trainable . ` );let o=!0,{value:a,grads:i}=z.gradients(e,t,null,o);V(i.some(u=>u!=null),()=>"Cannot find a connection between any variable and the result of the loss function y=f(x). Please make sure the operations that use variables are inside the function f passed to minimize()."),V(a.rank===0,()=> ` The f passed in variableGrads ( f ) must return a scalar , but it returned a rank - $ { a . rank } tensor ` );let l={};return t.forEach((u,c)=>{i[c]!=null&&(l[u.name]=i[c])}),r!=null&&r.forEach(u=>l[u.name]=null),{value:a,grads:l}}function wa(e){return z.customGrad(e)}function MLe(e){let n={x:O(e,"x","neg")};return z.runKernel(zp,n)}var jt=L({neg_:MLe});function zLe(e){let n={x:O(e,"x","softplus")};return z.runKernel(Lu,n)}var Ka=L({softplus_:zLe});function LLe(e){let t=O(e,"x","logSigmoid");return wa(r=>({value:jt(Ka(jt(r))),gradFunc:a=>Y(a,ra(jt(r)))}))(t)}var ED=L({logSigmoid_:LLe});function BLe(e,t=null,n=!1){let s={x:O(e,"x","max")},o={reductionIndices:t,keepDims:n};return z.runKernel(Dp,s,o)}var ms=L({max_:BLe});function VLe(e,t){let n=O(e,"a","sub"),r=O(t,"b","sub");[n,r]=_t(n,r);let s={a:n,b:r};return z.runKernel(Wu,s)}var je=L({sub_:VLe});function WLe(e,t=null,n=!1){let r=O(e,"x","sum");r.dtype==="bool"&&(r=we(r,"int32"));let s={x:r},o={axis:t,keepDims:n};return z.runKernel(Zp,s,o)}var ze=L({sum_:WLe});function GLe(e,t=-1){let n=O(e,"logits","logSoftmax");if(t===-1&&(t=n.rank-1),t!==n.rank-1)throw Error( ` Log Softmax along a non - last dimension is not yet supported . Logits was rank $ { n . rank } and axis was $ { t } ` );return wa((s,o)=>{let a=!0,i=ms(s,t,!0),l=je(s,i),u=je(we(l,"float32"),Ns(ze(Zr(l),t,a)));return o([u]),{value:u,gradFunc:(p,d)=>{let[f]=d,h=!0,m=Zr(f);return je(p,Y(ze(p,t,h),m))}}})(n)}var rm=L({logSoftmax_:GLe});function RD(e,t){for(let n=0;n<e.length;++n)if(e[e.length-n-1]!==t-1-n)return!1;return!0}function nq(e,t,n){let r=e.length+t.length,s=[],o=0,a=0;for(let i=0;i<r;i++)n.indexOf(i)===-1?s.push(e[o++]):s.push(t[a++]);return s}function DD(e,t){let n=[],r=e.length;for(let o=0;o<r;o++)t.indexOf(o)===-1&&n.push(e[o]);let s=t.map(o=>e[o]);return[n,s]}function Cl(e,t){let n=t.map(r=>1);return nq(e,n,t)}function ULe(e,t,n){V(RD(t,n),()=> ` $ { e } supports only inner - most axes for now . Got axes $ { t } and rank - $ { n } input . ` )}function $ D(e,t){if(RD(e,t))return null;let n=[];for(let r=0;r<t;++r)e.indexOf(r)===-1&&n.push(r);return e.forEach(r=>n.push(r)),n}function t2(e){return e.map((t,n)=>[n,t]).sort((t,n)=>t[1]-n[1]).map(t=>t[0])}function jLe(e,t){let n=[];for(let r=t-e;r<t;++r)n.push(r);return n}function HLe(e,t=null,n=!1){let r=O(e,"x","logSumExp"),s=Sr(t,r.shape),o=ms(r,s,!0),a=je(r,o),i=Zr(a),l=ze(i,s),u=Ns(l),c=Ie(X(o,u.shape),u);if(n){let p=Cl(c.shape,s);return X(c,p)}return c}var rq=L({logSumExp_:HLe});function qLe(e,t){let n=O(e,"a","logicalAnd","bool"),r=O(t,"b","logicalAnd","bool");kt(n.shape,r.shape);let s={a:n,b:r};return z.runKernel(_h,s)}var gs=L({logicalAnd_:qLe});function KLe(e){let n={x:O(e,"x","logicalNot","bool")};return z.runKernel(Eh,n)}var rc
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Values should be Tensor1D but received shape $ { o . shape } ` );if(a.rank!==1)throw new Error( ` Dense shape should be Tensor1D but received shape $ { a . shape } ` );if(i.rank!==0)throw new Error( ` Default value should be a scalar but received shape $ { i . shape } ` );let l={indices:s,values:o,denseShape:a,defaultValue:i},u=z.runKernel(Qx,l);return{outputIndices:u[0],outputValues:u[1],emptyRowIndicator:u[2],reverseIndexMap:u[3]}}var qq=L({sparseFillEmptyRows_:GVe});function UVe(e,t,n){let r=O(e,"inputIndices","sparseReshape"),s=O(t,"inputShape","sparseReshape"),o=O(n,"newShape","sparseReshape");if(r.rank!==2)throw new Error( ` Input indices should be Tensor2D but received shape
$ { r . shape } ` );if(s.rank!==1)throw new Error( ` Input shape should be Tensor1D but received shape $ { s . shape } ` );if(o.rank!==1)throw new Error( ` New shape should be Tensor1D but received shape $ { o . shape } ` );let a={inputIndices:r,inputShape:s,newShape:o},i=z.runKernel(ey,a);return{outputIndices:i[0],outputShape:i[1]}}var Kq=L({sparseReshape_:UVe});function jVe(e,t,n){let r=O(e,"data","sparseSegmentMean"),s=O(t,"indices","sparseSegmentMean"),o=O(n,"segmentIds","sparseSegmentMean");if(r.rank<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.rank!==1)throw new Error( ` Indices should be Tensor1D but received shape
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Segment ids should be Tensor1D but received shape
$ { o . shape } ` );let a={data:r,indices:s,segmentIds:o};return z.runKernel(ty,a)}var Xq=L({sparseSegmentMean_:jVe});function HVe(e,t,n){let r=O(e,"data","sparseSegmentSum"),s=O(t,"indices","sparseSegmentSum"),o=O(n,"segmentIds","sparseSegmentSum");if(r.rank<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.rank!==1)throw new Error( ` Indices should be Tensor1D but received shape
$ { s . shape } ` );if(o.rank!==1)throw new Error( ` Segment ids should be Tensor1D but received shape
$ { o . shape } ` );let a={data:r,indices:s,segmentIds:o};return z.runKernel(ny,a)}var Zq=L({sparseSegmentSum_:HVe});function qVe(e,t,n,r,s,o,a,i){let l=O(e,"data","stringNGrams","string");if(l.dtype!=="string")throw new Error("Data must be of datatype string");if(l.shape.length!==1)throw new Error( ` Data must be a vector , saw : $ { l . shape } ` );let u=O(t,"dataSplits","stringNGrams");if(u.dtype!=="int32")throw new Error("Data splits must be of datatype int32");let c={separator:n,nGramWidths:r,leftPad:s,rightPad:o,padWidth:a,preserveShortSequences:i},p={data:l,dataSplits:u},d=z.runKernel(ay,p,c);return{nGrams:d[0],nGramsSplits:d[1]}}var Yq=L({stringNGrams_:qVe});function KVe(e,t,n=!0){let r=O(e,"input","stringSplit","string"),s=O(t,"delimiter","stringSplit","string");if(r.rank!==1)throw new Error( ` Input should be Tensor1D but received shape $ { r . shape } ` );if(s.rank!==0)throw new Error( ` Delimiter should be a scalar but received shape $ { s . shape } ` );let o={skipEmpty:n},a={input:r,delimiter:s},i=z.runKernel(iy,a,o);return{indices:i[0],values:i[1],shape:i[2]}}var Jq=L({stringSplit_:KVe});function XVe(e,t){let n=O(e,"input","stringToHashBucketFast","string"),r={numBuckets:t};if(t<=0)throw new Error("Number of buckets must be at least 1");let s={input:n};return z.runKernel(ly,s,r)}var Qq=L({stringToHashBucketFast_:XVe});var so={flipLeftRight:Eq,grayscaleToRGB:Rq,resizeNearestNeighbor:H3,resizeBilinear:j3,rotateWithOffset:Dq,cropAndResize:_q,nonMaxSuppression: $ q,nonMaxSuppressionAsync:Pq,nonMaxSuppressionWithScore:Mq,nonMaxSuppressionWithScoreAsync:zq,nonMaxSuppressionPadded:Lq,nonMaxSuppressionPaddedAsync:Bq,threshold:Vq,transform:Wq},eK={bandPart:Gq,gramSchmidt:Uq,qr:Hq};var y2={sparseFillEmptyRows:qq,sparseReshape:Kq,sparseSegmentMean:Xq,sparseSegmentSum:Zq},q3={stringNGrams:Yq,stringSplit:Jq,stringToHashBucketFast:Qq};var Fs=class extends E3{minimize(t,n=!1,r){let{value:s,grads:o}=this.computeGradients(t,r);if(r!=null){let a=r.map(i=>({name:i.name,tensor:o[i.name]}));this.applyGradients(a)}else this.applyGradients(o);return ut(o),n?s:(s.dispose(),null)}get iterations(){return this.iterations_==null&&(this.iterations_=0),this.iterations_}incrementIterations(){this.iterations_=this.iterations+1}computeGradients(t,n){return _D(t,n)}dispose(){this.iterations_!=null&&ut(this.iterations_)}async saveIterations(){return this.iterations_==null&&(this.iterations_=0),{name:"iter",tensor:Ke(this.iterations_,"int32")}}async getWeights(){throw new Error("getWeights() is not implemented for this optimizer yet.")}async setWeights(t){throw new Error( ` setWeights ( ) is not implemented for this optimizer class $ { this . getClassName ( ) } ` )}async extractIterations(t){return this.iterations_=(await t[0].tensor.data())[0],t.slice(1)}};Object.defineProperty(Fs,Symbol.hasInstance,{value:e=>e.minimize!=null&&e.computeGradients!=null&&e.applyGradients!=null});var b2=class extends Fs{constructor(t,n,r=null){super();this.learningRate=t,this.rho=n,this.epsilon=r,this.accumulatedGrads=[],this.accumulatedUpdates=[],r==null&&(this.epsilon=z.backend.epsilon())}applyGradients(t){(Array.isArray(t)?t.map(r=>r.name):Object.keys(t)).forEach((r,s)=>{let o=z.registeredVariables[r],a=!1;this.accumulatedGrads[s]==null&&(this.accumulatedGrads[s]={originalName: ` $ { r } / accum _grad ` ,variable:le(()=>Ye(o).variable(a))}),this.accumulatedUpdates[s]==null&&(this.accumulatedUpdates[s]={originalName: ` $ { r } / accum _var ` ,variable:le(()=>Ye(o).variable(a))});let i=Array.isArray(t)?t[s].tensor:t[r];if(i==null)return;let l=this.accumulatedGrads[s].variable,u=this.accumulatedUpdates[s].variable;le(()=>{let c=Ie(Y(l,this.rho),Y(Nt(i),1-this.rho)),p=Y(Pe(hn(Ie(u,this.epsilon)),hn(Ie(l,this.epsilon))),i),d=Ie(Y(u,this.rho),Y(Nt(p),1-this.rho));l.assign(c),u.assign(d);let f=Ie(Y(p,-this.learningRate),o);o.assign(f)})}),this.incrementIterations()}dispose(){this.accumulatedUpdates!=null&&(ut(this.accumulatedGrads.map(t=>t.variable)),ut(this.accumulatedUpdates.map(t=>t.variable)))}async getWeights(){let t=[...this.accumulatedGrads,...this.accumulatedUpdates];return[await this.saveIterations()].concat(t.map(n
$ { o } ) . ` );if(n<r)throw new Error( ` batchDims ( $ { r } ) must be less than or equal to axis ( $ { n } ) . ` );for(let p=0;p<r;++p)if(e.shape[p]!==t.shape[p])throw new Error( ` x . shape [ $ { p } ] : $ { e . shape [ p ] } should be equal to indices . shape [ $ { p } ] : $ { t . shape [ p ] } . ` );let a=e.shape[n],i=[],l=1,u=1,c=1;for(let p=0;p<r;++p)i.push(e.shape[p]),l*=e.shape[p];for(let p=r;p<n;p++)i.push(e.shape[p]),u*=e.shape[p];for(let p=r;p<s;p++)i.push(t.shape[p]);for(let p=n+1;p<o;p++)i.push(e.shape[p]),c*=e.shape[p];return{batchSize:l,sliceSize:c,outerSize:u,dimSize:a,outputShape:i}}function RWe(e){try{return e.map(t=>xy(t))}catch(t){throw new Error( ` Failed to decode encoded string bytes into utf - 8 , error : $ { t } ` )}}function DWe(e){return e.map(t=>gy(t))}var Ts={};dn(Ts,{nonMaxSuppressionV3Impl:()=>W3,nonMaxSuppressionV4Impl:()=>G3,nonMaxSuppressionV5Impl:()=>U3,whereImpl:()=>z3});var X3={kernelName:pp,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Y(e,Bi(we(n,"float32"),-1))}}};var rK={kernelName:eu,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=Nt(we(n,"float32")),s=hn(je(Ke(1),r));return jt(Pe(e,s))}}}};var sK={kernelName:tu,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=hn(je(Nt(we(n,"float32")),1));return Pe(e,r)}}}};var oK={kernelName:Ri,inputsToSave:["a","b"],gradFunc:(e,t)=>{let[n,r]=t,s=kt(n.shape,r.shape);return{a:()=>{let i=e,l=An(n.shape,s);return l.length>0&&(i=ze(i,l)),X(i,n.shape)},b:()=>{let i=e,l=An(r.shape,s);return l.length>0&&(i=ze(i,l)),X(i,r.shape)}}}};var aK={kernelName:dp,saveAllInputs:!0,gradFunc:(e,t)=>{let n={};return t.forEach((r,s)=>{n[s]=()=>e.clone()}),n}};var iK={kernelName:fp,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Ye(n)}}};var lK={kernelName:hp,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Ye(n)}}};var uK={kernelName:nu,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Pe(e,hn(je(Ke(1),Nt(we(n,"float32")))))}}};var cK={kernelName:ru,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>{let r=hn(Ie(Ke(1),Nt(we(n,"float32"))));return Pe(e,r)}}}};var pK={kernelName:au,inputsToSave:["a","b"],gradFunc:(e,t)=>{let[n,r]=t,s=kt(n.shape,r.shape);return{a:()=>{let i=Ie(Nt(n),Nt(r)),l=Y(e,Pe(r,i)),u=An(n.shape,s);return u.length>0&&(l=ze(l,u)),X(l,n.shape)},b:()=>{let i=Ie(Nt(n),Nt(r)),l=jt(Y(e,Pe(n,i))),u=An(r.shape,s);return u.length>0&&(l=ze(l,u)),X(l,r.shape)}}}};var dK={kernelName:su,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Pe(e,Ie(Nt(we(n,"float32")),1))}}};var fK={kernelName:ou,inputsToSave:["x"],gradFunc:(e,t)=>{let[n]=t;return{x:()=>Pe(e,je(Ke(1),Nt(we(n,"float32"))))}}};function $ We(e,t,n,r,s,o){let a=O(e,"dy","avgPool3dGrad"),i=O(t,"input","avgPool3dGrad"),l=a,u=i,c=!1;i.rank===4&&(c=!0,l=X(a,[1,a.shape[0],a.shape[1],a.shape[2],a.shape[3]]),u=X(i,[1,i.shape[0],i.shape[1],i.shape[2],i.shape[3]])),V(l.rank===5,()=> ` Error in avgPool3dGrad : dy must be rank 5 but got rank $ { l . rank } . ` ),V(u.rank===5,()=> ` Error in avgPool3dGrad : input must be rank 5 but got rank $ { u . rank } . ` ),o!=null&&V(Jt(s),()=> ` Error in avgPool3dGrad : pad must be an integer when using , dimRoundingMode $ { o } but got pad $ { s } . ` );let p={dy:l,input:u},d={filterSize:n,strides:r,pad:s,dimRoundingMode:o},f=z.runKernel(bx,p,d);return c?X(f,[f.shape[1],f.shape[2],f.shape[3],f.shape[4]]):f}var hK=L({avgPool3dGrad_: $ We});var mK={kernelName:gp,inputsToSave:["x"],gradFunc:(e,t,n)=>{let[r]=t,{filterSize:s,strides:o,pad:a,dimRoundingMode:i}=n;return{x:()=>hK(e,r,s,o,a,i)}}};function FWe(e,t,n,r,s){let o=O(e,"dy","avgPoolGrad"),a=O(t,"input","avgPoolGrad");V(a.rank===o.rank,()=> ` Rank of input ( $ { a . rank } ) does not match rank of dy ( $ { o . rank } ) ` );let i=a,l=o,u=!1;a.rank===3&&(u=!0,i=X(a,[1,a.shape[0],a.shape[1],a.shape[2]]),l=X(o,[1,o.shape[0],o.shape[1],o.shape[2]])),V(l.rank===4,()=> ` Error in avgPoolGrad : dy must be rank 4 but got rank $ { l . rank } . ` ),V(i.rank===4,()=> ` Error in avgPoolGrad : input must be rank 4 but got rank $ { i . rank } . ` );let c={dy:l,input:i},p={filterSize:n,strides:r,pad:s},d=z.runKernel(yx,c,p);return u?X(d,[d.shape[1],d.shape[2],d.shape[3]]):d}var gK=L({avgPoolGra
1. The $ { r } is defined in Python , in which case it needs to be ported to TensorFlow . js or your JavaScript code .
2. The custom $ { r } is defined in JavaScript , but is not registered properly with tf . serialization . registerClass ( ) . ` );return a}else{let o=e;if(o.className==null||o.config==null)throw new ne( ` $ { r } : Improper config format : $ { JSON . stringify ( o ) } .
'className' and 'config' must set . ` );let a=o.className,i,l;if(a in n?[i,l]=n[a]:a in Qa?[i,l]=Qa.className:a in t&&([i,l]=t[a]),i==null)throw new ne( ` Unknown $ { r } : $ { a } . This may be due to one of the following reasons :
1. The $ { r } is defined in Python , in which case it needs to be ported to TensorFlow . js or your JavaScript code .
2. The custom $ { r } is defined in JavaScript , but is not registered properly with tf . serialization . registerClass ( ) . ` );if(l!=null){let u={};for(let f of Object.keys(Qa))u[f]=Qa[f];for(let f of Object.keys(n))u[f]=n[f];let c=o.config;c.customObjects=u;let p=Object.assign({},Qa);for(let f of Object.keys(n))Qa[f]=n[f];a $ (o.config);let d=l(i,o.config,n,s);return Qa=Object.assign({},p),d}else{let u=Object.assign({},Qa);for(let p of Object.keys(n))Qa[p]=n[p];let c=new i(o.config);return Qa=Object.assign({},u),c}}}function VWe(e,t){return e<t?-1:e>t?1:0}function N2(e,t){return-1*VWe(e,t)}function ei(e){if(e==null)return e;let t=[];for(let n of e)t.indexOf(n)===-1&&t.push(n);return t}function JX(e){if(e==null)throw new ne( ` Invalid value in obj : $ { JSON . stringify ( e ) } ` );for(let t in e)if(e.hasOwnProperty(t))return!1;return!0}function Tl(e,t,n){if(n!=null&&e.indexOf(n)<0)throw new ne( ` $ { n } is not a valid $ { t } . Valid values are $ { e } or null / undefined . ` )}function Y3(e,t,n=0,r=1/0){return ka(n>=0),ka(r>=n),Array.isArray(e)&&e.length>=n&&e.length<=r&&e.every(s=>typeof s===t)}function ar(e,t){Array.isArray(e)?(C.assert(e.length>0,()=> ` $ { t } is unexpectedly an empty array . ` ),e.forEach((n,r)=>ar(n, ` element $ { r + 1 } of $ { t } ` ))):C.assert(Number.isInteger(e)&&e>0,()=> ` Expected $ { t } to be a positive integer , but got $ { QX ( e ) } . ` )}function QX(e){return e===null?"null":Array.isArray(e)?"["+e.map(t=>QX(t)).join(",")+"]":typeof e=="string"? ` "${e}" ` : ` $ { e } ` }function eZ(e,t){let n=C.now(),r;return(...o)=>{let a=C.now();return a-n<t||(n=a,r=e(...o)),r}}function J3(e){return e==="relu"?"relu":e==="linear"?"linear":e==="elu"?"elu":null}function i $ (e,t){return le(()=>hn(ze(Y(e,e),t,!0)))}var Dy=class extends ke.Serializable{getConfig(){return{}}},Q3=class extends Dy{constructor(t){super();this.defaultMaxValue=2,this.defaultAxis=0,this.maxValue=t.maxValue!=null?t.maxValue:this.defaultMaxValue,this.axis=t.axis!=null?t.axis:this.defaultAxis}apply(t){return le(()=>{let n=i $ (t,this.axis),r=Pr(n,0,this.maxValue);return Y(t,Pe(r,Ie(br(),n)))})}getConfig(){return{maxValue:this.maxValue,axis:this.axis}}};Q3.className="MaxNorm";ke.registerClass(Q3);var eI=class extends Dy{constructor(t){super();this.defaultAxis=0,this.axis=t.axis!=null?t.axis:this.defaultAxis}apply(t){return le(()=>Pe(t,Ie(br(),i $ (t,this.axis))))}getConfig(){return{axis:this.axis}}};eI.className="UnitNorm";ke.registerClass(eI);var tI=class extends Dy{apply(t){return to(t)}};tI.className="NonNeg";ke.registerClass(tI);var nI=class extends Dy{constructor(t){super();this.defaultMinValue=0,this.defaultMaxValue=1,this.defaultRate=1,this.defaultAxis=0,this.minValue=t.minValue!=null?t.minValue:this.defaultMinValue,this.maxValue=t.maxValue!=null?t.maxValue:this.defaultMaxValue,this.rate=t.rate!=null?t.rate:this.defaultRate,this.axis=t.axis!=null?t.axis:this.defaultAxis}apply(t){return le(()=>{let n=i $ (t,this.axis),r=Ie(Y(this.rate,Pr(n,this.minValue,this.maxValue)),Y(1-this.rate,n));return Y(t,Pe(r,Ie(br(),n)))})}getConfig(){return{minValue:this.minValue,maxValue:this.maxValue,rate:this.rate,axis:this.axis}}};nI.className="MinMaxNorm";ke.registerClass(nI);var tZ={maxNorm:"MaxNorm",minMaxNorm:"MinMaxNorm",nonNeg:"NonNeg",unitNorm:"UnitNorm"};function Ln(e){return Ry(e)}function nZ(e,t={}){return Nl(e,ke.SerializationMap.getMap().classNameMap,t,"constraint")}function Bn(e){if(e==null)return null;if(typeof e=="string"){let n={className:e in tZ?tZ[e]:e,config:{}};return nZ(n)}else return e instanceof Dy?e:nZ(e)}var rZ=["channelsFirst","channelsLast"],sZ=["nearest","bilinear"],oZ=["valid","same","causal"],aZ=["max","avg"],iZ=["sum","mul","concat","ave"];var $ y=new Map;function Fn(e){Tl(rZ,"DataFormat",e)}function lZ(e){Tl(sZ,"InterpolationFormat",e)}function ao(e){Tl(oZ,"PaddingMode",e)}function l $ (e){Tl(aZ,"PoolMode",e)}var T2=[],uZ="/";function Vi(e,t){T2.push(e);try{let n=t();return T2.pop(),n}catch(n){throw T2.pop(),n}}function GWe(){return T2.length===0?"":T2.join(uZ)+uZ}function rI(e){if(!cZ(e))throw new Error("Not a valid tensor name: '"+e+"'");return GWe()+e}function sI(e){if(!cZ(e))throw new Error("Not a va
because the value dtype is $ { n . dtype } , but TensorArray dtype is $ { this . dtype } . ` );if(this.size()===0&&(this.elementShape==null||this.elementShape.length===0)&&(this.elementShape=n.shape),ia(this.elementShape,n.shape, ` TensorArray $ { this . name } : Could not write to TensorArray index $ { t } . ` ),r.read)throw new Error( ` TensorArray $ { this . name } : Could not write to TensorArray index $ { t } , because it has already been read . ` );if(r.written)throw new Error( ` TensorArray $ { this . name } : Could not write to TensorArray index $ { t } , because it has already been written . ` );r.tensor=n, $ n(n),r.written=!0,this.tensors[t]=r}writeMany(t,n){if(t.length!==n.length)throw new Error( ` TensorArray $ { this . name } : could not write multiple tensors , because the index size : $ { t . length } is not the same as tensors size : $ { n . length } . ` );t.forEach((r,s)=>this.write(r,n[s]))}gather(t,n){if(!!n&&n!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but gather requested dtype $ { n } ` );if(t)t=t.slice(0,this.size());else{t=[];for(let s=0;s<this.size();s++)t.push(s)}if(t.length===0)return Zs([],[0].concat(this.elementShape));let r=this.readMany(t);return ia(this.elementShape,r[0].shape,"TensorArray shape mismatch: "),or(r,0)}concat(t){if(!!t&&t!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but concat requested dtype $ { t } ` );if(this.size()===0)return Zs([],[0].concat(this.elementShape));let n=[];for(let s=0;s<this.size();s++)n.push(s);let r=this.readMany(n);return ia(this.elementShape,r[0].shape, ` TensorArray shape mismatch : tensor array shape ( $ { this . elementShape } ) vs first tensor shape ( $ { r [ 0 ] . shape } ) ` ),on(r,0)}scatter(t,n){if(n.dtype!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but tensor has dtype $ { n . dtype } ` );if(t.length!==n.shape[0])throw new Error( ` Expected len ( indices ) == tensor . shape [ 0 ] , but saw : $ { t . length } vs . $ { n . shape [ 0 ] } ` );let r=Math.max(...t);if(!this.dynamicSize&&r>=this.maxSize)throw new Error( ` Max index must be < array size ( $ { r } vs . $ { this . maxSize } ) ` );this.writeMany(t,zr(n,0))}split(t,n){if(n.dtype!==this.dtype)throw new Error( ` TensorArray dtype is $ { this . dtype } but tensor has dtype $ { n . dtype } ` );let r=0,s=t.map(l=>(r+=l,r));if(r!==n.shape[0])throw new Error( ` Expected sum of lengths to be equal to
tensor . shape [ 0 ] , but sum of lengths is
$ { r } , and tensor 's shape is: ${n.shape}`);if(!this.dynamicSize&&t.length!==this.maxSize)throw new Error(`TensorArray' s size is not equal to the size of lengths ( $ { this . maxSize } vs . $ { t . length } ) , and the TensorArray is not marked as dynamically resizeable ` );let o=r===0?0:n.size/r,a=[];le(()=>{n=X(n,[1,r,o]);for(let l=0;l<t.length;++l){let u=l===0?0:s[l-1],c=[0,u,0],p=[1,t[l],o];a[l]=X(Tt(n,c,p),this.elementShape)}return a});let i=[];for(let l=0;l<t.length;l++)i[l]=l;this.writeMany(i,a)}};var $ m=class{constructor(t,n,r,s=-1){this.tensors=t,this.elementShape=n,this.elementDtype=r,t!=null&&t.forEach(o=>{if(r!==o.dtype)throw new Error( ` Invalid data types ; op elements $ { r } , but list elements $ { o . dtype } ` );ia(n,o.shape,"TensorList shape mismatch: "), $ n(o)}),this.idTensor=Ke(0),this.maxNumElements=s, $ n(this.idTensor)}get id(){return this.idTensor.id}copy(){return new $ m([...this.tensors],this.elementShape,this.elementDtype)}clearAndClose(t){this.tensors.forEach(n=>{(t==null||!t.has(n.id))&&n.dispose()}),this.tensors.length=0,this.idTensor.dispose()}size(){return this.tensors.length}stack(t,n,r=-1){if(n!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { n } , but list elements $ { this . elementDtype } ` );if(r!==-1&&this.tensors.length!==r)throw new Error( ` Operation expected a list with $ { r } elements but got a list with $ { this . tensors . length } elements . ` );ia(t,this.elementShape,"TensorList shape mismatch: ");let s=Hy(this.elementShape,this.tensors,t);return le(()=>{let o=this.tensors.map(a=>X(a,s));return or(o,0)})}popBack(t,n){if(n!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { n } , but list elements $ { this . elementDtype } ` );if(this.size()===0)throw new Error("Trying to pop from an empty list.");let r=Hy(this.elementShape,this.tensors,t),s=this.tensors.pop();return ia(s.shape,t,"TensorList shape mismatch: "),X(s,r)}pushBack(t){if(t.dtype!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { t . dtype } , but list elements $ { this . elementDtype } ` );if(ia(t.shape,this.elementShape,"TensorList shape mismatch: "),this.maxNumElements===this.size())throw new Error("Trying to push element into a full list."); $ n(t),this.tensors.push(t)}resize(t){if(t<0)throw new Error( ` TensorListResize expects size to be non - negative . Got : $ { t } ` );if(this.maxNumElements!==-1&&t>this.maxNumElements)throw new Error( ` TensorListResize input size $ { t } is greater maxNumElement $ { this . maxNumElements } . ` );this.tensors.length=t}getItem(t,n,r){if(r!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { r } , but list elements $ { this . elementDtype } ` );if(t<0||t>this.tensors.length)throw new Error( ` Trying to access element $ { t } in a list with $ { this . tensors . length } elements . ` );if(this.tensors[t]==null)throw new Error( ` element at index $ { t } is null . ` );ia(this.tensors[t].shape,n,"TensorList shape mismatch: ");let s=Hy(this.elementShape,this.tensors,n);return X(this.tensors[t],s)}setItem(t,n){if(n.dtype!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { n . dtype } , but list elements $ { this . elementDtype } ` );if(t<0||this.maxNumElements!==-1&&t>=this.maxNumElements)throw new Error( ` Trying to set element $ { t } in a list with max $ { this . maxNumElements } elements . ` );ia(this.elementShape,n.shape,"TensorList shape mismatch: "), $ n(n),this.tensors[t]=n}gather(t,n,r){if(n!==this.elementDtype)throw new Error( ` Invalid data types ; op elements $ { n } , but list elements $ { this . elementDtype } ` );ia(this.elementShape,r,"TensorList shape mismatch: "),t=t.slice(0,this.size());let s=Hy(this.elementShape,this.tensors,r);return t.length===0?Zs([],[0].concat(s)):le(()=>{let o=t.map(a=>X(this.tensors[a],s));return or(o,0)})}concat(t,n){if(!!t&&t!==this.elementDtype)throw new Error( ` TensorList dtype is $ { this . elementDtype } but concat requested dtype $ { t } ` );ia(this.elementShape,n,"TensorList shape mismatch: ");let r=Hy(this.elementShape,this.tensors,n);return this.size()===0?Zs([],[0].concat(r)):le(()=>{let s=this.tensors.map(o=>X(o,r));return on(s,0)})}};function tY(e,t,n){let r=e.dtype;if(e.shape.length<1)throw new Error( `
tensor . shape [ 0 ] , but sum of lengths is
$ { r } , and tensor ' s shape is : $ { e . shape } ` );let o=e.shape.slice(1),a=HC(o,n),i=r===0?0:e.size/r,l=le(()=>{let c=[];e=X(e,[1,r,i]);for(let p=0;p<t.length;++p){let d=p===0?0:s[p-1],f=[0,d,0],h=[1,t[p],i];c[p]=X(Tt(e,f,h),a)}return e.dispose(),c}),u=new $ m([],n,e.dtype,t.length);for(let c=0;c<l.length;c++)u.setItem(c,l[c]);return u}var oY=async(e,t,n)=>{switch(e.op){case"If":case"StatelessIf":{let r=D("thenBranch",e,t,n),s=D("elseBranch",e,t,n),o=D("cond",e,t,n),a=D("args",e,t,n);return(await o.data())[0]?n.functionMap[r].executeFunctionAsync(a,n.tensorArrayMap,n.tensorListMap):n.functionMap[s].executeFunctionAsync(a,n.tensorArrayMap,n.tensorListMap)}case"While":case"StatelessWhile":{let r=D("body",e,t,n),s=D("cond",e,t,n),o=D("args",e,t,n),a=await n.functionMap[s].executeFunctionAsync(o,n.tensorArrayMap,n.tensorListMap),i=o.map(c=>c.id),l=await a[0].data();a.forEach(c=>{!c.kept&&i.indexOf(c.id)===-1&&c.dispose()});let u=o;for(;l[0];){let c=u;u=await n.functionMap[r].executeFunctionAsync(u,n.tensorArrayMap,n.tensorListMap);let p=u.map(f=>f.id);c.forEach(f=>{!f.kept&&i.indexOf(f.id)===-1&&p.indexOf(f.id)===-1&&f.dispose()});let d=await n.functionMap[s].executeFunctionAsync(u,n.tensorArrayMap,n.tensorListMap);l=await d[0].data(),d.forEach(f=>{!f.kept&&i.indexOf(f.id)===-1&&p.indexOf(f.id)===-1&&f.dispose()})}return u}case"LoopCond":{let r=D("pred",e,t,n);return[Xi(r)]}case"Switch":{let r=D("pred",e,t,n),s=D("data",e,t,n);return s.kept||(s=Xi(s)),(await r.data())[0]?[void 0,s]:[s,void 0]}case"Merge":{let r=e.inputNames.find(s=>Lr(s,t,n)!==void 0);if(r){let s=Lr(r,t,n);return[Xi(s)]}return}case"Enter":{let r=D("frameName",e,t,n),s=D("tensor",e,t,n);return n.enterFrame(r),[Xi(s)]}case"Exit":{let r=D("tensor",e,t,n);return n.exitFrame(),[Xi(r)]}case"NextIteration":{let r=D("tensor",e,t,n);return n.nextIteration(),[Xi(r)]}case"TensorArrayV3":{let r=D("size",e,t,n),s=D("dtype",e,t,n),o=D("elementShape",e,t,n),a=D("dynamicSize",e,t,n),i=D("clearAfterRead",e,t,n),l=D("identicalElementShapes",e,t,n),u=D("name",e,t,n),c=new b6(u,s,r,o,l,a,i);return n.addTensorArray(c),[c.idTensor,Ke(1)]}case"TensorArrayWriteV3":{let r=D("tensorArrayId",e,t,n),s=D("index",e,t,n),o=D("tensor",e,t,n),a=n.getTensorArray(r.id);return a.write(s,o),[a.idTensor]}case"TensorArrayReadV3":{let r=D("tensorArrayId",e,t,n),s=D("index",e,t,n);return[n.getTensorArray(r.id).read(s)]}case"TensorArrayGatherV3":{let r=D("tensorArrayId",e,t,n),s=D("indices",e,t,n),o=D("dtype",e,t,n);return[n.getTensorArray(r.id).gather(s,o)]}case"TensorArrayScatterV3":{let r=D("tensorArrayId",e,t,n),s=D("indices",e,t,n),o=D("tensor",e,t,n),a=n.getTensorArray(r.id);return a.scatter(s,o),[a.idTensor]}case"TensorArrayConcatV3":{let r=D("tensorArrayId",e,t,n),s=n.getTensorArray(r.id),o=D("dtype",e,t,n);return[s.concat(o)]}case"TensorArraySplitV3":{let r=D("tensorArrayId",e,t,n),s=D("tensor",e,t,n),o=D("lengths",e,t,n),a=n.getTensorArray(r.id);return a.split(o,s),[a.idTensor]}case"TensorArraySizeV3":{let r=D("tensorArrayId",e,t,n),s=n.getTensorArray(r.id);return[Ke(s.size(),"int32")]}case"TensorArrayCloseV3":{let r=D("tensorArrayId",e,t,n),s=n.getTensorArray(r.id);return s.clearAndClose(),[s.idTensor]}case"TensorListSetItem":{let r=D("tensorListId",e,t,n),s=D("index",e,t,n),o=D("tensor",e,t,n),a=n.getTensorList(r.id);return a.setItem(s,o),[a.idTensor]}case"TensorListGetItem":{let r=D("tensorListId",e,t,n),s=D("index",e,t,n),o=D("elementShape",e,t,n),a=D("elementDType",e,t,n);return[n.getTensorList(r.id).getItem(s,o,a)]}case"TensorListScatterV2":case"TensorListScatter":{let r=D("indices",e,t,n),s=D("tensor",e,t,n),o=D("elementShape",e,t,n),a=D("numElements",e,t,n),i=rY(s,r,o,a);return n.addTensorList(i),[i.idTensor]}case"TensorListReserve":case"EmptyTensorList":{let r=D("elementShape",e,t,n),s=D("elementDType",e,t,n),o;e.op==="TensorListReserve"?o="numElements":o="maxNumElements";let a=D(o,e,t,n),i=nY(r,s,a);return n.addTensorList(i),[i.idTensor]}case"TensorListGather":{let r=D("tensorListId",e,t,n),s=D("indices",e,t,n),o=D("elementShape",e,t,n),a=D("elementDType",e,t,n);return[n.getTenso
$ { t } ` );let s;return this.size===1/0||this.size==null?s=this.size:n?s=Math.ceil(this.size/t):s=Math.floor(this.size/t),Zi(async()=>(await r.iterator()).columnMajorBatch(t,n,LUe),s)}concatenate(t){let n=this,r;return this.size===1/0||t.size===1/0?r=1/0:this.size!=null&&t.size!=null?r=this.size+t.size:r=null,Zi(async()=>(await n.iterator()).concatenate(await t.iterator()),r)}filter(t){let n=this,r;return this.size===1/0?r=1/0:r=null,Zi(async()=>(await n.iterator()).filter(s=>le(()=>t(s))),r)}async forEachAsync(t){return(await this.iterator()).forEachAsync(t)}map(t){let n=this;return Zi(async()=>(await n.iterator()).map(r=>le(()=>t(r))),this.size)}mapAsync(t){let n=this;return Zi(async()=>(await n.iterator()).mapAsync(t),this.size)}prefetch(t){if(t==null)throw new RangeError(" ` Dataset . prefetch ( ) ` requires bufferSize to be specified.");let n=this;return Zi(async()=>(await n.iterator()).prefetch(t),this.size)}repeat(t){let n=this,r;return this.size!=null&&t>0?r=this.size*t:t===0?r=0:this.size!=null&&(t===void 0||t<0)?r=1/0:r=null,Zi(async()=>{let s=O6(async()=>({value:await n.iterator(),done:!1}));return KY(s.take(t))},r)}skip(t){let n=this,r;return this.size!=null&&t>=0&&this.size>=t?r=this.size-t:this.size!=null&&(this.size<t||t===void 0||t<0)?r=0:r=null,Zi(async()=>(await n.iterator()).skip(t),r)}shuffle(t,n,r=!0){if(t==null||t<0)throw this.size==null?new RangeError(" ` Dataset . shuffle ( ) ` requires bufferSize to be specified."):new RangeError( ` \ ` Dataset.shuffle() \` requires bufferSize to be specified. If your data fits in main memory (for regular JS objects), and/or GPU memory (for \` tf.Tensor \` s), consider setting bufferSize to the dataset size ( ${ this . size } elements) ` ) ; let s = this , o = aJ . alea ( n || C . now ( ) . toString ( ) ) ; return Zi ( async ( ) => { let a = o . int32 ( ) ; return r && ( a += o . int32 ( ) ) , ( await s . iterator ( ) ) . shuffle ( t , a . toString ( ) ) } , this . size ) } take ( t ) { let n = this , r ; return this . size != null && this . size > t ? r = t : this . size != null && this . size <= t ? r = this . size : r = null , Zi ( async ( ) => ( await n . iterator ( ) ) . take ( t ) , r ) } async toArray ( ) { if ( this . size === 1 / 0 ) throw new Error ( "Can not convert infinite data stream to array." ) ; return ( await this . iterator ( ) ) . toArray ( ) } async toArrayForTest ( ) { if ( this . size === 1 / 0 ) throw new Error ( "Can not convert infinite data stream to array." ) ; return ( await this . iterator ( ) ) . toArrayForTest ( ) } } ; Zy . MAX _BUFFER _SIZE = 1e4 ; function Zi ( e , t = null ) { return new class extends Zy { constructor ( ) { super ( ... arguments ) ; this . size = t } async iterator ( ) { return e ( ) } } } function LUe ( e ) { if ( e === null ) return null ; let t = e [ 0 ] ; return UY ( t ) ? { value : BUe ( e ) , recurse : ! 1 } : { value : null , recurse : ! 0 } } function BUe ( e ) { if ( e . length === 0 ) throw new Error ( "Can't make a batch of zero elements." ) ; return e [ 0 ] instanceof Ot ? or ( e ) : Zs ( e ) } var Qen = Symbol ( "out" ) , etn = Symbol ( "field" ) , ttn = Symbol ( "quote" ) , ntn = Symbol ( "quoteafterquote" ) , rtn = Symbol ( "quoteinquote" ) ; function Se ( e , t ) { Array . isArray ( e ) || ( e = [ e ] ) , e . forEach ( n => { n != null && C . assert ( n . dtype !== "complex64" , ( ) => ` ${ t } does not support complex64 tensors in the CPU backend. ` ) } ) } var ZUe = Ts . whereImpl , Yy = class extends up { constructor ( ) { super ( ) ; this . blockSize = 48 , this . firstUse = ! 0 , this . data = new ux ( this , Zu ( ) ) } nextDataId ( ) { return Yy . nextDataId ++ } write ( t , n , r ) { this . firstUse && ( this . firstUse = ! 1 , he ( ) . get ( "IS_NODE" ) && M . warn ( `
=== === === === === === === === === =
Hi there \ u { 1 F44B } . Looks like you are running TensorFlow . js in Node . js . To speed things up dramatically , install our node backend , which binds to TensorFlow C ++ , by running npm i @ tensorflow / tfjs - node , or npm i @ tensorflow / tfjs - node - gpu if you have CUDA . Then call require ( '@tensorflow/tfjs-node' ) ; ( - gpu suffix for CUDA ) at the start of your program . Visit https : //github.com/tensorflow/tfjs-node for more details.
=== === === === === === === === === = ` ));let s={id:this.nextDataId()};return this.data.set(s,{values:t,dtype:r,refCount:1}),s}makeTensorInfo(t,n,r){let s;if(n==="string"&&r!=null&&r.length>0&&C.isString(r[0])){let o=r.map(a=>C.encodeString(a));s=this.write(o,t,n)}else s=this.write(r,t,n);return{dataId:s,shape:t,dtype:n}}refCount(t){return this.data.has(t)?this.data.get(t).refCount:0}incRef(t){let n=this.data.get(t);n.refCount++}decRef(t){if(this.data.has(t)){let n=this.data.get(t);n.refCount--}}move(t,n,r,s,o){this.data.set(t,{values:n,dtype:s,refCount:o})}numDataIds(){return this.data.numDataIds()}async read(t){return this.readSync(t)}readSync(t){let{dtype:n,complexTensorInfos:r}=this.data.get(t);if(n==="complex64"){let s=this.readSync(r.real.dataId),o=this.readSync(r.imag.dataId);return M.mergeRealAndImagArrays(s,o)}return this.data.get(t).values}bufferSync(t){let n=this.readSync(t.dataId),r=n;if(t.dtype==="string")try{r=n.map(s=>C.decodeString(s))}catch(s){throw new Error("Failed to decode encoded string bytes into utf-8")}return et(t.shape,t.dtype,r)}makeOutput(t,n,r){let s=this.write(t,n,r);return Zu().makeTensorFromDataId(s,n,r,this)}disposeData(t,n=!1){if(this.data.has(t)){if(this.data.get(t).refCount--,!n&&this.data.get(t).refCount>0)return!1;let{complexTensorInfos:r}=this.data.get(t);r!=null&&(this.disposeData(r.real.dataId,!0),this.disposeData(r.imag.dataId,!0)),this.data.delete(t)}return!0}disposeIntermediateTensorInfo(t){this.disposeData(t.dataId)}async time(t){let n=C.now();return t(),{kernelMs:C.now()-n}}memory(){return{unreliable:!0,reasons:["The reported memory is an upper bound. Due to automatic garbage collection, the true allocated memory may be less."]}}where(t){Se([t],"where");let n=this.readSync(t.dataId);return ZUe(t.shape,n)}dispose(){}floatPrecision(){return 32}epsilon(){return super.epsilon()}};Yy.nextDataId=0;var cF={};dn(cF,{addImpl:()=>V6,bincountImpl:()=>e1,bincountReduceImpl:()=>YC,ceilImpl:()=>W6,concatImpl:()=>JC,equalImpl:()=>G6,expImpl:()=>j6,expm1Impl:()=>q6,floorImpl:()=>K6,gatherNdImpl:()=>QC,gatherV2Impl:()=>eS,greaterEqualImpl:()=>Z6,greaterImpl:()=>X6,lessEqualImpl:()=>J6,lessImpl:()=>Y6,linSpaceImpl:()=>tS,logImpl:()=>Q6,maxImpl:()=>nS,maximumImpl:()=>eF,minimumImpl:()=>tF,multiplyImpl:()=>ev,negImpl:()=>nF,notEqualImpl:()=>rF,prodImpl:()=>sF,rangeImpl:()=>rS,rsqrtImpl:()=>oF,sigmoidImpl:()=>RJ,simpleAbsImpl:()=>B6,sliceImpl:()=>iF,sparseFillEmptyRowsImpl:()=>sS,sparseReshapeImpl:()=>oS,sparseSegmentReductionImpl:()=>n1,sqrtImpl:()=>FJ,squaredDifferenceImpl:()=>lF,stridedSliceImpl:()=>aS,stringNGramsImpl:()=>iS,stringSplitImpl:()=>lS,stringToHashBucketFastImpl:()=>uS,subImpl:()=>uF,tileImpl:()=>cS,topKImpl:()=>pS,transposeImpl:()=>t1,uniqueImpl:()=>dS});function B6(e){let t=new Float32Array(e.length);for(let n=0;n<e.length;++n)t[n]=Math.abs(e[n]);return t}var YUe=e=>{let{x:t}=e.inputs,n=e.backend;Se(t,"abs");let r=new Float32Array(C.sizeFromShape(t.shape)),s=n.data.get(t.dataId).values;return r=B6(s),n.makeOutput(r,t.shape,"float32")},iJ={kernelName:pp,backendName:"cpu",kernelFunc:YUe};function Pt(e){return(t,n,r,s,o)=>{let a=M.assertAndGetBroadcastShape(t,n),i=a.length,l=C.computeStrides(a),u=C.sizeFromShape(a),c=C.getTypedArrayFromDType(o,u),p=t.length,d=n.length,f=C.computeStrides(t),h=C.computeStrides(n),m=M.getBroadcastDims(t,a),g=M.getBroadcastDims(n,a);if(m.length+g.length===0)for(let x=0;x<c.length;++x)c[x]=e(r[x%r.length],s[x%s.length]);else for(let x=0;x<c.length;++x){let y=C.indexToLoc(x,i,l),b=y.slice(-p);m.forEach(k=>b[k]=0);let A=C.locToIndex(b,p,f),v=y.slice(-d);g.forEach(k=>v[k]=0);let w=C.locToIndex(v,d,h);c[x]=e(r[A],s[w])}return[c,a]}}function Br(e){let{inputs:t,backend:n}=e,{real:r,imag:s}=t,o=n.data.get(r.dataId).values,a=n.data.get(s.dataId).values,i=n.makeTensorInfo(r.shape,"complex64"),l=n.data.get(i.dataId);return l.complexTensorInfos={real:n.makeTensorInfo(r.shape,"float32",o),imag:n.makeTensorInfo(s.shape,"float32",a)},i}var lJ={kernelName:vx,backendName:"cpu",kernelFunc:Br};function Jy(e,t,n="float32"){if(n==="complex64"){let s=Jy(e,t,"float32"),o=Jy(e,t,"floa
indices . shape [ 0 ] = $ { i } ` );let g=C.getArrayFromDType(n,0),x=C.getArrayFromDType(s,0);return[g,[0,p],x,u,c]}let d=!0,f=0,h=new Array(l).fill(0);for(let g=0;g<i;++g){let x=e[g*p];if(x<0)throw new Error( ` indices ( $ { g } , 0 ) is invalid : $ { x } < 0 ` );if(x>=l)throw new Error( ` indices ( $ { g } , 0 ) is invalid : $ { x } >= $ { l } ` );++h[x],d=d&&x>=f,f=x}let m=!0;for(let g=0;g<l;++g){let x=h[g]===0;u[g]=x,m=m&&!x,h[g]=Math.max(h[g],1),g>0&&(h[g]+=h[g-1])}if(m&&d){let g=e,x=r;for(let y=0;y<i;++y)c[y]=y;return[g,[i,p],x,u,c]}else{let g=h[l-1],x=C.getArrayFromDType(n,g*p),y=C.getArrayFromDType(s,g),b=new Array(l).fill(0);for(let A=0;A<i;++A){let v=e[A*p],w=b[v],k=(v===0?0:h[v-1])+w;b[v]++;for(let I=0;I<p;++I)x[k*p+I]=e[A*p+I];y[k]=r[A],c[A]=k}for(let A=0;A<l;++A)if(b[A]===0){let w=A===0?0:h[A-1];x[w*p+0]=A;for(let k=1;k<p;++k)x[w*p+k]=0;y[w]=a}return[x,[g,p],y,u,c]}}function oS(e,t,n,r,s){let o=C.sizeFromShape(r),a=t[0],i=s.length,l=[],u=1,c=-1;for(let g=0;g<i;++g){let x=s[g];if(x===-1){if(c!==-1)throw new Error( ` only one output dimension may be - 1 , not both $ { c } and $ { g } ` );c=g,l.push(1)}else{if(x<0)throw new Error( ` size $ { g } must be non - negative , not $ { x } ` );u*=x,l.push(x)}}if(c!==-1){if(u<=0)throw new Error("reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero");let g=Math.trunc(o/u);if(u*g!==o)throw new Error( ` Input to reshape is a SparseTensor with $ { o }
dense values , but the requested shape requires a multiple of $ { u } . inputShape = $ { r } outputShape = $ { l } ` );l[c]=g}let p=C.sizeFromShape(l);if(p!==o)throw new Error( ` Input to reshape is a tensor with $ { o } dense values , but the requested shape has $ { p } . inputShape = $ { r } outputShape = $ { l } ` );let d=r.length,f=[];if(d>0){f[d-1]=1;for(let g=d-2;g>=0;--g)f[g]=f[g+1]*r[g+1]}let h=[];if(i>0){h[i-1]=1;for(let g=i-2;g>=0;--g)h[g]=h[g+1]*l[g+1]}let m=C.getArrayFromDType(n,a*i);for(let g=0;g<a;++g){let x=0;for(let y=0;y<d;++y)x+=e[g*d+y]*f[y];for(let y=0;y<i;++y)m[g*i+y]=Math.trunc(x/h[y]),x%=h[y]}return[m,[a,i],l]}function n1(e,t,n,r,s,o=!1,a=0){let i=r.length;if(i!==s.length)throw new Error("segmentIds and indices should have same size.");let l=[t[0],e.length/t[0]],u=l[1],p=i>0?s[i-1]+1:0;if(p<0)throw new Error("segment ids must be >= 0");let d=t.slice();d[0]=p;let f=d.reduce((b,A)=>b*A,1),h=C.getArrayFromDType(n,f);if(i===0)return p>0&&h.fill(a),[h,d];if(p<=0)throw new Error("segment ids must be >= 0");let m=0,g=1,x=0,y=s[m];for(;;){let b=0;if(g<i){if(b=s[g],y===b){++g;continue}if(y>=b)throw new Error("segment ids are not increasing")}if(y<0||y>=p)throw new Error( ` Segment id $ { y } out of range [ 0 , $ { p } ) , possibly because segmentIds input is not sorted . ` );y>x&&h.fill(a,x*u,y*u);for(let A=m;A<g;++A){let v=r[A];if(v<0||v>=l[0])throw new Error( ` Bad : indices [ $ { A } ] == $ { r [ A ] } out of range [ 0 , $ { l [ 0 ] } ) ` );for(let w=0;w<u;w++)h[y*u+w]+=e[v*u+w]}if(o)for(let A=0;A<u;A++)h[y*u+A]/=g-m;if(m=g,++g,x=y+1,y=b,g>i)break}return x<p&&h.fill(a,x*u,p*u),[h,d]}var FJ=fo(e=>Math.sqrt(e)),hje=pt(Bu,e=>Math.sqrt(e)),OJ={kernelName:Bu,backendName:"cpu",kernelFunc:hje};var lF=Pt((e,t)=>{let n=e-t;return n*n}),mje=Ht(Vu,lF),PJ={kernelName:Vu,backendName:"cpu",kernelFunc:mje};function aS(e,t,n,r){let s=et(e,t.dtype);for(let o=0;o<s.size;o++){let a=s.indexToLoc(o),i=new Array(a.length);for(let l=0;l<i.length;l++)i[l]=a[l]*n[l]+r[l];s.set(t.get(...i),...a)}return s}var MJ=class{constructor(t,n,r,s,o,a){this.separator=C.encodeString(t),this.nGramWidths=n,this.leftPad=C.encodeString(r),this.rightPad=C.encodeString(s),this.padWidth=o,this.preserveShort=a}getPadWidth(t){return Math.min(this.padWidth<0?t-1:this.padWidth,t-1)}getNumNGrams(t,n){let r=this.getPadWidth(n);return Math.max(0,t+2*r-n+1)}createNGrams(t,n,r,s,o,a){for(let i=0;i<o;++i){let l=this.getPadWidth(a),u=Math.max(0,l-i),c=Math.max(0,l-(o-(i+1))),p=a-(u+c),d=n+(u>0?0:i-l),f=0;f+=u*this.leftPad.length;for(let y=0;y<p;++y)f+=t[d+y].length;f+=c*this.rightPad.length,f+=(u+c+p-1)*this.separator.length,r[s+i]=new Uint8Array(f);let m=r[s+i],g=0,x=y=>y.forEach(b=>m[g++]=b);for(let y=0;y<u;++y)x(this.leftPad),x(this.separator);for(let y=0;y<p-1;++y)x(t[d+y]),x(this.separator);if(p>0){x(t[d+p-1]);for(let y=0;y<c;++y)x(this.separator),x(this.rightPad)}else{for(let y=0;y<c-1;++y)x(this.rightPad),x(this.separator);x(this.rightPad)}}}compute(t,n){let r=t.length,s=n.length;if(s>0){let l=n[0];if(l!==0)throw new Error( ` First split value must be 0 , got $ { l } ` );for(let u=1;u<s;++u){let c=n[u]>=l;if(c=c&&n[u]<=r,!c)throw new Error( ` Invalid split value $ { n [ u ] } , must be in [ $ { l } , $ { r } ] ` );l=n[u]}if(l!==r)throw new Error( ` Last split value must be data size . Expected $ { r } , got $ { l } ` )}let o=s-1,a=C.getArrayFromDType("int32",s);if(r===0||s===0){let l=new Array(r);for(let u=0;u<=o;++u)a[u]=0;return[l,a]}a[0]=0;for(let l=1;l<=o;++l){let u=n[l]-n[l-1],c=0;this.nGramWidths.forEach(p=>{c+=this.getNumNGrams(u,p)}),this.preserveShort&&u>0&&c===0&&(c=1),a[l]=a[l-1]+c}let i=new Array(a[o]);for(let l=0;l<o;++l){let u=n[l],c=a[l];if(this.nGramWidths.forEach(p=>{let d=n[l+1]-n[l],f=this.getNumNGrams(d,p);this.createNGrams(t,u,i,c,f,p),c+=f}),this.preserveShort&&c===a[l]){let p=n[l+1]-n[l];if(p===0)continue;let d=p+2*this.padWidth,f=1;this.createNGrams(t,u,i,c,f,d)}}return[i,a]}};function iS(e,t,n,r,s,o,a,i){return new MJ(n,r,s,o,a,i).compute(e,t)}function gje(e,t,n,r){if(!e.length)return;if(t.length===0){for(let o=0;o<e.length;++o)r.push(e.subarray(o,o+1));return}if(t.length===1){let o=t[0],a=e.indexOf(o);for(;a!==-1;){let i=e.subarray(
$ { o . shape } ` );if(r.shape.length!==2)throw new Error( ` Indices must be a matrix , saw :
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Values must be a vector , saw :
$ { s . shape } ` );if(a.shape.length!==0)throw new Error( ` Default value must be a scalar , saw :
$ { a . shape } ` );let i=n.data.get(r.dataId).values,l=n.data.get(s.dataId).values,u=n.data.get(o.dataId).values,c=n.data.get(a.dataId).values[0],[p,d,f,h,m]=sS(i,r.shape,r.dtype,l,s.dtype,u,c);return[n.makeTensorInfo(d,r.dtype,p),n.makeTensorInfo([d[0]],s.dtype,f),n.makeTensorInfo([h.length],"bool",new Uint8Array(h.map(g=>Number(g)))),n.makeTensorInfo([m.length],r.dtype,new Int32Array(m))]}var Xee={kernelName:Qx,backendName:"cpu",kernelFunc:wqe};function kqe(e){let{inputs:t,backend:n}=e,{inputIndices:r,inputShape:s,newShape:o}=t;if(r.shape.length!==2)throw new Error( ` Input indices should be a matrix but received shape
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Input shape should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Target shape should be a vector but received shape $ { o . shape } ` );let a=Array.from(n.data.get(s.dataId).values),i=n.data.get(r.dataId).values,l=Array.from(n.data.get(o.dataId).values),[u,c,p]=oS(i,r.shape,r.dtype,a,l);return[n.makeTensorInfo(c,r.dtype,u),n.makeTensorInfo([p.length],o.dtype,new Int32Array(p))]}var Zee={kernelName:ey,backendName:"cpu",kernelFunc:kqe};function Iqe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.data.get(r.dataId).values,i=n.data.get(s.dataId).values,l=n.data.get(o.dataId).values,[u,c]=n1(a,r.shape,r.dtype,i,l,!0);return n.makeTensorInfo(c,r.dtype,u)}var Yee={kernelName:ty,backendName:"cpu",kernelFunc:Iqe};function Cqe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.data.get(r.dataId).values,i=n.data.get(s.dataId).values,l=n.data.get(o.dataId).values,[u,c]=n1(a,r.shape,r.dtype,i,l);return n.makeTensorInfo(c,r.dtype,u)}var Jee={kernelName:ny,backendName:"cpu",kernelFunc:Cqe};function Sqe(e){let{inputs:t,backend:n,attrs:r}=e,{sparseIndices:s,sparseValues:o,defaultValue:a}=t,{outputShape:i}=r,{sliceRank:l,numUpdates:u,sliceSize:c,strides:p,outputSize:d}=M.calculateShapes(o,s,i),f=!1,h=n.bufferSync(s),m=n.bufferSync(o),g=n.data.get(a.dataId).values[0],x=xS(h,m,i,d,c,u,l,p,g,f);return n.makeTensorInfo(i,x.dtype,x.values)}var Qee={kernelName:ry,backendName:"cpu",kernelFunc:Sqe};function Nqe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{numOrSizeSplits:o,axis:a}=r,i=C.parseAxisParam(a,s.shape)[0],l=M.prepareSplitSize(s,o,i),u=new Array(s.shape.length).fill(0),c=s.shape.slice();return l.map(p=>{let d=[...c];d[i]=p;let f=li({inputs:{x:s},backend:n,attrs:{begin:u,size:d}});return u[i]+=p,f})}var ete={kernelName:Jp,backendName:"cpu",kernelFunc:Nqe};var tte={kernelName:sy,backendName:"cpu",kernelFunc:({inputs:e,backend:t})=>{let{x:n}=e,r=t;Se(n,"square");let s=r.data.get(n.dataId).values,o=new Float32Array(s.length);for(let i=0;i<s.length;++i){let l=s[i];o[i]=l*l}return{dataId:r.write(o,n.shape,n.dtype),shape:n.shape,dtype:n.dtype}}};var Tqe=pt(ju,(e,t)=>{let n=t;return isNaN(e)?NaN:e>0?1:n.alpha}),nte={kernelName:ju,backendName:"cpu",kernelFunc:Tqe};function _qe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,end:a,strides:i,beginMask:l,endMask:u,ellipsisMask:c,newAxisMask:p,shrinkAxisMask:d}=r;Se(s,"stridedSlice");let{nonStrided:f, $ begin:h, $ strides:m,size:g,newShape:x,outShape:y}=Cs.sliceInfo(s.shape,o,a,i,l,u,c,p,d),b=Mt({inputs:{x:s},backend:n,attrs:{shape:x}}),A;if(f){let w=li({inputs:{x:b},backend:n,attrs:{begin:h,size:g}});A=Mt({inputs:{x:w},backend:n,attrs:{shape:y}}),n.disposeIntermediateTensorInfo(w)}else if(y.some(w=>w===0))A=n.makeTensorInfo(y,s.dtype,[]);else{let w=n.bufferSync(b),k=aS(y,w,m,h);A=n.makeTensorInfo(k.shape,k.dtype,k.values)}let v=Mt({inputs:{x:A},backend:n,attrs:{shape:y}});return n.disposeIntermediateTensorInfo(b),n.disposeIntermediateTensorInfo(A),v}var rte={kernelName:oy,backendName:"cpu",kernelFunc:_qe};function Eqe(e){let{inputs:t,backend:n,attrs:r}=e,{separator:s,nGramWidths:o,leftPad:a,rightPad:i,padWidth:l,preserveShortSequences:u}=r,{data:c,dataSplits:p}=t,d=n.data.get(c.dataId).values,f=n.data.get(p.dataId).values,[h,m]=iS(d,f,s,o,a,i,l,u);return[n.makeTensorInfo([h.length],"string",h),n.makeTensorInfo(p.shape,"int32",m)]}var ste={kernelName:ay,backendName:"cpu",kernelFunc:Eqe};function Rqe(e){let{inputs:t,backend:n,attrs:r}=e,{skipEmpty:s}=r,{input:o,delimiter:a}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(o.shape.length!==1)throw new Error( ` Input must be a vector , got shape : $ { o . shape } ` );if(a.shape.length!==0)throw new Error( ` Delimiter must be a scalar , got shape : $ { a . shape } ` );let i=n.data.get(o.dataId).values,l=n.data.get(a.dataId).values[0],[u,c,p]=lS(i,l,s),d=c.length;return[n.makeTensorInfo([d,2],"int32",u),n.makeTensorInfo([d],"string",c),n.makeTensorInfo([2],"int32",new Int32Array(p))]}var ote={kernelName:iy,backendName:"cpu",kernelFunc:Rqe};function Dqe(e){let{inputs:t,backend:n,attrs:r}=e,{numBuckets:s}=r,{input:o}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(s<=0)throw new Error("Number of buckets must be at least 1");let a=n.data.get(o.dataId).values,i=uS(a,s);return n.makeTensorInfo(o.shape,"int32",i)}var ate={kernelName:ly,backendName:"cpu",kernelFunc:Dqe};var $ qe=pt(Gu,e=>Math.tan(e)),ite={kernelName:Gu,backendName:"cpu",kernelFunc: $ qe};var Fqe=pt(Uu,e=>Math.tanh(e)),lte={kernelName:Uu,backendName:"cpu",kernelFunc:Fqe};function Oqe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reps:o}=r;Se(s,"tile");let a=cS(n.bufferSync(s),o);return n.makeTensorInfo(a.shape,a.dtype,a.values)}var ute={kernelName:vl,backendName:"cpu",kernelFunc:Oqe};function Pqe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{k:o,sorted:a}=r;Se(s,"topk");let i=n.data.get(s.d
` ),o=s.length.toString().length+2,a=s.map((p,d)=>C.rightPad((d+1).toString(),o)+p),i=0;for(let p=0;p<a.length;p++)i=Math.max(a[p].length,i);let l=a.slice(0,r-1),u=a.slice(r-1,r),c=a.slice(r);console.log(l.join( `
` )),console.log(t.split( `
` )[0]),console.log( ` % c $ { C . rightPad ( u [ 0 ] , i ) } ` ,"border:1px solid red; background-color:#e3d2d2; color:#a61717"),console.log(c.join( `
` ))}function wte(e){return bc(e,()=>e.createProgram(),"Unable to create WebGLProgram.")}function kte(e,t){if(ot(e,()=>e.linkProgram(t)),e.getProgramParameter(t,e.LINK_STATUS)===!1)throw console.log(e.getProgramInfoLog(t)),new Error("Failed to link vertex and fragment shaders.")}function yS(e,t){if(ot(e,()=>e.validateProgram(t)),e.getProgramParameter(t,e.VALIDATE_STATUS)===!1)throw console.log(e.getProgramInfoLog(t)),new Error("Shader program validation failed.")}function Ite(e,t){let n=bc(e,()=>e.createBuffer(),"Unable to create WebGLBuffer");return ot(e,()=>e.bindBuffer(e.ARRAY_BUFFER,n)),ot(e,()=>e.bufferData(e.ARRAY_BUFFER,t,e.STATIC_DRAW)),n}function Cte(e,t){let n=bc(e,()=>e.createBuffer(),"Unable to create WebGLBuffer");return ot(e,()=>e.bindBuffer(e.ELEMENT_ARRAY_BUFFER,n)),ot(e,()=>e.bufferData(e.ELEMENT_ARRAY_BUFFER,t,e.STATIC_DRAW)),n}function Ste(e){return bc(e,()=>e.createTexture(),"Unable to create WebGLTexture.")}function Nte(e,t){let n=he().getNumber("WEBGL_MAX_TEXTURE_SIZE");if(e<=0||t<=0){let r= ` [ $ { e } x$ { t } ] ` ;throw new Error("Requested texture size "+r+" is invalid.")}if(e>n||t>n){let r= ` [ $ { e } x$ { t } ] ` ,s= ` [ $ { n } x$ { n } ] ` ;throw new Error("Requested texture size "+r+" greater than WebGL maximum on this browser / GPU "+s+".")}}function Tte(e){return bc(e,()=>e.createFramebuffer(),"Unable to create WebGLFramebuffer.")}function IF(e,t,n,r,s,o,a){let i=e.getAttribLocation(t,n);return i===-1?!1:(ot(e,()=>e.bindBuffer(e.ARRAY_BUFFER,r)),ot(e,()=>e.vertexAttribPointer(i,s,e.FLOAT,!1,o,a)),ot(e,()=>e.enableVertexAttribArray(i)),!0)}function nKe(e,t,n){sKe(e,n),ot(e,()=>e.activeTexture(e.TEXTURE0+n)),ot(e,()=>e.bindTexture(e.TEXTURE_2D,t))}function _te(e,t,n){return bc(e,()=>e.getUniformLocation(t,n),'uniform "'+n+'" not present in program.')}function Ete(e,t,n){return e.getUniformLocation(t,n)}function Rte(e,t,n,r){ot(e,()=>nKe(e,t,r)),ot(e,()=>e.uniform1i(n,r))}function bS(e,t,n){ot(e,()=>e.bindFramebuffer(e.FRAMEBUFFER,n)),ot(e,()=>e.framebufferTexture2D(e.FRAMEBUFFER,e.COLOR_ATTACHMENT0,e.TEXTURE_2D,t,0))}function CF(e,t){ot(e,()=>e.bindFramebuffer(e.FRAMEBUFFER,t)),ot(e,()=>e.framebufferTexture2D(e.FRAMEBUFFER,e.COLOR_ATTACHMENT0,e.TEXTURE_2D,null,0))}function pv(e){let t=e.checkFramebufferStatus(e.FRAMEBUFFER);if(t!==e.FRAMEBUFFER_COMPLETE)throw new Error("Error binding framebuffer: "+rKe(e,t))}function rKe(e,t){switch(t){case e.FRAMEBUFFER_INCOMPLETE_ATTACHMENT:return"FRAMEBUFFER_INCOMPLETE_ATTACHMENT";case e.FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT:return"FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT";case e.FRAMEBUFFER_INCOMPLETE_DIMENSIONS:return"FRAMEBUFFER_INCOMPLETE_DIMENSIONS";case e.FRAMEBUFFER_UNSUPPORTED:return"FRAMEBUFFER_UNSUPPORTED";default:return ` unknown error $ { t } ` }}function bc(e,t,n){let r=ot(e,()=>t());if(r==null)throw new Error(n);return r}function sKe(e,t){let n=e.MAX_COMBINED_TEXTURE_IMAGE_UNITS-1,r=t+e.TEXTURE0;if(r<e.TEXTURE0||r>n){let s= ` [ gl . TEXTURE0 , gl . TEXTURE$ { n } ] ` ;throw new Error( ` textureUnit must be in $ { s } . ` )}}function Id(e,t=2){return C.sizeFromShape(e.slice(0,e.length-t))}function Cd(e){if(e.length===0)throw Error("Cannot get rows and columns of an empty shape array.");return[e.length>1?e[e.length-2]:1,e[e.length-1]]}function AS(e){let t=[1,1,1];return e.length===0||e.length===1&&e[0]===1||(t=[Id(e),...Cd(e)]),t}function Dte(e,t=!1){let n=he().getNumber("WEBGL_MAX_TEXTURE_SIZE");t&&(n=n*2,e=e.map((s,o)=>o>=e.length-2?C.nearestLargerEven(e[o]):e[o]),e.length===1&&(e=[2,e[0]])),e.length!==2&&(e=C.squeezeShape(e).newShape);let r=C.sizeFromShape(e);if(e.length<=1&&r<=n)return[1,r];if(e.length===2&&e[0]<=n&&e[1]<=n)return e;if(e.length===3&&e[0]*e[1]<=n&&e[2]<=n)return[e[0]*e[1],e[2]];if(e.length===3&&e[0]<=n&&e[1]*e[2]<=n)return[e[0],e[1]*e[2]];if(e.length===4&&e[0]*e[1]*e[2]<=n&&e[3]<=n)return[e[0]*e[1]*e[2],e[3]];if(e.length===4&&e[0]<=n&&e[1]*e[2]*e[3]<=n)return[e[0],e[1]*e[2]*e[3]];if(t){let s=Id(e),o=2,a=2;return e.length&&([o,a]=Cd(e)),r=s*(o/2)*(a/2),C.sizeToSquarishShape(r).map(i=>i*2)}return C.sizeToSquarishShape(r)}function vS(e){return e%2==0}function zm(e,t){if(e=e.slice(-2),t=
bool isnan _custom ( float val ) {
return ( val > 0.0 || val < 0.0 ) ? false : val != 0.0 ;
}
bvec4 isnan _custom ( vec4 val ) {
return bvec4 ( isnan _custom ( val . x ) ,
isnan _custom ( val . y ) , isnan _custom ( val . z ) , isnan _custom ( val . w ) ) ;
}
# define isnan ( value ) isnan _custom ( value )
` ,l="",u= `
# define round ( value ) newRound ( value )
int newRound ( float value ) {
return int ( floor ( value + 0.5 ) ) ;
}
ivec4 newRound ( vec4 value ) {
return ivec4 ( floor ( value + vec4 ( 0.5 ) ) ) ;
}
` ):(e="",t="attribute",n="varying",r="varying",s="texture2D",o="gl_FragColor",a="",i= `
# define isnan ( value ) isnan _custom ( value )
bool isnan _custom ( float val ) {
return ( val > 0. || val < 1. || val == 0. ) ? false : true ;
}
bvec4 isnan _custom ( vec4 val ) {
return bvec4 ( isnan ( val . x ) , isnan ( val . y ) , isnan ( val . z ) , isnan ( val . w ) ) ;
}
` ,l= `
uniform float INFINITY ;
bool isinf ( float val ) {
return abs ( val ) == INFINITY ;
}
bvec4 isinf ( vec4 val ) {
return equal ( abs ( val ) , vec4 ( INFINITY ) ) ;
}
` ,u= `
int round ( float value ) {
return int ( floor ( value + 0.5 ) ) ;
}
ivec4 round ( vec4 value ) {
return ivec4 ( floor ( value + vec4 ( 0.5 ) ) ) ;
}
` ),{version:e,attribute:t,varyingVs:n,varyingFs:r,texture2D:s,output:o,defineOutput:a,defineSpecialNaN:i,defineSpecialInf:l,defineRound:u}}function Yi(e,t,n="index"){let r=C.computeStrides(t);return r.map((s,o)=>{let a= ` int $ { e [ o ] } = $ { n } / $ { s } ` ,i=o===r.length-1? ` int $ { e [ o + 1 ] } = $ { n } - $ { e [ o ] } * $ { s } ` : ` index -= $ { e [ o ] } * $ { s } ` ;return ` $ { a } ; $ { i } ; ` }).join("")}function Lm(e,t,n="index"){let r=C.computeStrides(t);return r.map((s,o)=>{let a= ` int $ { e [ o ] } = $ { n } / outShapeStrides [ $ { o } ] ` ,i=o===r.length-1? ` int $ { e [ o + 1 ] } = $ { n } - $ { e [ o ] } * outShapeStrides [ $ { o } ] ` : ` index -= $ { e [ o ] } * outShapeStrides [ $ { o } ] ` ;return ` $ { a } ; $ { i } ; ` }).join("")}function aKe(e,t){let n=e.length,r=e.map(o=> ` $ { t } [ $ { o } ] ` ),s=new Array(n-1);s[n-2]=r[n-1];for(let o=n-3;o>=0;--o)s[o]= ` ( $ { s [ o + 1 ] } * $ { r [ o + 1 ] } ) ` ;return s}function Lte(e,t,n="index"){let r=e.map((o,a)=>a),s=aKe(r,t);return s.map((o,a)=>{let i= ` int $ { e [ a ] } = $ { n } / $ { s [ a ] } ` ,l=a===s.length-1? ` int $ { e [ a + 1 ] } = $ { n } - $ { e [ a ] } * $ { s [ a ] } ` : ` index -= $ { e [ a ] } * $ { s [ a ] } ` ;return ` $ { i } ; $ { l } ; ` }).join("")}function a1(e){let t=C.computeStrides(e).map(n=>n.toString());return `
int getFlatIndex ( ivec3 coords ) {
return coords . x * $ { t [ 0 ] } + coords . y * $ { t [ 1 ] } + coords . z ;
}
` }function i1(){return `
int getFlatIndex ( ivec3 coords ) {
return coords . x * outShapeStrides [ 0 ] + coords . y * outShapeStrides [ 1 ] + coords . z ;
}
` }var wS= `
const float FLOAT _MAX = 1.70141184 e38 ;
const float FLOAT _MIN = 1.17549435 e - 38 ;
lowp vec4 encode _float ( highp float v ) {
if ( isnan ( v ) ) {
return vec4 ( 255 , 255 , 255 , 255 ) ;
}
highp float av = abs ( v ) ;
if ( av < FLOAT _MIN ) {
return vec4 ( 0.0 , 0.0 , 0.0 , 0.0 ) ;
} else if ( v > FLOAT _MAX ) {
return vec4 ( 0.0 , 0.0 , 128.0 , 127.0 ) / 255.0 ;
} else if ( v < - FLOAT _MAX ) {
return vec4 ( 0.0 , 0.0 , 128.0 , 255.0 ) / 255.0 ;
}
highp vec4 c = vec4 ( 0 , 0 , 0 , 0 ) ;
highp float e = floor ( log2 ( av ) ) ;
highp float m = exp2 ( fract ( log2 ( av ) ) ) - 1.0 ;
c [ 2 ] = floor ( 128.0 * m ) ;
m -= c [ 2 ] / 128.0 ;
c [ 1 ] = floor ( 32768.0 * m ) ;
m -= c [ 1 ] / 32768.0 ;
c [ 0 ] = floor ( 8388608.0 * m ) ;
highp float ebias = e + 127.0 ;
c [ 3 ] = floor ( ebias / 2.0 ) ;
ebias -= c [ 3 ] * 2.0 ;
c [ 2 ] += floor ( ebias ) * 128.0 ;
c [ 3 ] += 128.0 * step ( 0.0 , - v ) ;
return c / 255.0 ;
}
` ;var{getBroadcastDims:Bte}=M;function Vte(e,t,n){let r=[];if(e.forEach(f=>{let h=C.sizeFromShape(f.shapeInfo.logicalShape);if(f.shapeInfo.isUniform?r.push( ` uniform float $ { f . name } $ { h > 1 ? ` [ ${ h } ] ` : "" } ; ` ):(r.push( ` uniform sampler2D $ { f . name } ; ` ),r.push( ` uniform int offset$ { f . name } ; ` )),n.enableShapeUniforms){let{uniformShape:m}=kS(n.packedInputs,f.shapeInfo.logicalShape,f.shapeInfo.texShape);switch(m.length){case 1:r.push( ` uniform int $ { f . name } Shape ; ` );break;case 2:r.push( ` uniform ivec2 $ { f . name } Shape ; ` );break;case 3:r.push( ` uniform ivec3 $ { f . name } Shape ; ` );break;case 4:r.push( ` uniform ivec4 $ { f . name } Shape ; ` );break;default:break}r.push( ` uniform ivec2 $ { f . name } TexShape ; ` )}}),n.enableShapeUniforms){switch(t.logicalShape.length){case 1:r.push("uniform int outShape;");break;case 2:r.push("uniform ivec2 outShape;"),r.push("uniform int outShapeStrides;");break;case 3:r.push("uniform ivec3 outShape;"),r.push("uniform ivec2 outShapeStrides;");break;case 4:r.push("uniform ivec4 outShape;"),r.push("uniform ivec3 outShapeStrides;");break;default:break}r.push("uniform ivec2 outTexShape;")}n.customUniforms&&n.customUniforms.forEach(f=>{r.push( ` uniform $ { f . type } $ { f . name } $ { f . arrayIndex ? ` [ ${ f . arrayIndex } ] ` : "" } ; ` )});let s=r.join( `
` ),o=e.map(f=>iKe(f,t,n.packedInputs,n.enableShapeUniforms)).join( `
` ),a=t.texShape,i=Vn(),l=cKe(i),u,c,p=fKe(i);return t.isPacked?(u=lKe(t.logicalShape,a,n.enableShapeUniforms),c=dKe(i)):(u=uKe(t.logicalShape,a,n.enableShapeUniforms),c=pKe(i)),n.packedInputs&&(p+=xKe),[p,l,c,s,u,o,n.userCode].join( `
` )}function l1(e,t=!1){let n=e.shapeInfo.logicalShape;switch(n.length){case 0:return _Ke(e,t);case 1:return RKe(e,t);case 2:return $ Ke(e,t);case 3:return OKe(e,t);case 4:return MKe(e,t);case 5:return zKe(e);case 6:return LKe(e);default:throw new Error( ` $ { n . length } - D input sampling is not yet supported ` )}}function Wte(e,t){switch(e.shapeInfo.logicalShape.length){case 0:return TKe(e);case 1:return EKe(e,t);case 2:return DKe(e,t);case 3:return FKe(e,t);default:return PKe(e,t)}}function iKe(e,t,n=!1,r){let s="";n?s+=Wte(e,r):s+=l1(e,r);let o=e.shapeInfo.logicalShape,a=t.logicalShape;return o.length<=a.length&&(n?s+=BKe(e,t):s+=VKe(e,t)),s}function lKe(e,t,n){switch(e.length){case 0:return Gte();case 1:return yKe(e,t,n);case 2:return SKe(e,t,n);case 3:return AKe(e,t,n);default:return wKe(e,t,n)}}function uKe(e,t,n){switch(e.length){case 0:return Gte();case 1:return bKe(e,t,n);case 2:return NKe(e,t,n);case 3:return vKe(e,t,n);case 4:return kKe(e,t,n);case 5:return IKe(e,t);case 6:return CKe(e,t);default:throw new Error( ` $ { e . length } - D output sampling is not yet supported ` )}}function cKe(e){return `
float sampleTexture ( sampler2D textureSampler , vec2 uv ) {
return $ { e . texture2D } ( textureSampler , uv ) . r ;
}
` }function pKe(e){return `
void setOutput ( float val ) {
$ { e . output } = vec4 ( val , 0 , 0 , 0 ) ;
}
` }function dKe(e){return `
void setOutput ( vec4 val ) {
$ { e . output } = val ;
}
` }function fKe(e){return ` $ { e . version }
precision highp float ;
precision highp int ;
precision highp sampler2D ;
$ { e . varyingFs } vec2 resultUV ;
$ { e . defineOutput }
const vec2 halfCR = vec2 ( 0.5 , 0.5 ) ;
struct ivec5
{
int x ;
int y ;
int z ;
int w ;
int u ;
} ;
struct ivec6
{
int x ;
int y ;
int z ;
int w ;
int u ;
int v ;
} ;
uniform float NAN ;
$ { e . defineSpecialNaN }
$ { e . defineSpecialInf }
$ { e . defineRound }
int imod ( int x , int y ) {
return x - y * ( x / y ) ;
}
int idiv ( int a , int b , float sign ) {
int res = a / b ;
int mod = imod ( a , b ) ;
if ( sign < 0. && mod != 0 ) {
res -= 1 ;
}
return res ;
}
//Based on the work of Dave Hoskins
//https://www.shadertoy.com/view/4djSRW
# define HASHSCALE1 443.8975
float random ( float seed ) {
vec2 p = resultUV * seed ;
vec3 p3 = fract ( vec3 ( p . xyx ) * HASHSCALE1 ) ;
p3 += dot ( p3 , p3 . yzx + 19.19 ) ;
return fract ( ( p3 . x + p3 . y ) * p3 . z ) ;
}
$ { hKe }
$ { mKe }
$ { gKe }
` }var hKe= `
vec2 uvFromFlat ( int texNumR , int texNumC , int index ) {
int texR = index / texNumC ;
int texC = index - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
vec2 packedUVfrom1D ( int texNumR , int texNumC , int index ) {
int texelIndex = index / 2 ;
int texR = texelIndex / texNumC ;
int texC = texelIndex - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
` ,mKe= `
vec2 packedUVfrom2D ( int texelsInLogicalRow , int texNumR ,
int texNumC , int row , int col ) {
int texelIndex = ( row / 2 ) * texelsInLogicalRow + ( col / 2 ) ;
int texR = texelIndex / texNumC ;
int texC = texelIndex - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
` ,gKe= `
vec2 packedUVfrom3D ( int texNumR , int texNumC ,
int texelsInBatch , int texelsInLogicalRow , int b ,
int row , int col ) {
int index = b * texelsInBatch + ( row / 2 ) * texelsInLogicalRow + ( col / 2 ) ;
int texR = index / texNumC ;
int texC = index - texR * texNumC ;
return ( vec2 ( texC , texR ) + halfCR ) / vec2 ( texNumC , texNumR ) ;
}
` ,xKe= `
float getChannel ( vec4 frag , vec2 innerDims ) {
vec2 modCoord = mod ( innerDims , 2. ) ;
return modCoord . x == 0. ?
( modCoord . y == 0. ? frag . r : frag . g ) :
( modCoord . y == 0. ? frag . b : frag . a ) ;
}
float getChannel ( vec4 frag , int dim ) {
float modCoord = mod ( float ( dim ) , 2. ) ;
return modCoord == 0. ? frag . r : frag . g ;
}
` ;function Gte(){return `
int getOutputCoords ( ) {
return 0 ;
}
` }function yKe(e,t,n){let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)];return r[0]===1?n? `
int getOutputCoords ( ) {
return 2 * int ( resultUV . x * ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
}
` : `
int getOutputCoords ( ) {
return 2 * int ( resultUV . x * $ { r [ 1 ] } . 0 ) ;
}
` :r[1]===1?n? `
int getOutputCoords ( ) {
return 2 * int ( resultUV . y * ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) ) ;
}
` : `
int getOutputCoords ( ) {
return 2 * int ( resultUV . y * $ { r [ 0 ] } . 0 ) ;
}
` :n? `
int getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
return 2 * ( resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ) ;
}
` : `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
return 2 * ( resTexRC . x * $ { r [ 1 ] } + resTexRC . y ) ;
}
` }function bKe(e,t,n){return t[0]===1?n? `
int getOutputCoords ( ) {
return int ( resultUV . x * float ( outTexShape [ 1 ] ) ) ;
}
` : `
int getOutputCoords ( ) {
return int ( resultUV . x * $ { t [ 1 ] } . 0 ) ;
}
` :t[1]===1?n? `
int getOutputCoords ( ) {
return int ( resultUV . y * float ( outTexShape [ 0 ] ) ) ;
}
` : `
int getOutputCoords ( ) {
return int ( resultUV . y * $ { t [ 0 ] } . 0 ) ;
}
` :n? `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
return resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
}
` : `
int getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
return resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
}
` }function AKe(e,t,n){if(n)return `
ivec3 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatch = texelsInLogicalRow * int ( ceil ( float ( outShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int b = index / texelsInBatch ;
index -= b * texelsInBatch ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec3 ( b , r , c ) ;
}
` ;let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)],s=Math.ceil(e[2]/2),o=s*Math.ceil(e[1]/2);return `
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
int b = index / $ { o } ;
index -= b * $ { o } ;
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
return ivec3 ( b , r , c ) ;
}
` }function vKe(e,t,n){if(n)return `
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
$ { Lm ( [ "r" , "c" , "d" ] , e ) }
return ivec3 ( r , c , d ) ;
}
` ;let r=Yi(["r","c","d"],e);return `
ivec3 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { r }
return ivec3 ( r , c , d ) ;
}
` }function wKe(e,t,n){if(n)return `
ivec4 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 3 ] ) / 2.0 ) ) ;
int texelsInBatch = texelsInLogicalRow * int ( ceil ( float ( outShape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatchN = texelsInBatch * outShape [ 1 ] ;
int b2 = index / texelsInBatchN ;
index -= b2 * texelsInBatchN ;
int b = index / texelsInBatch ;
index -= b * texelsInBatch ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec4 ( b2 , b , r , c ) ;
}
` ;let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)],s=Math.ceil(e[e.length-1]/2),o=s*Math.ceil(e[e.length-2]/2),a=o,i="",l="b, r, c";for(let u=2;u<e.length-1;u++)a*=e[e.length-u-1],i= `
int b$ { u } = index / $ { a } ;
index -= b$ { u } * $ { a } ;
` +i,l= ` b$ { u } , ` +l;return `
ivec$ { e . length } getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
$ { i }
int b = index / $ { o } ;
index -= b * $ { o } ;
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
return ivec$ { e . length } ( $ { l } ) ;
}
` }function kKe(e,t,n){if(n)return `
ivec4 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
$ { Lm ( [ "r" , "c" , "d" , "d2" ] , e ) }
return ivec4 ( r , c , d , d2 ) ;
}
` ;let r=Yi(["r","c","d","d2"],e);return `
ivec4 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { r }
return ivec4 ( r , c , d , d2 ) ;
}
` }function IKe(e,t){let n=Yi(["r","c","d","d2","d3"],e);return `
ivec5 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( $ { t [ 0 ] } ,
$ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { n }
ivec5 outShape = ivec5 ( r , c , d , d2 , d3 ) ;
return outShape ;
}
` }function CKe(e,t){let n=Yi(["r","c","d","d2","d3","d4"],e);return `
ivec6 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
$ { n }
ivec6 result = ivec6 ( r , c , d , d2 , d3 , d4 ) ;
return result ;
}
` }function SKe(e,t,n){let r=[Math.ceil(t[0]/2),Math.ceil(t[1]/2)];if(C.arraysEqual(e,t))return n? `
ivec2 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
return 2 * ivec2 ( resultUV . yx * vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
}
` : `
ivec2 getOutputCoords ( ) {
return 2 * ivec2 ( resultUV . yx * vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
}
` ;let s=Math.ceil(e[1]/2);return n? `
ivec2 getOutputCoords ( ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( outTexShape [ 0 ] ) / 2.0 ) , ceil ( float ( outTexShape [ 1 ] ) / 2.0 ) ) ;
int texelsInLogicalRow = int ( ceil ( float ( outShape [ 1 ] ) / 2.0 ) ) ;
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( packedTexShape [ 0 ] , packedTexShape [ 1 ] ) ) ;
int index = resTexRC . x * packedTexShape [ 1 ] + resTexRC . y ;
int r = 2 * ( index / texelsInLogicalRow ) ;
int c = imod ( index , texelsInLogicalRow ) * 2 ;
return ivec2 ( r , c ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { r [ 0 ] } , $ { r [ 1 ] } ) ) ;
int index = resTexRC . x * $ { r [ 1 ] } + resTexRC . y ;
int r = 2 * ( index / $ { s } ) ;
int c = imod ( index , $ { s } ) * 2 ;
return ivec2 ( r , c ) ;
}
` }function NKe(e,t,n){return C.arraysEqual(e,t)?n? `
ivec2 getOutputCoords ( ) {
return ivec2 ( resultUV . yx * vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
}
` : `
ivec2 getOutputCoords ( ) {
return ivec2 ( resultUV . yx * vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
}
` :e[1]===1?n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
return ivec2 ( index , 0 ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
return ivec2 ( index , 0 ) ;
}
` :e[0]===1?n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
return ivec2 ( 0 , index ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
return ivec2 ( 0 , index ) ;
}
` :n? `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( outTexShape [ 0 ] , outTexShape [ 1 ] ) ) ;
int index = resTexRC . x * outTexShape [ 1 ] + resTexRC . y ;
int r = index / outShape [ 1 ] ;
int c = index - r * outShape [ 1 ] ;
return ivec2 ( r , c ) ;
}
` : `
ivec2 getOutputCoords ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx *
vec2 ( $ { t [ 0 ] } , $ { t [ 1 ] } ) ) ;
int index = resTexRC . x * $ { t [ 1 ] } + resTexRC . y ;
int r = index / $ { e [ 1 ] } ;
int c = index - r * $ { e [ 1 ] } ;
return ivec2 ( r , c ) ;
}
` }function Bm(e){return ` offset$ { e } ` }function TKe(e){let t=e.name,n="get"+t.charAt(0).toUpperCase()+t.slice(1),r=Vn();return `
vec4 $ { n } ( ) {
return $ { r . texture2D } ( $ { t } , halfCR ) ;
}
` }function _Ke(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1);if(e.shapeInfo.isUniform)return ` float $ { r } ( ) { return $ { n } ; } ` ;let[s,o]=e.shapeInfo.texShape;if(s===1&&o===1)return `
float $ { r } ( ) {
return sampleTexture ( $ { n } , halfCR ) ;
}
` ;let a=Bm(n);if(t)return `
float $ { r } ( ) {
vec2 uv = uvFromFlat ( $ { n } TexShape [ 0 ] , $ { n } TexShape [ 1 ] , $ { a } ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` ;let[i,l]=e.shapeInfo.texShape;return `
float $ { r } ( ) {
vec2 uv = uvFromFlat ( $ { i } , $ { l } , $ { a } ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` }function EKe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=e.shapeInfo.texShape,o=Vn();if(t)return `
vec4 $ { r } ( int index ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { n } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { n } TexShape [ 1 ] ) / 2.0 ) ) ;
vec2 uv = packedUVfrom1D (
packedTexShape [ 0 ] , packedTexShape [ 1 ] , index ) ;
return $ { o . texture2D } ( $ { n } , uv ) ;
}
` ;let a=[Math.ceil(s[0]/2),Math.ceil(s[1]/2)];return `
vec4 $ { r } ( int index ) {
vec2 uv = packedUVfrom1D (
$ { a [ 0 ] } , $ { a [ 1 ] } , index ) ;
return $ { o . texture2D } ( $ { n } , uv ) ;
}
` }function RKe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1);if(e.shapeInfo.isUniform)return `
float $ { r } ( int index ) {
$ { u1 ( e ) }
}
` ;let s=e.shapeInfo.texShape,o=s[0],a=s[1];if(a===1&&o===1)return `
float $ { r } ( int index ) {
return sampleTexture ( $ { n } , halfCR ) ;
}
` ;let i=Bm(n);return a===1?t? `
float $ { r } ( int index ) {
vec2 uv = vec2 ( 0.5 , ( float ( index + $ { i } ) + 0.5 ) / float ( $ { n } TexShape [ 0 ] ) ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` : `
float $ { r } ( int index ) {
vec2 uv = vec2 ( 0.5 , ( float ( index + $ { i } ) + 0.5 ) / $ { o } . 0 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` :o===1?t? `
float $ { r } ( int index ) {
vec2 uv = vec2 ( ( float ( index + $ { i } ) + 0.5 ) / float ( $ { n } TexShape [ 1 ] ) , 0.5 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` : `
float $ { r } ( int index ) {
vec2 uv = vec2 ( ( float ( index + $ { i } ) + 0.5 ) / $ { a } . 0 , 0.5 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` :t? `
float $ { r } ( int index ) {
vec2 uv = uvFromFlat ( $ { n } TexShape [ 0 ] , $ { n } TexShape [ 1 ] , index + $ { i } ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` : `
float $ { r } ( int index ) {
vec2 uv = uvFromFlat ( $ { o } , $ { a } , index + $ { i } ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` }function DKe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape,a=o[0],i=o[1],l=Vn();if(o!=null&&C.arraysEqual(n,o))return t? `
vec4 $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return $ { l . texture2D } ( $ { r } , uv ) ;
}
` : `
vec4 $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { i } . 0 , $ { a } . 0 ) ;
return $ { l . texture2D } ( $ { r } , uv ) ;
}
` ;if(t)return `
vec4 $ { s } ( int row , int col ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { r } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { r } TexShape [ 1 ] ) / 2.0 ) ) ;
int valuesPerRow = int ( ceil ( float ( $ { r } Shape [ 1 ] ) / 2.0 ) ) ;
vec2 uv = packedUVfrom2D ( valuesPerRow , packedTexShape [ 0 ] , packedTexShape [ 1 ] , row , col ) ;
return $ { l . texture2D } ( $ { r } , uv ) ;
}
` ;let u=[Math.ceil(o[0]/2),Math.ceil(o[1]/2)],c=Math.ceil(n[1]/2);return `
vec4 $ { s } ( int row , int col ) {
vec2 uv = packedUVfrom2D ( $ { c } , $ { u [ 0 ] } , $ { u [ 1 ] } , row , col ) ;
return $ { l . texture2D } ( $ { r } , uv ) ;
}
` }function $ Ke(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape;if(o!=null&&C.arraysEqual(n,o)){if(t)return `
float $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;let d=o[0],f=o[1];return `
float $ { s } ( int row , int col ) {
vec2 uv = ( vec2 ( col , row ) + halfCR ) / vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` }let{newShape:a,keptDims:i}=C.squeezeShape(n),l=a;if(l.length<n.length){let d=c1(e,l),f=["row","col"];return `
$ { l1 ( d , t ) }
float $ { s } ( int row , int col ) {
return $ { s } ( $ { p1 ( f , i ) } ) ;
}
` }if(e.shapeInfo.isUniform)return `
float $ { s } ( int row , int col ) {
int index = round ( dot ( vec2 ( row , col ) , vec2 ( $ { n [ 1 ] } , 1 ) ) ) ;
$ { u1 ( e ) }
}
` ;let u=o[0],c=o[1],p=Bm(r);return c===1?t? `
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { r } Shape [ 1 ] , 1 , 1 ) ) ;
vec2 uv = vec2 ( 0.5 , ( index + 0.5 ) / float ( $ { r } TexShape [ 0 ] ) ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { n [ 1 ] } , 1 , 1 ) ) ;
vec2 uv = vec2 ( 0.5 , ( index + 0.5 ) / $ { u } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` :u===1?t? `
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { r } Shape [ 1 ] , 1 , 1 ) ) ;
vec2 uv = vec2 ( ( index + 0.5 ) / float ( $ { r } TexShape [ 1 ] ) , 0.5 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col ) {
float index = dot ( vec3 ( row , col , $ { p } ) , vec3 ( $ { n [ 1 ] } , 1 , 1 ) ) ;
vec2 uv = vec2 ( ( index + 0.5 ) / $ { c } . 0 , 0.5 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` :t? `
float $ { s } ( int row , int col ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { r } Shape [ 1 ] + col + $ { p } ;
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { n [ 1 ] } + col + $ { p } ;
vec2 uv = uvFromFlat ( $ { u } , $ { c } , index ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` }function FKe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=e.shapeInfo.texShape,a=[Math.ceil(o[0]/2),Math.ceil(o[1]/2)];if(n[0]===1){let d=n.slice(1),f=[1,2],h=c1(e,d),m=["b","row","col"];return `
$ { Wte ( h , t ) }
vec4 $ { s } ( int b , int row , int col ) {
return $ { s } ( $ { p1 ( m , f ) } ) ;
}
` }let i=Vn();if(t)return `
vec4 $ { s } ( int b , int row , int col ) {
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { r } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { r } TexShape [ 1 ] ) / 2.0 ) ) ;
int valuesPerRow = int ( ceil ( float ( $ { r } Shape [ 2 ] ) / 2.0 ) ) ;
int texelsInBatch = valuesPerRow * int ( ceil ( float ( $ { r } Shape [ 1 ] ) / 2.0 ) ) ;
vec2 uv = packedUVfrom3D (
packedTexShape [ 0 ] , packedTexShape [ 1 ] , texelsInBatch , valuesPerRow , b , row , col ) ;
return $ { i . texture2D } ( $ { r } , uv ) ;
}
` ;let l=a[0],u=a[1],c=Math.ceil(n[2]/2),p=c*Math.ceil(n[1]/2);return `
vec4 $ { s } ( int b , int row , int col ) {
vec2 uv = packedUVfrom3D (
$ { l } , $ { u } , $ { p } , $ { c } , b , row , col ) ;
return $ { i . texture2D } ( $ { r } , uv ) ;
}
` }function OKe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=n[1]*n[2],a=n[2],{newShape:i,keptDims:l}=C.squeezeShape(n),u=i;if(u.length<n.length){let m=c1(e,u),g=["row","col","depth"];return `
$ { l1 ( m , t ) }
float $ { s } ( int row , int col , int depth ) {
return $ { s } ( $ { p1 ( g , l ) } ) ;
}
` }if(e.shapeInfo.isUniform)return `
float $ { s } ( int row , int col , int depth ) {
int index = round ( dot ( vec3 ( row , col , depth ) ,
vec3 ( $ { o } , $ { a } , 1 ) ) ) ;
$ { u1 ( e ) }
}
` ;let c=e.shapeInfo.texShape,p=c[0],d=c[1],f=e.shapeInfo.flatOffset;if(d===o&&f==null)return t? `
float $ { s } ( int row , int col , int depth ) {
int stride1 = $ { r } Shape [ 2 ] ;
float texR = float ( row ) ;
float texC = dot ( vec2 ( col , depth ) , vec2 ( stride1 , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth ) {
float texR = float ( row ) ;
float texC = dot ( vec2 ( col , depth ) , vec2 ( $ { a } , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { d } . 0 , $ { p } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;if(d===a&&f==null)return t? `
float $ { s } ( int row , int col , int depth ) {
float texR = dot ( vec2 ( row , col ) , vec2 ( $ { r } Shape [ 1 ] , 1 ) ) ;
float texC = float ( depth ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth ) {
float texR = dot ( vec2 ( row , col ) , vec2 ( $ { n [ 1 ] } , 1 ) ) ;
float texC = float ( depth ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { d } . 0 , $ { p } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;let h=Bm(r);return t? `
float $ { s } ( int row , int col , int depth ) {
// Explicitly use integer operations as dot() only works on floats.
int stride0 = $ { r } Shape [ 1 ] * $ { r } Shape [ 2 ] ;
int stride1 = $ { r } Shape [ 2 ] ;
int index = row * $ { o } + col * $ { a } + depth + $ { h } ;
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { o } + col * $ { a } + depth + $ { h } ;
vec2 uv = uvFromFlat ( $ { p } , $ { d } , index ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` }function PKe(e,t){let n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=Vn();if(t)return `
vec4 $ { r } ( int b2 , int b , int row , int col ) {
int valuesPerRow = int ( ceil ( float ( $ { n } Shape [ 3 ] ) / 2.0 ) ) ;
int texelsInBatch = valuesPerRow * int ( ceil ( float ( $ { n } Shape [ 2 ] ) / 2.0 ) ) ;
int index = b * texelsInBatch + ( row / 2 ) * valuesPerRow + ( col / 2 ) ;
texelsInBatch *= $ { n } Shape [ 1 ] ;
index = b2 * texelsInBatch + index ;
ivec2 packedTexShape = ivec2 ( ceil ( float ( $ { n } TexShape [ 0 ] ) / 2.0 ) , ceil ( float ( $ { n } TexShape [ 1 ] ) / 2.0 ) ) ;
int texR = index / packedTexShape [ 1 ] ;
int texC = index - texR * packedTexShape [ 1 ] ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( packedTexShape [ 1 ] , packedTexShape [ 0 ] ) ; return $ { s . texture2D } ( $ { n } , uv ) ;
}
` ;let o=e.shapeInfo.logicalShape,a=o.length,i=e.shapeInfo.texShape,l=[Math.ceil(i[0]/2),Math.ceil(i[1]/2)],u=l[0],c=l[1],p=Math.ceil(o[a-1]/2),d=p*Math.ceil(o[a-2]/2),f="int b, int row, int col",h= ` b * $ { d } + ( row / 2 ) * $ { p } + ( col / 2 ) ` ;for(let m=2;m<a-1;m++)f= ` int b$ { m } , ` +f,d*=o[a-m-1],h= ` b$ { m } * $ { d } + ` +h;return `
vec4 $ { r } ( $ { f } ) {
int index = $ { h } ;
int texR = index / $ { c } ;
int texC = index - texR * $ { c } ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { c } , $ { u } ) ;
return $ { s . texture2D } ( $ { n } , uv ) ;
}
` }function MKe(e,t){let n=e.shapeInfo.logicalShape,r=e.name,s="get"+r.charAt(0).toUpperCase()+r.slice(1),o=n[3],a=n[2]*o,i=n[1]*a,{newShape:l,keptDims:u}=C.squeezeShape(n);if(l.length<n.length){let y=c1(e,l),b=["row","col","depth","depth2"];return `
$ { l1 ( y , t ) }
float $ { s } ( int row , int col , int depth , int depth2 ) {
return $ { s } ( $ { p1 ( b , u ) } ) ;
}
` }if(e.shapeInfo.isUniform)return `
float $ { s } ( int row , int col , int depth , int depth2 ) {
int index = round ( dot ( vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { i } , $ { a } , $ { o } , 1 ) ) ) ;
$ { u1 ( e ) }
}
` ;let c=e.shapeInfo.flatOffset,p=e.shapeInfo.texShape,d=p[0],f=p[1],h= ` int stride2 = $ { r } Shape [ 3 ] ; ` ,m= ` int stride1 = $ { r } Shape [ 2 ] * stride2 ; ` ,g= ` int stride0 = $ { r } Shape [ 1 ] * stride1 ; ` ;if(f===i&&c==null)return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
$ { h }
$ { m }
float texR = float ( row ) ;
float texC =
dot ( vec3 ( col , depth , depth2 ) ,
vec3 ( stride1 , stride2 , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth , int depth2 ) {
float texR = float ( row ) ;
float texC =
dot ( vec3 ( col , depth , depth2 ) ,
vec3 ( $ { a } , $ { o } , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;if(f===o&&c==null)return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
float texR = dot ( vec3 ( row , col , depth ) ,
vec3 ( $ { r } Shape [ 1 ] * $ { r } Shape [ 2 ] , $ { r } Shape [ 2 ] , 1 ) ) ;
float texC = float ( depth2 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { r } TexShape [ 1 ] , $ { r } TexShape [ 0 ] ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth , int depth2 ) {
float texR = dot ( vec3 ( row , col , depth ) ,
vec3 ( $ { n [ 1 ] * n [ 2 ] } , $ { n [ 2 ] } , 1 ) ) ;
float texC = float ( depth2 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` ;let x=Bm(r);return t? `
float $ { s } ( int row , int col , int depth , int depth2 ) {
// Explicitly use integer operations as dot() only works on floats.
$ { h }
$ { m }
$ { g }
int index = row * stride0 + col * stride1 +
depth * stride2 + depth2 ;
vec2 uv = uvFromFlat ( $ { r } TexShape [ 0 ] , $ { r } TexShape [ 1 ] , index + $ { x } ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` : `
float $ { s } ( int row , int col , int depth , int depth2 ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { i } + col * $ { a } +
depth * $ { o } + depth2 ;
vec2 uv = uvFromFlat ( $ { d } , $ { f } , index + $ { x } ) ;
return sampleTexture ( $ { r } , uv ) ;
}
` }function zKe(e){let t=e.shapeInfo.logicalShape,n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),s=t[4],o=t[3]*s,a=t[2]*o,i=t[1]*a,{newShape:l,keptDims:u}=C.squeezeShape(t);if(l.length<t.length){let m=c1(e,l),g=["row","col","depth","depth2","depth3"];return `
$ { l1 ( m ) }
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
return $ { r } ( $ { p1 ( g , u ) } ) ;
}
` }if(e.shapeInfo.isUniform)return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
float index = dot (
vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { i } , $ { a } , $ { o } , $ { s } ) ) +
depth3 ;
$ { u1 ( e ) }
}
` ;let c=e.shapeInfo.flatOffset,p=e.shapeInfo.texShape,d=p[0],f=p[1];if(f===i&&c==null)return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
int texR = row ;
float texC = dot ( vec4 ( col , depth , depth2 , depth3 ) ,
vec4 ( $ { a } , $ { o } , $ { s } , 1 ) ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` ;if(f===s&&c==null)return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
float texR = dot (
vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { t [ 1 ] * t [ 2 ] * t [ 3 ] } ,
$ { t [ 2 ] * t [ 3 ] } , $ { t [ 3 ] } , 1 ) ) ;
int texC = depth3 ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { f } . 0 , $ { d } . 0 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` ;let h=Bm(n);return `
float $ { r } ( int row , int col , int depth , int depth2 , int depth3 ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { i } + col * $ { a } + depth * $ { o } +
depth2 * $ { s } + depth3 + $ { h } ;
vec2 uv = uvFromFlat ( $ { d } , $ { f } , index ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` }function LKe(e){let t=e.shapeInfo.logicalShape,n=e.name,r="get"+n.charAt(0).toUpperCase()+n.slice(1),{newShape:s,keptDims:o}=C.squeezeShape(t);if(s.length<t.length){let g=c1(e,s),x=["row","col","depth","depth2","depth3","depth4"];return `
$ { l1 ( g ) }
float $ { r } ( int row , int col , int depth ,
int depth2 , int depth3 , int depth4 ) {
return $ { r } ( $ { p1 ( x , o ) } ) ;
}
` }let a=t[5],i=t[4]*a,l=t[3]*i,u=t[2]*l,c=t[1]*u;if(e.shapeInfo.isUniform)return `
float $ { r } ( int row , int col , int depth ,
int depth2 , int depth3 , int depth4 ) {
int index = round ( dot (
vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { c } , $ { u } , $ { l } , $ { i } ) ) +
dot (
vec2 ( depth3 , depth4 ) ,
vec2 ( $ { a } , 1 ) ) ) ;
$ { u1 ( e ) }
}
` ;let p=e.shapeInfo.flatOffset,d=e.shapeInfo.texShape,f=d[0],h=d[1];if(h===c&&p==null)return `
float $ { r } ( int row , int col , int depth ,
int depth2 , int depth3 , int depth4 ) {
int texR = row ;
float texC = dot ( vec4 ( col , depth , depth2 , depth3 ) ,
vec4 ( $ { u } , $ { l } , $ { i } , $ { a } ) ) +
float ( depth4 ) ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { h } . 0 , $ { f } . 0 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` ;if(h===a&&p==null)return `
float $ { r } ( int row , int col , int depth ,
int depth2 , int depth3 , int depth4 ) {
float texR = dot ( vec4 ( row , col , depth , depth2 ) ,
vec4 ( $ { t [ 1 ] * t [ 2 ] * t [ 3 ] * t [ 4 ] } ,
$ { t [ 2 ] * t [ 3 ] * t [ 4 ] } ,
$ { t [ 3 ] * t [ 4 ] } ,
$ { t [ 4 ] } ) ) + float ( depth3 ) ;
int texC = depth4 ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { h } . 0 , $ { f } . 0 ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` ;let m=Bm(n);return `
float $ { r } ( int row , int col , int depth ,
int depth2 , int depth3 , int depth4 ) {
// Explicitly use integer operations as dot() only works on floats.
int index = row * $ { c } + col * $ { u } + depth * $ { l } +
depth2 * $ { i } + depth3 * $ { a } + depth4 + $ { m } ;
vec2 uv = uvFromFlat ( $ { f } , $ { h } , index ) ;
return sampleTexture ( $ { n } , uv ) ;
}
` }function u1(e){let t=e.name,n=C.sizeFromShape(e.shapeInfo.logicalShape);return n<2? ` return $ { t } ; ` : `
for ( int i = 0 ; i < $ { n } ; i ++ ) {
if ( i == index ) {
return $ { t } [ i ] ;
}
}
` }function BKe(e,t){let n=e.name,r=n.charAt(0).toUpperCase()+n.slice(1),s="get"+r+"AtOutCoords",o=e.shapeInfo.logicalShape.length,a=t.logicalShape.length,i=Bte(e.shapeInfo.logicalShape,t.logicalShape),l=Ct(a),u=a-o,c,p=["x","y","z","w","u","v"];o===0?c="":a<2&&i.length>=1?c="coords = 0;":c=i.map(y=> ` coords . $ { p [ y + u ] } = 0 ; ` ).join( `
` );let d="";a<2&&o>0?d="coords":d=e.shapeInfo.logicalShape.map((y,b)=> ` coords . $ { p [ b + u ] } ` ).join(", ");let f="return outputValue;",m=C.sizeFromShape(e.shapeInfo.logicalShape)===1,x=C.sizeFromShape(t.logicalShape)===1;if(o===1&&!m&&!x)f= `
return vec4 ( outputValue . xy , outputValue . xy ) ;
` ;else if(m&&!x)a===1?f= `
return vec4 ( outputValue . x , outputValue . x , 0. , 0. ) ;
` :f= `
return vec4 ( outputValue . x ) ;
` ;else if(i.length){let y=o-2,b=o-1;i.indexOf(y)>-1&&i.indexOf(b)>-1?f="return vec4(outputValue.x);":i.indexOf(y)>-1?f="return vec4(outputValue.x, outputValue.y, outputValue.x, outputValue.y);":i.indexOf(b)>-1&&(f="return vec4(outputValue.xx, outputValue.zz);")}return `
vec4 $ { s } ( ) {
$ { l } coords = getOutputCoords ( ) ;
$ { c }
vec4 outputValue = get$ { r } ( $ { d } ) ;
$ { f }
}
` }function VKe(e,t){let n=e.name,r=n.charAt(0).toUpperCase()+n.slice(1),s="get"+r+"AtOutCoords",o=t.texShape,a=e.shapeInfo.texShape,i=e.shapeInfo.logicalShape.length,l=t.logicalShape.length;if(!e.shapeInfo.isUniform&&i===l&&e.shapeInfo.flatOffset==null&&C.arraysEqual(a,o))return `
float $ { s } ( ) {
return sampleTexture ( $ { n } , resultUV ) ;
}
` ;let u=Ct(l),c=Bte(e.shapeInfo.logicalShape,t.logicalShape),p=l-i,d,f=["x","y","z","w","u","v"];i===0?d="":l<2&&c.length>=1?d="coords = 0;":d=c.map(m=> ` coords . $ { f [ m + p ] } = 0 ; ` ).join( `
` );let h="";return l<2&&i>0?h="coords":h=e.shapeInfo.logicalShape.map((m,g)=> ` coords . $ { f [ g + p ] } ` ).join(", "), `
float $ { s } ( ) {
$ { u } coords = getOutputCoords ( ) ;
$ { d }
return get$ { r } ( $ { h } ) ;
}
` }function Ct(e){if(e<=1)return"int";if(e===2)return"ivec2";if(e===3)return"ivec3";if(e===4)return"ivec4";if(e===5)return"ivec5";if(e===6)return"ivec6";throw Error( ` GPU for rank $ { e } is not yet supported ` )}function kS(e,t,n){let{newShape:r,keptDims:s}=C.squeezeShape(t),o=t.length,a=e&&o===3&&t[0]===1,i=a?t.slice(1):r,l=!e&&o>1&&!C.arraysEqual(t,n)&&r.length<o||a;return{useSqueezeShape:l,uniformShape:l?i:t,keptDims:s}}function c1(e,t){let n=JSON.parse(JSON.stringify(e));return n.shapeInfo.logicalShape=t,n}function p1(e,t){return t.map(n=>e[n]).join(", ")}function Ute(e,t,n,r){let s=n.map((b,A)=>{let v={logicalShape:b.shape,texShape:b.isUniform?null:b.texData.texShape,isUniform:b.isUniform,isPacked:b.isUniform?!1:b.texData.isPacked,flatOffset:null};return b.texData!=null&&b.texData.slice!=null&&b.texData.slice.flatOffset>0&&(v.flatOffset=b.texData.slice.flatOffset),{name:t.variableNames[A],shapeInfo:v}}),o=s.map(b=>b.shapeInfo),a={logicalShape:r.shape,texShape:r.texData.texShape,isUniform:!1,isPacked:r.texData.isPacked,flatOffset:null},i=Vte(s,a,t),l=e.createProgram(i),u=null,c=e.getUniformLocation(l,"NAN",!1);he().getNumber("WEBGL_VERSION")===1&&(u=e.getUniformLocation(l,"INFINITY",!1));let p=!1,d={},f={},h={};for(let b=0;b<t.variableNames.length;b++){let A=t.variableNames[b];d[A]=e.getUniformLocation(l,A,p),d[ ` offset$ { A } ` ]=e.getUniformLocation(l, ` offset$ { A } ` ,p),t.enableShapeUniforms&&(f[ ` $ { A } Shape ` ]=e.getUniformLocation(l, ` $ { A } Shape ` ,p),h[ ` $ { A } TexShape ` ]=e.getUniformLocation(l, ` $ { A } TexShape ` ,p))}let m,g,x;t.enableShapeUniforms&&(m=e.getUniformLocation(l,"outShape",p),x=e.getUniformLocation(l,"outShapeStrides",p),g=e.getUniformLocation(l,"outTexShape",p));let y=[];return t.customUniforms&&t.customUniforms.forEach((b,A)=>{y[A]=e.getUniformLocation(l,b.name,p)}),{program:t,source:i,webGLProgram:l,uniformLocations:d,customUniformLocations:y,inShapeInfos:o,outShapeInfo:a,infLoc:u,nanLoc:c,inShapesLocations:f,inTexShapesLocations:h,outShapeLocation:m,outShapeStridesLocation:x,outTexShapeLocation:g}}function jte(e,t){if(e.length!==t.length)throw Error( ` Binary was compiled with $ { e . length } inputs , but was executed with $ { t . length } inputs ` );e.forEach((n,r)=>{let s=n.logicalShape,o=t[r],a=o.shape;if(!C.arraysEqual(s,a))throw Error( ` Binary was compiled with different shapes than the current args . Shapes $ { s } and $ { a } must match ` );if(n.isUniform&&o.isUniform)return;let i=n.texShape,l=o.isUniform?null:o.texData.texShape;if(!C.arraysEqual(i,l))throw Error( ` Binary was compiled with different texture shapes than the current args . Shape $ { i } and $ { l } must match ` )})}function Hte(e,t,n,r,s){t.program.enableShapeUniforms||(jte(t.inShapeInfos,n),jte([t.outShapeInfo],[r]));let o=r.texData.texture,a=r.texData.texShape;r.texData.isPacked?e.setOutputPackedMatrixTexture(o,a[0],a[1]):e.setOutputMatrixTexture(o,a[0],a[1]),e.setProgram(t.webGLProgram),he().getNumber("WEBGL_VERSION")===1&&t.infLoc!==null&&e.gl.uniform1f(t.infLoc,1/0),t.nanLoc!==null&&e.gl.uniform1f(t.nanLoc,NaN),n.forEach((l,u)=>{let c=t.program.variableNames[u],p=t.uniformLocations[c],d=t.uniformLocations[ ` offset$ { c } ` ],f=t.inShapesLocations[ ` $ { c } Shape ` ],h=t.inTexShapesLocations[ ` $ { c } TexShape ` ];if(f){let{uniformShape:m}=kS(t.program.packedInputs,l.shape,l.texData.texShape);switch(m.length){case 1:e.gl.uniform1iv(f,new Int32Array(m));break;case 2:e.gl.uniform2iv(f,new Int32Array(m));break;case 3:e.gl.uniform3iv(f,new Int32Array(m));break;case 4:e.gl.uniform4iv(f,new Int32Array(m));break;default:break}}if(h&&e.gl.uniform2i(h,l.texData.texShape[0],l.texData.texShape[1]),p!=null){if(l.isUniform){if(C.sizeFromShape(l.shape)<2)e.gl.uniform1f(p,l.uniformValues[0]);else{let m=l.uniformValues;m instanceof Float32Array||(m=new Float32Array(m)),e.gl.uniform1fv(p,m)}return}l.texData.slice!=null&&d!=null&&e.gl.uniform1i(d,l.texData.slice.flatOffset),e.setInputMatrixTexture(l.texData.texture,p,u)}});let i=t.outShapeLocation;if(i)switch(r.shape.length){case 1:e.gl.uniform1iv(i,new Int32Array(r.shape));break;case 2:e.gl.uniform2iv(i,new Int32Array(r.shape));break;case 3:e.gl.uniform3iv(i,new Int3
ivec3 outCoordsFromFlatIndex ( int index ) {
$ { this . enableShapeUniforms ? Lm ( [ "r" , "c" , "d" ] , t ) : Yi ( [ "r" , "c" , "d" ] , t ) }
return ivec3 ( r , c , d ) ;
}
void main ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( texShape [ 0 ] , texShape [ 1 ] ) ) ;
int index = 4 * ( resTexRC . x * texShape [ 1 ] + resTexRC . y ) ;
vec4 result = vec4 ( 0. ) ;
for ( int i = 0 ; i < 4 ; i ++ ) {
int flatIndex = index + i ;
ivec3 rc = outCoordsFromFlatIndex ( flatIndex ) ;
result [ i ] = getA ( rc . x , rc . y , rc . z ) ;
}
$ { n . output } = result ;
}
` }};var RF=class{constructor(t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outPackingScheme=kd.DENSE,this.customUniforms=[{name:"texShape",type:"ivec2"}];let n=Vn();this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length),this.userCode= `
ivec3 outCoordsFromFlatIndex ( int index ) {
$ { this . enableShapeUniforms ? Lm ( [ "r" , "c" , "d" ] , t ) : Yi ( [ "r" , "c" , "d" ] , t ) }
return ivec3 ( r , c , d ) ;
}
void main ( ) {
ivec2 resTexRC = ivec2 ( resultUV . yx * vec2 ( texShape [ 0 ] , texShape [ 1 ] ) ) ;
int index = 4 * ( resTexRC . x * texShape [ 1 ] + resTexRC . y ) ;
vec4 result = vec4 ( 0. ) ;
for ( int i = 0 ; i < 4 ; i ++ ) {
int flatIndex = index + i ;
ivec3 rc = outCoordsFromFlatIndex ( flatIndex ) ;
result [ i ] = getChannel ( getA ( rc . x , rc . y , rc . z ) , vec2 ( rc . y , rc . z ) ) ;
}
$ { n . output } = result ;
}
` }};var DF=class{constructor(t){this.variableNames=["A"],this.outTexUsage=Rs.DOWNLOAD;let n=Vn();this.outputShape=t,this.userCode= `
$ { wS }
void main ( ) {
float x = getAAtOutCoords ( ) ;
$ { n . output } = encode _float ( x ) ;
}
` }};var $ F=class{constructor(t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!1,this.outTexUsage=Rs.DOWNLOAD;let n=Vn();this.outputShape=t,this.userCode= `
$ { wS }
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
float x = getChannel ( getAAtOutCoords ( ) , vec2 ( coords . y , coords . z ) ) ;
$ { n . output } = encode _float ( x ) ;
}
` }};var FF=class{constructor(t,n=!1){this.variableNames=["A"],this.customUniforms=[{name:"texShape",type:"ivec2"}];let r=Vn();this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length);let s="result";n&&(s="floor(result * 255. + 0.5)"),this.userCode= `
$ { this . enableShapeUniforms ? i1 ( ) : a1 ( t ) }
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int flatIndex = getFlatIndex ( coords ) ;
int offset = imod ( flatIndex , 4 ) ;
flatIndex = idiv ( flatIndex , 4 , 1. ) ;
int r = flatIndex / texShape [ 1 ] ;
int c = imod ( flatIndex , texShape [ 1 ] ) ;
vec2 uv = ( vec2 ( c , r ) + halfCR ) / vec2 ( texShape [ 1 ] , texShape [ 0 ] ) ;
vec4 values = $ { r . texture2D } ( A , uv ) ;
float result ;
if ( offset == 0 ) {
result = values [ 0 ] ;
} else if ( offset == 1 ) {
result = values [ 1 ] ;
} else if ( offset == 2 ) {
result = values [ 2 ] ;
} else {
result = values [ 3 ] ;
}
$ { r . output } = vec4 ( $ { s } , 0. , 0. , 0. ) ;
}
` }};var OF=class{constructor(t,n=!1){this.variableNames=["A"],this.packedInputs=!1,this.packedOutput=!0,this.customUniforms=[{name:"texShape",type:"ivec2"}];let r=Vn();this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length);let s="",o="result";n&&(o="floor(result * 255. + 0.5)");for(let a=0;a<=1;a++)for(let i=0;i<=1;i++){let l=a*2+i;s+= `
localCoords = coords ;
if ( localCoords [ 2 ] + $ { i } < $ { this . enableShapeUniforms ? "outShape[2]" : ` ${ t [ 2 ] } ` } ) {
localCoords [ 2 ] += $ { i } ;
if ( localCoords [ 1 ] + $ { a } < $ { this . enableShapeUniforms ? "outShape[1]" : ` ${ t [ 1 ] } ` } ) {
localCoords [ 1 ] += $ { a } ;
flatIndex = getFlatIndex ( localCoords ) ;
offset = imod ( flatIndex , 4 ) ;
flatIndex = idiv ( flatIndex , 4 , 1. ) ;
int r = flatIndex / texShape [ 1 ] ;
int c = imod ( flatIndex , texShape [ 1 ] ) ;
vec2 uv = ( vec2 ( c , r ) + halfCR ) / vec2 ( texShape [ 1 ] , texShape [ 0 ] ) ;
values = $ { r . texture2D } ( A , uv ) ;
if ( offset == 0 ) {
result [ $ { l } ] = values [ 0 ] ;
} else if ( offset == 1 ) {
result [ $ { l } ] = values [ 1 ] ;
} else if ( offset == 2 ) {
result [ $ { l } ] = values [ 2 ] ;
} else {
result [ $ { l } ] = values [ 3 ] ;
}
}
}
` }this.userCode= `
$ { this . enableShapeUniforms ? i1 ( ) : a1 ( t ) }
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
int flatIndex , r , c , offset ;
ivec3 localCoords ;
vec2 uv ;
vec4 values ;
$ { s }
$ { r . output } = $ { o } ;
}
` }};function Kte(e){let t=Vn(),n= ` $ { t . version }
precision highp float ;
$ { t . attribute } vec3 clipSpacePos ;
$ { t . attribute } vec2 uv ;
$ { t . varyingVs } vec2 resultUV ;
void main ( ) {
gl _Position = vec4 ( clipSpacePos , 1 ) ;
resultUV = uv ;
} ` ;return Ate(e,n)}function Xte(e){let t=new Float32Array([-1,1,0,0,1,-1,-1,0,0,0,1,1,0,1,1,1,-1,0,1,0]);return Ite(e,t)}function Zte(e){let t=new Uint16Array([0,1,2,2,1,3]);return Cte(e,t)}function dv(e,t,n,r,s,o){Nte(t,n);let a=Ste(e),i=e.TEXTURE_2D;return ot(e,()=>e.bindTexture(i,a)),ot(e,()=>e.texParameteri(i,e.TEXTURE_WRAP_S,e.CLAMP_TO_EDGE)),ot(e,()=>e.texParameteri(i,e.TEXTURE_WRAP_T,e.CLAMP_TO_EDGE)),ot(e,()=>e.texParameteri(i,e.TEXTURE_MIN_FILTER,e.NEAREST)),ot(e,()=>e.texParameteri(i,e.TEXTURE_MAG_FILTER,e.NEAREST)),ot(e,()=>e.texImage2D(i,0,r,t,n,0,s,o,null)),ot(e,()=>e.bindTexture(e.TEXTURE_2D,null)),a}function PF(e){return e.internalFormatFloat}function Yte(e,t,n,r){let[s,o]=Mm(t,n);return dv(e,s,o,PF(r),r.textureFormatFloat,e.FLOAT)}function MF(e){return e.internalFormatHalfFloat}function Jte(e,t,n,r){let[s,o]=Mm(t,n);return dv(e,s,o,MF(r),r.textureFormatFloat,r.textureTypeHalfFloat)}function zF(e){return e.downloadTextureFormat}function Qte(e,t,n,r){let[s,o]=Mm(t,n);return dv(e,s,o,zF(r),e.RGBA,e.UNSIGNED_BYTE)}function LF(e){return e.internalFormatPackedFloat}function ene(e,t,n,r){let[s,o]=El(t,n);return dv(e,s,o,LF(r),e.RGBA,e.FLOAT)}function BF(e){return e.internalFormatPackedHalfFloat}function tne(e,t,n,r){let[s,o]=El(t,n);return dv(e,s,o,BF(r),e.RGBA,r.textureTypeHalfFloat)}function nne(e,t,n){let r=0,s=3*4,o=3*4+2*4;return ot(e,()=>e.bindBuffer(e.ARRAY_BUFFER,n)),IF(e,t,"clipSpacePos",n,3,o,r)&&IF(e,t,"uv",n,2,o,s)}function rne(e,t,n,r,s,o){ot(e,()=>e.bindTexture(e.TEXTURE_2D,t));let a,i,l;s instanceof Uint8Array?(a=new Uint8Array(n*r*4),i=e.UNSIGNED_BYTE,l=e.RGBA):(a=new Float32Array(n*r*4),i=e.FLOAT,l=o.internalFormatPackedFloat),a.set(s),ot(e,()=>e.texImage2D(e.TEXTURE_2D,0,l,n,r,0,e.RGBA,i,a)),ot(e,()=>e.bindTexture(e.TEXTURE_2D,null))}function sne(e,t,n){ot(e,()=>e.bindTexture(e.TEXTURE_2D,t)),n.data instanceof Uint8Array?ot(e,()=>e.texImage2D(e.TEXTURE_2D,0,e.RGBA,n.width,n.height,0,e.RGBA,e.UNSIGNED_BYTE,n.data)):ot(e,()=>e.texImage2D(e.TEXTURE_2D,0,e.RGBA,e.RGBA,e.UNSIGNED_BYTE,n)),ot(e,()=>e.bindTexture(e.TEXTURE_2D,null))}function one(e,t,n,r){let s=e.createBuffer();ot(e,()=>e.bindBuffer(e.PIXEL_PACK_BUFFER,s));let i=4*4*t*n;return ot(e,()=>e.bufferData(e.PIXEL_PACK_BUFFER,i,e.STREAM_READ)),ot(e,()=>e.readPixels(0,0,n,t,e.RGBA,e.FLOAT,0)),ot(e,()=>e.bindBuffer(e.PIXEL_PACK_BUFFER,null)),s}function ane(e,t,n){let r=e,s=new Float32Array(n);return r.bindBuffer(r.PIXEL_PACK_BUFFER,t),r.getBufferSubData(r.PIXEL_PACK_BUFFER,0,s),r.bindBuffer(r.PIXEL_PACK_BUFFER,null),s}function ine(e,t,n,r){let[s,o]=Mm(t,n),a=4,i=new Uint8Array(xte(t*n,a));return ot(e,()=>e.readPixels(0,0,s,o,r.downloadTextureFormat,e.UNSIGNED_BYTE,i)),new Float32Array(i.buffer)}function lne(e,t,n,r,s,o,a,i){let l=e,u=new Float32Array(yte(o,a));return l.bindBuffer(l.PIXEL_PACK_BUFFER,t),l.getBufferSubData(l.PIXEL_PACK_BUFFER,0,u),l.bindBuffer(l.PIXEL_PACK_BUFFER,null),u}function une(e,t,n){let r=new Float32Array(t*n*4);return ot(e,()=>e.readPixels(0,0,n,t,e.RGBA,e.FLOAT,r)),r}var VF=class{constructor(t){this.outputTexture=null,this.program=null,this.disposed=!1,this.vertexAttrsAreBound=!1,this.itemsToPoll=[];let n=he().getNumber("WEBGL_VERSION");t!=null?(this.gl=t,gte(n,t)):this.gl=Ca(n);let r="WEBGL_color_buffer_float",s="EXT_color_buffer_half_float";if(he().getNumber("WEBGL_VERSION")===1){let o="OES_texture_float",a="OES_texture_half_float";if(this.textureFloatExtension=cv(this.gl,o),Sa(this.gl,a))this.textureHalfFloatExtension=cv(this.gl,a);else if(he().get("WEBGL_FORCE_F16_TEXTURES"))throw new Error("GL context does not support half float textures, yet the environment flag WEBGL_FORCE_F16_TEXTURES is set to true.");if(this.colorBufferFloatExtension=this.gl.getExtension(r),Sa(this.gl,s))this.colorBufferHalfFloatExtension=cv(this.gl,s);else if(he().get("WEBGL_FORCE_F16_TEXTURES"))throw new Error("GL context does not support color renderable half floats, yet the environment flag WEBGL_FORCE_F16_TEXTURES is set to true.")}else if(r="EXT_color_buffer_float",Sa(this.gl,r))this.colorBufferFloatExtension=this.gl.getExtension(r);
void main ( ) {
setOutput ( vec4 ( getA ( ) , 0. , 0. , 0. ) ) ;
}
` ;else{let r=ir("rc",n),s=Ct(n),o=UKe(n,t,r),a=jKe(n,t[t.length-1],t[t.length-2],r),i=HKe(t,r);this.userCode= `
void main ( ) {
$ { s } rc = getOutputCoords ( ) ;
if ( $ { o } ) {
setOutput ( vec4 ( 0 ) ) ;
} else {
$ { a }
setOutput ( vec4 ( $ { i } ) ) ;
}
}
` }}};function GKe(e,t){let n=[];for(let r=0;r<=1;r++)for(let s=0;s<=1;s++){let o= ` $ { r === 0 ? "r" : "rp1" } , $ { s === 0 ? "c" : "cp1" } ` ;for(let a=2;a<e;a++)o= ` $ { t [ t . length - 1 - a ] } , ` +o;n.push(o)}return n}function UKe(e,t,n){if(e===1)return ` rc > $ { t [ 0 ] } ` ;let r="";for(let s=e-2;s<e;s++)r+= ` $ { n [ s ] } >= $ { t [ s ] } ` ,s<e-1&&(r+="||");return r}function jKe(e,t,n,r){if(e===1)return"";let s=r.slice(-2);return `
int r = $ { s [ 0 ] } ;
int c = $ { s [ 1 ] } ;
int rp1 = r + 1 ;
int cp1 = c + 1 ;
bool cEdge = cp1 >= $ { t } ;
bool rEdge = rp1 >= $ { n } ;
` }function HKe(e,t){let n=e.length,r=GKe(n,t);return n===1? ` getA ( rc ) ,
rc + 1 >= $ { e [ 0 ] } ? 0. : getA ( rc + 1 ) ,
0 , 0 ` : ` getA ( $ { r [ 0 ] } ) ,
cEdge ? 0. : getA ( $ { r [ 1 ] } ) ,
rEdge ? 0. : getA ( $ { r [ 2 ] } ) ,
rEdge || cEdge ? 0. : getA ( $ { r [ 3 ] } ) ` }var fv=class{constructor(t,n){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"inputShape",type:"ivec3"}],this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length);let r="";for(let s=0;s<4;s++){let o="thisRC = rc;";s%2==1&&(o+="thisRC.z += 1;"),s>1&&(o+="thisRC.y += 1;"),r+= `
$ { o }
$ { s > 0 ? "if(thisRC.y < rows && thisRC.z < cols){" : "" }
int flatIndex = getFlatIndex ( thisRC ) ;
ivec3 inputRC = inputCoordsFromReshapedOutCoords ( flatIndex ) ;
vec2 inputRCInnerDims = vec2 ( float ( inputRC . y ) , float ( inputRC . z ) ) ;
result [ $ { s } ] =
getChannel ( getA ( inputRC . x , inputRC . y , inputRC . z ) , inputRCInnerDims ) ;
$ { s > 0 ? "}" : "" }
` }this.userCode= `
$ { qKe ( n , this . enableShapeUniforms ) }
$ { this . enableShapeUniforms ? i1 ( ) : a1 ( t ) }
void main ( ) {
ivec3 rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
ivec3 thisRC ;
int rows = $ { this . enableShapeUniforms ? "outShape[1]" : t [ 1 ] } ;
int cols = $ { this . enableShapeUniforms ? "outShape[2]" : t [ 2 ] } ;
$ { r }
setOutput ( result ) ;
}
` }};function qKe(e,t){return `
ivec3 inputCoordsFromReshapedOutCoords ( int index ) {
$ { t ? Lte ( [ "r" , "c" , "d" ] , "inputShape" ) : Yi ( [ "r" , "c" , "d" ] , e ) }
return ivec3 ( r , c , d ) ;
}
` }var UF=class{constructor(t){this.gpgpu=t,this.numUsedTextures=0,this.numFreeTextures=0,this._numBytesAllocated=0,this._numBytesFree=0,this.freeTextures={},this.logEnabled=!1,this.usedTextures={}}acquireTexture(t,n,r){let s=Zne(n,r),o=Yne(t,s,r);o in this.freeTextures||(this.freeTextures[o]=[]),o in this.usedTextures||(this.usedTextures[o]=[]);let a=Xne(t,s,this.gpgpu.gl,this.gpgpu.textureConfig,r);if(this.freeTextures[o].length>0){this.numFreeTextures--,this.numUsedTextures++,this._numBytesFree-=a,this.log();let l=this.freeTextures[o].shift();return this.usedTextures[o].push(l),l}let i;return s===xs.PACKED_2X2_FLOAT32?i=this.gpgpu.createPackedMatrixTexture(t[0],t[1]):s===xs.PACKED_2X2_FLOAT16?i=this.gpgpu.createFloat16PackedMatrixTexture(t[0],t[1]):s===xs.UNPACKED_FLOAT32?i=this.gpgpu.createFloat32MatrixTexture(t[0],t[1]):s===xs.UNPACKED_FLOAT16?i=this.gpgpu.createFloat16MatrixTexture(t[0],t[1]):s===xs.PACKED_4X1_UNSIGNED_BYTE&&(i=this.gpgpu.createUnsignedBytesMatrixTexture(t[0],t[1])),this.usedTextures[o].push(i),this.numUsedTextures++,this._numBytesAllocated+=a,this.log(),i}releaseTexture(t,n,r,s){if(this.freeTextures==null)return;let o=Zne(r,s),a=Yne(n,o,s);a in this.freeTextures||(this.freeTextures[a]=[]);let i=Xne(n,o,this.gpgpu.gl,this.gpgpu.textureConfig,s),l=he().get("WEBGL_DELETE_TEXTURE_THRESHOLD");l!==-1&&this._numBytesAllocated>l?(this.gpgpu.deleteMatrixTexture(t),this._numBytesAllocated-=i):(this.freeTextures[a].push(t),this.numFreeTextures++,this._numBytesFree+=i),this.numUsedTextures--;let u=this.usedTextures[a],c=u.indexOf(t);if(c<0)throw new Error("Cannot release a texture that was never provided by this texture manager");u.splice(c,1),this.log()}log(){if(!this.logEnabled)return;let t=this.numFreeTextures+this.numUsedTextures;console.log("Free/Used", ` $ { this . numFreeTextures } / $ { this . numUsedTextures } ` , ` ( $ { t } ) ` );let n=this._numBytesFree/this._numBytesAllocated;console.log( ` Bytes allocated : $ { this . _numBytesAllocated } ` ),console.log( ` Bytes unused : $ { this . _numBytesFree } ( $ { Math . round ( 100 * n ) } % ) ` )}get numBytesAllocated(){return this._numBytesAllocated}get numBytesFree(){return this._numBytesFree}getNumUsedTextures(){return this.numUsedTextures}getNumFreeTextures(){return this.numFreeTextures}dispose(){if(this.freeTextures!=null){for(let t in this.freeTextures)this.freeTextures[t].forEach(n=>{this.gpgpu.deleteMatrixTexture(n)});for(let t in this.usedTextures)this.usedTextures[t].forEach(n=>{this.gpgpu.deleteMatrixTexture(n)});this.freeTextures=null,this.usedTextures=null,this.numUsedTextures=0,this.numFreeTextures=0,this._numBytesAllocated=0,this._numBytesFree=0}}};function KKe(e,t){let n=e;if(t===n.R32F)return 4;if(t===n.R16F)return 2;if(t===n.RGBA32F)return 16;if(t===e.RGBA)return 16;if(t===n.RGBA16F)return 8;throw new Error( ` Unknown internal format $ { t } ` )}function Xne(e,t,n,r,s){let o=XKe(t,r),a;if(s){let[l,u]=El(e[0],e[1]);a=l*u}else{let[l,u]=Mm(e[0],e[1]);a=l*u}let i=KKe(n,o);return a*i}function XKe(e,t){switch(e){case xs.PACKED_2X2_FLOAT32:return LF(t);case xs.PACKED_2X2_FLOAT16:return BF(t);case xs.UNPACKED_FLOAT32:return PF(t);case xs.UNPACKED_FLOAT16:return MF(t);case xs.PACKED_4X1_UNSIGNED_BYTE:return zF(t);default:throw new Error( ` Unknown physical texture type $ { e } ` )}}function ZKe(e){return he().getBool("WEBGL_RENDER_FLOAT32_ENABLED")?e?xs.PACKED_2X2_FLOAT32:xs.UNPACKED_FLOAT32:e?xs.PACKED_2X2_FLOAT16:xs.UNPACKED_FLOAT16}function Zne(e,t){if(e===Rs.UPLOAD)return xs.PACKED_2X2_FLOAT32;if(e===Rs.RENDER||e==null)return ZKe(t);if(e===Rs.DOWNLOAD||e===Rs.PIXELS)return xs.PACKED_4X1_UNSIGNED_BYTE;throw new Error( ` Unknown logical texture type $ { e } ` )}function Yne(e,t,n){return ` $ { e [ 0 ] } _$ { e [ 1 ] } _$ { t } _$ { n } ` }var $ o=class{constructor(t,n){this.variableNames=["A"],this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length),this.userCode= `
float unaryOperation ( float x ) {
$ { n }
}
void main ( ) {
float x = getAAtOutCoords ( ) ;
float y = unaryOperation ( x ) ;
setOutput ( y ) ;
}
` }},Qr="if (isnan(x)) return x;",Jne="return x;",jF="return abs(x);";var Qne="return (x >= 0.0) ? x : (exp(x) - 1.0);",ere=Qr+ `
return ( x < 0.0 ) ? 0.0 : x ;
` ,tre=Qr+ `
return ( x < 0.0 ) ? 0.0 : min ( 6.0 , x ) ;
` ,hv="return x;",nre="return 1.0 / (1.0 + exp(-1.0 * x));";var rre="return x;",sre= `
vec4 result ;
result . r = ( x . r >= 0.0 ) ? x . r : ( exp ( x . r ) - 1.0 ) ;
result . g = ( x . g >= 0.0 ) ? x . g : ( exp ( x . g ) - 1.0 ) ;
result . b = ( x . b >= 0.0 ) ? x . b : ( exp ( x . b ) - 1.0 ) ;
result . a = ( x . a >= 0.0 ) ? x . a : ( exp ( x . a ) - 1.0 ) ;
return result ;
` ,ore= `
vec4 result = x * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
` ,are= `
vec4 result = min ( x , vec4 ( 6. ) ) * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
` ,ire="return 1.0 / (1.0 + exp(-1.0 * x));",Ji=class{constructor(t,n){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length),this.userCode= `
vec4 unaryOperation ( vec4 x ) {
$ { n }
}
void main ( ) {
vec4 x = getAAtOutCoords ( ) ;
vec4 y = unaryOperation ( x ) ;
setOutput ( y ) ;
}
` }};var HF=class{constructor(t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!1,this.outputShape=t;let n=t.length,r=ir("rc",n),s=Ct(n),o=Kne(n,r),a=r.slice(-2),i=n<=1?"rc": ` vec2 ( $ { a . join ( "," ) } ) ` ;this.userCode= `
void main ( ) {
$ { s } rc = getOutputCoords ( ) ;
vec4 packedInput = getA ( $ { o } ) ;
setOutput ( getChannel ( packedInput , $ { i } ) ) ;
}
` }};var YKe=Ts.whereImpl,JKe=1e-7,QKe=1e-4,NS={};function eXe(e){return e in NS||(NS[e]={}),NS[e]}var tXe=he().getNumber("CPU_HANDOFF_SIZE_THRESHOLD"),nXe=600;function rXe(){return he().global.screen==null?1024:he().global.screen.height*he().global.screen.width*window.devicePixelRatio*nXe/1024/1024}var d1=class extends up{constructor(t){super();if(this.pendingRead=new WeakMap,this.pendingDisposal=new WeakSet,this.dataRefCount=new WeakMap,this.numBytesInGPU=0,this.uploadWaitMs=0,this.downloadWaitMs=0,this.lastGlFlushTime=0,this.warnedAboutMemory=!1,this.pendingDeletes=0,this.disposed=!1,!he().getBool("HAS_WEBGL"))throw new Error("WebGL is not supported on this device");if(t==null){let n=Ca(he().getNumber("WEBGL_VERSION"));this.binaryCache=eXe(he().getNumber("WEBGL_VERSION")),this.gpgpu=new VF(n),this.canvas=n.canvas,this.gpgpuCreatedLocally=!0}else this.gpgpu=t,this.binaryCache={},this.gpgpuCreatedLocally=!1,this.canvas=t.gl.canvas;this.textureManager=new UF(this.gpgpu),this.numMBBeforeWarning=rXe(),this.texData=new ux(this,Zu())}nextDataId(){return d1.nextDataId++}numDataIds(){return this.texData.numDataIds()-this.pendingDeletes}write(t,n,r){if((he().getBool("WEBGL_CHECK_NUMERICAL_PROBLEMS")||he().getBool("DEBUG"))&&this.checkNumericalProblems(t),r==="complex64"&&t!=null)throw new Error("Cannot write to a complex64 dtype. Please use tf.complex(real, imag).");let s={id:this.nextDataId()};return this.texData.set(s,{shape:n,dtype:r,values:t,usage:Rs.UPLOAD,refCount:1}),s}refCount(t){return this.texData.has(t)?this.texData.get(t).refCount:0}incRef(t){let n=this.texData.get(t);n.refCount++}decRef(t){if(this.texData.has(t)){let n=this.texData.get(t);n.refCount--}}move(t,n,r,s,o){if(he().getBool("DEBUG")&&this.checkNumericalProblems(n),s==="complex64")throw new Error("Cannot write to a complex64 dtype. Please use tf.complex(real, imag).");this.texData.set(t,{shape:r,dtype:s,values:n,usage:Rs.UPLOAD,refCount:o})}disposeIntermediateTensorInfo(t){this.disposeData(t.dataId)}readSync(t){let n=this.texData.get(t),{values:r,dtype:s,complexTensorInfos:o,slice:a,shape:i,isPacked:l}=n;if(a!=null){let d;l?d=new Ji(i,hv):d=new $ o(i,hv);let f=this.runWebGLProgram(d,[{dataId:t,shape:i,dtype:s}],s),h=this.readSync(f.dataId);return this.disposeIntermediateTensorInfo(f),h}if(r!=null)return this.convertAndCacheOnCPU(t);if(s==="string")return r;let u=this.activeTimers!=null,c;u&&(c=C.now());let p;if(s==="complex64"){let d=this.readSync(o.real.dataId),f=this.readSync(o.imag.dataId);p=M.mergeRealAndImagArrays(d,f)}else p=this.getValuesFromTexture(t);return u&&(this.downloadWaitMs+=C.now()-c),this.convertAndCacheOnCPU(t,p)}async read(t){if(this.pendingRead.has(t)){let h=this.pendingRead.get(t);return new Promise(m=>h.push(m))}let n=this.texData.get(t),{values:r,shape:s,slice:o,dtype:a,complexTensorInfos:i,isPacked:l}=n;if(o!=null){let h;l?h=new Ji(s,hv):h=new $ o(s,hv);let m=this.runWebGLProgram(h,[{dataId:t,shape:s,dtype:a}],a),g=this.read(m.dataId);return this.disposeIntermediateTensorInfo(m),g}if(r!=null)return this.convertAndCacheOnCPU(t);if(!he().getBool("WEBGL_DOWNLOAD_FLOAT_ENABLED")&&he().getNumber("WEBGL_VERSION")===2)throw new Error("tensor.data() with WEBGL_DOWNLOAD_FLOAT_ENABLED=false and WEBGL_VERSION=2 not yet supported.");let u=null,c;if(a!=="complex64"&&he().get("WEBGL_BUFFER_SUPPORTED")){c=this.decode(t);let h=this.texData.get(c.dataId);u=this.gpgpu.createBufferFromTexture(h.texture,...lv(s))}this.pendingRead.set(t,[]),a!=="complex64"&&await this.gpgpu.createAndWaitForFence();let p;if(a==="complex64"){let h=await Promise.all([this.read(i.real.dataId),this.read(i.imag.dataId)]),m=h[0],g=h[1];p=M.mergeRealAndImagArrays(m,g)}else if(u==null)p=this.getValuesFromTexture(t);else{let h=C.sizeFromShape(s);p=this.gpgpu.downloadFloat32MatrixFromBuffer(u,h)}if(c!=null&&this.disposeIntermediateTensorInfo(c),u!=null){let h=this.gpgpu.gl;ot(h,()=>h.deleteBuffer(u))}let d=this.convertAndCacheOnCPU(t,p),f=this.pendingRead.get(t);return this.pendingRead.delete(t),f.forEach(h=>h(d)),this.pendingDisposal.has(t)&&(this.pendingDisposal.delete(t),thi
if ( isnan ( a ) ) return a ;
if ( isnan ( b ) ) return b ;
` ;var ui=class{constructor(t,n,r){this.variableNames=["A","B"],this.outputShape=M.assertAndGetBroadcastShape(n,r),this.enableShapeUniforms=Xn(this.outputShape.length),this.userCode= `
float binaryOperation ( float a , float b ) {
$ { t }
}
void main ( ) {
float a = getAAtOutCoords ( ) ;
float b = getBAtOutCoords ( ) ;
setOutput ( binaryOperation ( a , b ) ) ;
}
` }};var Sd= `
result . r = isNaN . r > 0. ? NAN : result . r ;
result . g = isNaN . g > 0. ? NAN : result . g ;
result . b = isNaN . b > 0. ? NAN : result . b ;
result . a = isNaN . a > 0. ? NAN : result . a ;
` ;var Qi=class{constructor(t,n,r,s=!1){this.variableNames=["A","B"],this.supportsBroadcasting=!0,this.packedInputs=!0,this.packedOutput=!0,this.outputShape=M.assertAndGetBroadcastShape(n,r);let o=this.outputShape.length;this.enableShapeUniforms=Xn(o);let a="";if(s)if(o===0||C.sizeFromShape(this.outputShape)===1)a= `
result . y = 0. ;
result . z = 0. ;
result . w = 0. ;
` ;else if(a= `
$ { Ct ( o ) } coords = getOutputCoords ( ) ;
` ,o===1)this.enableShapeUniforms?a+= `
result . y = ( coords + 1 ) >= outShape ? 0. : result . y ;
result . z = 0. ;
result . w = 0. ;
` :a+= `
result . y = ( coords + 1 ) >= $ { this . outputShape [ 0 ] } ? 0. : result . y ;
result . z = 0. ;
result . w = 0. ;
` ;else{let l=ir("coords",o);this.enableShapeUniforms?a+= `
bool nextRowOutOfBounds =
( $ { l [ o - 2 ] } + 1 ) >= outShape [ $ { o } - 2 ] ;
bool nextColOutOfBounds =
( $ { l [ o - 1 ] } + 1 ) >= outShape [ $ { o } - 1 ] ;
result . y = nextColOutOfBounds ? 0. : result . y ;
result . z = nextRowOutOfBounds ? 0. : result . z ;
result . w = nextColOutOfBounds || nextRowOutOfBounds ? 0. : result . w ;
` :a+= `
bool nextRowOutOfBounds =
( $ { l [ o - 2 ] } + 1 ) >= $ { this . outputShape [ o - 2 ] } ;
bool nextColOutOfBounds =
( $ { l [ o - 1 ] } + 1 ) >= $ { this . outputShape [ o - 1 ] } ;
result . y = nextColOutOfBounds ? 0. : result . y ;
result . z = nextRowOutOfBounds ? 0. : result . z ;
result . w = nextColOutOfBounds || nextRowOutOfBounds ? 0. : result . w ;
` }this.userCode= `
vec4 binaryOperation ( vec4 a , vec4 b ) {
$ { t }
}
void main ( ) {
vec4 a = getAAtOutCoords ( ) ;
vec4 b = getBAtOutCoords ( ) ;
vec4 result = binaryOperation ( a , b ) ;
$ { a }
setOutput ( result ) ;
}
` }};function lr(e){let{inputs:t,backend:n}=e,{x:r}=t;return n.incRef(r.dataId),{dataId:r.dataId,shape:r.shape,dtype:r.dtype}}var lre={kernelName:Al,backendName:"webgl",kernelFunc:lr};function Fo(e){let{inputs:t,backend:n}=e,{real:r,imag:s}=t,o=n.makeTensorInfo(r.shape,"complex64"),a=n.texData.get(o.dataId),i=lr({inputs:{x:r},backend:n}),l=lr({inputs:{x:s},backend:n});return a.complexTensorInfos={real:i,imag:l},o}var ure={kernelName:vx,backendName:"webgl",kernelFunc:Fo};var qF="return (a < 0.) ? b * a : a;",KF= `
vec4 aLessThanZero = vec4 ( lessThan ( a , vec4 ( 0. ) ) ) ;
return ( aLessThanZero * ( b * a ) ) + ( ( vec4 ( 1.0 ) - aLessThanZero ) * a ) ;
` ;function oXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{alpha:o}=r,a=n.makeTensorInfo([],"float32",C.createScalarValue(o,"float32")),i=he().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new Qi(KF,s.shape,a.shape):new ui(qF,s.shape,a.shape),l=n.runWebGLProgram(i,[s,a],s.dtype);return n.disposeIntermediateTensorInfo(a),l}var cre={kernelName:Ep,backendName:"webgl",kernelFunc:oXe};var XF="return (a < 0.) ? b * a : a;",ZF= `
vec4 aLessThanZero = vec4 ( lessThan ( a , vec4 ( 0. ) ) ) ;
return ( aLessThanZero * ( b * a ) ) + ( ( vec4 ( 1.0 ) - aLessThanZero ) * a ) ;
` ;function aXe(e){let{inputs:t,backend:n}=e,{x:r,alpha:s}=t,o=he().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new Qi(ZF,r.shape,s.shape):new ui(XF,r.shape,s.shape);return n.runWebGLProgram(o,[r,s],r.dtype)}var pre={kernelName:Gp,backendName:"webgl",kernelFunc:aXe};var _S="if (isnan(x)) return x;",dre= `
if ( isnan ( a ) ) return a ;
if ( isnan ( b ) ) return b ;
` ,fre= `
result . r = isNaN . r > 0. ? NAN : result . r ;
result . g = isNaN . g > 0. ? NAN : result . g ;
result . b = isNaN . b > 0. ? NAN : result . b ;
result . a = isNaN . a > 0. ? NAN : result . a ;
` ;function Xe({opSnippet:e,packedOpSnippet:t,cpuKernelImpl:n,dtype:r}){return({inputs:s,backend:o})=>{let{x:a}=s,i=o,l=r||a.dtype;if(i.shouldExecuteOnCPU([a])&&n!=null){let p=i.texData.get(a.dataId),d=n(p.values,l);return i.makeTensorInfo(a.shape,l,d)}let u=he().getBool("WEBGL_PACK_UNARY_OPERATIONS")&&t!=null,c;return u?c=new Ji(a.shape,t):c=new $ o(a.shape,e),i.runWebGLProgram(c,[a],l)}}function Qt({opSnippet:e,packedOpSnippet:t,checkOutOfBounds:n=!1,supportsComplex:r=!1,cpuKernelImpl:s,dtype:o}){return({inputs:a,backend:i})=>{let{a:l,b:u}=a,c=i;if(r&&l.dtype==="complex64"){let h=c.texData.get(l.dataId),m=c.texData.get(u.dataId),[g,x]=[[h.complexTensorInfos.real,m.complexTensorInfos.real],[h.complexTensorInfos.imag,m.complexTensorInfos.imag]].map(b=>{let[A,v]=b,w={dataId:A.dataId,dtype:A.dtype,shape:l.shape},k={dataId:v.dataId,dtype:v.dtype,shape:u.shape},I=new ui(e,l.shape,u.shape);return c.runWebGLProgram(I,[w,k],Or(A.dtype,v.dtype))}),y=Fo({inputs:{real:g,imag:x},backend:c});return c.disposeIntermediateTensorInfo(g),c.disposeIntermediateTensorInfo(x),y}let p=o||Or(l.dtype,u.dtype);if((l.dtype==="string"||u.dtype==="string"||c.shouldExecuteOnCPU([l,u]))&&s!=null){let h=c.texData.get(l.dataId).values,m=c.texData.get(u.dataId).values,g=l.dtype==="string"?M.fromUint8ToStringArray(h):h,x=l.dtype==="string"?M.fromUint8ToStringArray(m):m,[y,b]=s(l.shape,u.shape,g,x,p),A=c.makeTensorInfo(b,p),v=c.texData.get(A.dataId);return v.values=y,A}let d=he().getBool("WEBGL_PACK_BINARY_OPERATIONS")&&t!=null,f;return d?f=new Qi(t,l.shape,u.shape,n):f=new ui(e,l.shape,u.shape),c.runWebGLProgram(f,[l,u],p)}}function Nd(e,t=!1){if(e==="linear")return t?rre:Jne;if(e==="relu")return t?ore:ere;if(e==="elu")return t?sre:Qne;if(e==="relu6")return t?are:tre;if(e==="prelu")return t?ZF:XF;if(e==="leakyrelu")return t?KF:qF;if(e==="sigmoid")return t?ire:nre;throw new Error( ` Activation $ { e } has not been implemented for the WebGL backend . ` )}var mv=class{constructor(t,n,r,s=!1,o=!1,a=!1,i=null,l=!1,u=!1){this.variableNames=["matrixA","matrixB"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=r,this.enableShapeUniforms=Xn(this.outputShape.length);let c=s?t[1]:t[2],p=Math.ceil(c/2),d=s?"i * 2, rc.y":"rc.y, i * 2",f=o?"rc.z, i * 2":"i * 2, rc.z",h=s?["a.xxyy","a.zzww"]:["a.xxzz","a.yyww"],m=o?["b.xzxz","b.ywyw"]:["b.xyxy","b.zwzw"],g="",x="";i&&(l?g= ` vec4 activation ( vec4 a ) {
vec4 b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { i }
} ` :u?g= ` vec4 activation ( vec4 a ) {
vec4 b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { i }
} ` :g= ` vec4 activation ( vec4 x ) {
$ { i }
} ` ,x="result = activation(result);");let y=a?"result += getBiasAtOutCoords();":"";a&&this.variableNames.push("bias"),l&&this.variableNames.push("preluActivationWeights"),u&&this.variableNames.push("leakyreluAlpha");let b="rc.x",A="rc.x";t[0]<n[0]?b= ` int ( min ( float ( rc . x ) , $ { t [ 0 ] - 1 } . ) ) ` :n[0]<t[0]&&(A= ` int ( min ( float ( rc . x ) , $ { n [ 0 ] - 1 } . ) ) ` ),this.userCode= `
$ { g }
// Don't use uniform for sharedDimensionPacked for performance.
const float sharedDimension = $ { p } . 0 ;
vec4 dot2x2ARowBCol ( ivec3 rc ) {
vec4 result = vec4 ( 0 ) ;
for ( int i = 0 ; i < $ { p } ; i ++ ) {
int batchA = $ { b } ;
int batchB = $ { A } ;
vec4 a = getMatrixA ( batchA , $ { d } ) ;
vec4 b = getMatrixB ( batchB , $ { f } ) ;
// These swizzled products need to be separately added.
// See: https://github.com/tensorflow/tfjs/issues/1735
result += ( $ { h [ 0 ] } * $ { m [ 0 ] } ) ;
result += ( $ { h [ 1 ] } * $ { m [ 1 ] } ) ;
}
return result ;
}
void main ( ) {
ivec3 rc = getOutputCoords ( ) ;
vec4 result = dot2x2ARowBCol ( rc ) ;
$ { y }
$ { x }
setOutput ( result ) ;
}
` }};var YF={REAL:"return areal * breal - aimag * bimag;",IMAG:"return areal * bimag + aimag * breal;"},ES=class{constructor(t,n,r){this.variableNames=["AReal","AImag","BReal","BImag"],this.outputShape=M.assertAndGetBroadcastShape(n,r),this.userCode= `
float binaryOpComplex (
float areal , float aimag , float breal , float bimag ) {
$ { t }
}
void main ( ) {
float areal = getARealAtOutCoords ( ) ;
float aimag = getAImagAtOutCoords ( ) ;
float breal = getBRealAtOutCoords ( ) ;
float bimag = getBImagAtOutCoords ( ) ;
setOutput ( binaryOpComplex ( areal , aimag , breal , bimag ) ) ;
}
` }};var hre="return a * b;";function gv(e){let{inputs:t,backend:n}=e,{a:r,b:s}=t,o=M.upcastType(r.dtype,s.dtype);if(r.dtype==="complex64"){let i=n.texData.get(r.dataId),l=n.texData.get(s.dataId),u=new ES(YF.REAL,r.shape,s.shape),c=new ES(YF.IMAG,r.shape,s.shape),p=[{dataId:i.complexTensorInfos.real.dataId,dtype:i.complexTensorInfos.real.dtype,shape:r.shape},{dataId:i.complexTensorInfos.imag.dataId,dtype:i.complexTensorInfos.imag.dtype,shape:r.shape},{dataId:l.complexTensorInfos.real.dataId,dtype:l.complexTensorInfos.real.dtype,shape:s.shape},{dataId:l.complexTensorInfos.imag.dataId,dtype:l.complexTensorInfos.imag.dtype,shape:s.shape}],d=n.runWebGLProgram(u,p,"float32"),f=n.runWebGLProgram(c,p,"float32"),h=Fo({inputs:{real:d,imag:f},backend:n});return n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}if(n.shouldExecuteOnCPU([r,s])){let i=n.texData.get(r.dataId),l=n.texData.get(s.dataId),[u,c]=_ne(r.shape,s.shape,i.values,l.values,o),p=n.makeTensorInfo(c,o),d=n.texData.get(p.dataId);return d.values=u,p}let a;return he().getBool("WEBGL_PACK_BINARY_OPERATIONS")?a=new Qi(hre,r.shape,s.shape):a=new ui(hre,r.shape,s.shape),n.runWebGLProgram(a,[r,s],o)}var mre={kernelName:Nu,backendName:"webgl",kernelFunc:gv};function gre(e,t,n){let r=[Id(e.shape),...Cd(e.shape)],s={dtype:e.dtype,shape:r,dataId:e.dataId},o=[Id(t),...Cd(t)],a=new fv(o,r),i=!0,l=[r],u=n.runWebGLProgram(a,[s],e.dtype,l,i);return{dataId:u.dataId,shape:t,dtype:u.dtype}}function $ e(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{shape:o}=r,a=n,i=C.sizeFromShape(s.shape),l=C.inferFromImplicitShape(o,i),u=C.sizeFromShape(l);C.assert(i===u,()=> ` The new shape ( $ { l } ) has $ { u } elements and the old shape ( $ { s . shape } ) has $ { i } elements . The new shape and old shape must have the same number of elements . ` );let c=a.texData.get(s.dataId);return c.isPacked&&!zm(s.shape,l)&&!(c.texture!==null&&zm(c.shape,l))?gre(s,l,a):(a.incRef(s.dataId),{dataId:s.dataId,shape:l,dtype:s.dtype})}var xre={kernelName:Up,backendName:"webgl",kernelFunc: $ e};var RS=class{constructor(t,n){this.variableNames=["x"];let{windowSize:r,batchSize:s,inSize:o,outSize:a}=t;this.outputShape=[s,a];let i=Math.floor(r/4)*4,l=r%4,u="sumValue += dot(values, ones);";if(n!=null){let p=1/n;u= ` sumValue += dot ( values * $ { C . isInt ( p ) ? p . toPrecision ( 2 ) : p } , ones ) ; ` }let c="";o%r>0&&(c= `
if ( inIdx < 0 || inIdx >= $ { o } ) {
return 0.0 ;
}
` ),this.userCode= `
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
float getValue ( int batch , int inIdx ) {
$ { c }
return getX ( batch , inIdx ) ;
}
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = outIdx * $ { r } ;
float sumValue = 0.0 ;
for ( int i = 0 ; i < $ { i } ; i += 4 ) {
int inIdx = inOffset + i ;
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
$ { u }
}
int inIdx = inOffset + $ { i } ;
if ( $ { l === 1 } ) {
vec4 values = vec4 ( getValue ( batch , inIdx ) , 0.0 , 0.0 , 0.0 ) ;
$ { u }
} else if ( $ { l === 2 } ) {
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) , 0.0 , 0.0 ) ;
$ { u }
} else if ( $ { l === 3 } ) {
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) , 0.0 ) ;
$ { u }
}
setOutput ( sumValue ) ;
}
` }};var JF=class{constructor(t,n){this.variableNames=["x"];let{windowSize:r,batchSize:s,inSize:o,outSize:a}=t;this.outputShape=[s,a];let i="0.0",l="";n==="prod"?i="1.0":n==="min"?(i="1.0 / 1e-20",l="min"):n==="max"&&(i="-1.0 / 1e-20",l="max");let u= ` $ { n } ( $ { n } ( $ { n } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;n==="sum"?u="sumValue":n==="prod"?u="prodValue":n==="all"?u="allValue":n==="any"&&(u="anyValue");let c=Math.floor(r/4)*4,p=r%4,d= `
if ( $ { n === "sum" } ) {
sumValue += dot ( values , ones ) ;
} else if ( $ { n === "prod" } ) {
vec2 tmp = vec2 ( values [ 0 ] , values [ 1 ] ) * vec2 ( values [ 2 ] , values [ 3 ] ) ;
prodValue *= tmp [ 0 ] * tmp [ 1 ] ;
} else {
minMaxValue = $ { l } ( values , minMaxValue ) ;
if ( $ { n === "min" } || $ { n === "max" } ) {
minMaxValue = $ { l } ( values , minMaxValue ) ;
bvec4 isNaN = isnan ( values ) ;
if ( isNaN . r || isNaN . g || isNaN . b || isNaN . a ) {
minMaxValue = vec4 ( NAN ) ;
}
}
}
` ,f="vec4";n==="all"?(i="1.0",d= `
bool reducedAllValue = all ( values ) ;
float floatedReducedAllValue = float ( reducedAllValue ) ;
allValue = float ( allValue >= 1.0 && floatedReducedAllValue >= 1.0 ) ;
` ,f="bvec4"):n==="any"&&(i="0.0",d= `
bool reducedAnyValue = any ( values ) ;
float floatedReducedAnyValue = float ( reducedAnyValue ) ;
anyValue = float ( anyValue >= 1.0 || floatedReducedAnyValue >= 1.0 ) ;
` ,f="bvec4");let h="";o%r>0&&(h= `
if ( inIdx < 0 || inIdx >= $ { o } ) {
return initializationValue ;
}
` ),this.userCode= `
const float initializationValue = $ { i } ;
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
float getValue ( int batch , int inIdx ) {
$ { h }
return getX ( batch , inIdx ) ;
}
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = outIdx * $ { r } ;
vec4 minMaxValue = vec4 ( $ { i } ) ;
float prodValue = 1.0 ;
float sumValue = 0.0 ;
float allValue = 1.0 ;
float anyValue = 0.0 ;
for ( int i = 0 ; i < $ { c } ; i += 4 ) {
int inIdx = inOffset + i ;
$ { f } values = $ { f } (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
$ { d }
}
int inIdx = inOffset + $ { c } ;
if ( $ { p === 1 } ) {
$ { f } values = $ { f } (
getValue ( batch , inIdx ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
$ { d }
} else if ( $ { p === 2 } ) {
$ { f } values = $ { f } (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
initializationValue ,
initializationValue
) ;
$ { d }
} else if ( $ { p === 3 } ) {
$ { f } values = $ { f } (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
initializationValue
) ;
$ { d }
}
setOutput ( $ { u } ) ;
}
` }};function iXe(e){let t=[];for(;t.length===0||t[t.length-1].outSize!==1;){let n=t.length?t[t.length-1].outSize:e[1],r=M.computeOptimalWindowSize(n);t.push({inSize:n,windowSize:r,outSize:Math.ceil(n/r)})}return t}function la(e,t,n,r){let s=iXe(e.shape),o=e;for(let a=0;a<s.length;a++){let{inSize:i,windowSize:l,outSize:u}=s[a],c,p;n==="mean"?c=a===0?new RS({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u},i):new RS({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u}):c=new JF({windowSize:l,inSize:i,batchSize:e.shape[0],outSize:u},n),p=o,o=r.runWebGLProgram(c,[o],t),p.dataId!==e.dataId&&r.disposeIntermediateTensorInfo(p)}return o}var QF=class{constructor(t,n){this.variableNames=["A"];let r=new Array(t.length);for(let a=0;a<r.length;a++)r[a]=t[n[a]];this.outputShape=r,this.rank=r.length;let s=Ct(this.rank),o=lXe(n);this.userCode= `
void main ( ) {
$ { s } resRC = getOutputCoords ( ) ;
setOutput ( getA ( $ { o } ) ) ;
}
` }};function lXe(e){let t=e.length;if(t>6)throw Error( ` Transpose for rank $ { t } is not yet supported ` );let n=["resRC.x","resRC.y","resRC.z","resRC.w","resRC.u","resRC.v"],r=new Array(t);for(let s=0;s<e.length;s++)r[e[s]]=n[s];return r.join()}var e8=class{constructor(t,n){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0;let r=new Array(t.length);for(let c=0;c<r.length;c++)r[c]=t[n[c]];if(this.outputShape=r,this.rank=r.length,this.rank>6)throw Error( ` Packed transpose for rank $ { this . rank } is not yet supported . ` );let s=Ct(this.rank),o=WF("rc",this.rank),a=new Array(this.rank);for(let c=0;c<n.length;c++)a[n[c]]=o[c];let i= ` vec2 ( $ { a . slice ( - 2 ) . join ( ) } ) ` ,l= ` ++ $ { o [ this . rank - 1 ] } < $ { r [ this . rank - 1 ] } ` ,u= ` getChannel ( getA ( $ { a . join ( ) } ) , $ { i } ) ` ;this.userCode= `
void main ( ) {
$ { s } rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
result [ 0 ] = $ { u } ;
if ( $ { l } ) {
result [ 1 ] = $ { u } ;
}
-- $ { o [ this . rank - 1 ] } ;
if ( ++ $ { o [ this . rank - 2 ] } < $ { r [ this . rank - 2 ] } ) {
result [ 2 ] = $ { u } ;
if ( $ { l } ) {
result [ 3 ] = $ { u } ;
}
}
setOutput ( result ) ;
}
` }};function Td(e,t,n){let r=he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new e8(e.shape,t):new QF(e.shape,t);return n.runWebGLProgram(r,[e],e.dtype)}function yre(e,t,n,r){let s=t,o=e.shape.length,a=C.parseAxisParam(s,e.shape),i=a,l=M.getAxesPermutation(i,o),u=l!=null,c=e;u&&(c=Td(e,l,r),i=M.getInnerMostAxes(i.length,o)),M.assertAxesAreInnerMostDims("sum",i,o);let[p,d]=M.computeOutAndReduceShapes(c.shape,i),f=p;n&&(f=M.expandShapeToKeepDim(p,a));let h=C.sizeFromShape(d),g=C.sizeFromShape(e.shape)/h,x= $ e({inputs:{x:c},attrs:{shape:[g,h]},backend:r}),y=Vh(e.dtype),b=la(x,y,"sum",r),A= $ e({inputs:{x:b},attrs:{shape:f},backend:r});return r.disposeIntermediateTensorInfo(x),r.disposeIntermediateTensorInfo(b),u&&r.disposeIntermediateTensorInfo(c),A}function Wm(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r;return yre(s,o,a,n)}var bre={kernelName:Zp,backendName:"webgl",kernelFunc:Wm};function On(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{perm:o}=r,a=n,i=s.shape.length,l=new Array(i);for(let c=0;c<l.length;c++)l[c]=s.shape[o[c]];let u;if(a.shouldExecuteOnCPU([s])){let p=a.texData.get(s.dataId).values,d=Vm(p,s.shape,s.dtype,o,l);u=a.makeTensorInfo(l,s.dtype);let f=a.texData.get(u.dataId);f.values=d}else u=Td(s,o,a);return u}var Are={kernelName:ed,backendName:"webgl",kernelFunc:On};var t8=1e3;function Gm({a:e,b:t,transposeA:n,transposeB:r,backend:s,bias:o=null,preluActivationWeights:a=null,leakyreluAlpha:i=0,activation:l=null}){let u=e.shape.length,c=t.shape.length,p=n?e.shape[u-2]:e.shape[u-1],d=r?t.shape[c-1]:t.shape[c-2],f=n?e.shape[u-1]:e.shape[u-2],h=r?t.shape[c-2]:t.shape[c-1],m=e.shape.slice(0,-2),g=t.shape.slice(0,-2),x=C.sizeFromShape(m),y=C.sizeFromShape(g),b=x===y||x===1||y===1;C.assert(u>=2&&c>=2&&b,()=> ` Error in matMul : the input batch dimensions must either be the same or at least one input batch dimension must be 1. Got input batch dimensions of ( $ { m } ) and ( $ { g } ) . ` );let v=(x>y?e.shape.slice(0,-2):t.shape.slice(0,-2)).concat([f,h]);C.assert(p===d,()=> ` Error in matMul : inner shapes ( $ { p } ) and ( $ { d } ) of Tensors with shapes $ { e . shape } and $ { t . shape } and transposeA = $ { n } and transposeB = $ { r } must match . ` );let w=n?[x,p,f]:[x,f,p],k=r?[y,h,d]:[y,d,h],I= $ e({inputs:{x:e},backend:s,attrs:{shape:w}}),N= $ e({inputs:{x:t},backend:s,attrs:{shape:k}}),T=[I,N],_=Math.max(x,y),R=n?I.shape[1]:I.shape[2],S=o!=null,F=a!=null,B=l==="leakyrelu",W=l!=null?Nd(l,!0):null,q=S||F||B||W!=null,K;if((f===1||h===1)&&R>t8&&q===!1){let Q=I,ee=N;n&&(Q=On({inputs:{x:I},backend:s,attrs:{perm:[0,2,1]}}),T.push(Q)),r&&(ee=On({inputs:{x:N},backend:s,attrs:{perm:[0,2,1]}}),T.push(ee));let ue=h!==1,re=h===1,de=Q;ue&&(de= $ e({inputs:{x:Q},backend:s,attrs:{shape:[_,R,1]}}),T.push(de));let fe=h===1?2:1,me=ee;re&&(me= $ e({inputs:{x:ee},backend:s,attrs:{shape:[_,1,R]}}),T.push(me));let xe=gv({inputs:{a:de,b:me},backend:s});K=Wm({inputs:{x:xe},backend:s,attrs:{axis:fe,keepDims:!0}}),T.push(xe)}else{let Q=Or(e.dtype,t.dtype),ee=new mv(w,k,[_,f,h],n,r,S,W,F,B),ue=[I,N];if(o!=null&&ue.push(o),F&&ue.push(a),B){let re=s.makeTensorInfo([],"float32",C.createScalarValue(i,"float32"));ue.push(re),T.push(re)}K=s.runWebGLProgram(ee,ue,Q)}let J= $ e({inputs:{x:K},backend:s,attrs:{shape:v}});T.push(K);for(let Q of T)s.disposeIntermediateTensorInfo(Q);return J}function uXe(e){let{inputs:t,backend:n,attrs:r}=e,{a:s,b:o,bias:a,preluActivationWeights:i}=t,{transposeA:l,transposeB:u,activation:c,leakyreluAlpha:p}=r;return Gm({a:s,b:o,transposeA:l,transposeB:u,backend:n,bias:a,preluActivationWeights:i,leakyreluAlpha:p,activation:c})}var vre={kernelName: $ h,backendName:"webgl",kernelFunc:uXe};var wre="return abs(x);";function cXe(e){let{inputs:t,backend:n}=e,{x:r}=t;if(n.shouldExecuteOnCPU([r])&&r.dtype!=="complex64"){let o=n.texData.get(r.dataId),a=CS(o.values);return n.makeTensorInfo(r.shape,r.dtype,a)}let s;return he().getBool("WEBGL_PACK_UNARY_OPERATIONS")?s=new Ji(r.shape,wre):s=new $ o(r.shape,wre),n.runWebGLProgram(s,[r],r.dtype)}var kre={kernelName:pp,backendName:"webgl",kernelFunc:cXe};var pXe=Qr+ `
if ( abs ( x ) > 1. ) {
return NAN ;
}
return acos ( x ) ;
` ,dXe=Xe({opSnippet:pXe}),Ire={kernelName:eu,backendName:"webgl",kernelFunc:dXe};var fXe=Qr+ `
if ( x < 1.0 ) return NAN ;
return log ( x + sqrt ( x * x - 1.0 ) ) ; ` ,hXe=Xe({opSnippet:fXe}),Cre={kernelName:tu,backendName:"webgl",kernelFunc:hXe};var Sre="return a + b;",mXe=Qt({opSnippet:Sre,packedOpSnippet:Sre,supportsComplex:!0,cpuKernelImpl:cne}),Nre={kernelName:Ri,backendName:"webgl",kernelFunc:mXe};var n8=class{constructor(t,n){this.outputShape=[],this.outputShape=t,this.variableNames=n.map((o,a)=> ` T$ { a } ` );let r=[];this.variableNames.forEach(o=>{r.push( ` float v$ { o } = get$ { o } AtOutCoords ( ) ; ` )});let s=this.variableNames.map(o=> ` v$ { o } ` ).join(" + ");this.userCode= `
void main ( ) {
$ { r . join ( `
` )}
float result = $ { s } ;
setOutput ( result ) ;
}
` }};var r8=class{constructor(t,n){this.outputShape=[],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=t,this.variableNames=n.map((o,a)=> ` T$ { a } ` );let r=[];this.variableNames.forEach(o=>{r.push( ` vec4 v$ { o } = get$ { o } AtOutCoords ( ) ; ` )});let s=this.variableNames.map(o=> ` v$ { o } ` ).join(" + ");this.userCode= `
void main ( ) {
$ { r . join ( `
` )}
vec4 result = $ { s } ;
setOutput ( result ) ;
}
` }};function DS(e){let{inputs:t,backend:n}=e,r=t;if(r.length===1)return lr({inputs:{x:r[0]},backend:n});if(r.length>he().get("WEBGL_MAX_TEXTURES_IN_SHADER")){let l=Math.floor(r.length/2),u=DS({inputs:r.slice(0,l),backend:n}),c=DS({inputs:r.slice(l),backend:n});return DS({inputs:[u,c],backend:n})}let s=r.map(l=>l.dtype).reduce((l,u)=>Or(l,u)),o=r.map(l=>l.shape),i=he().getBool("WEBGL_PACK")?new r8(r[0].shape,o):new n8(r[0].shape,o);return n.runWebGLProgram(i,r,s)}var Tre={kernelName:dp,backendName:"webgl",kernelFunc:DS};function gXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=C.parseAxisParam(o,s.shape),u=l,c=M.getAxesPermutation(u,i),p=s;c!=null&&(p=On({inputs:{x:s},backend:n,attrs:{perm:c}}),u=M.getInnerMostAxes(u.length,i)),M.assertAxesAreInnerMostDims("all",u,i);let[d,f]=M.computeOutAndReduceShapes(p.shape,u),h=C.sizeFromShape(f),m= $ e({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=la(m,m.dtype,"all",n),x;if(a){let y=M.expandShapeToKeepDim(d,l);x= $ e({inputs:{x:g},backend:n,attrs:{shape:y}})}else x= $ e({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var _re={kernelName:gx,backendName:"webgl",kernelFunc:gXe};function xXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=C.parseAxisParam(o,s.shape),u=l,c=M.getAxesPermutation(u,i),p=s;c!=null&&(p=On({inputs:{x:s},backend:n,attrs:{perm:c}}),u=M.getInnerMostAxes(u.length,i)),M.assertAxesAreInnerMostDims("any",u,i);let[d,f]=M.computeOutAndReduceShapes(p.shape,u),h=C.sizeFromShape(f),m= $ e({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=la(m,m.dtype,"any",n),x;if(a){let y=M.expandShapeToKeepDim(d,l);x= $ e({inputs:{x:g},backend:n,attrs:{shape:y}})}else x= $ e({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var Ere={kernelName:xx,backendName:"webgl",kernelFunc:xXe};var s8=class{constructor(t,n,r){this.variableNames=["A"];let{windowSize:s,batchSize:o,outSize:a}=t;r||this.variableNames.push("bestIndicesA"),this.outputShape=[o,a];let i=n==="max"?">":"<",l=r?"inOffset + i;":"round(getBestIndicesA(batch, inOffset + i));";this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = outIdx * $ { s } ;
int bestIndex = inOffset ;
float bestValue = getA ( batch , bestIndex ) ;
for ( int i = 0 ; i < $ { s } ; i ++ ) {
int inIdx = $ { l } ;
float candidate = getA ( batch , inIdx ) ;
if ( candidate $ { i } bestValue ) {
bestValue = candidate ;
bestIndex = inIdx ;
}
}
setOutput ( float ( bestIndex ) ) ;
}
` }};var o8=class{constructor(t,n,r,s){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,C.assert(t.length>2,()=> ` Packed arg$ { r . charAt ( 0 ) . toUpperCase ( ) + r . slice ( 1 ) } supports only inputs with rank above 2. ` );let o=t[t.length-1],a=Math.ceil(o/n);this.outputShape=t.slice(0,-1),a>1&&this.outputShape.push(a),s||this.variableNames.push("bestIndicesA");let i=this.outputShape,l=i.length,u=Ct(l),c=ir("coords",l),p,d;if(a===1){d=l+1;let I=Ct(d);p= `
$ { I } sourceLocR = $ { I } ( $ { c . join ( ) } , 0 ) ;
++ $ { c [ l - 1 ] } ;
$ { I } sourceLocG = $ { I } ( $ { c . join ( ) } , 0 ) ;
++ $ { c [ l - 2 ] } ;
$ { I } sourceLocA = $ { I } ( $ { c . join ( ) } , 0 ) ;
-- $ { c [ l - 1 ] } ;
$ { I } sourceLocB = $ { I } ( $ { c . join ( ) } , 0 ) ;
-- $ { c [ l - 2 ] } ; ` }else d=l,p= `
$ { u } sourceLocR = coords ;
++ $ { c [ l - 1 ] } ;
$ { u } sourceLocG = coords ;
++ $ { c [ l - 2 ] } ;
$ { u } sourceLocA = coords ;
-- $ { c [ l - 1 ] } ;
$ { u } sourceLocB = coords ;
-- $ { c [ l - 2 ] } ; ` ;let f=["x","y","z","w","u","v"].slice(0,d),h="."+f[d-1],m=f.map(I=>"int "+I),g=ir("sourceLocR",d-1).concat("inIdx.r"),x=ir("sourceLocG",d-1).concat("inIdx.g"),y=ir("sourceLocB",d-1).concat("inIdx.b"),b=ir("sourceLocA",d-1).concat("inIdx.a"),A=r==="max"?"greaterThan":"lessThan",v=s?"": `
inIdx = round ( vec4 ( getBestIndicesAChannel ( $ { g . join ( ) } ) ,
getBestIndicesAChannel ( $ { x . join ( ) } ) ,
getBestIndicesAChannel ( $ { y . join ( ) } ) ,
getBestIndicesAChannel ( $ { b . join ( ) } ) ) ) ; ` ,w= ` vec4 (
getAChannel ( $ { g . join ( ) } ) ,
hasNextCol ? getAChannel ( $ { x . join ( ) } ) : 0. ,
hasNextRow ? getAChannel ( $ { y . join ( ) } ) : 0. ,
hasNextRow && hasNextCol ? getAChannel ( $ { b . join ( ) } ) : 0. ) ` ,k=s?"": `
float getBestIndicesAChannel ( $ { m . join ( ) } ) {
return getChannel ( getBestIndicesA ( $ { f . join ( ) } ) ,
vec2 ( $ { f . slice ( - 2 ) . join ( ) } ) ) ;
} ` ;this.userCode= `
float getAChannel ( $ { m . join ( ) } ) {
return getChannel ( getA ( $ { f . join ( ) } ) ,
vec2 ( $ { f . slice ( - 2 ) . join ( ) } ) ) ;
}
$ { k }
void main ( ) {
$ { u } coords = getOutputCoords ( ) ;
bool hasNextCol = $ { c [ l - 1 ] } < $ { i [ l - 1 ] - 1 } ;
bool hasNextRow = $ { c [ l - 2 ] } < $ { i [ l - 2 ] - 1 } ;
$ { p }
ivec4 srcIdx = ivec4 ( sourceLocR$ { h } , sourceLocG$ { h } ,
sourceLocB$ { h } , sourceLocA$ { h } ) * $ { n } ;
ivec4 inIdx = srcIdx ;
vec4 bestIndex = vec4 ( inIdx ) ;
vec4 bestValue = $ { w } ;
for ( int i = 0 ; i < $ { n } ; i ++ ) {
inIdx = srcIdx ;
$ { v }
vec4 candidate = $ { w } ;
bvec4 nan = isnan ( candidate ) ;
bvec4 replace = bvec4 (
vec4 ( $ { A } ( candidate , bestValue ) ) * ( vec4 ( 1.0 ) - vec4 ( nan ) ) ) ;
bestValue = vec4 ( replace . x ? candidate . x : bestValue . x ,
replace . y ? candidate . y : bestValue . y ,
replace . z ? candidate . z : bestValue . z ,
replace . w ? candidate . w : bestValue . w ) ;
bestIndex = mix ( bestIndex , vec4 ( inIdx ) , vec4 ( replace ) ) ;
srcIdx ++ ;
}
setOutput ( bestIndex ) ;
}
` }};function Rre(e,t,n,r=null){let s=t.shape[0],o=t.shape[1];r!=null&&(s=r.shape[0],o=r.shape[1]);let a=M.computeOptimalWindowSize(o),i={windowSize:a,inSize:o,batchSize:s,outSize:Math.ceil(o/a)},l=new s8(i,n,r==null),u=[t];r!=null&&u.push(r);let c=e.runWebGLProgram(l,u,"int32");if(c.shape[1]===1)return c;let p=Rre(e,t,n,c);return e.disposeIntermediateTensorInfo(c),p}function Dre(e,t,n,r=null){let s=r!=null?r.shape:t.shape,o=s[s.length-1],a=M.computeOptimalWindowSize(o),i=new o8(s,a,n,r==null),l=r==null?[t]:[t,r],u=e.runWebGLProgram(i,l,"int32");if(u.shape.length===t.shape.length){let c=Dre(e,t,n,u);return e.disposeIntermediateTensorInfo(u),c}return u}function $ S(e,t,n,r){let s=[n];if(M.assertAxesAreInnerMostDims("arg"+r.charAt(0).toUpperCase()+r.slice(1),s,t.shape.length),!he().getBool("WEBGL_PACK_REDUCE")||t.shape.length<=2){let o=[],a=e.texData.get(t.dataId),i=a!==null&&a.isPacked,l=t;i&&(l=e.unpackTensor(t),o.push(l));let[u,c]=M.computeOutAndReduceShapes(l.shape,s),p=C.sizeFromShape(c),d= $ e({inputs:{x:l},backend:e,attrs:{shape:[-1,p]}});o.push(d);let f=Rre(e,d,r);o.push(f);let h= $ e({inputs:{x:f},backend:e,attrs:{shape:u}});return o.forEach(m=>e.disposeIntermediateTensorInfo(m)),h}return Dre(e,t,r)}function yXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o}=r,a=C.parseAxisParam(o,s.shape),i=M.getAxesPermutation(a,s.shape.length),l=s,u=[];i!=null&&(l=On({inputs:{x:s},backend:n,attrs:{perm:i}}),u.push(l),a=M.getInnerMostAxes(a.length,l.shape.length)),M.assertAxesAreInnerMostDims("argMax",[a[0]],l.shape.length);let c= $ S(n,l,a[0],"max");return u.forEach(p=>n.disposeIntermediateTensorInfo(p)),c}var $ re={kernelName:fp,backendName:"webgl",kernelFunc:yXe};function bXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o}=r,a=C.parseAxisParam(o,s.shape),i=M.getAxesPermutation(a,s.shape.length),l=s,u=[];i!=null&&(l=On({inputs:{x:s},backend:n,attrs:{perm:i}}),u.push(l),a=M.getInnerMostAxes(a.length,l.shape.length)),M.assertAxesAreInnerMostDims("argMin",[a[0]],l.shape.length);let c= $ S(n,l,a[0],"min");return u.forEach(p=>n.disposeIntermediateTensorInfo(p)),c}var Fre={kernelName:hp,backendName:"webgl",kernelFunc:bXe};var AXe=Qr+ `
if ( abs ( x ) > 1. ) {
return NAN ;
}
return asin ( x ) ;
` ,vXe=Xe({opSnippet:AXe}),Ore={kernelName:nu,backendName:"webgl",kernelFunc:vXe};var wXe=Qr+"return log(x + sqrt(x * x + 1.0));",kXe=Xe({opSnippet:wXe}),Pre={kernelName:ru,backendName:"webgl",kernelFunc:kXe};var IXe=Qr+ `
return atan ( x ) ;
` ,CXe=Xe({opSnippet:IXe}),Mre={kernelName:su,backendName:"webgl",kernelFunc:CXe};var SXe=dre+ `
return atan ( a , b ) ;
` ,NXe= `
vec4 result = atan ( a , b ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
` +fre+ `
return result ;
` ,TXe=Qt({opSnippet:SXe,packedOpSnippet:NXe}),zre={kernelName:au,backendName:"webgl",kernelFunc:TXe};var _Xe=Qr+ `
if ( ( x < - 1.0 ) || ( x > 1.0 ) ) return NAN ;
return ( log ( 1.0 + x ) - log ( 1.0 - x ) ) / 2.0 ; ` ,EXe=Xe({opSnippet:_Xe}),Lre={kernelName:ou,backendName:"webgl",kernelFunc:EXe};var Dl=class{constructor(t,n,r,s=!1,o=!1){if(this.variableNames=["x"],n==="avg"&&r)throw new Error("Cannot compute positions for average pool.");let a=t.filterWidth,i=t.strideHeight,l=t.strideWidth,u=t.dilationHeight,c=t.dilationWidth,p=t.effectiveFilterHeight,d=t.effectiveFilterWidth,f=t.padInfo.top,h=t.padInfo.left;this.outputShape=t.outShape;let m=n==="avg",g= ` ( ( batch * $ { t . inHeight } + xR ) * $ { t . inWidth } + xC ) * $ { t . inChannels } + d ` ,x= ` ( xR * $ { t . inWidth } + xC ) * $ { t . inChannels } + d ` ,y="0.0";if(m||(y="-1.0 / 1e-20"),r){let I=">=";this.userCode= `
const ivec2 strides = ivec2 ( $ { i } , $ { l } ) ;
const ivec2 pads = ivec2 ( $ { f } , $ { h } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
// max/min x(?, ?, d) to get y(yR, yC, d).
// ? = to be determined
float minMaxValue = 0.0 ;
float minMaxValueFound = 0.0 ;
int minMaxPosition = 0 ;
float avgValue = 0.0 ;
for ( int wR = 0 ; wR < $ { p } ;
wR += $ { u } ) {
int xR = xRCorner + wR ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { d } ;
wC += $ { c } ) {
int xC = xCCorner + wC ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
float value = getX ( batch , xR , xC , d ) ;
// If a min / max value has already been found, use it. If not,
// use the current value.
float currMinMaxValue = mix (
value , minMaxValue , minMaxValueFound ) ;
if ( value $ { I } currMinMaxValue ) {
minMaxValue = value ;
minMaxValueFound = 1.0 ;
minMaxPosition = $ { s ? o ? g : x : ` wR * ${ d } + wC ` } ;
}
}
}
setOutput ( float ( minMaxPosition ) ) ;
}
` ;return}let b="max",A= ` $ { n } ( $ { n } ( $ { n } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;n==="avg"&&(A="avgValue / count");let v=Math.floor(a/4)*4,w=a%4,k= `
if ( $ { m } ) {
avgValue += dot ( values , ones ) ;
} else {
minMaxValue = $ { b } ( values , minMaxValue ) ;
}
` ;this.userCode= `
const ivec2 strides = ivec2 ( $ { i } , $ { l } ) ;
const ivec2 pads = ivec2 ( $ { f } , $ { h } ) ;
const float initializationValue = $ { y } ;
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
float count = 0.0 ;
float getValue ( int batch , int xR , int xC , int d ) {
if ( xC < 0 || xC >= $ { t . inWidth } ) {
return initializationValue ;
}
count += 1.0 ;
return getX ( batch , xR , xC , d ) ;
}
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
// max/min x(?, ?, d) to get y(yR, yC, d).
// ? = to be determined
vec4 minMaxValue = vec4 ( $ { y } ) ;
float avgValue = 0.0 ;
count = 0.0 ;
for ( int wR = 0 ; wR < $ { p } ;
wR += $ { u } ) {
int xR = xRCorner + wR ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { v } ; wC += 4 ) {
int xC = xCCorner + wC * $ { c } ;
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { c } , d ) ,
getValue ( batch , xR , xC + 2 * $ { c } , d ) ,
getValue ( batch , xR , xC + 3 * $ { c } , d )
) ;
$ { k }
}
int xC = xCCorner + $ { v } ;
if ( $ { w === 1 } ) {
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
$ { k }
} else if ( $ { w === 2 } ) {
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { c } , d ) ,
initializationValue ,
initializationValue
) ;
$ { k }
} else if ( $ { w === 3 } ) {
vec4 values = vec4 (
getValue ( batch , xR , xC , d ) ,
getValue ( batch , xR , xC + $ { c } , d ) ,
getValue ( batch , xR , xC + 2 * $ { c } , d ) ,
initializationValue
) ;
$ { k }
}
}
setOutput ( $ { A } ) ;
}
` }},Um=class{constructor(t,n,r,s=!1,o=!1){if(this.variableNames=["x"],n==="avg"&&r)throw new Error("Cannot compute positions for average pool.");let a=t.filterWidth,i=t.strideDepth,l=t.strideHeight,u=t.strideWidth,c=t.dilationDepth,p=t.dilationHeight,d=t.dilationWidth,f=t.effectiveFilterDepth,h=t.effectiveFilterHeight,m=t.effectiveFilterWidth,g=t.padInfo.front,x=t.padInfo.top,y=t.padInfo.left;this.outputShape=t.outShape;let b=n==="avg",A="0.0";if(b||(A="-1.0 / 1e-20"),r){let T=">=";this.userCode= `
const ivec3 strides =
ivec3 ( $ { i } , $ { l } , $ { u } ) ;
const ivec3 pads = ivec3 ( $ { g } , $ { x } , $ { y } ) ;
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 xCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xDCorner = xCorner . x ;
int xRCorner = xCorner . y ;
int xCCorner = xCorner . z ;
// max/min x(?, ?, ?, ch) to get y(yD, yR, yC, ch).
// ? = to be determined
float minMaxValue = 0.0 ;
float minMaxValueFound = 0.0 ;
int minMaxPosition = 0 ;
for ( int wD = 0 ; wD < $ { f } ;
wD += $ { c } ) {
int xD = xDCorner + wD ;
if ( xD < 0 || xD >= $ { t . inDepth } ) {
continue ;
}
for ( int wR = 0 ; wR < $ { h } ;
wR += $ { p } ) {
int xR = xRCorner + wR ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { m } ;
wC += $ { d } ) {
int xC = xCCorner + wC ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
float value = getX ( batch , xD , xR , xC , ch ) ;
// If a min / max value has already been found, use it. If not,
// use the current value.
float currMinMaxValue = mix (
value , minMaxValue , minMaxValueFound ) ;
if ( value $ { T } currMinMaxValue ) {
minMaxValue = value ;
minMaxValueFound = 1.0 ;
minMaxPosition = $ { s ? o ? ` (((batch * ${ t . inDepth } + xD) * ${ t . inHeight } + xR) * ${ t . inWidth } + xC) * ${ t . inChannels } + ch ` : ` ((xD * ${ t . inHeight } + xR) * ${ t . inWidth } + xC) * ${ t . inChannels } + ch ` : ` wD * ${ h } * ${ m } +
wR * $ { m } + wC ` };
}
}
}
}
setOutput ( float ( minMaxPosition ) ) ;
}
` ;return}let v="max",w= ` $ { n } ( $ { n } ( $ { n } ( minMaxValue [ 0 ] , minMaxValue [ 1 ] ) , minMaxValue [ 2 ] ) , minMaxValue [ 3 ] ) ` ;n==="avg"&&(w="avgValue / count");let k=Math.floor(a/4)*4,I=a%4,N= `
if ( $ { b } ) {
avgValue += dot ( values , ones ) ;
} else {
minMaxValue = $ { v } ( values , minMaxValue ) ;
}
` ;this.userCode= `
const ivec3 strides =
ivec3 ( $ { i } , $ { l } , $ { u } ) ;
const ivec3 pads = ivec3 ( $ { g } , $ { x } , $ { y } ) ;
const float initializationValue = $ { A } ;
const vec4 ones = vec4 ( 1.0 , 1.0 , 1.0 , 1.0 ) ;
float count = 0.0 ;
float getValue ( int batch , int xD , int xR , int xC , int ch ) {
if ( xC < 0 || xC >= $ { t . inWidth } ) {
return initializationValue ;
}
count += 1.0 ;
return getX ( batch , xD , xR , xC , ch ) ;
}
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 xCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xDCorner = xCorner . x ;
int xRCorner = xCorner . y ;
int xCCorner = xCorner . z ;
// max/min x(?, ?, ?, d) to get y(yD, yR, yC, ch).
// ? = to be determined
vec4 minMaxValue = vec4 ( $ { A } ) ;
float avgValue = 0.0 ;
count = 0.0 ;
for ( int wD = 0 ; wD < $ { f } ;
wD += $ { c } ) {
int xD = xDCorner + wD ;
if ( xD < 0 || xD >= $ { t . inDepth } ) {
continue ;
}
for ( int wR = 0 ; wR < $ { h } ;
wR += $ { p } ) {
int xR = xRCorner + wR ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { k } ; wC += 4 ) {
int xC = xCCorner + wC * $ { d } ;
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
getValue ( batch , xD , xR , xC + $ { d } , ch ) ,
getValue ( batch , xD , xR , xC + 2 * $ { d } , ch ) ,
getValue ( batch , xD , xR , xC + 3 * $ { d } , ch )
) ;
$ { N }
}
int xC = xCCorner + $ { k } ;
if ( $ { I === 1 } ) {
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
$ { N }
} else if ( $ { I === 2 } ) {
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
getValue ( batch , xD , xR , xC + $ { d } , ch ) ,
initializationValue ,
initializationValue
) ;
$ { N }
} else if ( $ { I === 3 } ) {
vec4 values = vec4 (
getValue ( batch , xD , xR , xC , ch ) ,
getValue ( batch , xD , xR , xC + $ { d } , ch ) ,
getValue ( batch , xD , xR , xC + 2 * $ { d } , ch ) ,
initializationValue
) ;
$ { N }
}
}
setOutput ( $ { w } ) ;
}
}
` }};function RXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t;Rl(s,"avgPool");let{filterSize:o,strides:a,pad:i,dimRoundingMode:l}=r,u=1;C.assert(M.eitherStridesOrDilationsAreOne(a,u),()=> ` Error in avgPool : Either strides or dilations must be 1. Got strides $ { a } and dilations '${u}' ` );let c=M.computePool2DInfo(s.shape,o,a,u,i,l);if(c.filterWidth===1&&c.filterHeight===1&&C.arraysEqual(c.inShape,c.outShape))return lr({inputs:{x:s},backend:n});let p=new Dl(c,"avg",!1);return n.runWebGLProgram(p,[s],"float32")}var Bre={kernelName:mp,backendName:"webgl",kernelFunc:RXe};function DXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{filterSize:o,strides:a,pad:i,dimRoundingMode:l,dataFormat:u}=r,c=[1,1,1],p=M.computePool3DInfo(s.shape,o,a,c,i,l,u),d=new Um(p,"avg",!1);return n.runWebGLProgram(d,[s],"float32")}var Vre={kernelName:gp,backendName:"webgl",kernelFunc:DXe};var a8=class{constructor(t){this.variableNames=["dy"],this.outputShape=t.inShape;let n=t.filterHeight,r=t.filterWidth,s=t.strideHeight,o=t.strideWidth,a=t.dilationHeight,i=t.dilationWidth,l=t.effectiveFilterHeight,u=t.effectiveFilterWidth,c=l-1-t.padInfo.top,p=u-1-t.padInfo.left,d=1/(n*r);this.userCode= `
const ivec2 pads = ivec2 ( $ { c } , $ { p } ) ;
const float avgMultiplier = float ( $ { d } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 dyRCCorner = coords . yz - pads ;
int dyRCorner = dyRCCorner . x ;
int dyCCorner = dyRCCorner . y ;
// Convolve dy(?, ?, d) with pos mask(:, :, d) to get dx(xR, xC, d).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { l } ;
wR += $ { a } ) {
float dyR = float ( dyRCorner + wR ) / $ { s } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
for ( int wC = 0 ; wC < $ { u } ;
wC += $ { i } ) {
float dyC = float ( dyCCorner + wC ) / $ { o } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( b , idyR , idyC , d ) ;
dotProd += dyValue * avgMultiplier ;
}
}
setOutput ( dotProd ) ;
}
` }},i8=class{constructor(t){this.variableNames=["dy"],this.outputShape=t.inShape;let n=t.filterDepth,r=t.filterHeight,s=t.filterWidth,o=t.strideDepth,a=t.strideHeight,i=t.strideWidth,l=t.dilationDepth,u=t.dilationHeight,c=t.dilationWidth,p=t.effectiveFilterDepth,d=t.effectiveFilterHeight,f=t.effectiveFilterWidth,h=p-1-t.padInfo.front,m=d-1-t.padInfo.top,g=f-1-t.padInfo.left,x=1/(n*r*s);this.userCode= `
const ivec3 pads = ivec3 ( $ { h } , $ { m } , $ { g } ) ;
const float avgMultiplier = float ( $ { x } ) ;
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyDCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
// Convolve dy(?, ?, ?, d) with pos mask(:, :, :, ch) to get
// dx(xD, xR, xC, ch).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wD = 0 ; wD < $ { p } ;
wD += $ { l } ) {
float dyD = float ( dyDCorner + wD ) / $ { o } . 0 ;
if ( dyD < 0.0 || dyD >= $ { t . outDepth } . 0 || fract ( dyD ) > 0.0 ) {
continue ;
}
int idyD = int ( dyD ) ;
for ( int wR = 0 ; wR < $ { d } ;
wR += $ { u } ) {
float dyR = float ( dyRCorner + wR ) / $ { a } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
for ( int wC = 0 ; wC < $ { f } ;
wC += $ { c } ) {
float dyC = float ( dyCCorner + wC ) / $ { i } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( batch , idyD , idyR , idyC , ch ) ;
dotProd += dyValue * avgMultiplier ;
}
}
}
setOutput ( dotProd ) ;
}
` }};function $ Xe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o,{filterSize:i,strides:l,pad:u,dimRoundingMode:c}=r,p=[1,1,1],d=M.computePool3DInfo(a.shape,i,l,p,u,c),f=new i8(d);return n.runWebGLProgram(f,[s],a.dtype)}var Wre={kernelName:bx,backendName:"webgl",kernelFunc: $ Xe};function FXe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o;Rl([s,o],"avgPoolGrad");let{filterSize:i,strides:l,pad:u}=r,c=M.computePool2DInfo(a.shape,i,l,1,u),p=new a8(c);return n.runWebGLProgram(p,[s],a.dtype)}var Gre={kernelName:yx,backendName:"webgl",kernelFunc:FXe};function OXe(e){let{inputs:t,backend:n,attrs:r}=e,{a:s,b:o}=t,{transposeA:a,transposeB:i}=r;return Gm({a:s,b:o,transposeA:a,transposeB:i,backend:n})}var Ure={kernelName:xp,backendName:"webgl",kernelFunc:OXe};var l8=class{constructor(t,n,r,s,o,a){this.outputShape=[],this.variableNames=["x","mean","variance"],M.assertAndGetBroadcastShape(t,n),M.assertAndGetBroadcastShape(t,r);let i="0.0";s!=null&&(M.assertAndGetBroadcastShape(t,s),this.variableNames.push("offset"),i="getOffsetAtOutCoords()");let l="1.0";o!=null&&(M.assertAndGetBroadcastShape(t,o),this.variableNames.push("scale"),l="getScaleAtOutCoords()"),this.outputShape=t,this.userCode= `
void main ( ) {
float x = getXAtOutCoords ( ) ;
float mean = getMeanAtOutCoords ( ) ;
float variance = getVarianceAtOutCoords ( ) ;
float offset = $ { i } ;
float scale = $ { l } ;
float inv = scale * inversesqrt ( variance + float ( $ { a } ) ) ;
setOutput ( dot ( vec3 ( x , - mean , offset ) , vec3 ( inv , inv , 1 ) ) ) ;
}
` }};var u8=class{constructor(t,n,r,s,o,a){this.packedInputs=!0,this.packedOutput=!0,this.variableNames=["x","mean","variance"],M.assertAndGetBroadcastShape(t,n),M.assertAndGetBroadcastShape(t,r);let i="vec4(0.0)";s!=null&&(M.assertAndGetBroadcastShape(t,s),this.variableNames.push("offset"),i="getOffsetAtOutCoords()");let l="vec4(1.0)";o!=null&&(M.assertAndGetBroadcastShape(t,o),this.variableNames.push("scale"),l="getScaleAtOutCoords()"),this.outputShape=t,this.userCode= `
void main ( ) {
vec4 offset = $ { i } ;
vec4 scale = $ { l } ;
vec4 x = getXAtOutCoords ( ) ;
vec4 mean = getMeanAtOutCoords ( ) ;
vec4 variance = getVarianceAtOutCoords ( ) ;
vec4 inv = scale * inversesqrt ( variance + vec4 ( $ { a } ) ) ;
setOutput ( ( x - mean ) * inv + offset ) ;
}
` }};var PXe=({inputs:e,backend:t,attrs:n})=>{let{x:r,mean:s,variance:o,offset:a,scale:i}=e;C.assert(s.shape.length===o.shape.length,()=>"Batch normalization gradient requires mean and variance to have equal ranks."),C.assert(a==null||s.shape.length===a.shape.length,()=>"Batch normalization gradient requires mean and offset to have equal ranks."),C.assert(i==null||s.shape.length===i.shape.length,()=>"Batch normalization gradient requires mean and scale to have equal ranks.");let{varianceEpsilon:l}=n;l==null&&(l=.001);let u=[r,s,o],c=null;a!=null&&(c=a.shape,u.push(a));let p=null;i!=null&&(p=i.shape,u.push(i));let d=he().getBool("WEBGL_PACK_NORMALIZATION")?new u8(r.shape,s.shape,o.shape,c,p,l):new l8(r.shape,s.shape,o.shape,c,p,l);return t.runWebGLProgram(d,u,u[0].dtype)},jre={kernelName:Tp,backendName:"webgl",kernelFunc:PXe};var c8=class{constructor(t){this.variableNames=["source"],this.outputShape=t,this.rank=t.length;let n=Ct(this.rank);this.customUniforms=[{name:"start",arrayIndex:this.rank,type:"int"}];let r=MXe(this.rank),s,o=t.map((a,i)=> ` sourceLoc . $ { p8 [ i ] } = start [ $ { i } ] + coords . $ { p8 [ i ] } ; ` );s= `
$ { n } sourceLoc ;
$ { n } coords = getOutputCoords ( ) ;
$ { o . join ( `
` )}
` ,this.userCode= `
void main ( ) {
$ { s }
setOutput ( getSource ( $ { r } ) ) ;
}
` }},p8=["x","y","z","w","u","v"];function MXe(e){if(e===1)return"sourceLoc";if(e<=6)return p8.slice(0,e).map(t=>"sourceLoc."+t).join(",");throw Error( ` Slicing for rank $ { e } is not yet supported ` )}var d8=class{constructor(t){this.variableNames=["source"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=t,this.rank=t.length,this.customUniforms=[{name:"start",arrayIndex:this.rank,type:"int"}];let n=Ct(this.rank),r=ir("coords",this.rank),s=ir("sourceLoc",this.rank),o=this.rank===1?"sourceLoc": ` vec2 ( $ { s . slice ( - 2 ) . join ( ) } ) ` ,a= ` getChannel ( getSource ( $ { s . join ( ) } ) , $ { o } ) ` ,i= `
result . x = $ { a } ;
if ( ++ $ { r [ this . rank - 1 ] } < $ { t [ this . rank - 1 ] } ) {
++ $ { s [ this . rank - 1 ] } ;
result . y = $ { a } ;
-- $ { s [ this . rank - 1 ] } ;
}
` ,l=this.rank===1?"": `
-- $ { r [ this . rank - 1 ] } ;
if ( ++ $ { r [ this . rank - 2 ] } < $ { t [ this . rank - 2 ] } ) {
++ $ { s [ this . rank - 2 ] } ;
result . z = $ { a } ;
if ( ++ $ { r [ this . rank - 1 ] } < $ { t [ this . rank - 1 ] } ) {
++ $ { s [ this . rank - 1 ] } ;
result . w = $ { a } ;
}
}
` ,u=this.rank<=4? ` sourceLoc = coords +
$ { n } ( $ { t . map ( ( c , p ) => ` start[ ${ p } ] ` ) . join ( ) } ) ; ` :t.map((c,p)=> ` $ { s [ p ] } = $ { r [ p ] } + start [ $ { p } ] ; ` ).join( `
` );this.userCode= `
void main ( ) {
$ { n } coords = getOutputCoords ( ) ;
$ { n } sourceLoc ;
$ { u }
vec4 result = vec4 ( 0. ) ;
$ { i }
$ { l }
setOutput ( result ) ;
}
` }};function zXe(e,t,n,r){let s=r.texData.get(e.dataId),o=r.makeTensorInfo(n,e.dtype),a=r.texData.get(o.dataId);Object.assign(a,s),a.refCount=1,a.shape=n,a.dtype=e.dtype;let i=Cs.computeFlatOffset(t,C.computeStrides(e.shape));s.slice&&(i+=s.slice.flatOffset),a.slice={flatOffset:i,origDataId:s.slice&&s.slice.origDataId||e.dataId};let l=r.dataRefCount.get(a.slice.origDataId)||1;return r.dataRefCount.set(a.slice.origDataId,l+1),o}function el(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,size:a}=r,[i,l]=Cs.parseSliceParams(s,o,a);if(Cs.assertParamsValid(s,i,l),C.sizeFromShape(l)===0)return n.makeTensorInfo(l,s.dtype,[]);if(n.shouldExecuteOnCPU([s])||s.dtype==="string"){let p=n.texData.get(s.dataId),d=Pne(p.values,i,l,s.shape,s.dtype);return n.makeTensorInfo(l,s.dtype,d)}let{isPacked:u}=n.texData.get(s.dataId),c=Cs.isSliceContinous(s.shape,i,l);if(u||!c){let p=he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new d8(l):new c8(l),d=[i];return n.runWebGLProgram(p,[s],s.dtype,d)}return n.uploadToGPU(s.dataId),zXe(s,i,l,n)}var Hre={kernelName:Xp,backendName:"webgl",kernelFunc:el};var LXe=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockShape:o,crops:a}=r;C.assert(s.shape.length<=4,()=>"batchToSpaceND for rank > 4 with a WebGL backend not implemented yet");let i=o.reduce((y,b)=>y*b),l=M.getReshaped(s.shape,o,i),u=M.getPermuted(l.length,o.length),c=M.getReshapedPermuted(s.shape,o,i),p=M.getSliceBeginCoords(a,o.length),d=M.getSliceSize(c,a,o.length),f=[],h= $ e({inputs:{x:s},backend:n,attrs:{shape:l}}),m=On({inputs:{x:h},backend:n,attrs:{perm:u}}),g= $ e({inputs:{x:m},backend:n,attrs:{shape:c}}),x=el({inputs:{x:g},backend:n,attrs:{begin:p,size:d}});return f.push(h),f.push(m),f.push(g),f.forEach(y=>n.disposeIntermediateTensorInfo(y)),x},qre={kernelName:yp,backendName:"webgl",kernelFunc:LXe};function BXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,weights:o}=t,{size:a}=r,i=n.readSync(s.dataId),l=n.readSync(o.dataId),u=IS(i,l,o.dtype,o.shape,a);return n.makeTensorInfo([a],o.dtype,u)}var Kre={kernelName:Ax,backendName:"webgl",kernelFunc:BXe};var VXe="return float(a != b);",f8=Qt({opSnippet:VXe,cpuKernelImpl:Rne,dtype:"bool"}),Xre={kernelName:Dh,backendName:"webgl",kernelFunc:f8};function Ac(e){let{inputs:t,backend:n}=e,{input:r}=t,s=n.texData.get(r.dataId);return lr({inputs:{x:s.complexTensorInfos.real},backend:n})}var Zre={kernelName:Xx,backendName:"webgl",kernelFunc:Ac};var WXe="return float(int(x));";function Yre(e,t){let n=new $ o(e.shape,WXe),r=t.runWebGLProgram(n,[e],"int32");return{dataId:r.dataId,shape:r.shape,dtype:r.dtype}}function h8(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{dtype:o}=r;if(o==="complex64"){if(s.dtype==="complex64")return lr({inputs:{x:s},backend:n});let a=fn(s.shape),i=h8({inputs:{x:s},backend:n,attrs:{dtype:"float32"}}),l=Fo({inputs:{real:i,imag:a},backend:n});return a.dispose(),n.disposeIntermediateTensorInfo(i),l}if(s.dtype==="complex64"){let a=Ac({inputs:{input:s},backend:n}),i=h8({inputs:{x:a},backend:n,attrs:{dtype:o}});return n.disposeIntermediateTensorInfo(a),i}if(!C.hasEncodingLoss(s.dtype,o)){let a=lr({inputs:{x:s},backend:n});return{dataId:a.dataId,shape:a.shape,dtype:o}}if(o==="int32")return Yre(s,n);if(o==="bool"){let a=n.makeTensorInfo([],"bool",C.getTypedArrayFromDType("bool",1)),l=f8({inputs:{a:s,b:a},backend:n});return n.disposeIntermediateTensorInfo(a),l}throw new Error( ` Error in Cast : failed to cast $ { s . dtype } to $ { o } ` )}var Jre={kernelName:bl,backendName:"webgl",kernelFunc:h8};var Qre="return ceil(x);",GXe=Xe({opSnippet:Qre,packedOpSnippet:Qre,cpuKernelImpl:dne}),ese={kernelName:iu,backendName:"webgl",kernelFunc:GXe};var m8=class{constructor(t){this.variableNames=["A"],this.customUniforms=[{name:"minVal",type:"float"},{name:"maxVal",type:"float"}],this.outputShape=t,this.userCode= `
void main ( ) {
float value = getAAtOutCoords ( ) ;
if ( isnan ( value ) ) {
setOutput ( value ) ;
return ;
}
setOutput ( clamp ( value , minVal , maxVal ) ) ;
}
` }};var g8=class{constructor(t){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"minVal",type:"float"},{name:"maxVal",type:"float"}],this.outputShape=t,this.userCode= `
void main ( ) {
vec4 value = getAAtOutCoords ( ) ;
if ( any ( isnan ( value ) ) ) {
setOutput ( value ) ;
return ;
}
setOutput ( clamp ( value , vec4 ( minVal ) , vec4 ( maxVal ) ) ) ;
}
` }};function UXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{clipValueMin:o,clipValueMax:a}=r,i;he().getBool("WEBGL_PACK_CLIP")?i=new g8(s.shape):i=new m8(s.shape);let l=[[o],[a]];return n.runWebGLProgram(i,[s],s.dtype,l)}var tse={kernelName:lu,backendName:"webgl",kernelFunc:UXe};var x8=class{constructor(t){this.variableNames=["real","imag"],this.outputShape=t,this.userCode= `
void main ( ) {
float re = abs ( getRealAtOutCoords ( ) ) ;
float im = abs ( getImagAtOutCoords ( ) ) ;
float mx = max ( re , im ) ;
// sadly the length function in glsl is not underflow-safe
// (at least not on Intel GPUs). So the safe solution is
// to ensure underflow-safety in all cases.
setOutput (
mx == 0.0 ? 0.0 : mx * length ( vec2 ( 1 , min ( re , im ) / mx ) )
) ;
}
` }};function nse(e,t){return{dataId:t.dataId,dtype:t.dtype,shape:e.shape}}function jXe(e){let{inputs:t,backend:n}=e,{x:r}=t,s=n.texData.get(r.dataId),o=new x8(r.shape),a=[nse(r,s.complexTensorInfos.real),nse(r,s.complexTensorInfos.imag)];return n.runWebGLProgram(o,a,a[0].dtype)}var rse={kernelName:bp,backendName:"webgl",kernelFunc:jXe};var y8=class{constructor(t){this.outputShape=[],this.outputShape=M.computeOutShape(t,1),this.variableNames=t.map((a,i)=> ` T$ { i } ` );let n=new Array(t.length-1);n[0]=t[0][1];for(let a=1;a<n.length;a++)n[a]=n[a-1]+t[a][1];let r=[ ` if ( yC < $ { n [ 0 ] } ) setOutput ( getT0 ( yR , yC ) ) ; ` ];for(let a=1;a<n.length;a++){let i=n[a-1];r.push( ` else if ( yC < $ { n [ a ] } ) setOutput ( getT$ { a } ( yR , yC - $ { i } ) ) ; ` )}let s=n.length,o=n[n.length-1];r.push( ` else setOutput ( getT$ { s } ( yR , yC - $ { o } ) ) ; ` ),this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int yR = coords . x ;
int yC = coords . y ;
$ { r . join ( `
` )}
}
` }};var b8=class{constructor(t,n){this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[],this.outputShape=M.computeOutShape(t,n);let r=this.outputShape,s=r.length,o=Ct(s),a=ir("coords",s),i=["x","y","z","w","u","v"].slice(0,s);this.variableNames=t.map((m,g)=> ` T$ { g } ` );let l=new Array(t.length-1);l[0]=t[0][n];for(let m=1;m<l.length;m++)l[m]=l[m-1]+t[m][n];let u=i[n],c=i.slice(-2),p=i.join(),d= ` if ( $ { u } < $ { l [ 0 ] } ) {
return getChannel (
getT0 ( $ { p } ) , vec2 ( $ { c . join ( ) } ) ) ;
} ` ;for(let m=1;m<l.length;m++){let g=l[m-1];d+= `
if ( $ { u } < $ { l [ m ] } && $ { u } >= $ { l [ m - 1 ] } ) {
return getChannel (
getT$ { m } ( $ { FS ( i , u , g ) } ) ,
vec2 ( $ { FS ( c , u , g ) } ) ) ;
} ` }let f=l.length,h=l[l.length-1];d+= `
return getChannel (
getT$ { f } ( $ { FS ( i , u , h ) } ) ,
vec2 ( $ { FS ( c , u , h ) } ) ) ; ` ,this.userCode= `
float getValue ( $ { i . map ( m => "int " + m ) } ) {
$ { d }
}
void main ( ) {
$ { o } coords = getOutputCoords ( ) ;
vec4 result = vec4 ( getValue ( $ { a } ) , 0. , 0. , 0. ) ;
$ { a [ s - 1 ] } = $ { a [ s - 1 ] } + 1 ;
if ( $ { a [ s - 1 ] } < $ { r [ s - 1 ] } ) {
result . g = getValue ( $ { a } ) ;
}
$ { a [ s - 2 ] } = $ { a [ s - 2 ] } + 1 ;
if ( $ { a [ s - 2 ] } < $ { r [ s - 2 ] } ) {
result . a = getValue ( $ { a } ) ;
}
$ { a [ s - 1 ] } = $ { a [ s - 1 ] } - 1 ;
if ( $ { a [ s - 2 ] } < $ { r [ s - 2 ] } &&
$ { a [ s - 1 ] } < $ { r [ s - 1 ] } ) {
result . b = getValue ( $ { a } ) ;
}
setOutput ( result ) ;
}
` }};function FS(e,t,n){let r=e.indexOf(t);return e.map((o,a)=>a===r? ` $ { o } - $ { n } ` :o).join()}function jm(e){let{inputs:t,backend:n}=e,{input:r}=t,s=n.texData.get(r.dataId);return lr({inputs:{x:s.complexTensorInfos.imag},backend:n})}var sse={kernelName:Mx,backendName:"webgl",kernelFunc:jm};function Hm(e,t,n){let r=e[0].dtype;if(r==="complex64"){let c=e.map(m=>Ac({inputs:{input:m},backend:n})),p=e.map(m=>jm({inputs:{input:m},backend:n})),d=Hm(c,t,n),f=Hm(p,t,n),h=Fo({inputs:{real:d,imag:f},backend:n});return c.forEach(m=>n.disposeIntermediateTensorInfo(m)),p.forEach(m=>n.disposeIntermediateTensorInfo(m)),n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}let s=n.shouldExecuteOnCPU(e);if(r==="string"&&(s=!0),s){let c=e.map(x=>{let y=C.sizeFromShape(x.shape.slice(t));return $ e({inputs:{x},backend:n,attrs:{shape:[-1,y]}})}),p=c.map(x=>({vals:n.readSync(x.dataId),shape:x.shape})),d=M.computeOutShape(c.map(x=>x.shape),1),f=c[0].shape[0]===1,h=fne(p,d,r,f),m=M.computeOutShape(e.map(x=>x.shape),t),g=n.makeTensorInfo(m,r,h);return c.forEach(x=>n.disposeIntermediateTensorInfo(x)),g}if(e.length>he().getNumber("WEBGL_MAX_TEXTURES_IN_SHADER")){let c=Math.floor(e.length/2),p=Hm(e.slice(0,c),t,n),d=Hm(e.slice(c),t,n),f=Hm([p,d],t,n);return n.disposeIntermediateTensorInfo(p),n.disposeIntermediateTensorInfo(d),f}if(he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")&&e[0].shape.length>1){let c=new b8(e.map(p=>p.shape),t);return n.runWebGLProgram(c,e,r)}let{tensors2D:o,outShape:a}=HXe(e,t,n),i=new y8(o.map(c=>c.shape)),l=n.runWebGLProgram(i,o,r);o.forEach(c=>n.disposeIntermediateTensorInfo(c));let u= $ e({inputs:{x:l},attrs:{shape:a},backend:n});return n.disposeIntermediateTensorInfo(l),u}function HXe(e,t,n){let r=M.computeOutShape(e.map(o=>o.shape),t);return{tensors2D:e.map(o=> $ e({inputs:{x:o},attrs:{shape:[-1,C.sizeFromShape(o.shape.slice(t))]},backend:n})),outShape:r}}function A8(e){let{inputs:t,backend:n,attrs:r}=e,{axis:s}=r,o=C.parseAxisParam(s,t[0].shape)[0],a=M.computeOutShape(t.map(u=>u.shape),o);if(C.sizeFromShape(a)===0)return n.makeTensorInfo(a,t[0].dtype,[]);let i=t.filter(u=>C.sizeFromShape(u.shape)>0);if(i.length===1)return lr({inputs:{x:i[0]},backend:n});let l=i.map(u=>u.shape);return M.assertParamsConsistent(l,o),Hm(i,o,n)}var ose={kernelName:Ap,backendName:"webgl",kernelFunc:A8};var xv=class{constructor(t,n=!1,r=null,s=!1,o=!1){this.variableNames=["x","W"],this.outputShape=t.outShape;let a=t.padInfo.top,i=t.padInfo.left,l=t.strideHeight,u=t.strideWidth,c=t.dilationHeight,p=t.dilationWidth,d=t.filterHeight,f=t.filterWidth,h=Math.floor(t.inChannels/4)*4,m=t.inChannels%4,g=t.dataFormat==="channelsLast",x=g?1:2,y=g?2:3,b=g?3:1,A="",v="";r&&(s?A= ` float activation ( float a ) {
float b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { r }
} ` :o?A= ` float activation ( float a ) {
float b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { r }
} ` :A= `
float activation ( float x ) {
$ { r }
}
` ,v="result = activation(result);");let w=n?"result += getBiasAtOutCoords();":"";n&&this.variableNames.push("bias"),s&&this.variableNames.push("preluActivationWeights"),o&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
$ { A }
const ivec2 strides = ivec2 ( $ { l } , $ { u } ) ;
const ivec2 pads = ivec2 ( $ { a } , $ { i } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d2 = coords [ $ { b } ] ;
ivec2 xRCCorner =
ivec2 ( coords [ $ { x } ] , coords [ $ { y } ] ) * strides - pads ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
// Convolve x(?, ?, d1) with w(:, :, d1, d2) to get y(yR, yC, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { d } ; wR ++ ) {
int xR = xRCorner + wR * $ { c } ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { f } ; wC ++ ) {
int xC = xCCorner + wC * $ { p } ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
for ( int d1 = 0 ; d1 < $ { h } ; d1 += 4 ) {
vec4 wValues = vec4 (
getW ( wR , wC , d1 , d2 ) ,
getW ( wR , wC , d1 + 1 , d2 ) ,
getW ( wR , wC , d1 + 2 , d2 ) ,
getW ( wR , wC , d1 + 3 , d2 )
) ;
if ( $ { g } ) {
vec4 xValues = vec4 (
getX ( batch , xR , xC , d1 ) ,
getX ( batch , xR , xC , d1 + 1 ) ,
getX ( batch , xR , xC , d1 + 2 ) ,
getX ( batch , xR , xC , d1 + 3 )
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec4 xValues = vec4 (
getX ( batch , d1 , xR , xC ) ,
getX ( batch , d1 + 1 , xR , xC ) ,
getX ( batch , d1 + 2 , xR , xC ) ,
getX ( batch , d1 + 3 , xR , xC )
) ;
dotProd += dot ( xValues , wValues ) ;
}
}
if ( $ { m === 1 } ) {
if ( $ { g } ) {
dotProd +=
getX ( batch , xR , xC , $ { h } ) *
getW ( wR , wC , $ { h } , d2 ) ;
} else {
dotProd +=
getX ( batch , $ { h } , xR , xC ) *
getW ( wR , wC , $ { h } , d2 ) ;
}
} else if ( $ { m === 2 } ) {
vec2 wValues = vec2 (
getW ( wR , wC , $ { h } , d2 ) ,
getW ( wR , wC , $ { h } + 1 , d2 )
) ;
if ( $ { g } ) {
vec2 xValues = vec2 (
getX ( batch , xR , xC , $ { h } ) ,
getX ( batch , xR , xC , $ { h } + 1 )
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec2 xValues = vec2 (
getX ( batch , $ { h } , xR , xC ) ,
getX ( batch , $ { h } + 1 , xR , xC )
) ;
dotProd += dot ( xValues , wValues ) ;
}
} else if ( $ { m === 3 } ) {
vec3 wValues = vec3 (
getW ( wR , wC , $ { h } , d2 ) ,
getW ( wR , wC , $ { h } + 1 , d2 ) ,
getW ( wR , wC , $ { h } + 2 , d2 )
) ;
if ( $ { g } ) {
vec3 xValues = vec3 (
getX ( batch , xR , xC , $ { h } ) ,
getX ( batch , xR , xC , $ { h } + 1 ) ,
getX ( batch , xR , xC , $ { h } + 2 )
) ;
dotProd += dot ( xValues , wValues ) ;
} else {
vec3 xValues = vec3 (
getX ( batch , $ { h } , xR , xC ) ,
getX ( batch , $ { h } + 1 , xR , xC ) ,
getX ( batch , $ { h } + 2 , xR , xC )
) ;
dotProd += dot ( xValues , wValues ) ;
}
}
}
}
float result = dotProd ;
$ { w }
$ { v }
setOutput ( result ) ;
}
` }},v8=class{constructor(t){this.variableNames=["x","W"],this.outputShape=t.outShape;let n=t.padInfo.front,r=t.padInfo.top,s=t.padInfo.left,o=t.strideDepth,a=t.strideHeight,i=t.strideWidth,l=t.dilationDepth,u=t.dilationHeight,c=t.dilationWidth,p=t.filterDepth,d=t.filterHeight,f=t.filterWidth,h=Math.floor(t.inChannels/4)*4,m=t.inChannels%4;this.userCode= `
const ivec3 strides = ivec3 ( $ { o } , $ { a } , $ { i } ) ;
const ivec3 pads = ivec3 ( $ { n } , $ { r } , $ { s } ) ;
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d2 = coords . u ;
ivec3 xFRCCorner = ivec3 ( coords . y , coords . z , coords . w ) * strides - pads ;
int xFCorner = xFRCCorner . x ;
int xRCorner = xFRCCorner . y ;
int xCCorner = xFRCCorner . z ;
// Convolve x(?, ?, ?, d1) with w(:, :, :, d1, d2) to get
// y(yF, yR, yC, d2). ? = to be determined. : = across all
// values in that axis.
float dotProd = 0.0 ;
for ( int wF = 0 ; wF < $ { p } ; wF ++ ) {
int xF = xFCorner + wF * $ { l } ;
if ( xF < 0 || xF >= $ { t . inDepth } ) {
continue ;
}
for ( int wR = 0 ; wR < $ { d } ; wR ++ ) {
int xR = xRCorner + wR * $ { u } ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int wC = 0 ; wC < $ { f } ; wC ++ ) {
int xC = xCCorner + wC * $ { c } ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
for ( int d1 = 0 ; d1 < $ { h } ; d1 += 4 ) {
vec4 xValues = vec4 (
getX ( batch , xF , xR , xC , d1 ) ,
getX ( batch , xF , xR , xC , d1 + 1 ) ,
getX ( batch , xF , xR , xC , d1 + 2 ) ,
getX ( batch , xF , xR , xC , d1 + 3 )
) ;
vec4 wValues = vec4 (
getW ( wF , wR , wC , d1 , d2 ) ,
getW ( wF , wR , wC , d1 + 1 , d2 ) ,
getW ( wF , wR , wC , d1 + 2 , d2 ) ,
getW ( wF , wR , wC , d1 + 3 , d2 )
) ;
dotProd += dot ( xValues , wValues ) ;
}
if ( $ { m === 1 } ) {
dotProd +=
getX ( batch , xF , xR , xC , $ { h } ) *
getW ( wF , wR , wC , $ { h } , d2 ) ;
} else if ( $ { m === 2 } ) {
vec2 xValues = vec2 (
getX ( batch , xF , xR , xC , $ { h } ) ,
getX ( batch , xF , xR , xC , $ { h } + 1 )
) ;
vec2 wValues = vec2 (
getW ( wF , wR , wC , $ { h } , d2 ) ,
getW ( wF , wR , wC , $ { h } + 1 , d2 )
) ;
dotProd += dot ( xValues , wValues ) ;
} else if ( $ { m === 3 } ) {
vec3 xValues = vec3 (
getX ( batch , xF , xR , xC , $ { h } ) ,
getX ( batch , xF , xR , xC , $ { h } + 1 ) ,
getX ( batch , xF , xR , xC , $ { h } + 2 )
) ;
vec3 wValues = vec3 (
getW ( wF , wR , wC , $ { h } , d2 ) ,
getW ( wF , wR , wC , $ { h } + 1 , d2 ) ,
getW ( wF , wR , wC , $ { h } + 2 , d2 )
) ;
dotProd += dot ( xValues , wValues ) ;
}
}
}
}
setOutput ( dotProd ) ;
}
` }};var w8=class{constructor(t,n){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"inputShape",type:"ivec3"},{name:"pad",type:"ivec2"},{name:"stride",type:"ivec2"},{name:"dilation",type:"ivec2"},{name:"inChannels",type:"int"},{name:"itemsPerBlockRow",type:"int"},{name:"outWidth",type:"int"}],this.outputShape=t,this.enableShapeUniforms=Xn(this.outputShape.length);let{dataFormat:r}=n,s=Vn(),o=r==="channelsLast",a=o?0:1,i=o?1:2,l=this.enableShapeUniforms?"if(blockIndex < outShape[1] && pos < outShape[0]) {": ` if ( blockIndex < $ { t [ 1 ] } && pos < $ { t [ 0 ] } ) { ` ,u="";for(let c=0;c<=1;c++)for(let p=0;p<=1;p++)u+= `
blockIndex = rc . y + $ { p } ;
pos = rc . x + $ { c } ;
$ { l }
offsetY = int ( blockIndex / outWidth ) * stride [ 0 ] - pad [ 0 ] ;
d0 = offsetY + dilation [ 0 ] * ( pos / itemsPerBlockRow ) ;
if ( d0 < inputShape [ $ { a } ] && d0 >= 0 ) {
// Use custom imod instead mod. On Intel GPU, mod may generate
// unexpected value.
// https://github.com/tensorflow/tfjs/issues/5447
offsetX = imod ( blockIndex , outWidth ) * stride [ 1 ] - pad [ 1 ] ;
d1 = offsetX + dilation [ 1 ] * ( imod ( pos , itemsPerBlockRow ) /
inChannels ) ;
if ( d1 < inputShape [ $ { i } ] && d1 >= 0 ) {
ch = imod ( pos , inChannels ) ;
if ( $ { o } ) {
innerDims = vec2 ( d1 , ch ) ;
result [ $ { c * 2 + p } ] = getChannel (
getA ( d0 , int ( innerDims . x ) ,
int ( innerDims . y ) ) , innerDims ) ;
} else {
innerDims = vec2 ( d0 , d1 ) ;
result [ $ { c * 2 + p } ] = getChannel (
getA ( ch , int ( innerDims . x ) ,
int ( innerDims . y ) ) , innerDims ) ;
}
}
}
}
` ;this.userCode= `
void main ( ) {
ivec2 rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0 ) ;
int blockIndex , pos , offsetY , d0 , offsetX , d1 , ch ;
vec2 innerDims ;
$ { u }
$ { s . output } = result ;
}
` }};function OS({x:e,filter:t,convInfo:n,backend:r,bias:s=null,preluActivationWeights:o=null,leakyreluAlpha:a=0,activation:i=null}){let l=e.shape,u=r.texData.get(e.dataId),c=n.inChannels,p=l[0]*l[1]*l[2],d=n.outChannels,f=n.dataFormat==="channelsLast",h=!1,m=!1,g,x=[];if(!((p===1||d===1)&&c>t8)&&u.isPacked&&f&&u.texture!=null&&l[2]%2!=0&&C.arraysEqual(u.shape.slice(-3),l.slice(-3))){let A=l[0]*l[1]*(l[2]+1),v={dataId:e.dataId,shape:[1,A,n.inChannels],dtype:e.dtype},w=u.shape;u.shape=u.shape.slice(),u.shape[u.shape.length-2]++,C.assert(zm(u.shape,v.shape),()=> ` packed reshape $ { u . shape } to $ { v . shape } isn ' t free ` );let k= $ e({inputs:{x:t},backend:r,attrs:{shape:[1,n.inChannels,n.outChannels]}});x.push(k);let I=Gm({a:v,b:k,backend:r,transposeA:h,transposeB:m,bias:s,activation:i,preluActivationWeights:o,leakyreluAlpha:a}),N=r.texData.get(I.dataId);C.assert(N.isPacked,()=>"batchMatMul result is expected to be packed"),u.shape=w,N.shape=n.outShape,g=lr({inputs:{x:I},backend:r}),g.shape=n.outShape,x.push(I)}else{let A=f?l[0]*l[1]*l[2]:l[0]*l[2]*l[3],v= $ e({inputs:{x:e},backend:r,attrs:{shape:[1,A,n.inChannels]}}),w= $ e({inputs:{x:t},backend:r,attrs:{shape:[1,n.inChannels,n.outChannels]}}),k=Gm({a:v,b:w,transposeA:h,transposeB:m,backend:r,bias:s,activation:i,preluActivationWeights:o,leakyreluAlpha:a});g= $ e({inputs:{x:k},backend:r,attrs:{shape:n.outShape}}),x.push(v),x.push(w),x.push(k)}for(let A of x)r.disposeIntermediateTensorInfo(A);return g}function PS({x:e,filter:t,convInfo:n,backend:r,bias:s=null,preluActivationWeights:o=null,leakyreluAlpha:a=0,activation:i=null}){let{filterWidth:l,filterHeight:u,inChannels:c,outWidth:p,outHeight:d,dataFormat:f}=n,h=f==="channelsLast",m=l*u*c,g=d*p,x=[m,g],y=!0,b=!1,A=[],v= $ e({inputs:{x:e},backend:r,attrs:{shape:e.shape.slice(1)}}),w= $ e({inputs:{x:t},backend:r,attrs:{shape:[1,m,C.sizeFromShape(t.shape)/m]}});A.push(v),A.push(w);let k=new w8(x,n),I=[v.shape,[n.padInfo.top,n.padInfo.left],[n.strideHeight,n.strideWidth],[n.dilationHeight,n.dilationWidth],[n.inChannels],[n.filterWidth*n.inChannels],[n.outWidth]],N=r.runWebGLProgram(k,[v],"float32",I),T= $ e({inputs:{x:N},backend:r,attrs:{shape:[1,x[0],x[1]]}});A.push(N),A.push(T);let _=s!=null,R=o!=null,S=i==="leakyrelu",F=i?Nd(i,!0):null,B=new mv(T.shape,w.shape,[1,g,n.outChannels],y,b,_,F,R,S),W=[T,w];if(s&&W.push(s),R&&W.push(o),S){let Q=r.makeTensorInfo([],"float32",C.createScalarValue(a,"float32"));W.push(Q),A.push(Q)}let q=r.runWebGLProgram(B,W,"float32"),K=h?[1,d,p,n.outChannels]:[1,n.outChannels,d,p],J= $ e({inputs:{x:q},backend:r,attrs:{shape:K}});A.push(q);for(let Q of A)r.disposeIntermediateTensorInfo(Q);return J}function qXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dataFormat:l,dilations:u,dimRoundingMode:c}=r,p=M.convertConv2DDataFormat(l),d=M.computeConv2DInfo(s.shape,o.shape,a,u,i,c,!1,p),f;if(d.filterHeight===1&&d.filterWidth===1&&d.dilationHeight===1&&d.dilationWidth===1&&d.strideHeight===1&&d.strideWidth===1&&(d.padInfo.type==="SAME"||d.padInfo.type==="VALID"))f=OS({x:s,filter:o,convInfo:d,backend:n});else if(he().getBool("WEBGL_CONV_IM2COL")&&s.shape[0]===1)f=PS({x:s,filter:o,convInfo:d,backend:n});else{let m=new xv(d);f=n.runWebGLProgram(m,[s,o],"float32")}let h= $ e({inputs:{x:f},backend:n,attrs:{shape:d.outShape}});return n.disposeIntermediateTensorInfo(f),h}var ase={kernelName:vp,backendName:"webgl",kernelFunc:qXe};var k8=class{constructor(t){this.variableNames=["x","dy"],this.outputShape=t.filterShape;let n=t.strideHeight,r=t.strideWidth,s=t.padInfo.top,o=t.padInfo.left,a=t.dataFormat==="channelsLast";this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int wR = coords . x ;
int wC = coords . y ;
int d1 = coords . z ;
int d2 = coords . w ;
// Convolve x(?, ?, d1) with dy(:, :, d2) to get dw(wR, wC, d1, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int b = 0 ; b < $ { t . batchSize } ; b ++ ) {
for ( int yR = 0 ; yR < $ { t . outHeight } ; yR ++ ) {
int xR = wR + yR * $ { n } - $ { s } ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { t . outWidth } ; yC ++ ) {
int xC = wC + yC * $ { r } - $ { o } ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
if ( $ { a } ) {
float dyValue = getDy ( b , yR , yC , d2 ) ;
float xValue = getX ( b , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
} else {
float dyValue = getDy ( b , d2 , yR , yC ) ;
float xValue = getX ( b , d1 , xR , xC ) ;
dotProd += ( xValue * dyValue ) ;
}
}
}
}
setOutput ( dotProd ) ;
}
` }},I8=class{constructor(t){this.variableNames=["dy","W"],this.outputShape=t.inShape;let n=t.filterHeight,r=t.filterWidth,s=t.strideHeight,o=t.strideWidth,a=t.dataFormat==="channelsLast",i=n-1-t.padInfo.top,l=r-1-t.padInfo.left,u=a?1:2,c=a?2:3,p=a?3:1;this.userCode= `
const ivec2 pads = ivec2 ( $ { i } , $ { l } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d1 = coords [ $ { p } ] ;
ivec2 dyCorner = ivec2 ( coords [ $ { u } ] , coords [ $ { c } ] ) - pads ;
int dyRCorner = dyCorner . x ;
int dyCCorner = dyCorner . y ;
// Convolve dy(?, ?, d2) with w(:, :, d1, d2) to compute dx(xR, xC, d1).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { n } ; wR ++ ) {
float dyR = float ( dyRCorner + wR ) / $ { s } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { n } - 1 - wR ;
for ( int wC = 0 ; wC < $ { r } ; wC ++ ) {
float dyC = float ( dyCCorner + wC ) / $ { o } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
int wCPerm = $ { r } - 1 - wC ;
for ( int d2 = 0 ; d2 < $ { t . outChannels } ; d2 ++ ) {
if ( $ { a } ) {
float xValue = getDy ( batch , idyR , idyC , d2 ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
} else {
float xValue = getDy ( batch , d2 , idyR , idyC ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
}
}
}
}
setOutput ( dotProd ) ;
}
` }},C8=class{constructor(t){this.variableNames=["x","dy"],this.outputShape=t.filterShape;let n=t.strideDepth,r=t.strideHeight,s=t.strideWidth,o=t.padInfo.front,a=t.padInfo.top,i=t.padInfo.left;this.userCode= `
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int wF = coords . x ;
int wR = coords . y ;
int wC = coords . z ;
int d1 = coords . w ;
int d2 = coords . u ;
float dotProd = 0.0 ;
for ( int b = 0 ; b < $ { t . batchSize } ; b ++ ) {
for ( int yF = 0 ; yF < $ { t . outDepth } ; yF ++ ) {
int xF = wF + yF * $ { n } - $ { o } ;
if ( xF < 0 || xF >= $ { t . inDepth } ) {
continue ;
}
for ( int yR = 0 ; yR < $ { t . outHeight } ; yR ++ ) {
int xR = wR + yR * $ { r } - $ { a } ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { t . outWidth } ; yC ++ ) {
int xC = wC + yC * $ { s } - $ { i } ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
float dyValue = getDy ( b , yF , yR , yC , d2 ) ;
float xValue = getX ( b , xF , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
}
}
}
}
setOutput ( dotProd ) ;
}
` }},S8=class{constructor(t){this.variableNames=["dy","W"],this.outputShape=t.inShape;let n=t.filterDepth,r=t.filterHeight,s=t.filterWidth,o=t.strideDepth,a=t.strideHeight,i=t.strideWidth,l=n-1-t.padInfo.front,u=r-1-t.padInfo.top,c=s-1-t.padInfo.left;this.userCode= `
const ivec3 pads = ivec3 ( $ { l } , $ { u } , $ { c } ) ;
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d1 = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyFCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
float dotProd = 0.0 ;
for ( int wF = 0 ; wF < $ { n } ; wF ++ ) {
float dyF = float ( dyFCorner + wF ) / $ { o } . 0 ;
if ( dyF < 0.0 || dyF >= $ { t . outDepth } . 0 || fract ( dyF ) > 0.0 ) {
continue ;
}
int idyF = int ( dyF ) ;
int wFPerm = $ { n } - 1 - wF ;
for ( int wR = 0 ; wR < $ { r } ; wR ++ ) {
float dyR = float ( dyRCorner + wR ) / $ { a } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { r } - 1 - wR ;
for ( int wC = 0 ; wC < $ { s } ; wC ++ ) {
float dyC = float ( dyCCorner + wC ) / $ { i } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
int wCPerm = $ { s } - 1 - wC ;
for ( int d2 = 0 ; d2 < $ { t . outChannels } ; d2 ++ ) {
float xValue = getDy ( batch , idyF , idyR , idyC , d2 ) ;
float wValue = getW ( wFPerm , wRPerm , wCPerm , d1 , d2 ) ;
dotProd += xValue * wValue ;
}
}
}
}
setOutput ( dotProd ) ;
}
` }};function KXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,pad:i,dataFormat:l,dimRoundingMode:u,filterShape:c}=r,p=M.convertConv2DDataFormat(l),d=M.computeConv2DInfo(s.shape,c,a,1,i,u,!1,p),f=new k8(d);return n.runWebGLProgram(f,[s,o],"float32")}var ise={kernelName:wx,backendName:"webgl",kernelFunc:KXe};function XXe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{inputShape:a,strides:i,pad:l,dataFormat:u,dimRoundingMode:c}=r,p=M.convertConv2DDataFormat(u),d=M.computeConv2DInfo(a,o.shape,i,1,l,c,!1,p),f=new I8(d);return n.runWebGLProgram(f,[s,o],"float32")}var lse={kernelName:wp,backendName:"webgl",kernelFunc:XXe};function ZXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l}=r,u=M.computeConv3DInfo(s.shape,o.shape,a,l,i),c=new v8(u);return n.runWebGLProgram(c,[s,o],"float32")}var use={kernelName:kp,backendName:"webgl",kernelFunc:ZXe};function YXe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,pad:i,filterShape:l}=r,u=M.computeConv3DInfo(s.shape,l,a,1,i),c=new C8(u);return n.runWebGLProgram(c,[s,o],"float32")}var cse={kernelName:kx,backendName:"webgl",kernelFunc:YXe};function JXe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{pad:a,strides:i,inputShape:l}=r,u=M.computeConv3DInfo(l,o.shape,i,1,a),c=new S8(u);return n.runWebGLProgram(c,[s,o],"float32")}var pse={kernelName:Ix,backendName:"webgl",kernelFunc:JXe};var QXe=_S+ `
return cos ( x ) ;
` ,eZe=Xe({opSnippet:QXe}),dse={kernelName:uu,backendName:"webgl",kernelFunc:eZe};var tZe= `
float e2x = exp ( - x ) ;
return ( e2x + 1.0 / e2x ) / 2.0 ;
` ,nZe=Xe({opSnippet:tZe}),fse={kernelName:cu,backendName:"webgl",kernelFunc:nZe};var N8=class{constructor(t,n,r,s,o){this.variableNames=["Image","Boxes","BoxInd"],this.outputShape=[];let[a,i,l,u]=t,[c]=n,[p,d]=r;this.outputShape=[c,p,d,u];let f=s==="bilinear"?1:0,[h,m]=[ ` $ { i - 1 } . 0 ` , ` $ { l - 1 } . 0 ` ],[g,x,y]=p>1?[ ` $ { ( i - 1 ) / ( p - 1 ) } ` ,"(y2-y1) * height_ratio", ` y1 * $ { h } + float ( y ) * ( height _scale ) ` ]:["0.0","0.0", ` 0.5 * ( y1 + y2 ) * $ { h } ` ],[b,A,v]=d>1?[ ` $ { ( l - 1 ) / ( d - 1 ) } ` ,"(x2-x1) * width_ratio", ` x1 * $ { m } + float ( x ) * ( width _scale ) ` ]:["0.0","0.0", ` 0.5 * ( x1 + x2 ) * $ { m } ` ];this.userCode= `
const float height _ratio = float ( $ { g } ) ;
const float width _ratio = float ( $ { b } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int y = coords [ 1 ] ;
int x = coords [ 2 ] ;
int d = coords [ 3 ] ;
// get box vals
float y1 = getBoxes ( b , 0 ) ;
float x1 = getBoxes ( b , 1 ) ;
float y2 = getBoxes ( b , 2 ) ;
float x2 = getBoxes ( b , 3 ) ;
// get image in batch index
int bInd = round ( getBoxInd ( b ) ) ;
if ( bInd < 0 || bInd >= $ { a } ) {
return ;
}
float height _scale = $ { x } ;
float width _scale = $ { A } ;
float in _y = $ { y } ;
if ( in _y < 0.0 || in _y > $ { h } ) {
setOutput ( float ( $ { o } ) ) ;
return ;
}
float in _x = $ { v } ;
if ( in _x < 0.0 || in _x > $ { m } ) {
setOutput ( float ( $ { o } ) ) ;
return ;
}
vec2 sourceFracIndexCR = vec2 ( in _x , in _y ) ;
if ( $ { f } == 1 ) {
// Compute the four integer indices.
ivec2 sourceFloorCR = ivec2 ( sourceFracIndexCR ) ;
ivec2 sourceCeilCR = ivec2 ( ceil ( sourceFracIndexCR ) ) ;
float topLeft = getImage ( b , sourceFloorCR . y , sourceFloorCR . x , d ) ;
float bottomLeft = getImage ( b , sourceCeilCR . y , sourceFloorCR . x , d ) ;
float topRight = getImage ( b , sourceFloorCR . y , sourceCeilCR . x , d ) ;
float bottomRight = getImage ( b , sourceCeilCR . y , sourceCeilCR . x , d ) ;
vec2 fracCR = sourceFracIndexCR - vec2 ( sourceFloorCR ) ;
float top = topLeft + ( topRight - topLeft ) * fracCR . x ;
float bottom = bottomLeft + ( bottomRight - bottomLeft ) * fracCR . x ;
float newValue = top + ( bottom - top ) * fracCR . y ;
setOutput ( newValue ) ;
} else {
// Compute the coordinators of nearest neighbor point.
ivec2 sourceNearestCR = ivec2 ( floor (
sourceFracIndexCR + vec2 ( 0.5 , 0.5 ) ) ) ;
float newValue = getImage ( b , sourceNearestCR . y , sourceNearestCR . x , d ) ;
setOutput ( newValue ) ;
}
}
` }};var rZe=e=>{let{inputs:t,backend:n,attrs:r}=e,{image:s,boxes:o,boxInd:a}=t,{cropSize:i,method:l,extrapolationValue:u}=r,c=new N8(s.shape,o.shape,i,l,u);return n.runWebGLProgram(c,[s,o,a],"float32")},hse={kernelName:Cx,backendName:"webgl",kernelFunc:rZe};var MS=class{constructor(t,n,r){this.variableNames=["x"],this.customUniforms=[{name:"index",type:"float"}],this.outputShape=t;let s=t.length,o=n?"0.0": ` getX ( $ { mse ( s , "coords" ) } ) ` ,a=t[t.length-1],i="",l="";n?(i=r? ` end != $ { a - 1 } ` :"end != 0",l=r?"end + 1":"end - 1"):(i=r? ` end + pow2 < $ { a } ` :"end >= pow2",l=r?"end + pow2":"end - pow2"),this.userCode= `
void main ( ) {
$ { Ct ( s ) } coords = getOutputCoords ( ) ;
int end = $ { gse ( s , "coords" ) } ;
float val = $ { o } ;
int pow2 = int ( pow ( 2.0 , index ) ) ;
if ( $ { i } ) {
int idx = $ { l } ;
$ { gse ( s , "coords" ) } = idx ;
val += getX ( $ { mse ( s , "coords" ) } ) ;
}
setOutput ( val ) ;
}
` }};function mse(e,t){if(e===1)return ` $ { t } ` ;if(e===2)return ` $ { t } . x , $ { t } . y ` ;if(e===3)return ` $ { t } . x , $ { t } . y , $ { t } . z ` ;if(e===4)return ` $ { t } . x , $ { t } . y , $ { t } . z , $ { t } . w ` ;throw Error( ` Cumulative sum for rank $ { e } is not yet supported ` )}function gse(e,t){if(e===1)return ` $ { t } ` ;if(e===2)return ` $ { t } . y ` ;if(e===3)return ` $ { t } . z ` ;if(e===4)return ` $ { t } . w ` ;throw Error( ` Cumulative sum for rank $ { e } is not yet supported ` )}function sZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,exclusive:a,reverse:i}=r,l=s.shape.length,u=M.getAxesPermutation([o],l),c=s;u!=null&&(c=On({inputs:{x:s},backend:n,attrs:{perm:u}}));let p=M.getInnerMostAxes(1,l)[0];if(p!==l-1)throw new Error( ` WebGL cumsum shader expects an inner - most axis = $ { s . shape . length - 1 } but got axis = $ { o } ` );let d=c.shape[p],f=lr({inputs:{x:c},backend:n});for(let h=0;h<=Math.ceil(Math.log2(d))-1;h++){let m=new MS(c.shape,!1,i),g=[[h]],x=f;f=n.runWebGLProgram(m,[f],f.dtype,g),n.disposeIntermediateTensorInfo(x)}if(a){let h=new MS(c.shape,a,i),m=f;f=n.runWebGLProgram(h,[f],f.dtype),n.disposeIntermediateTensorInfo(m)}if(u!=null){let h=M.getUndoAxesPermutation(u),m=On({inputs:{x:f},backend:n,attrs:{perm:h}});return n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(c),m}return f}var xse={kernelName:Ip,backendName:"webgl",kernelFunc:sZe};function oZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,weights:o}=t,{size:a,binaryOutput:i}=r;if(s.shape.length===1){let l=n.readSync(s.dataId),u=n.readSync(o.dataId),c=IS(l,u,o.dtype,o.shape,a);return n.makeTensorInfo([a],o.dtype,c)}else if(s.shape.length===2){let l=n.bufferSync(s),u=n.bufferSync(o),c=pne(l,u,a,i);return n.makeTensorInfo(c.shape,o.dtype,c.values)}throw new Error( ` Error in denseBincount : input must be at most rank 2 , but got rank$ { s . shape . length } . ` )}var yse={kernelName:Sx,backendName:"webgl",kernelFunc:oZe};var T8=class{constructor(t,n,r){this.variableNames=["x"],this.outputShape=[],this.outputShape=t,this.blockSize=n,this.dataFormat=r,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int h = $ { this . getHeightCoordString ( ) } ;
int w = $ { this . getWidthCoordString ( ) } ;
int d = $ { this . getDepthCoordString ( ) } ;
int in _h = h / $ { n } ;
int offset _h = imod ( h , $ { n } ) ;
int in _w = w / $ { n } ;
int offset _w = imod ( w , $ { n } ) ;
int offset _d = ( offset _h * $ { n } + offset _w ) *
$ { this . getOutputDepthSize ( ) } ;
int in _d = d + offset _d ;
float result = $ { this . getInputSamplingString ( ) } ;
setOutput ( result ) ;
}
` }getHeightCoordString(){return this.dataFormat==="NHWC"?"coords[1]":"coords[2]"}getWidthCoordString(){return this.dataFormat==="NHWC"?"coords[2]":"coords[3]"}getDepthCoordString(){return this.dataFormat==="NHWC"?"coords[3]":"coords[1]"}getOutputDepthSize(){return this.dataFormat==="NHWC"?this.outputShape[3]:this.outputShape[1]}getInputSamplingString(){return this.dataFormat==="NHWC"?"getX(b, in_h, in_w, in_d)":"getX(b, in_d, in_h, in_w)"}};function aZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockSize:o,dataFormat:a}=r;C.assert(o>1,()=> ` blockSize should be > 1 for depthToSpace , but was : $ { o } ` );let i=s.shape[0],l=a==="NHWC"?s.shape[1]:s.shape[2],u=a==="NHWC"?s.shape[2]:s.shape[3],c=a==="NHWC"?s.shape[3]:s.shape[1],p=l*o,d=u*o,f=c/(o*o),h=a==="NHWC"?[i,p,d,f]:[i,f,p,d],m=new T8(h,o,a);return n.runWebGLProgram(m,[s],s.dtype)}var bse={kernelName:Nx,backendName:"webgl",kernelFunc:aZe};var yv=class{constructor(t,n=!1,r=null,s=!1,o=!1){this.variableNames=["x","W"],this.customUniforms=[{name:"pads",type:"ivec2"},{name:"strides",type:"ivec2"},{name:"dilations",type:"ivec2"},{name:"inDims",type:"ivec2"}],this.outputShape=t.outShape,this.enableShapeUniforms=Xn(this.outputShape.length);let a=t.filterHeight,i=t.filterWidth,l=t.outChannels/t.inChannels,u="",c="";r&&(s?u= ` float activation ( float a ) {
float b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { r }
} ` :o?u= ` float activation ( float a ) {
float b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { r }
} ` :u= `
float activation ( float x ) {
$ { r }
}
` ,c="result = activation(result);");let p=n?"result += getBiasAtOutCoords();":"";n&&this.variableNames.push("bias"),s&&this.variableNames.push("preluActivationWeights"),o&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
$ { u }
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int d2 = coords . w ;
int d1 = d2 / $ { l } ;
int q = d2 - d1 * $ { l } ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
// Convolve x(?, ?, d1) with w(:, :, d1, q) to get y(yR, yC, d2).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
// TO DO(dsmilkov): Flatten the two for loops and vec4 the operations.
for ( int wR = 0 ; wR < $ { a } ; wR ++ ) {
int xR = xRCorner + wR * dilations [ 0 ] ;
if ( xR < 0 || xR >= inDims [ 0 ] ) {
continue ;
}
for ( int wC = 0 ; wC < $ { i } ; wC ++ ) {
int xC = xCCorner + wC * dilations [ 1 ] ;
if ( xC < 0 || xC >= inDims [ 1 ] ) {
continue ;
}
float xVal = getX ( batch , xR , xC , d1 ) ;
float wVal = getW ( wR , wC , d1 , q ) ;
dotProd += xVal * wVal ;
}
}
float result = dotProd ;
$ { p }
$ { c }
setOutput ( result ) ;
}
` }};var bv=class{constructor(t,n=!1,r=null,s=!1,o=!1){this.variableNames=["x","W"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"pads",type:"ivec2"},{name:"strides",type:"ivec2"},{name:"dilations",type:"ivec2"},{name:"inDims",type:"ivec2"}],this.outputShape=t.outShape,this.enableShapeUniforms=Xn(this.outputShape.length);let a=t.outChannels/t.inChannels,i=t.padInfo.left,l=t.strideWidth,u=t.dilationWidth,c=t.filterHeight,p=t.filterWidth,d=p,f= `
int xR ; int xC ; int xCOffset ;
vec4 wTexel ; vec4 previous ; vec4 final ; ` ;for(let x=0;x<p;x++)f+= `
vec4 xTexelC$ { x * 2 } ;
int xTexelC$ { x * 2 } Ready ;
vec4 xTexelC$ { x * 2 + 1 } ;
int xTexelC$ { x * 2 + 1 } Ready ;
vec4 xC$ { x } ; ` ;for(let x=0;x<c;x++){for(let y=0;y<p;y++)f+= `
xTexelC$ { y * 2 } = vec4 ( 0.0 ) ;
xTexelC$ { y * 2 } Ready = 0 ;
xTexelC$ { y * 2 + 1 } = vec4 ( 0.0 ) ;
xTexelC$ { y * 2 + 1 } Ready = 0 ;
xC$ { y } = vec4 ( 0.0 ) ; ` ;f+= `
xR = xRCorner + $ { x } * dilations [ 0 ] ;
if ( xR >= 0 && xR < inDims [ 0 ] ) {
` ;for(let y=0;y<(d+1)/2;y++){let b=y*2;if(f+= `
xC = xCCorner + $ { b * u } ;
` ,l===1){if(b<p&&(i%2==1?(f+= `
xCOffset = xC + 1 ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b } Ready == 0 ) {
xTexelC$ { b } = getX ( batch , xR , xCOffset , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { b } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b } Ready = 1 ;
}
` ,u===1&&b>0?f+= `
xC$ { b } = vec4 ( xTexelC$ { b - 2 } . zw , xTexelC$ { b } . xy ) ;
` :f+= `
xCOffset = xC + 1 - 2 ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] ) {
previous = getX ( batch , xR , xCOffset , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
previous . zw = vec2 ( 0.0 ) ;
}
xC$ { b } = vec4 ( previous . zw , xTexelC$ { b } . xy ) ;
} else {
xC$ { b } = vec4 ( 0.0 , 0.0 , xTexelC$ { b } . xy ) ;
}
` ):f+= `
if ( xC >= 0 && xC < inDims [ 1 ] && xTexelC$ { b } Ready == 0 ) {
xTexelC$ { b } = getX ( batch , xR , xC , d1 ) ;
if ( xC + 1 >= inDims [ 1 ] ) {
xTexelC$ { b } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b } Ready = 1 ;
}
xC$ { b } = xTexelC$ { b } ;
` ,b+1<p)){let A=i%2==0?C.nearestLargerEven(u):u;u%2==0&&i%2==1||u%2!=0&&i%2!=1?(f+= `
xCOffset = xC + imod ( pads [ 1 ] , 2 ) + $ { A } ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b + 1 } Ready == 0 ) {
xTexelC$ { b + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { b + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b + 1 } Ready = 1 ;
}
` ,u>1&&(f+= `
xCOffset -= 2 ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b } Ready == 0 ) {
xTexelC$ { b } = getX ( batch , xR , xCOffset , d1 ) ;
xTexelC$ { b } Ready = 1 ;
}
` ),f+= `
xC$ { b + 1 } = vec4 ( xTexelC$ { b } . zw , xTexelC$ { b + 1 } . xy ) ;
` ):A===1?f+= `
xC$ { b + 1 } = xTexelC$ { b } ;
` :f+= `
xCOffset = xC + $ { A } ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b + 1 } Ready == 0 ) {
xTexelC$ { b + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { b + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b + 1 } Ready = 1 ;
}
xC$ { b + 1 } = xTexelC$ { b + 1 } ;
` }}else b<p&&(i%2==1?(f+= `
xCOffset = xC + 1 - strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b } Ready == 0 ) {
xTexelC$ { b } = getX ( batch , xR , xCOffset , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { b } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b } Ready = 1 ;
}
if ( xC + 1 >= 0 && xC + 1 < inDims [ 1 ] && xTexelC$ { b + 1 } Ready == 0 ) {
xTexelC$ { b + 1 } = getX ( batch , xR , xC + 1 , d1 ) ;
// Need to manually clear unused channels in case
// we're reading from recycled texture.
if ( xC + 2 >= inDims [ 1 ] ) {
xTexelC$ { b + 1 } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b + 1 } Ready = 1 ;
}
xC$ { b } = vec4 ( xTexelC$ { b } . zw , xTexelC$ { b + 1 } . zw ) ;
` ,b+1<p&&(f+= `
final = vec4 ( 0.0 ) ;
xCOffset = xC + 1 + strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] ) {
final = getX ( batch , xR , xCOffset , d1 ) ;
}
xC$ { b + 1 } = vec4 ( xTexelC$ { b + 1 } . xy , final . xy ) ;
` )):(f+= `
if ( xC >= 0 && xC < inDims [ 1 ] && xTexelC$ { b } Ready == 0 ) {
xTexelC$ { b } = getX ( batch , xR , xC , d1 ) ;
if ( xC + 1 >= inDims [ 1 ] ) {
xTexelC$ { b } . zw = vec2 ( 0.0 ) ;
}
xTexelC$ { b } Ready = 1 ;
}
xCOffset = xC + strides [ 1 ] ;
if ( xCOffset >= 0 && xCOffset < inDims [ 1 ] && xTexelC$ { b + 1 } Ready == 0 ) {
xTexelC$ { b + 1 } = getX ( batch , xR , xCOffset , d1 ) ;
if ( xCOffset + 1 >= inDims [ 1 ] ) {
xTexelC$ { b + 1 } . zw = vec2 ( 0. ) ;
}
xTexelC$ { b + 1 } Ready = 1 ;
}
xC$ { b } = vec4 (
xTexelC$ { b } . xy , xTexelC$ { b + 1 } . xy ) ;
` ,b+1<p&&(f+= `
xC$ { b + 1 } = vec4 ( xTexelC$ { b } . zw , xTexelC$ { b + 1 } . zw ) ;
` )));b<p&&(f+= `
wTexel = getW ( $ { x } , $ { b } , d1 , q ) ;
dotProd += xC$ { b } * vec4 ( wTexel . xz , wTexel . xz ) ;
` ,b+1<p&&(f+= `
wTexel = getW ( $ { x } , $ { b + 1 } , d1 , q ) ;
dotProd += xC$ { b + 1 } * vec4 ( wTexel . xz , wTexel . xz ) ;
` ))}f+= `
}
` }let h="",m="";r&&(s?h= ` vec4 activation ( vec4 a ) {
vec4 b = getPreluActivationWeightsAtOutCoords ( ) ;
$ { r }
} ` :o?h= ` vec4 activation ( vec4 a ) {
vec4 b = getLeakyreluAlphaAtOutCoords ( ) ;
$ { r }
} ` :h= ` vec4 activation ( vec4 x ) {
$ { r }
} ` ,m="result = activation(result);");let g=n?"result += getBiasAtOutCoords();":"";n&&this.variableNames.push("bias"),s&&this.variableNames.push("preluActivationWeights"),o&&this.variableNames.push("leakyreluAlpha"),this.userCode= `
$ { h }
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
ivec2 xRCCorner = coords . yz * strides - pads ;
int d2 = coords . w ;
int d1 = d2 / $ { a } ;
int q = d2 - d1 * $ { a } ;
int xRCorner = xRCCorner . x ;
int xCCorner = xRCCorner . y ;
//intialize dotProd with a small epsilon seems to reduce GPU accuracy loss.
vec4 dotProd = vec4 ( 0.000000000000001 ) ;
$ { f }
vec4 result = dotProd - vec4 ( 0.000000000000001 ) ;
$ { g }
$ { m }
setOutput ( result ) ;
}
` }};function iZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l,dimRoundingMode:u}=r,c=l;c==null&&(c=[1,1]),C.assert(M.eitherStridesOrDilationsAreOne(a,c),()=> ` Error in depthwiseConv2d : Either strides or dilations must be 1. Got strides $ { a } and dilations '${c}' ` );let p=M.computeConv2DInfo(s.shape,o.shape,a,c,i,u,!0),d;he().getBool("WEBGL_PACK_DEPTHWISECONV")&&p.strideWidth<=2&&p.outChannels/p.inChannels==1?d=new bv(p):d=new yv(p);let f=[[p.padInfo.top,p.padInfo.left],[p.strideHeight,p.strideWidth],[p.dilationHeight,p.dilationWidth],[p.inHeight,p.inWidth]];return n.runWebGLProgram(d,[s,o],"float32",f)}var Ase={kernelName:Cp,backendName:"webgl",kernelFunc:iZe};var _8=class{constructor(t){this.variableNames=["x","dy"],this.outputShape=t.filterShape;let n=t.strideHeight,r=t.strideWidth,s=t.padInfo.top,o=t.padInfo.left,a=t.outChannels/t.inChannels;this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int wR = coords . x ;
int wC = coords . y ;
int d1 = coords . z ;
int dm = coords . w ;
int d2 = d1 * $ { a } + dm ;
float dotProd = 0.0 ;
// TO DO: Vec4 over the batch size
for ( int b = 0 ; b < $ { t . batchSize } ; b ++ ) {
for ( int yR = 0 ; yR < $ { t . outHeight } ; yR ++ ) {
int xR = wR + yR * $ { n } - $ { s } ;
if ( xR < 0 || xR >= $ { t . inHeight } ) {
continue ;
}
for ( int yC = 0 ; yC < $ { t . outWidth } ; yC ++ ) {
int xC = wC + yC * $ { r } - $ { o } ;
if ( xC < 0 || xC >= $ { t . inWidth } ) {
continue ;
}
float dyValue = getDy ( b , yR , yC , d2 ) ;
float xValue = getX ( b , xR , xC , d1 ) ;
dotProd += ( xValue * dyValue ) ;
}
}
}
setOutput ( dotProd ) ;
}
` }},E8=class{constructor(t){this.variableNames=["dy","W"],this.outputShape=t.inShape;let n=t.filterHeight,r=t.filterWidth,s=t.strideHeight,o=t.strideWidth,a=n-1-t.padInfo.top,i=r-1-t.padInfo.left,l=t.outChannels/t.inChannels;this.userCode= `
const ivec2 pads = ivec2 ( $ { a } , $ { i } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int d1 = coords [ 3 ] ;
ivec2 dyCorner = coords . yz - pads ;
int dyRCorner = dyCorner . x ;
int dyCCorner = dyCorner . y ;
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { n } ; wR ++ ) {
float dyR = float ( dyRCorner + wR ) / $ { s } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
int wRPerm = $ { n } - 1 - wR ;
for ( int wC = 0 ; wC < $ { r } ; wC ++ ) {
float dyC = float ( dyCCorner + wC ) / $ { o } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
int wCPerm = $ { r } - 1 - wC ;
// TO DO: Vec4 over the channelMul
for ( int dm = 0 ; dm < $ { l } ; dm ++ ) {
int d2 = d1 * $ { l } + dm ;
float xValue = getDy ( batch , idyR , idyC , d2 ) ;
float wValue = getW ( wRPerm , wCPerm , d1 , dm ) ;
dotProd += xValue * wValue ;
}
}
}
setOutput ( dotProd ) ;
}
` }};function lZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,dy:o}=t,{strides:a,dilations:i,pad:l,dimRoundingMode:u,filterShape:c}=r,p=M.computeConv2DInfo(s.shape,c,a,i,l,u,!0),d=new _8(p);return n.runWebGLProgram(d,[s,o],"float32")}var vse={kernelName:Tx,backendName:"webgl",kernelFunc:lZe};function uZe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,filter:o}=t,{strides:a,dilations:i,pad:l,dimRoundingMode:u,inputShape:c}=r,p=M.computeConv2DInfo(c,o.shape,a,i,l,u,!0),d=new E8(p);return n.runWebGLProgram(d,[s,o],"float32")}var wse={kernelName:_x,backendName:"webgl",kernelFunc:uZe};var R8=class{constructor(t){this.variableNames=["X"],this.outputShape=[t,t],this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
float val = coords [ 0 ] == coords [ 1 ] ? getX ( coords [ 0 ] ) : 0.0 ;
setOutput ( val ) ;
}
` }};function cZe(e){let{inputs:t,backend:n}=e,{x:r}=t,s=[...r.shape,...r.shape],o=C.sizeFromShape(r.shape),a= $ e({inputs:{x:r},backend:n,attrs:{shape:[o]}}),i=new R8(o),l=n.runWebGLProgram(i,[a],a.dtype),u= $ e({inputs:{x:l},backend:n,attrs:{shape:s}});return n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(l),u}var kse={kernelName:h3,backendName:"webgl",kernelFunc:cZe};var D8=class{constructor(t){this.variableNames=["x","W"],this.outputShape=t.outShape;let{inHeight:n,inWidth:r,padInfo:s,strideHeight:o,strideWidth:a,filterHeight:i,filterWidth:l,dilationHeight:u,dilationWidth:c}=t,{top:p,left:d}=s;this.userCode= `
const ivec2 strides = ivec2 ( $ { o } , $ { a } ) ;
const ivec2 pads = ivec2 ( $ { p } , $ { d } ) ;
const float neg _infinity = - 3.4 e38 ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int d1 = coords . w ;
ivec2 outTopLeftCorner =
coords . yz * strides - pads ;
int hBeg = outTopLeftCorner . x ;
int wBeg = outTopLeftCorner . y ;
float curVal = neg _infinity ;
for ( int h = 0 ; h < $ { i } ; h ++ ) {
int hIn = hBeg + h * $ { u } ;
if ( hIn >= 0 && hIn < $ { n } ) {
for ( int w = 0 ; w < $ { l } ; w ++ ) {
int wIn = wBeg + w * $ { c } ;
if ( wIn >= 0 && wIn < $ { r } ) {
float xVal = getX ( batch , hIn , wIn , d1 ) ;
float wVal = getW ( h , w , d1 ) ;
float val = xVal + wVal ;
if ( val > curVal ) {
curVal = val ;
}
}
}
}
}
float result = curVal ;
setOutput ( result ) ;
}
` }};function pZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o}=t,{strides:a,pad:i,dilations:l}=r,u=M.computeDilation2DInfo(s.shape,o.shape,a,i,"NHWC",l),c,p=new D8(u);c=n.runWebGLProgram(p,[s,o],"float32");let d= $ e({inputs:{x:c},backend:n,attrs:{shape:u.outShape}});return n.disposeIntermediateTensorInfo(c),d}var Ise={kernelName:Sp,backendName:"webgl",kernelFunc:pZe};function dZe(e){let{inputs:t,backend:n,attrs:r}=e,{equation:s}=r,o=t,{allDims:a,summedDims:i,idDims:l}=M.decodeEinsumEquation(s,o.length);M.checkEinsumDimSizes(a.length,l,o);let{path:u,steps:c}=M.getEinsumComputePath(i,l),p=c.length,d=null,f=a.length,h=[];for(let m=0;m<p;++m){for(let g of c[m]){let{permutationIndices:x,expandDims:y}=M.getEinsumPermutation(f,l[g]),b;M.isIdentityPermutation(x)?b=o[g]:(b=On({inputs:{x:o[g]},backend:n,attrs:{perm:x}}),h.push(b));let A=b.shape.slice();for(let v=0;v<y.length;++v)A.splice(y[v],0,1);C.arraysEqual(b.shape,A)||(b= $ e({inputs:{x:b},backend:n,attrs:{shape:A}}),h.push(b)),d===null?d=b:(d=gv({inputs:{a:b,b:d},backend:n}),h.push(d))}m<p-1&&(u[m]>=0&&(d=Wm({inputs:{x:d},backend:n,attrs:{axis:u[m]-(a.length-f),keepDims:!1}}),h.push(d)),f--)}for(let m of h)m!==d&&n.disposeIntermediateTensorInfo(m);return d}var Cse={kernelName:Ex,backendName:"webgl",kernelFunc:dZe};var fZe="return (x >= 0.0) ? x : (exp(x) - 1.0);",hZe= `
vec4 result ;
result . r = ( x . r >= 0.0 ) ? x . r : ( exp ( x . r ) - 1.0 ) ;
result . g = ( x . g >= 0.0 ) ? x . g : ( exp ( x . g ) - 1.0 ) ;
result . b = ( x . b >= 0.0 ) ? x . b : ( exp ( x . b ) - 1.0 ) ;
result . a = ( x . a >= 0.0 ) ? x . a : ( exp ( x . a ) - 1.0 ) ;
return result ;
` ,mZe=Xe({opSnippet:fZe,packedOpSnippet:hZe}),Sse={kernelName:du,backendName:"webgl",kernelFunc:mZe};var gZe="return (b >= 1.0) ? a : a * (b + 1.0);",xZe= `
vec4 bGTEZero = vec4 ( greaterThanEqual ( b , vec4 ( 0. ) ) ) ;
return ( bGTEZero * a ) + ( ( vec4 ( 1.0 ) - bGTEZero ) * ( a * ( b + vec4 ( 1.0 ) ) ) ) ;
` ,yZe=e=>{let{inputs:t,backend:n}=e,{dy:r,y:s}=t,o=he().getBool("WEBGL_PACK_BINARY_OPERATIONS")?new Qi(xZe,r.shape,s.shape):new ui(gZe,r.shape,s.shape);return n.runWebGLProgram(o,[r,s],r.dtype)},Nse={kernelName:Rx,backendName:"webgl",kernelFunc:yZe};var bZe= `
return vec4 ( equal ( a , b ) ) ;
` ,AZe="return float(a == b);",vZe=Qt({opSnippet:AZe,packedOpSnippet:bZe,dtype:"bool",cpuKernelImpl:hne}),Tse={kernelName:Ch,backendName:"webgl",kernelFunc:vZe};var wZe= `
// Error function is calculated approximately with elementary function.
// See "Handbook of Mathematical Functions with Formulas,
// Graphs, and Mathematical Tables", Abramowitz and Stegun.
float p = $ { M . ERF _P } ;
float a1 = $ { M . ERF _A1 } ;
float a2 = $ { M . ERF _A2 } ;
float a3 = $ { M . ERF _A3 } ;
float a4 = $ { M . ERF _A4 } ;
float a5 = $ { M . ERF _A5 } ;
float sign = sign ( x ) ;
x = abs ( x ) ;
float t = 1.0 / ( 1.0 + p * x ) ;
return sign * ( 1.0 - ( ( ( ( ( a5 * t + a4 ) * t ) + a3 ) * t + a2 ) * t + a1 ) * t * exp ( - x * x ) ) ;
` ,kZe=Xe({opSnippet:wZe}),_se={kernelName:fu,backendName:"webgl",kernelFunc:kZe};var Ese="return exp(x);", $ 8=Xe({opSnippet:Ese,packedOpSnippet:Ese,cpuKernelImpl:mne}),Rse={kernelName:hu,backendName:"webgl",kernelFunc: $ 8};function zS(e){let{inputs:t,attrs:n,backend:r}=e,{dim:s}=n,{input:o}=t,a=o.shape.length,i=o.shape.slice(),l=s;return s<0&&(C.assert(-(a+1)<=s,()=> ` Axis must be in the interval [ $ { - ( a + 1 ) } , $ { a } ] ` ),l=a+s+1),i.splice(l,0,1), $ e({inputs:{x:o},backend:r,attrs:{shape:i}})}var Dse={kernelName:Np,backendName:"webgl",kernelFunc:zS};var $ se="return exp(x) - 1.0;",IZe=Xe({opSnippet: $ se,packedOpSnippet: $ se,cpuKernelImpl:gne}),Fse={kernelName:mu,backendName:"webgl",kernelFunc:IZe};var LS=class{constructor(t,n,r){this.variableNames=["real","imag"];let s=n[1];this.outputShape=n;let o=r? ` 2.0 * $ { Math . PI } ` : ` - 2.0 * $ { Math . PI } ` ,a=r? ` $ { s } . 0 ` :"1.0",i;if(t==="real")i="return real * expR - imag * expI;";else if(t==="imag")i="return real * expI + imag * expR;";else throw new Error( ` FFT component must be either "real" or "imag" , got $ { t } . ` );this.userCode= `
const float exponentMultiplier = $ { o } ;
float unaryOpComplex ( float real , float expR , float imag , float expI ) {
$ { i }
}
float mulMatDFT ( int batch , int index ) {
float indexRatio = float ( index ) / float ( $ { s } ) ;
float exponentMultiplierTimesIndexRatio =
exponentMultiplier * indexRatio ;
float result = 0.0 ;
for ( int i = 0 ; i < $ { s } ; i ++ ) {
// x = (-2|2 * PI / N) * index * i;
float x = exponentMultiplierTimesIndexRatio * float ( i ) ;
float expR = cos ( x ) ;
float expI = sin ( x ) ;
float real = getReal ( batch , i ) ;
float imag = getImag ( batch , i ) ;
result +=
unaryOpComplex ( real , expR , imag , expI ) / $ { a } ;
}
return result ;
}
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
setOutput ( mulMatDFT ( coords [ 0 ] , coords [ 1 ] ) ) ;
}
` }};function BS(e,t,n){let r=n.texData.get(e.dataId),s=C.sizeFromShape(e.shape),o=e.shape[e.shape.length-1],a=s/o,i= $ e({inputs:{x:e},backend:n,attrs:{shape:[a,o]}}),l=i.shape,u=new LS("real",l,t),c=new LS("imag",l,t),p=[{dataId:r.complexTensorInfos.real.dataId,dtype:r.complexTensorInfos.real.dtype,shape:l},{dataId:r.complexTensorInfos.imag.dataId,dtype:r.complexTensorInfos.imag.dtype,shape:l}],d=n.runWebGLProgram(u,p,"float32"),f=n.runWebGLProgram(c,p,"float32"),h=Fo({inputs:{real:d,imag:f},backend:n});n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f);let m= $ e({inputs:{x:h},backend:n,attrs:{shape:e.shape}});return n.disposeIntermediateTensorInfo(i),n.disposeIntermediateTensorInfo(h),m}function CZe(e){let{inputs:t,backend:n}=e,{input:r}=t;return BS(r,!1,n)}var Ose={kernelName:Dx,backendName:"webgl",kernelFunc:CZe};var F8=class{constructor(t,n){this.outputShape=[],this.customUniforms=[{name:"value",type:"float"}],this.variableNames=["x"],this.outputShape=t,this.userCode= `
void main ( ) {
// Input can be obtained from uniform value.
setOutput ( value ) ;
}
` }};function vc(e){let{backend:t,attrs:n}=e,{shape:r,value:s}=n,{dtype:o}=n;if(o=o||C.inferDtype(s),o==="string"){let a=C.getArrayFromDType(o,C.sizeFromShape(r));return a.fill(s),t.makeTensorInfo(r,o,a)}else{let a=new F8(r,s),i=[[s]];return t.runWebGLProgram(a,[],o,i)}}var Pse={kernelName: $ x,backendName:"webgl",kernelFunc:vc};var O8=class{constructor(t){this.variableNames=["Image"],this.outputShape=[];let n=t[2];this.outputShape=t,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int x = coords [ 2 ] ;
int coordX = $ { n } - x - 1 ;
float outputValue ;
if ( coordX >= 0 && coordX < $ { n } ) {
outputValue = getImage ( coords [ 0 ] , coords [ 1 ] , coordX , coords [ 3 ] ) ;
} else {
outputValue = getImage ( coords [ 0 ] , coords [ 1 ] , coords [ 2 ] , coords [ 3 ] ) ;
}
setOutput ( outputValue ) ;
}
` }};var Mse={kernelName:Fx,backendName:"webgl",kernelFunc:({inputs:e,backend:t})=>{let{image:n}=e,r=t,s=new O8(n.shape);return r.runWebGLProgram(s,[n],n.dtype)}};var zse="return floor(x);",SZe=Xe({opSnippet:zse,packedOpSnippet:zse,cpuKernelImpl:xne}),Lse={kernelName:gu,backendName:"webgl",kernelFunc:SZe};var NZe= `
float s = sign ( a ) * sign ( b ) ;
int ia = round ( a ) ;
int ib = round ( b ) ;
if ( ib != 0 ) {
// Windows (D3D) wants guaranteed non-zero int division at compile-time.
return float ( idiv ( ia , ib , s ) ) ;
} else {
return NAN ;
}
` ,TZe= `
ivec4 ia = round ( a ) ;
ivec4 ib = round ( b ) ;
bvec4 cond = notEqual ( ib , ivec4 ( 0 ) ) ;
ivec4 result = ivec4 ( 0 ) ;
vec4 s = sign ( a ) * sign ( b ) ;
// Windows (D3D) wants guaranteed non-zero int division at compile-time.
if ( cond [ 0 ] ) {
result [ 0 ] = idiv ( ia [ 0 ] , ib [ 0 ] , s [ 0 ] ) ;
}
if ( cond [ 1 ] ) {
result [ 1 ] = idiv ( ia [ 1 ] , ib [ 1 ] , s [ 1 ] ) ;
}
if ( cond [ 2 ] ) {
result [ 2 ] = idiv ( ia [ 2 ] , ib [ 2 ] , s [ 2 ] ) ;
}
if ( cond [ 3 ] ) {
result [ 3 ] = idiv ( ia [ 3 ] , ib [ 3 ] , s [ 3 ] ) ;
}
return vec4 ( result ) ;
` ,_Ze=Qt({opSnippet:NZe,packedOpSnippet:TZe,dtype:"int32"}),Bse={kernelName:xu,backendName:"webgl",kernelFunc:_Ze};var P8=class{constructor(t){this.variableNames=["A"];let n=Vn(),[r,s]=t;this.outputShape=t,this.userCode= `
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int texR = coords [ 0 ] ;
int texC = coords [ 1 ] ;
int depth = coords [ 2 ] ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) / vec2 ( $ { s } . 0 , $ { r } . 0 ) ;
vec4 values = $ { n . texture2D } ( A , uv ) ;
float value ;
if ( depth == 0 ) {
value = values . r ;
} else if ( depth == 1 ) {
value = values . g ;
} else if ( depth == 2 ) {
value = values . b ;
} else if ( depth == 3 ) {
value = values . a ;
}
setOutput ( floor ( value * 255.0 + 0.5 ) ) ;
}
` }};var M8=class{constructor(t){this.variableNames=["A"],this.packedInputs=!1,this.packedOutput=!0;let n=Vn(),[r,s]=t;this.outputShape=t,this.userCode= `
void main ( ) {
ivec3 coords = getOutputCoords ( ) ;
int texR = coords [ 0 ] ;
int texC = coords [ 1 ] ;
int depth = coords [ 2 ] ;
vec4 result = vec4 ( 0. ) ;
for ( int row = 0 ; row <= 1 ; row ++ ) {
for ( int col = 0 ; col <= 1 ; col ++ ) {
texC = coords [ 1 ] + row ;
depth = coords [ 2 ] + col ;
vec2 uv = ( vec2 ( texC , texR ) + halfCR ) /
vec2 ( $ { s } . 0 , $ { r } . 0 ) ;
vec4 values = $ { n . texture2D } ( A , uv ) ;
float value ;
if ( depth == 0 ) {
value = values . r ;
} else if ( depth == 1 ) {
value = values . g ;
} else if ( depth == 2 ) {
value = values . b ;
} else if ( depth == 3 ) {
value = values . a ;
}
result [ row * 2 + col ] = floor ( value * 255.0 + 0.5 ) ;
}
}
$ { n . output } = result ;
}
` }};var Vse={kernelName:Dj,backendName:"webgl",kernelFunc:EZe},f1;function EZe(e){let{inputs:t,backend:n,attrs:r}=e,{pixels:s}=t,{numChannels:o}=r,a=typeof HTMLVideoElement!="undefined"&&s instanceof HTMLVideoElement,i=typeof HTMLImageElement!="undefined"&&s instanceof HTMLImageElement,[l,u]=a?[s.videoWidth,s.videoHeight]:[s.width,s.height],c=[u,l],p=[u,l,o];(i||a)&&(f1==null&&(f1=document.createElement("canvas").getContext("2d")),f1.canvas.width=l,f1.canvas.height=u,f1.drawImage(s,0,0,l,u),s=f1.canvas);let d=n.makeTensorInfo(c,"int32");n.texData.get(d.dataId).usage=Rs.PIXELS,n.gpgpu.uploadPixelDataToTexture(n.getTexture(d.dataId),s);let f=he().getBool("WEBGL_PACK")?new M8(p):new P8(p),h=n.runWebGLProgram(f,[d],"int32");return n.disposeData(d.dataId),h}function RZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o,bias:a,preluActivationWeights:i}=t,{strides:l,pad:u,dataFormat:c,dilations:p,dimRoundingMode:d,activation:f,leakyreluAlpha:h}=r,m=M.convertConv2DDataFormat(c),g=M.computeConv2DInfo(s.shape,o.shape,l,p,u,d,!1,m),x,y=[];if(g.filterHeight===1&&g.filterWidth===1&&g.dilationHeight===1&&g.dilationWidth===1&&g.strideHeight===1&&g.strideWidth===1&&(g.padInfo.type==="SAME"||g.padInfo.type==="VALID"))x=OS({x:s,filter:o,convInfo:g,backend:n,bias:a,activation:f,preluActivationWeights:i,leakyreluAlpha:h});else if(he().getBool("WEBGL_CONV_IM2COL")&&s.shape[0]===1)x=PS({x:s,filter:o,convInfo:g,backend:n,bias:a,activation:f,preluActivationWeights:i,leakyreluAlpha:h});else{let A=a!=null,v=i!=null,w=f==="leakyrelu",k=f?Nd(f,!1):null,I=new xv(g,A,k,v,w),N=[s,o];if(a&&N.push(a),i&&N.push(i),w){let T=n.makeTensorInfo([],"float32",C.createScalarValue(h,"float32"));N.push(T),y.push(T)}x=n.runWebGLProgram(I,N,"float32")}let b= $ e({inputs:{x},backend:n,attrs:{shape:g.outShape}});return y.push(x),y.forEach(A=>n.disposeIntermediateTensorInfo(A)),b}var Wse={kernelName:Fh,backendName:"webgl",kernelFunc:RZe};function DZe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,filter:o,bias:a,preluActivationWeights:i}=t,{strides:l,pad:u,dilations:c,dimRoundingMode:p,activation:d,leakyreluAlpha:f}=r,h=[],m=c;m==null&&(m=[1,1]),C.assert(M.eitherStridesOrDilationsAreOne(l,m),()=> ` Error in depthwiseConv2d : Either strides or dilations must be 1. Got strides $ { l } and dilations '${m}' ` );let g=M.computeConv2DInfo(s.shape,o.shape,l,m,u,p,!0),x=he().getBool("WEBGL_PACK_DEPTHWISECONV")&&g.strideWidth<=2&&g.outChannels/g.inChannels==1,y=d?Nd(d,x):null,b=[s,o],A=a!=null,v=i!=null,w=d==="leakyrelu";if(A&&b.push(a),v&&b.push(i),w){let T=n.makeTensorInfo([],"float32",C.createScalarValue(f,"float32"));b.push(T),h.push(T)}let k;x?k=new bv(g,A,y,v,w):k=new yv(g,A,y,v,w);let I=[[g.padInfo.top,g.padInfo.left],[g.strideHeight,g.strideWidth],[g.dilationHeight,g.dilationWidth],[g.inHeight,g.inWidth]],N=n.runWebGLProgram(k,b,"float32",I);return h.forEach(T=>n.disposeIntermediateTensorInfo(T)),N}var Gse={kernelName:Oh,backendName:"webgl",kernelFunc:DZe};var z8=class{constructor(t,n,r){this.sliceDim=t,this.strides=n,this.variableNames=["x","indices"],this.outputShape=r;let s=Ct(n.length),o=Ct(r.length),a=this.sliceDim>1?"strides[j]":"strides";this.userCode= `
$ { s } strides = $ { s } ( $ { this . strides } ) ;
void main ( ) {
$ { o } coords = getOutputCoords ( ) ;
int flattenIndex = 0 ;
for ( int j = 0 ; j < $ { this . sliceDim } ; j ++ ) {
int index = round ( getIndices ( coords [ 0 ] , j ) ) ;
flattenIndex += index * $ { a } ;
}
setOutput ( getX ( flattenIndex , coords [ 1 ] ) ) ;
}
` }};function $ Ze(e){let{inputs:t,backend:n}=e,{params:r,indices:s}=t,o=s.shape,a=o[o.length-1],i=C.sizeFromShape(r.shape),[l,u,c,p]=M.prepareAndValidate(r,s),d= $ e({inputs:{x:s},backend:n,attrs:{shape:[u,a]}}),f= $ e({inputs:{x:r},backend:n,attrs:{shape:[C.sizeFromShape(r.shape)/c,c]}});if(n.shouldExecuteOnCPU([r,s])||r.dtype==="string"){let x=n.readSync(s.dataId),y=n.bufferSync(r),b=yne(x,y,r.dtype,u,a,c,p,r.shape,i);return n.makeTensorInfo(l,r.dtype,b.values)}let h=new z8(a,p,[u,c]),m=n.runWebGLProgram(h,[f,d],f.dtype),g= $ e({inputs:{x:m},backend:n,attrs:{shape:l}});return n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(m),g}var Use={kernelName:Ox,backendName:"webgl",kernelFunc: $ Ze};var L8=class{constructor(t,n){this.variableNames=["A","indices"],this.outputShape=n,this.rank=n.length;let r=Ct(this.rank),s=FZe(t,2);this.userCode= `
void main ( ) {
$ { r } resRC = getOutputCoords ( ) ;
setOutput ( getA ( $ { s } ) ) ;
}
` }};function FZe(e,t){let n=["resRC.x","resRC.y","resRC.z","resRC.w"],r=[];for(let s=0;s<e.length;s++)s===2?r.push("int(getIndices(resRC.x, resRC.z))"):r.push( ` $ { n [ s ] } ` );return r.join()}function B8(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,indices:o}=t,{axis:a,batchDims:i}=r,l=C.parseAxisParam(a,s.shape)[0],u=M.segment_util.collectGatherOpShapeInfo(s,o,l,i),c=C.sizeFromShape(o.shape),p=[],d= $ e({inputs:{x:s},backend:n,attrs:{shape:[u.batchSize,u.outerSize,u.dimSize,u.sliceSize]}}),f= $ e({inputs:{x:o},backend:n,attrs:{shape:[u.batchSize,c/u.batchSize]}});p.push(d),p.push(f);let h=[u.batchSize,u.outerSize,c/u.batchSize,u.sliceSize];if(n.shouldExecuteOnCPU([s,o])||s.dtype==="string"){let y=n.bufferSync(f),b=n.bufferSync(d),A=bne(b,y,h);return p.forEach(v=>n.disposeIntermediateTensorInfo(v)),n.makeTensorInfo(u.outputShape,A.dtype,A.values)}let m=new L8(d.shape,h),g=n.runWebGLProgram(m,[d,f],d.dtype);p.push(g);let x= $ e({inputs:{x:g},backend:n,attrs:{shape:u.outputShape}});return p.forEach(y=>n.disposeIntermediateTensorInfo(y)),x}var jse={kernelName:_p,backendName:"webgl",kernelFunc:B8};var OZe="return float(a > b);",PZe= `
return vec4 ( greaterThan ( a , b ) ) ;
` ,MZe=Qt({opSnippet:OZe,packedOpSnippet:PZe,cpuKernelImpl:Ane,dtype:"bool"}),Hse={kernelName:Sh,backendName:"webgl",kernelFunc:MZe};var zZe="return float(a >= b);",LZe= `
return vec4 ( greaterThanEqual ( a , b ) ) ;
` ,BZe=Qt({opSnippet:zZe,packedOpSnippet:LZe,dtype:"bool",cpuKernelImpl:vne}),qse={kernelName:yu,backendName:"webgl",kernelFunc:BZe};function VZe(e){let{inputs:t,backend:n}=e,{input:r}=t;return BS(r,!0,n)}var Kse={kernelName:Px,backendName:"webgl",kernelFunc:VZe};var WZe="return float(!isnan(x) && !isinf(x));",GZe=Xe({opSnippet:WZe,dtype:"bool"}),Xse={kernelName:bu,backendName:"webgl",kernelFunc:GZe};var UZe="return float(isinf(x));",jZe=Xe({opSnippet:UZe,dtype:"bool"}),Zse={kernelName:Au,backendName:"webgl",kernelFunc:jZe};var HZe="return float(isnan(x));",qZe=Xe({opSnippet:HZe,dtype:"bool"}),Yse={kernelName:vu,backendName:"webgl",kernelFunc:qZe};var KZe="return float(a < b);",XZe= `
return vec4 ( lessThan ( a , b ) ) ;
` ,ZZe=Qt({opSnippet:KZe,packedOpSnippet:XZe,cpuKernelImpl:wne,dtype:"bool"}),Jse={kernelName:Nh,backendName:"webgl",kernelFunc:ZZe};var YZe="return float(a <= b);",JZe= `
return vec4 ( lessThanEqual ( a , b ) ) ;
` ,QZe=Qt({opSnippet:YZe,packedOpSnippet:JZe,cpuKernelImpl:kne,dtype:"bool"}),Qse={kernelName:Th,backendName:"webgl",kernelFunc:QZe};function eYe(e){let{backend:t,attrs:n}=e,{start:r,stop:s,num:o}=n,a=Ine(r,s,o);return t.makeTensorInfo([a.length],"float32",a)}var eoe={kernelName:zx,backendName:"webgl",kernelFunc:eYe};var tYe= ` if ( x < 0.0 ) return NAN ;
return log ( x ) ; ` ,nYe= `
vec4 result = log ( x ) ;
vec4 isNaN = vec4 ( lessThan ( x , vec4 ( 0.0 ) ) ) ;
result . r = isNaN . r == 1.0 ? NAN : result . r ;
result . g = isNaN . g == 1.0 ? NAN : result . g ;
result . b = isNaN . b == 1.0 ? NAN : result . b ;
result . a = isNaN . a == 1.0 ? NAN : result . a ;
return result ;
` ,rYe=Xe({opSnippet:tYe,packedOpSnippet:nYe,cpuKernelImpl:Cne}),toe={kernelName:wu,backendName:"webgl",kernelFunc:rYe};var sYe="return log(1.0 + x);",oYe=Xe({opSnippet:sYe}),noe={kernelName:ku,backendName:"webgl",kernelFunc:oYe};var aYe="return float(a >= 1.0 && b >= 1.0);",iYe= `
return vec4 (
vec4 ( greaterThanEqual ( a , vec4 ( 1.0 ) ) ) *
vec4 ( greaterThanEqual ( b , vec4 ( 1.0 ) ) ) ) ;
` ,lYe=Qt({opSnippet:aYe,packedOpSnippet:iYe,dtype:"bool"}),roe={kernelName:_h,backendName:"webgl",kernelFunc:lYe};var uYe="return float(!(x >= 1.0));",cYe=Xe({opSnippet:uYe}),soe={kernelName:Eh,backendName:"webgl",kernelFunc:cYe};var pYe="return float(a >= 1.0 || b >= 1.0);",dYe= `
return min (
vec4 ( greaterThanEqual ( a , vec4 ( 1.0 ) ) ) +
vec4 ( greaterThanEqual ( b , vec4 ( 1.0 ) ) ) ,
vec4 ( 1.0 ) ) ;
` ,fYe=Qt({opSnippet:pYe,packedOpSnippet:dYe,dtype:"bool"}),ooe={kernelName:Rh,backendName:"webgl",kernelFunc:fYe};var V8=class{constructor(t,n,r,s,o){this.variableNames=["x"],this.outputShape=[];let a=n,i=t[3]-1;this.outputShape=t;let l,u= ` float ( $ { r } ) + float ( $ { s } ) * sum ` ;o===.5?l= ` inversesqrt ( $ { u } ) ` :o===1?l= ` 1.0 / ( $ { u } ) ` :l= ` exp ( log ( $ { u } ) * float ( - $ { o } ) ) ; ` ,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
int d = coords [ 3 ] ;
float x = getX ( b , r , c , d ) ;
float sum = 0.0 ;
for ( int j = - $ { a } ; j <= $ { a } ; j ++ ) {
int idx = d + j ;
if ( idx >= 0 && idx <= $ { i } ) {
float z = getX ( b , r , c , idx ) ;
sum += z * z ;
}
}
float val = x * $ { l } ;
setOutput ( val ) ;
}
` }};var W8=class{constructor(t,n,r,s,o){this.variableNames=["x"],this.outputShape=[],this.packedInputs=!0,this.packedOutput=!0;let a=n,i=t[3]-1;this.outputShape=t;let l,u= ` float ( $ { r } ) + float ( $ { s } ) * sum ` ;o===.5?l= ` inversesqrt ( $ { u } ) ` :o===1?l= ` 1.0 / ( $ { u } ) ` :l= ` exp ( log ( $ { u } ) * float ( - $ { o } ) ) ; ` ,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords . x ;
int r = coords . y ;
int c = coords . z ;
int d = coords . w ;
bool hasNextCol = d < $ { this . outputShape [ 3 ] } ;
bool hasNextRow = c < $ { this . outputShape [ 2 ] } ;
vec4 sum = vec4 ( 0. ) ;
vec4 xFragAtOutputCoords = getX ( b , r , c , d ) ;
vec4 xAtOutputCoords = vec4 (
getChannel ( xFragAtOutputCoords , vec2 ( c , d ) ) ,
hasNextCol ?
getChannel ( xFragAtOutputCoords , vec2 ( c , d + 1 ) ) : 0.0 ,
hasNextRow ?
getChannel ( xFragAtOutputCoords , vec2 ( c + 1 , d ) ) : 0.0 ,
( hasNextRow && hasNextCol ) ?
getChannel ( xFragAtOutputCoords , vec2 ( c + 1 , d + 1 ) ) : 0.0
) ;
int firstChannel = d - $ { a } ;
vec2 cache = vec2 ( 0. ) ;
if ( firstChannel >= 0 ) {
vec4 firstChannelFrag = getX ( b , r , c , firstChannel ) ;
cache . x = getChannel ( firstChannelFrag , vec2 ( c , firstChannel ) ) ;
if ( hasNextRow ) {
cache . y = getChannel ( firstChannelFrag , vec2 ( c + 1 , firstChannel ) ) ;
}
}
ivec2 depth = ivec2 ( d , d + 1 ) ;
for ( int j = - $ { a } ; j <= $ { a } ; j ++ ) {
ivec2 idx = depth + j ;
bvec2 aboveLowerBound = greaterThanEqual ( idx , ivec2 ( 0 ) ) ;
bvec2 belowUpperBound = lessThanEqual ( idx , ivec2 ( $ { i } ) ) ;
bool depthInRange = aboveLowerBound . x && belowUpperBound . x ;
bool depthPlusOneInRange = aboveLowerBound . y && belowUpperBound . y ;
if ( depthInRange || depthPlusOneInRange ) {
vec4 z = vec4 ( 0. ) ;
vec4 xFragAtCurrentDepth ;
z . xz = cache . xy ;
if ( depthPlusOneInRange && hasNextCol ) {
xFragAtCurrentDepth = idx . y != d ?
getX ( b , r , c , idx . y ) : xFragAtOutputCoords ;
z . y = getChannel ( xFragAtCurrentDepth , vec2 ( c , idx . y ) ) ;
if ( hasNextRow ) {
z . w = getChannel ( xFragAtCurrentDepth , vec2 ( c + 1 , idx . y ) ) ;
}
}
cache . xy = z . yw ;
sum += z * z ;
}
}
vec4 result = xAtOutputCoords * $ { l } ;
setOutput ( result ) ;
}
` }};var hYe=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{depthRadius:o,bias:a,alpha:i,beta:l}=r,u=he().getBool("WEBGL_PACK_NORMALIZATION")?new W8(s.shape,o,a,i,l):new V8(s.shape,o,a,i,l);return n.runWebGLProgram(u,[s],s.dtype)},aoe={kernelName:Rp,backendName:"webgl",kernelFunc:hYe};var G8=class{constructor(t,n,r,s,o){this.variableNames=["inputImage","outputImage","dy"],this.outputShape=[],this.outputShape=t,this.depth=t[3],this.depthRadius=n,this.bias=r,this.alpha=s,this.beta=o,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
float result = 0.0 ;
for ( int d = 0 ; d < $ { this . depth } ; ++ d ) {
int depthBegin = int ( max ( 0.0 , float ( d - $ { n } ) ) ) ;
int depthEnd = int ( min ( float ( $ { this . depth } ) ,
float ( d + $ { n } + 1 ) ) ) ;
const int MIN _DEPTH _BEGIN = 0 ;
const int MAX _DEPTH _END = $ { this . depth } ;
float norm = 0.0 ;
for ( int k = MIN _DEPTH _BEGIN ; k < MAX _DEPTH _END ; ++ k ) {
if ( k < depthBegin ) {
continue ;
}
else if ( k >= depthBegin && k < depthEnd ) {
norm += getInputImage ( b , r , c , k ) * getInputImage ( b , r , c , k ) ;
}
else {
break ;
}
}
norm = float ( $ { s } ) * norm + float ( $ { r } ) ;
for ( int k = MIN _DEPTH _BEGIN ; k < MAX _DEPTH _END ; ++ k ) {
if ( k < depthBegin ) {
continue ;
}
else if ( k >= depthBegin && k < depthEnd ) {
float dyi = - 2.0 * float ( $ { s } )
* float ( $ { o } )
* getInputImage ( b , r , c , k ) * getOutputImage ( b , r , c , d )
/ n o r m ;
if ( k == d ) {
dyi += pow ( norm , - 1.0 * $ { o } ) ;
}
if ( k == coords [ 3 ] ) {
dyi *= getDy ( b , r , c , d ) ;
result += dyi ;
}
}
else {
break ;
}
}
}
setOutput ( result ) ;
}
` }};var mYe=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s,y:o,dy:a}=t,{depthRadius:i,bias:l,alpha:u,beta:c}=r,p=new G8(s.shape,i,l,u,c);return n.runWebGLProgram(p,[s,o,a],s.dtype)},ioe={kernelName:Lx,backendName:"webgl",kernelFunc:mYe};function loe(e,t,n,r){let s=C.sizeFromShape(t),a=C.sizeFromShape(e.shape)/s,i= $ e({inputs:{x:e},attrs:{shape:[a,s]},backend:r}),l=la(i,e.dtype,"max",r),u= $ e({inputs:{x:l},attrs:{shape:n},backend:r});return r.disposeIntermediateTensorInfo(i),r.disposeIntermediateTensorInfo(l),u}function U8(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reductionIndices:o,keepDims:a}=r,i=s.shape.length,l=C.parseAxisParam(o,s.shape),u=l,c=M.getAxesPermutation(u,i),p=c!=null,d=n.shouldExecuteOnCPU([s]),f=s;if(p){if(d){let b=n.texData.get(f.dataId).values,A=new Array(i);for(let k=0;k<A.length;k++)A[k]=s.shape[c[k]];let v=Vm(b,s.shape,s.dtype,c,A);f=n.makeTensorInfo(A,s.dtype);let w=n.texData.get(f.dataId);w.values=v}else f=Td(s,c,n);u=M.getInnerMostAxes(u.length,i)}M.assertAxesAreInnerMostDims("max",u,i);let[h,m]=M.computeOutAndReduceShapes(f.shape,u),g=h;a&&(g=M.expandShapeToKeepDim(h,l));let x;if(d){let b=n.texData.get(f.dataId).values,A=Sne(b,C.sizeFromShape(m),g,s.dtype);x=n.makeTensorInfo(g,s.dtype);let v=n.texData.get(x.dataId);v.values=A}else x=loe(f,m,g,n);return p&&n.disposeIntermediateTensorInfo(f),x}var uoe={kernelName:Dp,backendName:"webgl",kernelFunc:U8};var gYe=TS+ `
return max ( a , b ) ;
` ,xYe= `
vec4 result = vec4 ( max ( a , b ) ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
` +Sd+ `
return result ;
` ,yYe=Qt({opSnippet:gYe,packedOpSnippet:xYe,cpuKernelImpl:Nne}),coe={kernelName:Iu,backendName:"webgl",kernelFunc:yYe};function bYe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t;Rl(s,"maxPool");let{filterSize:o,strides:a,pad:i,dimRoundingMode:l}=r,u=1;C.assert(M.eitherStridesOrDilationsAreOne(a,u),()=> ` Error in maxPool : Either strides or dilations must be 1. Got strides $ { a } and dilations '${u}' ` );let c=M.computePool2DInfo(s.shape,o,a,u,i,l);if(c.filterWidth===1&&c.filterHeight===1&&C.arraysEqual(c.inShape,c.outShape))return lr({inputs:{x:s},backend:n});let p=new Dl(c,"max",!1);return n.runWebGLProgram(p,[s],s.dtype)}var poe={kernelName: $ p,backendName:"webgl",kernelFunc:bYe};function AYe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{filterSize:o,strides:a,pad:i,dataFormat:l,dimRoundingMode:u}=r,c=[1,1,1],p=M.computePool3DInfo(s.shape,o,a,c,i,u,l),d=new Um(p,"max",!1);return n.runWebGLProgram(d,[s],s.dtype)}var doe={kernelName:Fp,backendName:"webgl",kernelFunc:AYe};var j8=class{constructor(t){this.variableNames=["dy","maxPos"],this.outputShape=t.inShape;let n=t.strideHeight,r=t.strideWidth,s=t.dilationHeight,o=t.effectiveFilterHeight,a=t.effectiveFilterWidth,i=o-1-t.padInfo.top,l=a-1-t.padInfo.left,u=o*a-1;this.userCode= `
const ivec2 pads = ivec2 ( $ { i } , $ { l } ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 dyRCCorner = coords . yz - pads ;
int dyRCorner = dyRCCorner . x ;
int dyCCorner = dyRCCorner . y ;
// Convolve dy(?, ?, d) with pos mask(:, :, d) to get dx(xR, xC, d).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wR = 0 ; wR < $ { o } ;
wR += $ { s } ) {
float dyR = float ( dyRCorner + wR ) / $ { n } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 || fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
for ( int wC = 0 ; wC < $ { a } ; wC ++ ) {
float dyC = float ( dyCCorner + wC ) / $ { r } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( b , idyR , idyC , d ) ;
int maxPosValue = $ { u } - int ( getMaxPos ( b , idyR , idyC , d ) ) ;
// Get the current value, check it against the value from the
// position matrix.
int curPosValue = wR * $ { a } + wC ;
float mask = float ( maxPosValue == curPosValue ? 1.0 : 0.0 ) ;
dotProd += dyValue * mask ;
}
}
setOutput ( dotProd ) ;
}
` }},H8=class{constructor(t){this.variableNames=["dy","maxPos"],this.outputShape=t.inShape;let n=t.strideDepth,r=t.strideHeight,s=t.strideWidth,o=t.dilationDepth,a=t.dilationHeight,i=t.dilationWidth,l=t.effectiveFilterDepth,u=t.effectiveFilterHeight,c=t.effectiveFilterWidth,p=l-1-t.padInfo.front,d=u-1-t.padInfo.top,f=c-1-t.padInfo.left,h=l*u*c-1;this.userCode= `
const ivec3 pads = ivec3 ( $ { p } , $ { d } , $ { f } ) ;
void main ( ) {
ivec5 coords = getOutputCoords ( ) ;
int batch = coords . x ;
int ch = coords . u ;
ivec3 dyCorner = ivec3 ( coords . y , coords . z , coords . w ) - pads ;
int dyDCorner = dyCorner . x ;
int dyRCorner = dyCorner . y ;
int dyCCorner = dyCorner . z ;
// Convolve dy(?, ?, ?, ch) with pos mask(:, :, :, d) to get
// dx(xD, xR, xC, ch).
// ? = to be determined. : = across all values in that axis.
float dotProd = 0.0 ;
for ( int wD = 0 ; wD < $ { l } ;
wD += $ { o } ) {
float dyD = float ( dyDCorner + wD ) / $ { n } . 0 ;
if ( dyD < 0.0 || dyD >= $ { t . outDepth } . 0 || fract ( dyD ) > 0.0 ) {
continue ;
}
int idyD = int ( dyD ) ;
for ( int wR = 0 ; wR < $ { u } ;
wR += $ { a } ) {
float dyR = float ( dyRCorner + wR ) / $ { r } . 0 ;
if ( dyR < 0.0 || dyR >= $ { t . outHeight } . 0 ||
fract ( dyR ) > 0.0 ) {
continue ;
}
int idyR = int ( dyR ) ;
for ( int wC = 0 ; wC < $ { c } ;
wC += $ { i } ) {
float dyC = float ( dyCCorner + wC ) / $ { s } . 0 ;
if ( dyC < 0.0 || dyC >= $ { t . outWidth } . 0 ||
fract ( dyC ) > 0.0 ) {
continue ;
}
int idyC = int ( dyC ) ;
float dyValue = getDy ( batch , idyD , idyR , idyC , ch ) ;
int maxPosValue = $ { h } -
int ( getMaxPos ( batch , idyD , idyR , idyC , ch ) ) ;
// Get the current value, check it against the value from the
// position matrix.
int curPosValue =
wD * $ { u } * $ { c } +
wR * $ { c } + wC ;
float mask = float ( maxPosValue == curPosValue ? 1.0 : 0.0 ) ;
dotProd += dyValue * mask ;
}
}
}
setOutput ( dotProd ) ;
}
` }};function vYe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o}=t,a=o,{filterSize:i,strides:l,pad:u,dimRoundingMode:c}=r,p=[1,1,1],d=M.computePool3DInfo(a.shape,i,l,p,u,c),f=new Um(d,"max",!0),h=n.runWebGLProgram(f,[a],a.dtype),m=new H8(d),g=n.runWebGLProgram(m,[s,h],a.dtype);return n.disposeIntermediateTensorInfo(h),g}var foe={kernelName:Vx,backendName:"webgl",kernelFunc:vYe};function wYe(e){let{inputs:t,backend:n,attrs:r}=e,{dy:s,input:o,output:a}=t,i=o;Rl([o,a],"maxPoolGrad");let{filterSize:l,strides:u,pad:c,dimRoundingMode:p}=r,d=M.computePool2DInfo(i.shape,l,u,1,c,p),f=!0,h=new Dl(d,"max",f),m=n.runWebGLProgram(h,[i],i.dtype),g=new j8(d),x=n.runWebGLProgram(g,[s,m],i.dtype);return n.disposeIntermediateTensorInfo(m),x}var hoe={kernelName:Bx,backendName:"webgl",kernelFunc:wYe};function moe(e,t,n,r){let s=new Dl(n,"max",!1),o=r.runWebGLProgram(s,[e],"float32");s=new Dl(n,"max",!0,!0,t);let a=r.runWebGLProgram(s,[e],"float32");return[o,a]}var goe={kernelName:Wx,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{x:r}=e,{filterSize:s,strides:o,pad:a,includeBatchInIndex:i}=t,l=n;C.assert(r.shape.length===4,()=> ` Error in maxPool : input must be rank 4 but got rank $ { r . shape . length } . ` );let u=[1,1];C.assert(M.eitherStridesOrDilationsAreOne(o,u),()=> ` Error in maxPool : Either strides or dilations must be 1. Got strides $ { o } and dilations '${u}' ` );let c=M.computePool2DInfo(r.shape,s,o,u,a),[p,d]=moe(r,i,c,l);return[p,d]}};function xoe(e,t,n,r){let s=C.sizeFromShape(t),a=C.sizeFromShape(e.shape)/s,i= $ e({inputs:{x:e},attrs:{shape:[a,s]},backend:r}),l=la(i,"float32","mean",r),u= $ e({inputs:{x:l},attrs:{shape:n},backend:r});return r.disposeIntermediateTensorInfo(i),r.disposeIntermediateTensorInfo(l),u}var yoe={kernelName:Op,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{x:r}=e,{keepDims:s,axis:o}=t,a=n,i=r.shape.length,l=C.parseAxisParam(o,r.shape),u=l,c=M.getAxesPermutation(u,i),p=c!=null,d=a.shouldExecuteOnCPU([r]),f=[],h=r;if(p){if(d){let A=a.texData.get(h.dataId).values,v=new Array(i);for(let I=0;I<v.length;I++)v[I]=r.shape[c[I]];let w=Vm(A,r.shape,r.dtype,c,v);h=a.makeTensorInfo(v,r.dtype);let k=a.texData.get(h.dataId);k.values=w}else h=Td(r,c,a);f.push(h),u=M.getInnerMostAxes(u.length,i)}M.assertAxesAreInnerMostDims("sum",u,i);let[m,g]=M.computeOutAndReduceShapes(h.shape,u),x=m;s&&(x=M.expandShapeToKeepDim(m,l));let y=xoe(h,g,x,a);for(let b of f)a.disposeIntermediateTensorInfo(b);return y}};function kYe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=C.parseAxisParam(o,s.shape),u=l,c=M.getAxesPermutation(u,i),p=s;c!=null&&(p=On({inputs:{x:s},backend:n,attrs:{perm:c}}),u=M.getInnerMostAxes(u.length,s.shape.length)),M.assertAxesAreInnerMostDims("min",u,i);let[d,f]=M.computeOutAndReduceShapes(p.shape,u),h=C.sizeFromShape(f),m= $ e({inputs:{x:p},backend:n,attrs:{shape:[-1,h]}}),g=la(m,m.dtype,"min",n),x;if(a){let y=M.expandShapeToKeepDim(d,l);x= $ e({inputs:{x:g},backend:n,attrs:{shape:y}})}else x= $ e({inputs:{x:g},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(m),n.disposeIntermediateTensorInfo(g),c!=null&&n.disposeIntermediateTensorInfo(p),x}var boe={kernelName:Pp,backendName:"webgl",kernelFunc:kYe};var IYe=TS+ `
return min ( a , b ) ;
` ,CYe= `
vec4 result = vec4 ( min ( a , b ) ) ;
vec4 isNaN = min ( vec4 ( isnan ( a ) ) + vec4 ( isnan ( b ) ) , vec4 ( 1.0 ) ) ;
` +Sd+ `
return result ;
` ,SYe=Qt({opSnippet:IYe,packedOpSnippet:CYe,cpuKernelImpl:Tne}),Aoe={kernelName:Cu,backendName:"webgl",kernelFunc:SYe};var q8=class{constructor(t,n,r){this.variableNames=["x"],this.outputShape=n.map((c,p)=>c[0]+t[p]+c[1]);let s=t.length,o=Ct(s),a=n.map(c=>c[0]).join(","),i=n.map((c,p)=>c[0]+t[p]).join(","),l=["coords[0]","coords[1]","coords[2]","coords[3]"].slice(0,s),u=r==="reflect"?0:1;if(s===1){this.userCode= `
int start = $ { a } ;
int end = $ { i } ;
void main ( ) {
int outC = getOutputCoords ( ) ;
if ( outC < start ) {
outC = start * 2 - outC - $ { u } ;
} else if ( outC >= end ) {
outC = ( end - 1 ) * 2 - outC + $ { u } ;
}
setOutput ( getX ( outC - start ) ) ;
}
` ;return}this.userCode= `
$ { o } start = $ { o } ( $ { a } ) ;
$ { o } end = $ { o } ( $ { i } ) ;
void main ( ) {
$ { o } outC = getOutputCoords ( ) ;
for ( int i = 0 ; i < $ { s } ; i ++ ) {
if ( outC [ i ] < start [ i ] ) {
outC [ i ] = start [ i ] * 2 - outC [ i ] - $ { u } ;
} else if ( outC [ i ] >= end [ i ] ) {
outC [ i ] = ( end [ i ] - 1 ) * 2 - outC [ i ] + $ { u } ;
}
}
$ { o } coords = outC - start ;
setOutput ( getX ( $ { l } ) ) ;
}
` }};var K8=class{constructor(t,n,r){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=n.map((h,m)=>h[0]+t[m]+h[1]);let s=t.length,o=Ct(s),a=n.map(h=>h[0]).join(","),i=n.map((h,m)=>h[0]+t[m]).join(","),l=ir("rc",s),u=ir("source",s),c= ` $ { l [ s - 1 ] } < $ { this . outputShape [ s - 1 ] } ` ,p=s===1?"source": ` vec2 ( $ { u . slice ( - 2 ) . join ( ) } ) ` ,d=r==="reflect"?0:1,f="";if(s===1){let h= `
$ { o } source = rc ;
if ( source < start ) {
source = start * 2 - source - $ { d } ;
} else if ( source >= end ) {
source = ( end - 1 ) * 2 - source + $ { d } ;
}
source -= start ;
` ;f= `
$ { o } rc = outputLoc ;
$ { h }
result [ 0 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
$ { l [ s - 1 ] } += 1 ;
if ( $ { c } ) {
$ { h }
result [ 1 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
}
` }else{let h= `
$ { o } source = rc ;
$ { o } lt = $ { o } ( lessThan ( source , start ) ) ;
$ { o } gte = $ { o } ( greaterThanEqual ( source , end ) ) ;
$ { o } orig = 1 - ( lt + gte ) ;
source = orig * source +
lt * ( start * 2 - source - $ { d } ) +
gte * ( ( end - 1 ) * 2 - source + $ { d } ) ;
source -= start ;
` ;f= `
$ { o } rc = outputLoc ;
$ { h }
result [ 0 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
$ { l [ s - 1 ] } += 1 ;
if ( $ { c } ) {
$ { h }
result [ 1 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
}
rc = outputLoc ;
$ { l [ s - 2 ] } += 1 ;
if ( $ { l [ s - 2 ] } < $ { this . outputShape [ s - 2 ] } ) {
$ { h }
result [ 2 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
$ { l [ s - 1 ] } += 1 ;
if ( $ { c } ) {
$ { h }
result [ 3 ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
}
}
` }this.userCode= `
const $ { o } start = $ { o } ( $ { a } ) ;
const $ { o } end = $ { o } ( $ { i } ) ;
void main ( ) {
$ { o } outputLoc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
$ { f }
setOutput ( result ) ;
}
` }};var NYe=({inputs:e,backend:t,attrs:n})=>{let{x:r}=e,{paddings:s,mode:o}=n,a=he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new K8(r.shape,s,o):new q8(r.shape,s,o);return t.runWebGLProgram(a,[r],r.dtype)},voe={kernelName:Mp,backendName:"webgl",kernelFunc:NYe};var TYe= ` if ( b == 0.0 ) return NAN ;
return mod ( a , b ) ; ` ,_Ye= `
vec4 result = mod ( a , b ) ;
vec4 isNaN = vec4 ( equal ( b , vec4 ( 0.0 ) ) ) ;
` +Sd+ `
return result ;
` ,EYe=Qt({opSnippet:TYe,packedOpSnippet:_Ye}),woe={kernelName:Su,backendName:"webgl",kernelFunc:EYe};var X8=class{constructor(t,n,r){this.variableNames=["probs"],this.customUniforms=[{name:"seed",type:"float"}],this.outputShape=[t,r],this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
float r = random ( seed ) ;
float cdf = 0.0 ;
for ( int i = 0 ; i < $ { n - 1 } ; i ++ ) {
cdf += getProbs ( batch , i ) ;
if ( r < cdf ) {
setOutput ( float ( i ) ) ;
return ;
}
}
// If no other event happened, last event happened.
setOutput ( float ( $ { n - 1 } ) ) ;
}
` }};var RYe= `
if ( a == b ) {
return 1.0 ;
} ;
return a / b ; ` ,DYe= `
// vec4 one = vec4(equal(a, b));
// return one + (vec4(1.0) - one) * a / b;
vec4 result = a / b ;
if ( a . x == b . x ) {
result . x = 1. ;
}
if ( a . y == b . y ) {
result . y = 1. ;
}
if ( a . z == b . z ) {
result . z = 1. ;
}
if ( a . w == b . w ) {
result . w = 1. ;
}
return result ;
` ,Z8=Qt({opSnippet:RYe,packedOpSnippet:DYe,checkOutOfBounds:!0}),koe={kernelName:pu,backendName:"webgl",kernelFunc:Z8};var Ioe="return a - b;",Y8=Qt({opSnippet:Ioe,packedOpSnippet:Ioe,supportsComplex:!0,cpuKernelImpl:Une}),Coe={kernelName:Wu,backendName:"webgl",kernelFunc:Y8};function J8(e){let{inputs:t,backend:n,attrs:r}=e,{logits:s}=t,{dim:o}=r,a=C.parseAxisParam([o],s.shape),i=U8({inputs:{x:s},backend:n,attrs:{reductionIndices:a,keepDims:!1}}),l=M.expandShapeToKeepDim(i.shape,a),u= $ e({inputs:{x:i},backend:n,attrs:{shape:l}}),c=Y8({inputs:{a:s,b:u},backend:n}),p= $ 8({inputs:{x:c},backend:n}),d=Wm({inputs:{x:p},backend:n,attrs:{axis:a,keepDims:!1}}),f= $ e({inputs:{x:d},backend:n,attrs:{shape:l}}),h=Z8({inputs:{a:p,b:f},backend:n});return n.disposeIntermediateTensorInfo(i),n.disposeIntermediateTensorInfo(u),n.disposeIntermediateTensorInfo(c),n.disposeIntermediateTensorInfo(p),n.disposeIntermediateTensorInfo(d),n.disposeIntermediateTensorInfo(f),h}var Soe={kernelName:Qp,backendName:"webgl",kernelFunc:J8};function $ Ye(e){let{inputs:t,backend:n,attrs:r}=e,{logits:s}=t,{numSamples:o,seed:a,normalized:i}=r,l=i?s:J8({inputs:{logits:s},backend:n,attrs:{dim:s.shape.length-1}}),u=l.shape[0],c=l.shape[1],p=new X8(u,c,o),d=[[a]],f=n.runWebGLProgram(p,[l],"int32",d);return i||n.disposeIntermediateTensorInfo(l),f}var Noe={kernelName:Gx,backendName:"webgl",kernelFunc: $ Ye};var Toe="return -x;";function FYe(e){let{inputs:t,backend:n}=e,{x:r}=t;if(n.shouldExecuteOnCPU([r])){let o=n.texData.get(r.dataId),[a,i]=Ene(o.values,r.shape,r.dtype);return n.makeTensorInfo(i,r.dtype,a)}let s;return he().getBool("WEBGL_PACK_UNARY_OPERATIONS")?s=new Ji(r.shape,Toe):s=new $ o(r.shape,Toe),n.runWebGLProgram(s,[r],r.dtype)}var _oe={kernelName:zp,backendName:"webgl",kernelFunc:FYe};var OYe=Ts.nonMaxSuppressionV3Impl;function PYe(e){M.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l}=r,u=n.readSync(s.dataId),c=n.readSync(o.dataId),{selectedIndices:p}=OYe(u,c,a,i,l);return n.makeTensorInfo([p.length],"int32",new Int32Array(p))}var Eoe={kernelName:Ux,backendName:"webgl",kernelFunc:PYe};var MYe=Ts.nonMaxSuppressionV4Impl;function zYe(e){M.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l,padToMaxOutputSize:u}=r,c=n.readSync(s.dataId),p=n.readSync(o.dataId),{selectedIndices:d,validOutputs:f}=MYe(c,p,a,i,l,u);return[n.makeTensorInfo([d.length],"int32",new Int32Array(d)),n.makeTensorInfo([],"int32",new Int32Array([f]))]}var Roe={kernelName:jx,backendName:"webgl",kernelFunc:zYe};var LYe=Ts.nonMaxSuppressionV5Impl;function BYe(e){M.warn("tf.nonMaxSuppression() in webgl locks the UI thread. Call tf.nonMaxSuppressionAsync() instead");let{inputs:t,backend:n,attrs:r}=e,{boxes:s,scores:o}=t,{maxOutputSize:a,iouThreshold:i,scoreThreshold:l,softNmsSigma:u}=r,c=n.readSync(s.dataId),p=n.readSync(o.dataId),d=a,f=i,h=l,m=u,{selectedIndices:g,selectedScores:x}=LYe(c,p,d,f,h,m);return[n.makeTensorInfo([g.length],"int32",new Int32Array(g)),n.makeTensorInfo([x.length],"float32",new Float32Array(x))]}var Doe={kernelName:Hx,backendName:"webgl",kernelFunc:BYe};var Q8=class{constructor(t,n,r,s){this.variableNames=["indices"],this.outputShape=[t,n],this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int index = round ( getIndices ( coords . x ) ) ;
setOutput ( mix ( float ( $ { s } ) , float ( $ { r } ) ,
float ( index == coords . y ) ) ) ;
}
` }};var VYe=e=>{let{inputs:t,backend:n,attrs:r}=e,{indices:s}=t,{depth:o,onValue:a,offValue:i}=r,l=C.sizeFromShape(s.shape),u=new Q8(l,o,a,i),c= $ e({inputs:{x:s},backend:n,attrs:{shape:[l]}}),p=n.runWebGLProgram(u,[c],s.dtype);n.disposeIntermediateTensorInfo(c);let d=[...s.shape,o],f= $ e({inputs:{x:p},backend:n,attrs:{shape:d}});return n.disposeIntermediateTensorInfo(p),f}, $ oe={kernelName:Bp,backendName:"webgl",kernelFunc:VYe};function Av(e){let{inputs:t,backend:n}=e,{x:r}=t;if(r.dtype==="complex64"){let s=Ac({inputs:{input:r},backend:n}),o=Av({inputs:{x:s},backend:n}),a=jm({inputs:{input:r},backend:n}),i=Av({inputs:{x:a},backend:n}),l=Fo({inputs:{real:o,imag:i},backend:n});return n.disposeIntermediateTensorInfo(s),n.disposeIntermediateTensorInfo(o),n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(i),l}else return vc({attrs:{shape:r.shape,dtype:r.dtype,value:r.dtype==="string"?"":0},backend:n})}var Foe={kernelName:rd,backendName:"webgl",kernelFunc:Av};function Ooe(e){let{inputs:t,backend:n}=e,{x:r}=t;if(r.dtype==="string")throw new Error("onesLike is not supported under string dtype");if(r.dtype==="complex64"){let s=Ac({inputs:{input:r},backend:n}),o=Ooe({inputs:{x:s},backend:n}),a=jm({inputs:{input:r},backend:n}),i=Av({inputs:{x:a},backend:n}),l=Fo({inputs:{real:o,imag:i},backend:n});return n.disposeIntermediateTensorInfo(s),n.disposeIntermediateTensorInfo(o),n.disposeIntermediateTensorInfo(a),n.disposeIntermediateTensorInfo(i),l}else return vc({attrs:{shape:r.shape,dtype:r.dtype,value:1},backend:n})}var Poe={kernelName:Lp,backendName:"webgl",kernelFunc:Ooe};function WYe(e){let{inputs:t,backend:n,attrs:r}=e,{axis:s}=r;if(t.length===1)return zS({inputs:{input:t[0]},backend:n,attrs:{dim:s}});let o=t[0].shape,a=t[0].dtype;t.forEach(c=>{C.assertShapesMatch(o,c.shape,"All tensors passed to stack must have matching shapes"),C.assert(a===c.dtype,()=>"All tensors passed to stack must have matching dtypes")});let i=[],l=t.map(c=>{let p=zS({inputs:{input:c},backend:n,attrs:{dim:s}});return i.push(p),p}),u=A8({inputs:l,backend:n,attrs:{axis:s}});return i.forEach(c=>n.disposeIntermediateTensorInfo(c)),u}var Moe={kernelName:Vp,backendName:"webgl",kernelFunc:WYe};var eO=class{constructor(t,n,r){this.variableNames=["x"],this.customUniforms=[{name:"value",type:"float"}],this.outputShape=n.map((u,c)=>u[0]+t[c]+u[1]);let s=t.length,o=Ct(s),a=n.map(u=>u[0]).join(","),i=n.map((u,c)=>u[0]+t[c]).join(","),l=["coords[0]","coords[1]","coords[2]","coords[3]"].slice(0,s);if(s===1){this.userCode= `
int start = $ { a } ;
int end = $ { i } ;
void main ( ) {
int outC = getOutputCoords ( ) ;
if ( outC < start || outC >= end ) {
setOutput ( value ) ;
} else {
setOutput ( getX ( outC - start ) ) ;
}
}
` ;return}this.userCode= `
$ { o } start = $ { o } ( $ { a } ) ;
$ { o } end = $ { o } ( $ { i } ) ;
void main ( ) {
$ { o } outC = getOutputCoords ( ) ;
if ( any ( lessThan ( outC , start ) ) || any ( greaterThanEqual ( outC , end ) ) ) {
setOutput ( value ) ;
} else {
$ { o } coords = outC - start ;
setOutput ( getX ( $ { l } ) ) ;
}
}
` }};var tO=class{constructor(t,n,r){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0,this.customUniforms=[{name:"value",type:"float"}],this.outputShape=n.map((m,g)=>m[0]+t[g]+m[1]);let s=t.length,o=Ct(s),a=n.map(m=>m[0]).join(","),i=n.map((m,g)=>m[0]+t[g]).join(","),l=ir("rc",s),u=ir("source",s),c= ` $ { l [ s - 1 ] } < $ { this . outputShape [ s - 1 ] } ` ,p=s===1?"source": ` vec2 ( $ { u . slice ( - 2 ) . join ( ) } ) ` ,d=[ ` $ { o } rc = outputLoc ; ` , ` $ { l [ s - 1 ] } += 1 ;
if ( $ { c } ) {
` ,s===1?"": ` }
rc = outputLoc ;
$ { l [ s - 2 ] } += 1 ;
if ( $ { l [ s - 2 ] } < $ { this . outputShape [ s - 2 ] } ) { ` ,s===1?"": ` $ { l [ s - 1 ] } += 1 ;
if ( $ { c } ) { ` ],f=s===1?"rc < start || rc >= end":"any(lessThan(rc, start)) || any(greaterThanEqual(rc, end))",h="";for(let m=0,g=s===1?2:4;m<g;m++)h+= `
$ { d [ m ] }
if ( $ { f } ) {
result [ $ { m } ] = float ( value ) ;
} else {
$ { o } source = rc - start ;
result [ $ { m } ] = getChannel ( getX ( $ { u . join ( ) } ) , $ { p } ) ;
}
` ;h+=s===1?"} ":"}}",this.userCode= `
const $ { o } start = $ { o } ( $ { a } ) ;
const $ { o } end = $ { o } ( $ { i } ) ;
void main ( ) {
$ { o } outputLoc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
$ { h }
setOutput ( result ) ;
}
` }};var nO=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{paddings:o,constantValue:a}=r;if(C.sizeFromShape(s.shape)===0){let u=o.map((c,p)=>c[0]+s.shape[p]+c[1]);return vc({backend:n,attrs:{shape:u,value:a,dtype:s.dtype}})}let i=he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new tO(s.shape,o,a):new eO(s.shape,o,a),l=[[a]];return n.runWebGLProgram(i,[s],s.dtype,l)},zoe={kernelName:Wp,backendName:"webgl",kernelFunc:nO};var GYe= `
if ( a < 0.0 && floor ( b ) < b ) {
return NAN ;
}
if ( b == 0.0 ) {
return 1.0 ;
}
return ( round ( mod ( b , 2.0 ) ) != 1 ) ?
pow ( abs ( a ) , b ) : sign ( a ) * pow ( abs ( a ) , b ) ;
` ,UYe= `
// isModRound1 has 1 for components with round(mod(b, 2.0)) == 1, 0 otherwise.
vec4 isModRound1 = vec4 ( equal ( round ( mod ( b , 2.0 ) ) , ivec4 ( 1 ) ) ) ;
vec4 multiplier = sign ( a ) * isModRound1 + ( vec4 ( 1.0 ) - isModRound1 ) ;
vec4 result = multiplier * pow ( abs ( a ) , b ) ;
// Ensure that a^0 = 1, including 0^0 = 1 as this correspond to TF and JS
bvec4 isExpZero = equal ( b , vec4 ( 0.0 ) ) ;
result . r = isExpZero . r ? 1.0 : result . r ;
result . g = isExpZero . g ? 1.0 : result . g ;
result . b = isExpZero . b ? 1.0 : result . b ;
result . a = isExpZero . a ? 1.0 : result . a ;
vec4 isNaN = vec4 ( lessThan ( a , vec4 ( 0.0 ) ) ) * vec4 ( lessThan ( floor ( b ) , b ) ) ;
` +Sd+ `
return result ;
` ,jYe=Qt({opSnippet:GYe,packedOpSnippet:UYe}),Loe={kernelName:Tu,backendName:"webgl",kernelFunc:jYe};function HYe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{axis:o,keepDims:a}=r,i=s.shape.length,l=[],u=C.parseAxisParam(o,s.shape),c=u,p=M.getAxesPermutation(c,i),d=s;p!=null&&(d=On({inputs:{x:s},backend:n,attrs:{perm:p}}),c=M.getInnerMostAxes(c.length,i),l.push(d)),M.assertAxesAreInnerMostDims("prod",c,i);let f;if(n.shouldExecuteOnCPU([d])){let h=n.texData.get(d.dataId).values,{outVals:m,outShape:g,outDtype:x}=Dne(d.shape,d.dtype,h,c);f=n.makeTensorInfo(g,x,m)}else{let[h,m]=M.computeOutAndReduceShapes(d.shape,c),g=C.sizeFromShape(m),x= $ e({inputs:{x:d},backend:n,attrs:{shape:[-1,g]}}),y=Vh(s.dtype),b=la(x,y,"prod",n);f= $ e({inputs:{x:b},backend:n,attrs:{shape:h}}),l.push(x),l.push(b)}if(a){l.push(f);let h=M.expandShapeToKeepDim(f.shape,u);f= $ e({inputs:{x:f},backend:n,attrs:{shape:h}})}return l.forEach(h=>n.disposeIntermediateTensorInfo(h)),f}var Boe={kernelName:qx,backendName:"webgl",kernelFunc:HYe};var rO=e=>{let{backend:t,attrs:n}=e,{start:r,stop:s,step:o,dtype:a}=n,i= $ ne(r,s,o,a);return t.makeTensorInfo([i.length],a,i)},Voe={kernelName:Kx,backendName:"webgl",kernelFunc:rO};var qYe="return 1.0 / x;",KYe=Xe({opSnippet:qYe}),Woe={kernelName:_u,backendName:"webgl",kernelFunc:KYe};var XYe=Qr+ `
return ( x < 0.0 ) ? 0.0 : x ;
` ,ZYe= `
vec4 result = x * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
` ,YYe=Xe({opSnippet:XYe,packedOpSnippet:ZYe}),Goe={kernelName:Eu,backendName:"webgl",kernelFunc:YYe};var JYe=Qr+ `
return ( x < 0.0 ) ? 0.0 : min ( 6.0 , x ) ;
` ,QYe= `
vec4 result = min ( x , vec4 ( 6. ) ) * vec4 ( greaterThanEqual ( x , vec4 ( 0.0 ) ) ) ;
bvec4 isNaN = isnan ( x ) ;
result . r = isNaN . r ? x . r : result . r ;
result . g = isNaN . g ? x . g : result . g ;
result . b = isNaN . b ? x . b : result . b ;
result . a = isNaN . a ? x . a : result . a ;
return result ;
` ,eJe=Xe({opSnippet:JYe,packedOpSnippet:QYe}),Uoe={kernelName:Ru,backendName:"webgl",kernelFunc:eJe};var sO=class{constructor(t,n,r,s,o){this.variableNames=["A"],this.outputShape=[];let[a,i,l,u]=t;this.outputShape=[a,n,r,u];let c=[s&&n>1?i-1:i,s&&r>1?l-1:l],p=[s&&n>1?n-1:n,s&&r>1?r-1:r],d;o?d="(vec2(yRC) + vec2(0.5)) * effectiveInputOverOutputRatioRC - vec2(0.5)":d="vec2(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
const vec2 effectiveInputOverOutputRatioRC = vec2 (
$ { c [ 0 ] / p [ 0 ] } ,
$ { c [ 1 ] / p [ 1 ] } ) ;
const vec2 inputShapeRC = vec2 ( $ { i } . 0 , $ { l } . 0 ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 yRC = coords . yz ;
// Fractional source index.
vec2 sourceFracIndexRC = $ { d } ;
// Compute the four integer indices.
ivec2 sourceFloorRC = ivec2 ( max ( sourceFracIndexRC , vec2 ( 0.0 ) ) ) ;
ivec2 sourceCeilRC = ivec2 (
min ( inputShapeRC - 1.0 , ceil ( sourceFracIndexRC ) ) ) ;
float topLeft = getA ( b , sourceFloorRC . x , sourceFloorRC . y , d ) ;
float bottomLeft = getA ( b , sourceCeilRC . x , sourceFloorRC . y , d ) ;
float topRight = getA ( b , sourceFloorRC . x , sourceCeilRC . y , d ) ;
float bottomRight = getA ( b , sourceCeilRC . x , sourceCeilRC . y , d ) ;
vec2 fracRC = sourceFracIndexRC - vec2 ( sourceFloorRC ) ;
float top = topLeft + ( topRight - topLeft ) * fracRC . y ;
float bottom = bottomLeft + ( bottomRight - bottomLeft ) * fracRC . y ;
float newValue = top + ( bottom - top ) * fracRC . x ;
setOutput ( newValue ) ;
}
` }};var oO=class{constructor(t,n,r,s,o){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[];let[a,i,l,u]=t;this.outputShape=[a,n,r,u];let c=[s&&n>1?i-1:i,s&&r>1?l-1:l],p=[s&&n>1?n-1:n,s&&r>1?r-1:r],d;o?d="(vec3(yRC) + vec3(0.5)) * effectiveInputOverOutputRatioRC - vec3(0.5)":d="vec3(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
const vec3 effectiveInputOverOutputRatioRC = vec3 (
$ { c [ 0 ] / p [ 0 ] } ,
$ { c [ 1 ] / p [ 1 ] } ,
$ { c [ 1 ] / p [ 1 ] } ) ;
const vec3 inputShapeRC = vec3 ( $ { i } . 0 , $ { l } . 0 ,
$ { l } . 0 ) ;
float getAValue ( int b , int r , int c , int d ) {
return getChannel ( getA ( b , r , c , d ) , vec2 ( c , d ) ) ;
}
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
// Calculate values for next column in yRC.z.
ivec3 yRC = coords . yzz + ivec3 ( 0 , 0 , 1 ) ;
// Fractional source index.
vec3 sourceFracIndexRC = $ { d } ;
// Compute the four integer indices.
ivec3 sourceFloorRC = ivec3 ( max ( sourceFracIndexRC , vec3 ( 0.0 ) ) ) ;
ivec3 sourceCeilRC = ivec3 (
min ( inputShapeRC - 1.0 , ceil ( sourceFracIndexRC ) ) ) ;
// Should we calculate next column and row elements in 2x2 packed cell.
bool hasNextCol = d < $ { u - 1 } ;
bool hasNextRow = coords . z < $ { r - 1 } ;
// In parallel, construct four corners for all four components in
// packed 2x2 cell.
vec4 topLeft = vec4 (
getAValue ( b , sourceFloorRC . x , sourceFloorRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceFloorRC . x , sourceFloorRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceFloorRC . x , sourceFloorRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceFloorRC . x , sourceFloorRC . z , d + 1 ) : 0.0 ) ;
vec4 bottomLeft = vec4 (
getAValue ( b , sourceCeilRC . x , sourceFloorRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceCeilRC . x , sourceFloorRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceCeilRC . x , sourceFloorRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceCeilRC . x , sourceFloorRC . z , d + 1 ) : 0.0 ) ;
vec4 topRight = vec4 (
getAValue ( b , sourceFloorRC . x , sourceCeilRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceFloorRC . x , sourceCeilRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceFloorRC . x , sourceCeilRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceFloorRC . x , sourceCeilRC . z , d + 1 ) : 0.0 ) ;
vec4 bottomRight = vec4 (
getAValue ( b , sourceCeilRC . x , sourceCeilRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceCeilRC . x , sourceCeilRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceCeilRC . x , sourceCeilRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceCeilRC . x , sourceCeilRC . z , d + 1 ) : 0.0 ) ;
vec3 fracRC = sourceFracIndexRC - vec3 ( sourceFloorRC ) ;
vec4 top = mix ( topLeft , topRight , fracRC . yyzz ) ;
vec4 bottom = mix ( bottomLeft , bottomRight , fracRC . yyzz ) ;
vec4 newValue = mix ( top , bottom , fracRC . x ) ;
setOutput ( newValue ) ;
}
` }};function tJe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s}=t,{alignCorners:o,halfPixelCenters:a,size:i}=r,[l,u]=i,c=he().getBool("WEBGL_PACK_IMAGE_OPERATIONS")?new oO(s.shape,l,u,o,a):new sO(s.shape,l,u,o,a);return n.runWebGLProgram(c,[s],"float32")}var joe={kernelName:Hp,backendName:"webgl",kernelFunc:tJe};var aO=class{constructor(t,n,r){this.variableNames=["dy"],this.outputShape=[],this.outputShape=n;let[,s,o]=n,[,a,i]=t,l=[r&&a>1?s-1:s,r&&i>1?o-1:o],u=[r&&a>1?a-1:a,r&&i>1?i-1:i],c=l[0]/u[0],p=l[1]/u[1],d=1/c,f=1/p,h=Math.ceil(d)*2+2,m=Math.ceil(f)*2+2;this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
float accumulator = 0.0 ;
const float heightScale = float ( $ { c } ) ;
const float widthScale = float ( $ { p } ) ;
const float invHeightScale = float ( $ { d } ) ;
const float invWidthScale = float ( $ { f } ) ;
const int winHeight = int ( $ { h } ) ;
const int winWidth = int ( $ { m } ) ;
// Compute bounds for where in dy we will look
float startRLerp = floor ( float ( r ) * invHeightScale ) ;
int startDyR = int ( startRLerp - float ( winHeight / 2 ) ) ;
float startCLerp = floor ( float ( c ) * invWidthScale ) ;
int startDyC = int ( startCLerp - float ( winWidth / 2 ) ) ;
// Loop over dy
for ( int dyROffset = 0 ; dyROffset < winHeight ; dyROffset ++ ) {
int dyR = dyROffset + startDyR ;
// Guard against the window exceeding the bounds of dy
if ( dyR < 0 || dyR >= $ { a } ) {
continue ;
}
for ( int dyCOffset = 0 ; dyCOffset < winWidth ; dyCOffset ++ ) {
int dyC = dyCOffset + startDyC ;
// Guard against the window exceeding the bounds of dy
if ( dyC < 0 || dyC >= $ { i } ) {
continue ;
}
float dxR = float ( dyR ) * heightScale ;
int topDxRIndex = int ( floor ( dxR ) ) ;
int bottomDxRIndex = int ( min ( ceil ( dxR ) , $ { s - 1 } . 0 ) ) ;
float dxRLerp = dxR - float ( topDxRIndex ) ;
float inverseDxRLerp = 1.0 - dxRLerp ;
float dxC = float ( dyC ) * widthScale ;
int leftDxCIndex = int ( floor ( dxC ) ) ;
int rightDxCIndex = int ( min ( ceil ( dxC ) , $ { o - 1 } . 0 ) ) ;
float dxCLerp = dxC - float ( leftDxCIndex ) ;
float inverseDxCLerp = 1.0 - dxCLerp ;
if ( r == topDxRIndex && c == leftDxCIndex ) {
// topLeft
accumulator +=
getDy ( b , dyR , dyC , d ) * inverseDxRLerp * inverseDxCLerp ;
}
if ( r == topDxRIndex && c == rightDxCIndex ) {
// topRight
accumulator += getDy ( b , dyR , dyC , d ) * inverseDxRLerp * dxCLerp ;
}
if ( r == bottomDxRIndex && c == leftDxCIndex ) {
// bottomLeft
accumulator += getDy ( b , dyR , dyC , d ) * dxRLerp * inverseDxCLerp ;
}
if ( r == bottomDxRIndex && c == rightDxCIndex ) {
// bottomRight
accumulator += getDy ( b , dyR , dyC , d ) * dxRLerp * dxCLerp ;
}
}
}
// End loop over dy
setOutput ( accumulator ) ;
}
` }};function nJe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s,dy:o}=t,{alignCorners:a}=r,i=new aO(o.shape,s.shape,a);return n.runWebGLProgram(i,[o],o.dtype)}var Hoe={kernelName:Yx,backendName:"webgl",kernelFunc:nJe};var iO=class{constructor(t,n,r,s,o){this.variableNames=["A"],this.outputShape=[];let[a,i,l,u]=t;this.outputShape=[a,n,r,u];let c=[s&&n>1?i-1:i,s&&r>1?l-1:l],p=[s&&n>1?n-1:n,s&&r>1?r-1:r],d=s?"0.5":"0.0",f;o?f="max((vec2(yRC) + vec2(0.5)) * effectiveInputOverOutputRatioRC, vec2(0.0))":f="vec2(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
const vec2 effectiveInputOverOutputRatioRC = vec2 (
$ { c [ 0 ] / p [ 0 ] } ,
$ { c [ 1 ] / p [ 1 ] } ) ;
const vec2 inputShapeRC = vec2 ( $ { i } . 0 , $ { l } . 0 ) ;
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
ivec2 yRC = coords . yz ;
// Fractional source index.
vec2 sourceFracIndexRC = $ { f } ;
// Compute the coordinators of nearest neighbor point.
ivec2 sourceNearestRC = ivec2 (
min ( inputShapeRC - 1.0 , floor ( sourceFracIndexRC + $ { d } ) ) ) ;
float newValue = getA ( b , sourceNearestRC . x , sourceNearestRC . y , d ) ;
setOutput ( newValue ) ;
}
` }};var lO=class{constructor(t,n,r,s,o){this.variableNames=["A"],this.packedInputs=!0,this.packedOutput=!0,this.outputShape=[];let[a,i,l,u]=t;this.outputShape=[a,n,r,u];let c=[s&&n>1?i-1:i,s&&r>1?l-1:l],p=[s&&n>1?n-1:n,s&&r>1?r-1:r],d=s?"0.5":"0.0",f;o?f="max((vec3(yRC) + vec3(0.5)) * effectiveInputOverOutputRatioRC, vec3(0.0))":f="vec3(yRC) * effectiveInputOverOutputRatioRC",this.userCode= `
const vec3 effectiveInputOverOutputRatioRC = vec3 (
$ { c [ 0 ] / p [ 0 ] } ,
$ { c [ 1 ] / p [ 1 ] } ,
$ { c [ 1 ] / p [ 1 ] } ) ;
const vec3 inputShapeRC = vec3 ( $ { i } . 0 , $ { l } . 0 ,
$ { l } . 0 ) ;
float getAValue ( int b , int r , int c , int d ) {
return getChannel ( getA ( b , r , c , d ) , vec2 ( c , d ) ) ;
}
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
// Calculate values for next column in yRC.z.
ivec3 yRC = coords . yzz + ivec3 ( 0 , 0 , 1 ) ;
// Fractional source index.
vec3 sourceFracIndexRC = $ { f } ;
// Compute the coordinators of nearest neighbor point.
ivec3 sourceNearestRC = ivec3 (
min ( inputShapeRC - 1.0 , floor ( sourceFracIndexRC + $ { d } ) ) ) ;
// Should we calculate next column and row elements in 2x2 packed cell.
bool hasNextCol = d < $ { u - 1 } ;
bool hasNextRow = coords . z < $ { r - 1 } ;
vec4 newValue = vec4 (
getAValue ( b , sourceNearestRC . x , sourceNearestRC . y , d ) ,
hasNextCol ? getAValue ( b , sourceNearestRC . x , sourceNearestRC . y , d + 1 )
: 0.0 ,
hasNextRow ? getAValue ( b , sourceNearestRC . x , sourceNearestRC . z , d )
: 0.0 ,
( hasNextRow && hasNextCol ) ?
getAValue ( b , sourceNearestRC . x , sourceNearestRC . z , d + 1 ) : 0.0 ) ;
setOutput ( newValue ) ;
}
` }};function rJe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s}=t,{alignCorners:o,halfPixelCenters:a,size:i}=r,[l,u]=i,c=he().getBool("WEBGL_PACK_IMAGE_OPERATIONS")?new lO(s.shape,l,u,o,a):new iO(s.shape,l,u,o,a);return n.runWebGLProgram(c,[s],s.dtype)}var qoe={kernelName:jp,backendName:"webgl",kernelFunc:rJe};var uO=class{constructor(t,n,r){this.variableNames=["dy"],this.outputShape=[],this.outputShape=n;let[,s,o]=n,[,a,i]=t,l=[r&&a>1?s-1:s,r&&i>1?o-1:o],u=[r&&a>1?a-1:a,r&&i>1?i-1:i],c=l[0]/u[0],p=l[1]/u[1],d=1/c,f=1/p,h=Math.ceil(d)*2+2,m=Math.ceil(f)*2+2;this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int b = coords [ 0 ] ;
int d = coords [ 3 ] ;
int r = coords [ 1 ] ;
int c = coords [ 2 ] ;
float accumulator = 0.0 ;
const float heightScale = float ( $ { c } ) ;
const float widthScale = float ( $ { p } ) ;
const float invHeightScale = float ( $ { d } ) ;
const float invWidthScale = float ( $ { f } ) ;
const int winHeight = int ( $ { h } ) ;
const int winWidth = int ( $ { m } ) ;
// Compute bounds for where in dy we will look
float startRLerp = floor ( float ( r ) * invHeightScale ) ;
int startDyR = int ( floor ( startRLerp - float ( winHeight / 2 ) ) ) ;
float startCLerp = floor ( float ( c ) * invWidthScale ) ;
int startDyC = int ( floor ( startCLerp - float ( winWidth / 2 ) ) ) ;
// Loop over dy
for ( int dyROffset = 0 ; dyROffset < winHeight ; dyROffset ++ ) {
int dyR = dyROffset + startDyR ;
// Guard against the window exceeding the bounds of dy
if ( dyR < 0 || dyR >= $ { a } ) {
continue ;
}
for ( int dyCOffset = 0 ; dyCOffset < winWidth ; dyCOffset ++ ) {
int dyC = dyCOffset + startDyC ;
// Guard against the window exceeding the bounds of dy
if ( dyC < 0 || dyC >= $ { i } ) {
continue ;
}
float sourceFracRow =
float ( $ { l [ 0 ] } ) *
( float ( dyR ) / float ( $ { u [ 0 ] } ) ) ;
float sourceFracCol =
float ( $ { l [ 1 ] } ) *
( float ( dyC ) / float ( $ { u [ 1 ] } ) ) ;
int sourceNearestRow = int ( min (
float ( int ( $ { s } ) - 1 ) ,
$ { r } ? float ( round ( sourceFracRow ) ) :
float ( floor ( sourceFracRow ) ) ) ) ;
int sourceNearestCol = int ( min (
float ( int ( $ { o } ) - 1 ) ,
$ { r } ? float ( round ( sourceFracCol ) ) :
float ( floor ( sourceFracCol ) ) ) ) ;
if ( r == sourceNearestRow && c == sourceNearestCol ) {
accumulator += getDy ( b , dyR , dyC , d ) ;
}
}
}
// End loop over dy
setOutput ( accumulator ) ;
}
` }};function sJe(e){let{inputs:t,backend:n,attrs:r}=e,{images:s,dy:o}=t,{alignCorners:a}=r,i=new uO(o.shape,s.shape,a);return n.runWebGLProgram(i,[o],o.dtype)}var Koe={kernelName:Zx,backendName:"webgl",kernelFunc:sJe};var cO=class{constructor(t,n){this.variableNames=["x"];let r=t.length;if(r>4)throw new Error( ` WebGL backend : Reverse of rank - $ { r } tensor is not yet supported ` );if(this.outputShape=t,r===1){this.userCode= `
void main ( ) {
int coord = getOutputCoords ( ) ;
setOutput ( getX ( $ { t [ 0 ] } - coord - 1 ) ) ;
}
` ;return}let s=i=>n.indexOf(i)!==-1&&t[i]!==1? ` $ { t [ i ] } - coords [ $ { i } ] - 1 ` : ` coords [ $ { i } ] ` ,o=t.map((i,l)=>s(l)).join(","),a=Ct(r);this.userCode= `
void main ( ) {
$ { a } coords = getOutputCoords ( ) ;
setOutput ( getX ( $ { o } ) ) ;
}
` }};var pO=class{constructor(t,n){this.variableNames=["x"],this.packedInputs=!0,this.packedOutput=!0;let r=t.length;if(r>4)throw new Error( ` WebGL backend : Reverse of rank - $ { r } tensor is not yet supported ` );this.outputShape=t;let s=ir("rc",r),o= ` $ { s [ r - 1 ] } + 1 < $ { this . outputShape [ r - 1 ] } ` ,a= ` $ { s [ r - 2 ] } + 1 < $ { this . outputShape [ r - 2 ] } ` ,i=Ct(r);r===1?this.userCode= `
void main ( ) {
int rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
result . r = getChannel ( getX ( $ { t [ 0 ] } - rc - 1 ) ,
$ { t [ 0 ] } - rc - 1 ) ;
if ( $ { o } ) {
result . g = getChannel ( getX ( $ { t [ 0 ] } - ( rc + 1 ) - 1 ) ,
$ { t [ 0 ] } - ( rc + 1 ) - 1 ) ;
}
setOutput ( result ) ;
}
` :this.userCode= `
void main ( ) {
$ { i } rc = getOutputCoords ( ) ;
vec4 result = vec4 ( 0. ) ;
result . r = $ { l ( s . slice ( ) ) } ;
if ( $ { o } ) {
result . g = $ { u ( s . slice ( ) ) } ;
}
if ( $ { a } ) {
result . b = $ { c ( s . slice ( ) ) } ;
if ( $ { o } ) {
result . a = $ { p ( s . slice ( ) ) } ;
}
}
setOutput ( result ) ;
}
` ;function l(h){return d(h)}function u(h){return h[r-1]="("+h[r-1]+" + 1)",d(h)}function c(h){return h[r-2]="("+h[r-2]+" + 1)",d(h)}function p(h){return h[r-1]="("+h[r-1]+" + 1)",h[r-2]="("+h[r-2]+" + 1)",d(h)}function d(h){let m=t.map((y,b)=>f(b,h)),g=m.join(","),x=m.slice(-2).join(",");return ` getChannel ( getX ( $ { g } ) , vec2 ( $ { x } ) ) ` }function f(h,m){return n.indexOf(h)!==-1&&t[h]!==1? ` $ { t [ h ] } - $ { m [ h ] } - 1 ` : ` $ { m [ h ] } ` }}};function oJe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{dims:o}=r,a=s.shape.length,i=C.parseAxisParam(o,s.shape);if(a===0)return lr({inputs:{x:s},backend:n});let l=he().getBool("WEBGL_PACK_ARRAY_OPERATIONS")?new pO(s.shape,i):new cO(s.shape,i);return n.runWebGLProgram(l,[s],s.dtype)}var Xoe={kernelName:qp,backendName:"webgl",kernelFunc:oJe};var dO=class{constructor(t,n){this.variableNames=["Image"],this.outputShape=[],this.customUniforms=[{name:"params",type:"vec4"}];let r=t[1],s=t[2];this.outputShape=t;let o="";typeof n=="number"?o= ` float outputValue = $ { n . toFixed ( 2 ) } ; ` :o= `
vec3 fill = vec3 ( $ { n . join ( "," ) } ) ;
float outputValue = fill [ coords [ 3 ] ] ; ` ,this.userCode= `
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
int x = coords [ 2 ] ;
int y = coords [ 1 ] ;
float coordXFloat = ( float ( x ) - params [ 0 ] ) * params [ 3 ] -
( float ( y ) - params [ 1 ] ) * params [ 2 ] ;
float coordYFloat = ( float ( x ) - params [ 0 ] ) * params [ 2 ] +
( float ( y ) - params [ 1 ] ) * params [ 3 ] ;
int coordX = int ( round ( coordXFloat + params [ 0 ] ) ) ;
int coordY = int ( round ( coordYFloat + params [ 1 ] ) ) ;
$ { o }
if ( coordX >= 0 && coordX < $ { s } && coordY >= 0 && coordY < $ { r } ) {
outputValue = getImage ( coords [ 0 ] , coordY , coordX , coords [ 3 ] ) ;
}
setOutput ( outputValue ) ;
}
` }};var Zoe={kernelName:dy,backendName:"webgl",kernelFunc:({inputs:e,attrs:t,backend:n})=>{let{image:r}=e,{radians:s,fillValue:o,center:a}=t,i=n,l=new dO(r.shape,o),[u,c]=M.getImageCenter(a,r.shape[1],r.shape[2]),p=[[u,c,Math.sin(s),Math.cos(s)]];return i.runWebGLProgram(l,[r],r.dtype,p)}};var aJe= `
// OpenGL ES does not support round function.
// The algorithm is based on banker's rounding.
float base = floor ( x ) ;
if ( ( x - base ) < 0.5 ) {
return floor ( x ) ;
} else if ( ( x - base ) > 0.5 ) {
return ceil ( x ) ;
} else {
if ( mod ( base , 2.0 ) == 0.0 ) {
return base ;
} else {
return base + 1.0 ;
}
}
` ,iJe=Xe({opSnippet:aJe}),Yoe={kernelName:Du,backendName:"webgl",kernelFunc:iJe};var lJe="return inversesqrt(x);",uJe=Xe({opSnippet:lJe,cpuKernelImpl:Fne}),Joe={kernelName: $ u,backendName:"webgl",kernelFunc:uJe};var vv=class{constructor(t,n,r,s,o,a,i=!0){this.variableNames=["updates","indices","defaultValue"],this.outputShape=a;let l=Ct(o.length),u=Ct(a.length),c="";r===1?c="i":r===2&&(c="i, j");let p= ` getIndices ( $ { c } ) ` ,d="";s===1?d="i":s===2&&(d="i, coords[1]");let f= ` getUpdates ( $ { d } ) ` ,h=n>1?"strides[j]":"strides";this.userCode= `
$ { l } strides = $ { l } ( $ { o } ) ;
void main ( ) {
$ { u } coords = getOutputCoords ( ) ;
float sum = 0.0 ;
bool found = false ;
for ( int i = 0 ; i < $ { t } ; i ++ ) {
int flattenedIndex = 0 ;
for ( int j = 0 ; j < $ { n } ; j ++ ) {
int index = round ( $ { p } ) ;
flattenedIndex += index * $ { h } ;
}
if ( flattenedIndex == coords [ 0 ] ) {
sum += $ { f } ;
found = true ;
}
}
setOutput ( mix ( getDefaultValue ( ) , sum , float ( found ) ) ) ;
}
` }};function cJe(e){let{inputs:t,backend:n,attrs:r}=e,{indices:s,updates:o}=t,{shape:a}=r,{sliceRank:i,numUpdates:l,sliceSize:u,strides:c,outputSize:p}=M.calculateShapes(o,s,a),d=[p/u,u];if(p===0)return n.makeTensorInfo(a,s.dtype);let f= $ e({inputs:{x:s},backend:n,attrs:{shape:[l,i]}}),h= $ e({inputs:{x:o},backend:n,attrs:{shape:[l,u]}}),m=n.makeTensorInfo([],"float32",new Float32Array([0])),g=new vv(l,i,f.shape.length,h.shape.length,c,d),x=n.runWebGLProgram(g,[h,f,m],h.dtype),y= $ e({inputs:{x},backend:n,attrs:{shape:a}});return n.disposeIntermediateTensorInfo(f),n.disposeIntermediateTensorInfo(h),n.disposeIntermediateTensorInfo(x),n.disposeIntermediateTensorInfo(m),y}var Qoe={kernelName:Jx,backendName:"webgl",kernelFunc:cJe};var fO=class{constructor(t,n,r){this.variableNames=["c","a","b"],this.outputShape=n;let s,o;if(r>4)throw Error( ` Where for rank $ { r } is not yet supported ` );if(r===1)o="resRC",s="resRC";else{let i=["resRC.x","resRC.y","resRC.z","resRC.w"],l=[],u=[];for(let c=0;c<n.length;c++)u.push( ` $ { i [ c ] } ` ),c<t&&l.push( ` $ { i [ c ] } ` );s=l.join(),o=u.join()}let a=Ct(r);this.userCode= `
void main ( ) {
$ { a } resRC = getOutputCoords ( ) ;
float cVal = getC ( $ { s } ) ;
if ( cVal >= 1.0 ) {
setOutput ( getA ( $ { o } ) ) ;
} else {
setOutput ( getB ( $ { o } ) ) ;
}
}
` }};function pJe(e){let{inputs:t,backend:n}=e,{condition:r,t:s,e:o}=t,a=new fO(r.shape.length,s.shape,s.shape.length);return n.runWebGLProgram(a,[r,s,o],Or(s.dtype,o.dtype))}var eae={kernelName:Kp,backendName:"webgl",kernelFunc:pJe};var dJe= `
// Stable and Attracting Fixed Point (0, 1) for Normalized Weights.
// see: https://arxiv.org/abs/1706.02515
float scaleAlpha = $ { M . SELU _SCALEALPHA } ;
float scale = $ { M . SELU _SCALE } ;
return ( x >= 0.0 ) ? scale * x : scaleAlpha * ( exp ( x ) - 1.0 ) ;
` ,fJe=Xe({opSnippet:dJe}),tae={kernelName:Fu,backendName:"webgl",kernelFunc:fJe};var nae="return 1.0 / (1.0 + exp(-1.0 * x));",hJe=Xe({opSnippet:nae,packedOpSnippet:nae,cpuKernelImpl:One}),rae={kernelName:zu,backendName:"webgl",kernelFunc:hJe};var mJe= `
if ( isnan ( x ) ) { return 0.0 ; }
return sign ( x ) ;
` ,gJe=Xe({opSnippet:mJe}),sae={kernelName:Mu,backendName:"webgl",kernelFunc:gJe};var xJe=_S+ `
return sin ( x ) ;
` ,yJe=Xe({opSnippet:xJe}),oae={kernelName:Ou,backendName:"webgl",kernelFunc:yJe};var bJe= `
float e2x = exp ( x ) ;
return ( e2x - 1.0 / e2x ) / 2.0 ;
` ,AJe=Xe({opSnippet:bJe}),aae={kernelName:Pu,backendName:"webgl",kernelFunc:AJe};var vJe= `
float epsilon = 1.1920928955078125 e - 7 ;
float threshold = log ( epsilon ) + 2.0 ;
bool too _large = x > - threshold ;
bool too _small = x < threshold ;
float result ;
float exp _x = exp ( x ) ;
if ( too _large ) {
result = x ;
}
else if ( too _small ) {
result = exp _x ;
}
else {
result = log ( exp _x + 1.0 ) ;
}
return result ;
` ,wJe=Xe({opSnippet:vJe}),iae={kernelName:Lu,backendName:"webgl",kernelFunc:wJe};var kJe=e=>{let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{blockShape:o,paddings:a}=r;C.assert(s.shape.length<=4,()=>"spaceToBatchND for rank > 4 with a WebGL backend not implemented yet");let i=o.reduce((x,y)=>x*y),l=[[0,0]];l.push(...a);for(let x=1+o.length;x<s.shape.length;++x)l.push([0,0]);let u=[],c=nO({inputs:{x:s},backend:n,attrs:{paddings:l,constantValue:0}}),p=M.getReshaped(c.shape,o,i,!1),d=M.getPermuted(p.length,o.length,!1),f=M.getReshapedPermuted(c.shape,o,i,!1),h= $ e({inputs:{x:c},backend:n,attrs:{shape:p}}),m=On({inputs:{x:h},backend:n,attrs:{perm:d}}),g= $ e({inputs:{x:m},backend:n,attrs:{shape:f}});return u.push(c),u.push(h),u.push(m),u.forEach(x=>n.disposeIntermediateTensorInfo(x)),g},lae={kernelName:Yp,backendName:"webgl",kernelFunc:kJe};function IJe(e){let{inputs:t,backend:n}=e,{indices:r,values:s,denseShape:o,defaultValue:a}=t;if(o.shape.length!==1)throw new Error( ` Dense shape must be a vector , saw :
$ { o . shape } ` );if(r.shape.length!==2)throw new Error( ` Indices must be a matrix , saw :
$ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Values must be a vector , saw :
$ { s . shape } ` );if(a.shape.length!==0)throw new Error( ` Default value must be a scalar , saw :
$ { a . shape } ` );let i=n.readSync(r.dataId),l=n.readSync(s.dataId),u=n.readSync(o.dataId),c=n.readSync(a.dataId)[0],[p,d,f,h,m]=Mne(i,r.shape,r.dtype,l,s.dtype,u,c);return[n.makeTensorInfo(d,r.dtype,p),n.makeTensorInfo([d[0]],s.dtype,f),n.makeTensorInfo([h.length],"bool",new Uint8Array(h.map(g=>Number(g)))),n.makeTensorInfo([m.length],r.dtype,new Int32Array(m))]}var uae={kernelName:Qx,backendName:"webgl",kernelFunc:IJe};function CJe(e){let{inputs:t,backend:n}=e,{inputIndices:r,inputShape:s,newShape:o}=t;if(r.shape.length!==2)throw new Error( ` Input indices should be a matrix but received shape $ { r . shape } ` );if(s.shape.length!==1)throw new Error( ` Input shape should be a vector but received shape $ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Target shape should be a vector but received shape $ { o . shape } ` );let a=Array.from(n.readSync(s.dataId)),i=n.readSync(r.dataId),l=Array.from(n.readSync(o.dataId)),[u,c,p]=zne(i,r.shape,r.dtype,a,l);return[n.makeTensorInfo(c,r.dtype,u),n.makeTensorInfo([p.length],o.dtype,new Int32Array(p))]}var cae={kernelName:ey,backendName:"webgl",kernelFunc:CJe};function SJe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.readSync(r.dataId),i=n.readSync(s.dataId),l=n.readSync(o.dataId),[u,c]=SS(a,r.shape,r.dtype,i,l,!0);return n.makeTensorInfo(c,r.dtype,u)}var pae={kernelName:ty,backendName:"webgl",kernelFunc:SJe};function NJe(e){let{inputs:t,backend:n}=e,{data:r,indices:s,segmentIds:o}=t;if(r.shape.length<1)throw new Error("Data should be at least 1 dimensional but received scalar");if(s.shape.length!==1)throw new Error( ` Indices should be a vector but received shape
$ { s . shape } ` );if(o.shape.length!==1)throw new Error( ` Segment ids should be a vector but received shape
$ { o . shape } ` );let a=n.readSync(r.dataId),i=n.readSync(s.dataId),l=n.readSync(o.dataId),[u,c]=SS(a,r.shape,r.dtype,i,l);return n.makeTensorInfo(c,r.dtype,u)}var dae={kernelName:ny,backendName:"webgl",kernelFunc:NJe};function TJe(e){let{inputs:t,backend:n,attrs:r}=e,{sparseIndices:s,sparseValues:o,defaultValue:a}=t,{outputShape:i}=r,{sliceRank:l,numUpdates:u,strides:c,outputSize:p}=M.calculateShapes(o,s,i),d=!1,f=new vv(u,l,s.shape.length,o.shape.length,c,[p,1],d),h=n.runWebGLProgram(f,[o,s,a],o.dtype),m= $ e({inputs:{x:h},backend:n,attrs:{shape:i}});return n.disposeIntermediateTensorInfo(h),m}var fae={kernelName:ry,backendName:"webgl",kernelFunc:TJe};function _Je(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{numOrSizeSplits:o,axis:a}=r,i=C.parseAxisParam(a,s.shape)[0],l=M.prepareSplitSize(s,o,i),u=s.shape.length,c=new Array(u).fill(0),p=s.shape.slice();return l.map(d=>{let f=[...p];f[i]=d;let h=el({inputs:{x:s},backend:n,attrs:{begin:c,size:f}});return c[i]+=d,h})}var hae={kernelName:Jp,backendName:"webgl",kernelFunc:_Je};var mae="return sqrt(x);",EJe=Xe({opSnippet:mae,packedOpSnippet:mae,cpuKernelImpl:Lne}),gae={kernelName:Bu,backendName:"webgl",kernelFunc:EJe};var RJe="return x * x;",DJe=Xe({opSnippet:RJe}),xae={kernelName:sy,backendName:"webgl",kernelFunc:DJe};var yae="return (a - b) * (a - b);", $ Je=Qt({opSnippet:yae,packedOpSnippet:yae}),bae={kernelName:Vu,backendName:"webgl",kernelFunc: $ Je};function FJe({inputs:e,attrs:t,backend:n}){let{x:r}=e,s=Qr+ `
return x > 0.0 ? 1.0 : float ( $ { t . alpha } ) ;
` ,o=new $ o(r.shape,s);return n.runWebGLProgram(o,[r],r.dtype)}var Aae={kernelName:ju,backendName:"webgl",kernelFunc:FJe};var hO=class{constructor(t,n,r){this.variableNames=["x"],this.outputShape=r;let s=r.length,o=Ct(r.length),a=Ct(r.length),i="";if(s===1)i="coords * strides + begin";else{let l=0;i=r.map((u,c)=>(l++,r.length===1? ` coords * strides [ $ { c } ] + begin [ $ { c } ] ` : ` coords [ $ { l - 1 } ] * strides [ $ { c } ] + begin [ $ { c } ] ` )).join(",")}this.userCode= `
$ { o } begin = $ { o } ( $ { t } ) ;
$ { o } strides = $ { o } ( $ { n } ) ;
void main ( ) {
$ { a } coords = getOutputCoords ( ) ;
setOutput ( getX ( $ { i } ) ) ;
}
` }};function OJe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{begin:o,end:a,strides:i,beginMask:l,endMask:u,ellipsisMask:c,newAxisMask:p,shrinkAxisMask:d}=r,{nonStrided:f, $ begin:h, $ strides:m,size:g,newShape:x,outShape:y}=Cs.sliceInfo(s.shape,o,a,i,l,u,c,p,d),b= $ e({inputs:{x:s},backend:n,attrs:{shape:x}}),A;if(f){let w=el({inputs:{x:b},backend:n,attrs:{begin:h,size:g}});A= $ e({inputs:{x:w},backend:n,attrs:{shape:y}}),n.disposeIntermediateTensorInfo(w)}else if(y.some(w=>w===0))A=n.makeTensorInfo(y,s.dtype,[]);else if(n.shouldExecuteOnCPU([b])){let I=n.texData.get(b.dataId).values,N=et(b.shape,b.dtype,I),T=Bne(y,N,m,h);A=n.makeTensorInfo(y,b.dtype,T.values)}else{let k=new hO(h,m,y);A=n.runWebGLProgram(k,[b],b.dtype)}let v= $ e({inputs:{x:A},backend:n,attrs:{shape:y}});return n.disposeIntermediateTensorInfo(b),n.disposeIntermediateTensorInfo(A),v}var vae={kernelName:oy,backendName:"webgl",kernelFunc:OJe};function PJe(e){let{inputs:t,backend:n,attrs:r}=e,{separator:s,nGramWidths:o,leftPad:a,rightPad:i,padWidth:l,preserveShortSequences:u}=r,{data:c,dataSplits:p}=t,d=n.readSync(c.dataId),f=n.readSync(p.dataId),[h,m]=Vne(d,f,s,o,a,i,l,u);return[n.makeTensorInfo([h.length],"string",h),n.makeTensorInfo(p.shape,"int32",m)]}var wae={kernelName:ay,backendName:"webgl",kernelFunc:PJe};function MJe(e){let{inputs:t,backend:n,attrs:r}=e,{skipEmpty:s}=r,{input:o,delimiter:a}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(o.shape.length!==1)throw new Error( ` Input must be a vector , got shape : $ { o . shape } ` );if(a.shape.length!==0)throw new Error( ` Delimiter must be a scalar , got shape : $ { a . shape } ` );let i=n.readSync(o.dataId),l=n.readSync(a.dataId)[0],[u,c,p]=Wne(i,l,s),d=c.length;return[n.makeTensorInfo([d,2],"int32",u),n.makeTensorInfo([d],"string",c),n.makeTensorInfo([2],"int32",new Int32Array(p))]}var kae={kernelName:iy,backendName:"webgl",kernelFunc:MJe};function zJe(e){let{inputs:t,backend:n,attrs:r}=e,{numBuckets:s}=r,{input:o}=t;if(o.dtype!=="string")throw new Error("Input must be of datatype string");if(s<=0)throw new Error("Number of buckets must be at least 1");let a=n.readSync(o.dataId),i=Gne(a,s);return n.makeTensorInfo(o.shape,"int32",i)}var Iae={kernelName:ly,backendName:"webgl",kernelFunc:zJe};var LJe="return tan(x);",BJe=Xe({opSnippet:LJe}),Cae={kernelName:Gu,backendName:"webgl",kernelFunc:BJe};var VJe= `
float e2x = exp ( - 2.0 * abs ( x ) ) ;
return sign ( x ) * ( 1.0 - e2x ) / ( 1.0 + e2x ) ;
` ,WJe=Xe({opSnippet:VJe}),Sae={kernelName:Uu,backendName:"webgl",kernelFunc:WJe};var mO=class{constructor(t,n){this.variableNames=["A"];let r=new Array(t.length);for(let a=0;a<r.length;a++)r[a]=t[a]*n[a];this.outputShape=r,this.rank=r.length;let s=Ct(this.rank),o=GJe(t);this.userCode= `
void main ( ) {
$ { s } resRC = getOutputCoords ( ) ;
setOutput ( getA ( $ { o } ) ) ;
}
` }};function GJe(e){let t=e.length;if(t>5)throw Error( ` Tile for rank $ { t } is not yet supported ` );if(t===1)return ` imod ( resRC , $ { e [ 0 ] } ) ` ;let n=["resRC.x","resRC.y","resRC.z","resRC.w","resRC.u"],r=[];for(let s=0;s<e.length;s++)r.push( ` imod ( $ { n [ s ] } , $ { e [ s ] } ) ` );return r.join()}function gO(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{reps:o}=r;if(s.dtype==="string"||s.shape.length>5){let l=n.readSync(s.dataId),u=s.dtype==="string"?l.map(d=>C.decodeString(d)):l,c=et(s.shape,s.dtype,u),p=jne(c,o);return n.makeTensorInfo(p.shape,p.dtype,p.values)}let a=new mO(s.shape,o);return n.runWebGLProgram(a,[s],s.dtype)}var Nae={kernelName:vl,backendName:"webgl",kernelFunc:gO};var xO=class{constructor(t){this.variableNames=["x","indices"],this.customUniforms=[{name:"n",type:"int"},{name:"firstPass",type:"int"},{name:"negativeInf",type:"float"},{name:"dir",type:"int"},{name:"inc",type:"int"}],this.outputShape=t,this.userCode= `
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int elemIdx = coords [ 1 ] ;
// We compare elements pair-wise within a group of size 2 * inc.
// The comparing rule for each group alternates between ascending
// and descending. Within each group, we compare each pair at
// positions i and i+inc. To decide whether an element at position i
// is x0 or x1, we mod it by 2 * inc, if the result is smaller than
// inc, it is in the first half of the group, we denote it as x0,
// otherwise we denote it as x1.
// For example, as shown in the Bitonic top K paper referenced above,
// Figure5(a) shows that element[1] is in the
// second half of the group when group size is 2, but it is in the
// first half of the group when group size is 4.
bool isFirstInPair = imod ( elemIdx , 2 * inc ) < inc ;
int i = isFirstInPair ? elemIdx : elemIdx - inc ;
int i0 = firstPass == 1 ? i : int ( getIndices ( batch , i ) ) ;
int i1 = firstPass == 1 ? i + inc : int ( getIndices ( batch , i + inc ) ) ;
float x0 = i0 < n ? getX ( batch , i0 ) : negativeInf ;
float x1 = i1 < n ? getX ( batch , i1 ) : negativeInf ;
// Denotes which direction indices are in (ascending or descending).
bool reverse = imod ( elemIdx , 2 * dir ) >= dir ;
bool isGreater = x0 > x1 || ( x0 == x1 && i1 > i0 ) ;
if ( reverse == isGreater ) { // Elements in opposite order of direction
int iTemp = i0 ;
i0 = i1 ;
i1 = iTemp ;
}
if ( isFirstInPair ) {
setOutput ( float ( i0 ) ) ;
} else {
setOutput ( float ( i1 ) ) ;
}
}
` }},yO=class{constructor(t){this.variableNames=["x","indices"],this.customUniforms=[{name:"n",type:"int"},{name:"firstPass",type:"int"},{name:"k",type:"int"}],this.outputShape=t,this.userCode= `
void main ( ) {
// Takes max of indices (0, k), (1, k + 1), (2, k + 2) ...
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int elemIdx = coords [ 1 ] ;
// The output size is half of the previous size.
// If the previous sequence is | | | | _ _ _ _ | | | | _ _ _ _ (k=4),
// we only need to output the indices at positions |, the indices at
// positions _ can be thrown away, see Figure5(b) After Phase 2
// (Merge phase) in the Bitonic Top K paper referenced above.
// For example, the paper shows we only need to output the orange bars.
// The output sequence should look like this | | | | | | | |.
// Because the sequence is halved, to map the output index back
// to the previous sequence to find the corresponding value,
// we need to double the index. When we double the index,
// we basically interpolate a position, so 2i looks like
// | _ | _ | _ | _ | _ | _ | _. We move the | to the first k position
// of each 2k positions by - elemIdx % k. E.g. for output at
// index 4,5,6,7, we want to get the corresponding element at
// original index 8,9,10,11, for output at index 8,9,10,11,
// we want to get the corresponding element at original index
// 16,17,18,19, so on and so forth.
int i = elemIdx < k ? elemIdx : ( elemIdx * 2 - imod ( elemIdx , k ) ) ;
int i0 = firstPass == 1 ? i : int ( getIndices ( batch , i ) ) ;
int i1 = firstPass == 1 ? i + k : int ( getIndices ( batch , i + k ) ) ;
float x0 = getX ( batch , i0 ) ;
float x1 = i1 < n ? getX ( batch , i1 ) : x0 ;
setOutput ( x0 >= x1 ? float ( i0 ) : float ( i1 ) ) ;
}
` }};function qm(e,t){t!==null&&e.disposeIntermediateTensorInfo(t)}function Tae(e){let t=1;for(;t<e;)t*=2;return t}function UJe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s}=t,{k:o,sorted:a}=r,i=he().getNumber("TOPK_LAST_DIM_CPU_HANDOFF_SIZE_THRESHOLD"),l=he().getNumber("TOPK_K_CPU_HANDOFF_THRESHOLD"),u=s.shape,c=u[u.length-1];if(n.shouldExecuteOnCPU([s])||c<i||o>l){let T=n.readSync(s.dataId),[_,R]=Hne(T,u,s.dtype,o,a);return[n.makeTensorInfo(_.shape,_.dtype,_.values),n.makeTensorInfo(R.shape,R.dtype,R.values)]}if(o===0)return u[u.length-1]=0,[n.makeTensorInfo(u,s.dtype,[]),n.makeTensorInfo(u,"int32",[])];if(c===1)return[s,vc({attrs:{shape:u,dtype:"int32",value:0},backend:n})];let p=n.texData.get(s.dataId),d=p!==null&&p.isPacked,f=d?n.unpackTensor(s):s,m=C.sizeFromShape(u)/c,g= $ e({inputs:{x:f},attrs:{shape:[m,c]},backend:n});d&&qm(n,f);let x=Tae(o),y=Tae(c),b=null,A=()=>b===null?[g,g]:[g,b],v=(T,_,R)=>{let S=A(),F=new xO(R),W=[[c],[b===null?1:0],[Number.NEGATIVE_INFINITY],[T],[_]],q=b;b=n.runWebGLProgram(F,S,"int32",W),qm(n,q)};for(let T=1;T<x;T*=2){let _=T*2;for(let R=T;R>=1;R/=2)v(_,R,[m,y])}for(let T=y;T>x;T/=2){let _=A(),R=new yO([m,T/2]),F=[[c],[b===null?1:0],[x]],B=b;b=n.runWebGLProgram(R,_,"int32",F),qm(n,B);let W=x/2,q=W*2;for(let K=W;K>=1;K/=2)v(q,K,b.shape)}let w=b;b=el({inputs:{x:b},backend:n,attrs:{begin:0,size:[m,o]}}),qm(n,w);let k=B8({inputs:{x:g,indices:b},backend:n,attrs:{axis:1,batchDims:1}});qm(n,g);let I=u.slice(0,-1);I.push(o),w=b,b= $ e({inputs:{x:b},attrs:{shape:I},backend:n}),qm(n,w);let N=k;return k= $ e({inputs:{x:k},attrs:{shape:I},backend:n}),qm(n,N),[k,b]}var _ae={kernelName:uy,backendName:"webgl",kernelFunc:UJe};var bO=class{constructor(t,n,r,s,o,a){this.variableNames=["Image","Transforms"],this.outputShape=a;let i=r==="nearest"?1:2,l;switch(s){case"constant":l=1;break;case"reflect":l=2;break;case"wrap":l=3;break;case"nearest":l=4;break;default:l=1;break}this.userCode= `
float mapCoord ( float outCoord , float len ) {
float inCoord = outCoord ;
if ( $ { l } == 2 ) {
if ( inCoord < 0.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz2 = 2.0 * len ;
if ( inCoord < sz2 ) {
inCoord = sz2 * float ( int ( float ( - inCoord / sz2 ) ) ) +
inCoord ;
}
inCoord = inCoord < - len ? inCoord + sz2 : - inCoord - 1.0 ;
}
} else if ( inCoord > len - 1.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz2 = 2.0 * len ;
inCoord -= sz2 * float ( int ( float ( inCoord / sz2 ) ) ) ;
if ( inCoord >= len ) {
inCoord = sz2 - inCoord - 1.0 ;
}
}
}
return clamp ( inCoord , 0.0 , len - 1.0 ) ;
} else if ( $ { l } == 3 ) {
if ( inCoord < 0.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz = len - 1.0 ;
inCoord += len * ( float ( int ( float ( - inCoord / sz ) ) ) + 1.0 ) ;
}
} else if ( inCoord > len - 1.0 ) {
if ( len <= 1.0 ) {
inCoord = 0.0 ;
} else {
float sz = len - 1.0 ;
inCoord -= len * float ( int ( float ( inCoord / sz ) ) ) ;
}
}
return clamp ( inCoord , 0.0 , len - 1.0 ) ;
} else if ( $ { l } == 4 ) {
return clamp ( outCoord , 0.0 , len - 1.0 ) ;
} else {
return outCoord ;
}
}
float readWithFillValue ( int batch , int coordY , int coordX ,
int channel ) {
float outputValue ;
if ( 0 <= coordY && coordY < $ { t } && 0 <= coordX && coordX < $ { n } ) {
outputValue = getImage ( batch , coordY , coordX , channel ) ;
} else {
outputValue = float ( $ { o } ) ;
}
return outputValue ;
}
void main ( ) {
ivec4 coords = getOutputCoords ( ) ;
float outputValue ;
int batch = coords [ 0 ] ;
int x = coords [ 2 ] ;
int y = coords [ 1 ] ;
int channel = coords [ 3 ] ;
float xf = float ( x ) ;
float yf = float ( y ) ;
float a1 = getTransforms ( batch , 0 ) ;
float a2 = getTransforms ( batch , 1 ) ;
float a3 = getTransforms ( batch , 2 ) ;
float b1 = getTransforms ( batch , 3 ) ;
float b2 = getTransforms ( batch , 4 ) ;
float b3 = getTransforms ( batch , 5 ) ;
float c1 = getTransforms ( batch , 6 ) ;
float c2 = getTransforms ( batch , 7 ) ;
float projection = c1 * xf + c2 * yf + 1.0 ;
if ( projection == 0.0 ) {
outputValue = float ( $ { o } ) ;
} else {
float inX = ( a1 * xf + a2 * yf + a3 ) / projection ;
float inY = ( b1 * xf + b2 * yf + b3 ) / projection ;
float mapX = mapCoord ( inX , float ( $ { n } ) ) ;
float mapY = mapCoord ( inY , float ( $ { t } ) ) ;
if ( $ { i } == 1 ) {
int coordY = int ( round ( mapY ) ) ;
int coordX = int ( round ( mapX ) ) ;
outputValue = readWithFillValue ( batch , coordY , coordX ,
channel ) ;
} else {
float yFloor = floor ( mapY ) ;
float xFloor = floor ( mapX ) ;
float yCeil = yFloor + 1.0 ;
float xCeil = xFloor + 1.0 ;
float valueYFloor = ( xCeil - mapX ) *
readWithFillValue ( batch , int ( yFloor ) , int ( xFloor ) , channel ) +
( mapX - xFloor ) *
readWithFillValue ( batch , int ( yFloor ) , int ( xCeil ) , channel ) ;
float valueYCeil = ( xCeil - mapX ) *
readWithFillValue ( batch , int ( yCeil ) , int ( xFloor ) , channel ) +
( mapX - xFloor ) *
readWithFillValue ( batch , int ( yCeil ) , int ( xCeil ) , channel ) ;
outputValue = ( yCeil - mapY ) * valueYFloor +
( mapY - yFloor ) * valueYCeil ;
}
}
setOutput ( outputValue ) ;
}
` }};function jJe(e){let{inputs:t,backend:n,attrs:r}=e,{image:s,transforms:o}=t,{interpolation:a,fillMode:i,fillValue:l,outputShape:u}=r,[c,p,d,f]=s.shape,[h,m]=u!=null?u:[p,d],g=[c,h,m,f],x=new bO(p,d,a,i,l,g);return n.runWebGLProgram(x,[s,o],"float32")}var Eae={kernelName:cy,backendName:"webgl",kernelFunc:jJe};function HJe(e){let{inputs:t,attrs:n,backend:r}=e,{axis:s}=n,{x:o}=t;Rl(o,"unique"),console.warn("WARNING: ","UI might be locked temporarily as data is being downloaded");let a=r.readSync(o.dataId),{outputValues:i,outputShape:l,indices:u}=qne(a,s,o.shape,o.dtype);return[r.makeTensorInfo(l,o.dtype,i),r.makeTensorInfo([u.length],"int32",u)]}var Rae={kernelName:py,backendName:"webgl",kernelFunc:HJe};function qJe(e){let{inputs:t,backend:n,attrs:r}=e,{value:s}=t,{axis:o}=r;o<0&&(o+=s.shape.length);let a=s,i=a.shape.length,l=s.shape[o],u=new Array(i-1),c=0;for(let m=0;m<i;m++)m!==o&&(u[c++]=a.shape[m]);let p=[],d=new Array(i).fill(0),f=a.shape.slice();f[o]=1;let h=new Array(l);for(let m=0;m<h.length;m++){d[o]=m;let g=el({inputs:{x:a},backend:n,attrs:{begin:d,size:f}}),x= $ e({inputs:{x:g},backend:n,attrs:{shape:u}});h[m]=x,p.push(g)}return p.forEach(m=>n.disposeIntermediateTensorInfo(m)),h}var Dae={kernelName:td,backendName:"webgl",kernelFunc:qJe};var AO=class{constructor(t,n){this.variableNames=["x","segmentIds"];let r=t.windowSize,s=t.batchSize,o=t.inSize,a=t.numSegments,i=a*Math.ceil(o/r);this.outputShape=[s,i];let l="0.0",u="sumValue",c=Math.floor(r/4)*4,p=r%4,d= `
sumValue += dot ( values , segFilter ) ;
` ,f="";o%r>0&&(f= `
if ( inIdx < 0 || inIdx >= $ { o } ) {
return initializationValue ;
}
` );let h="";o%r>0&&(h= `
if ( inIdx < 0 || inIdx >= $ { o } ) {
return - 1.0 ;
}
` ),this.userCode= `
const float initializationValue = $ { l } ;
float getValue ( int batch , int inIdx ) {
$ { f }
return getX ( batch , inIdx ) ;
}
float getSegmentIdAtIndex ( int inIdx ) {
$ { h }
return getSegmentIds ( inIdx ) ;
}
void main ( ) {
ivec2 coords = getOutputCoords ( ) ;
int batch = coords [ 0 ] ;
int outIdx = coords [ 1 ] ;
int inOffset = int ( floor ( float ( outIdx ) / float (
$ { a } ) ) * float ( $ { r } ) ) ;
int currentSeg = int ( mod ( float ( outIdx ) , float ( $ { a } ) ) ) ;
float sumValue = 0.0 ;
for ( int i = 0 ; i < $ { c } ; i += 4 ) {
int inIdx = inOffset + i ;
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
getValue ( batch , inIdx + 3 )
) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 2 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 3 ) ) == currentSeg ? 1 : 0
) ;
$ { d }
}
int inIdx = inOffset + $ { c } ;
if ( $ { p === 1 } ) {
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
initializationValue ,
initializationValue ,
initializationValue
) ;
int inIdxSeg = int ( getSegmentIdAtIndex ( inIdx ) ) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
0 ,
0 ,
0
) ;
$ { d }
} else if ( $ { p === 2 } ) {
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
initializationValue ,
initializationValue
) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
0 ,
0
) ;
$ { d }
} else if ( $ { p === 3 } ) {
vec4 values = vec4 (
getValue ( batch , inIdx ) ,
getValue ( batch , inIdx + 1 ) ,
getValue ( batch , inIdx + 2 ) ,
initializationValue
) ;
vec4 segFilter = vec4 (
int ( getSegmentIdAtIndex ( inIdx ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 1 ) ) == currentSeg ? 1 : 0 ,
int ( getSegmentIdAtIndex ( inIdx + 2 ) ) == currentSeg ? 1 : 0 ,
0
) ;
$ { d }
}
setOutput ( $ { u } ) ;
}
` }};function KJe(e){let{inputs:t,backend:n,attrs:r}=e,{x:s,segmentIds:o}=t,{numSegments:a}=r,i=s.shape.length,l=[],u=0,c=M.getAxesPermutation([u],i),p=s;c!=null&&(p=On({inputs:{x:s},backend:n,attrs:{perm:c}}),l.push(p),u=M.getInnerMostAxes(1,i)[0]);let d=M.segment_util.computeOutShape(p.shape,u,a),f=C.sizeFromShape([p.shape[u]]),h= $ e({inputs:{x:p},backend:n,attrs:{shape:[-1,f]}});l.push(h);let m=Vh(s.dtype),g=(A,v,w,k,I)=>{let N=A.shape[0],T=A.shape[1],_=M.segment_util.segOpComputeOptimalWindowSize(T,I),R={windowSize:_,inSize:T,batchSize:N,numSegments:I},S=new AO(R,v),F=n.compileAndRun(S,[A,w],k);if(l.push(F),F.shape[1]===I)return F;let B=rO({backend:n,attrs:{start:0,stop:I,step:1,dtype:"float32"}}),W=gO({inputs:{x:B},backend:n,attrs:{reps:[T/_]}});return l.push(B),l.push(W),g(F,v,W,k,I)},x=g(h,"unsortedSegmentSum",o,m,a),y= $ e({inputs:{x},backend:n,attrs:{shape:d}}),b=y;if(c!=null){l.push(y);let A=M.getUndoAxesPermutation(c);b=On({inputs:{x:b},backend:n,attrs:{perm:A}})}return l.forEach(A=>n.disposeIntermediateTensorInfo(A)),b}var $ ae={kernelName:nd,backendName:"webgl",kernelFunc:KJe};var XJe=[aoe,ioe,vre,kre,Ire,Cre,Nre,Tre,_re,Ere, $ re,Fre,Ore,Pre,zre,Mre,Lre,Vre,Bre,Wre,Gre,Ure,jre,qre,Kre,Jre,ese,tse,rse,ure,ose,ise,lse,ase,cse,pse,use,dse,fse,hse,xse,yse,bse,vse,wse,Ase,kse,Ise,Cse,Sse,Nse,Tse,_se,Rse,Dse,Fse,Ose,Pse,Mse,Lse,Bse,Vse,Wse,Gse,Use,jse,Hse,qse,lre,Kse,sse,Xse,Zse,Yse,cre,Jse,Qse,eoe,noe,toe,roe,soe,ooe,uoe,doe,poe,foe,hoe,goe,coe,yoe,boe,Aoe,voe,woe,Noe,mre,_oe,Eoe,Roe,Doe,Xre, $ oe,Poe,Moe,zoe,Loe,pre,Boe,Voe,Zre,koe,Woe,Uoe,Goe,xre,joe,Hoe,qoe,Koe,Xoe,Zoe,Yoe,Joe,Qoe,eae,tae,rae,sae,oae,aae,Hre,Soe,iae,lae,uae,cae,pae,dae,fae,hae,gae,xae,bae,Aae,vae,wae,kae,Iae,Coe,bre,Cae,Sae,Nae,_ae,Eae,Are,Rae,Dae, $ ae,Foe];for(let e of XJe)g3(e);var Fae=["nose","leftEyeInside","leftEye","leftEyeOutside","rightEyeInside","rightEye","rightEyeOutside","leftEar","rightEar","leftMouth","rightMouth","leftShoulder","rightShoulder","leftElbow","rightElbow","leftWrist","rightWrist","leftPalm","rightPalm","leftIndex","rightIndex","leftPinky","rightPinky","leftHip","rightHip","leftKnee","rightKnee","leftAnkle","rightAnkle","leftHeel","rightHeel","leftFoot","rightFoot","bodyCenter","bodyTop","leftThumb","leftHand","rightThumb","rightHand"],Oae={leftLeg:["leftHip","leftKnee","leftAnkle","leftHeel","leftFoot"],rightLeg:["rightHip","rightKnee","rightAnkle","rightHeel","rightFoot"],torso:["leftShoulder","rightShoulder","rightHip","leftHip","leftShoulder"],leftArm:["leftShoulder","leftElbow","leftWrist","leftPalm"],rightArm:["rightShoulder","rightElbow","rightWrist","rightPalm"],leftHand:[],rightHand:[],head:[]};var Pae={initial:!0},es=[null,null],_d=[[0,0],[0,0]],vO=Number.MAX_SAFE_INTEGER,wO,kO=null,Ed=[[0,0],[0,0],[0,0],[0,0]];async function Mae(e){var t,n;if(Pae.initial&&(es[0]=null),!es[0]&&((t=e.body.detector)==null?void 0:t.modelPath)){es[0]=await KC(an(e.modelBasePath,((n=e.body.detector)==null?void 0:n.modelPath)||""));let r=Object.values(es[0].modelSignature.inputs);_d[0][0]=Array.isArray(r)?parseInt(r[0].tensorShape.dim[1].size):0,_d[0][1]=Array.isArray(r)?parseInt(r[0].tensorShape.dim[2].size):0,!es[0]||!es[0].modelUrl?Ce("load model failed:",e.object.modelPath):e.debug&&Ce("load model:",es[0].modelUrl)}else e.debug&&es[0]&&Ce("cached model:",es[0].modelUrl);return es[0]}async function zae(e){var t;if(Pae.initial&&(es[1]=null),es[1])e.debug&&Ce("cached model:",es[1].modelUrl);else{es[1]=await KC(an(e.modelBasePath,e.body.modelPath||""));let n=Object.values(es[1].modelSignature.inputs);_d[1][0]=Array.isArray(n)?parseInt(n[0].tensorShape.dim[1].size):0,_d[1][1]=Array.isArray(n)?parseInt(n[0].tensorShape.dim[2].size):0,((t=e.body.modelPath)==null?void 0:t.includes("lite"))?wO=["ld_3d","output_segmentation","output_heatmap","world_3d","output_poseflag"]:wO=["Identity","Identity_2","Identity_3","Identity_4","Identity_1"],!es[1]||!es[1].modelUrl?Ce("load model failed:",e.object.modelPath):e.debug&&Ce("load model:",es[1].modelUrl)}return es[1]}function ZJe(e,t){let n=e.map(a=>a.position[0]),r=e.map(a=>a.position[1]),s=[Math.
2021-03-10 00:32:35 +01:00
/ 9 j / 4 A A Q S k Z J R g A B A Q E A Y A B g A A D / 4 Q B o R X h p Z g A A T U 0 A K g A A A A g A B A E a A A U A A A A B A A A A P g E b A A U A
AAABAAAARgEoAAMAAAABAAIAAAExAAIAAAARAAAATgAAAAAAAABgAAAAAQAAAGAAAAABcGFpbnQu
bmV0IDQuMi4xMwAA / 9 sAQwAGBAUGBQQGBgUGBwcGCAoQCgoJCQoUDg8MEBcUGBgXFBYWGh0lHxob
IxwWFiAsICMmJykqKRkfLTAtKDAlKCko / 9 sAQwEHBwcKCAoTCgoTKBoWGigoKCgoKCgoKCgoKCgo
KCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgo / 8 AAEQgBAAEAAwEhAAIRAQMRAf / E
AB8AAAEFAQEBAQEBAAAAAAAAAAABAgMEBQYHCAkKC //EALUQAAIBAwMCBAMFBQQEAAABfQECAwAE
EQUSITFBBhNRYQcicRQygZGhCCNCscEVUtHwJDNicoIJChYXGBkaJSYnKCkqNDU2Nzg5OkNERUZH
SElKU1RVVldYWVpjZGVmZ2hpanN0dXZ3eHl6g4SFhoeIiYqSk5SVlpeYmZqio6Slpqeoqaqys7S1
tre4ubrCw8TFxsfIycrS09TV1tfY2drh4uPk5ebn6Onq8fLz9PX29 / j5 + v / EAB8BAAMBAQEBAQEB
AQEAAAAAAAABAgMEBQYHCAkKC //EALURAAIBAgQEAwQHBQQEAAECdwABAgMRBAUhMQYSQVEHYXET
IjKBCBRCkaGxwQkjM1LwFWJy0QoWJDThJfEXGBkaJicoKSo1Njc4OTpDREVGR0hJSlNUVVZXWFla
Y2RlZmdoaWpzdHV2d3h5eoKDhIWGh4iJipKTlJWWl5iZmqKjpKWmp6ipqrKztLW2t7i5usLDxMXG
x8jJytLT1NXW19jZ2uLj5OXm5 + jp6vLz9PX29 / j5 + v / aAAwDAQACEQMRAD8A + qaKACigApGOKAML
Xp8xlF5A7V4X8RtYs7PzfNImnx8sa8Kp9z3q2tEgp6angWs62ZZ5CTGoJ6DArGNz5p + UrID6EUrF
PUlW1EuN0XNW7PQ2L5j3JnoKXN0KijqNP0eYoqXBdgPuuo + ZPeupisWn2Jd4 + 0 r924XgsQOCff3 /
AJ1FzRKxDqGii6m3siiQ8F1XGfXI6YNWLfRbiRQMkcZI9fpTDluT2 / h6Qy8gDPbtmtG38JeY480Z
5 zSLUTZg8M28YwYxjAArXtdPt402qgHbpSaLWhma3o0Uqk7Nx9DWLaaVblgPs6qRyds2M / gRSQp9
zZOni2iWS2hlQ + kjYz9OMGrdjq89vIPPVhj + 8 M / lQyDq9P1WOYBlMZz1AOD + VdDaTiReOKulK0jO
tHmi0WDTlr0TyxRVhT8tJjIX + 9 SUxHXUV553BRQAVBcPhSBTSuxPY86 + IGti0s5I7dsORy9fM3i6
8e8 mfDO5P90ZrWWiJicNPpZZtxV / xrW0jQt4DOv6Vk2dEEdTY6BHuB25rpbPSo0QARjP0qTRI17W
wA / hFaMWmoQMgflQXYsDS142rU9tpqqenfNA7GgtihxkdKuRW6qMY / GkDZY8sY4Ap4hXbyB + VArk
EtuH4wPyrk / EGkOm + a3jw3suRQLc5i38SX9hJ9nnY + XnBUdPyNdFY6pa3KkkAE9l6f8AfJ / pSJT6
GhDmI + Zb4ZRycdv6ium0nUhKFydrelTsNnS2829RnrVgV6NKXNG55lWPLIM81Op + WrZkRMfmNNzT
A7GivPO4KKAEY4XNYWt3vkwPg4OK0giJdjw / xrqhm87Zs8tc7pX5A + leSajf6aHYJ50kn4AZpTep
rBWRm2Vobm4BXfyehPFdnpmnBFUY5rI2SN63tlToK0YI + KZpFF + 3 QdavwoKTLtoW0Toaswpk5pCb
LCxipAhoIuP2dKevHXoaYDylRyxhlwRQI4nxVoCXWZI1GfpXGtbSWjYPGP73 + NIGupt6TqMsLruZ
ih4xnP5V09mQ + JLd8gn0xSYJnVaVdkook69K34zuUGunDS3Rx4qOzHVIp4rrOMY3NJQI7GivPO8K
KAILt9kZrz3xlebYiu8KCCWb0XvW0NFch6ysfO3jLVjfXLIn + pQkKorl7WxNxIPl71g2dUUdpo + l
pBGvHPet23iC8ihFosrxirkHQUFo0IF4FXI1O726CpKLacCrMJoJLYHAPpTwucHpSRJJ5e4AZI9x
UqpxzVpCuOC8cUpQUMRnXttuB4rjNdsYyeVwfXpmpGmcvcQyafMCFJjPY10eg34BUg4DcZP8jUO4
HaRq3lLNF + IHet7R7jz7c56rwa2wz9 + xhiVeFy / T1PFegeaNPWigDsc0ZrzzvDNIaAM7VpNqdegr
xL4l6kywyRhseZ19lrdfAZL4jxYg3Fw20d63tJsdrDI5rm3Z3R0R0Mce1eKnQYAplIkWrMJ45oZS
NO3PHbNXIyfpSGWowSOasxLUiZdjFSqtNEMkUemKlAGKsRJjAppFAiORMjmsTVrNZEO4cfSoZSOD
1 eJ7WXBUzQZ + 7 nkfSo7e2Ei + ZaMzxntjBX2NSU1Y6 / wxqojiEFzkA8KTXYaUoWRyv3W5rSjpNHPX
+ BmpSg8V6J5gUUAdhRXnneFFAGHrTfu5PpXzj8S70 / aZtxzztXFbv4DKHxHI + H4GZiz9zxXXW8G3
GBXMjvLRXAx0oPGPSmMVeOnWrMTYpFI0bcg1fh54xmgovRcD3qxETSIZcRvzp + / B p E k q s B U q s M 9 K
q4Em4Gkxk0yRGXrVW6i8yFhkg + tJjRxGsWrxllkUMh9eK5uMz6bcebbnfG33kPcVkay2OntPKuo0
nhXI67c8qa7Lw3c + adjcEDGK1paSRhVV4s6A0or0jyRRQ1AHX0V553hRQBz + vNtt5z3xXzX8Qbdm
uic5YnOMdK3l8JnTXvlbwpYl + WySOgrp5YfLOOB9O1c62O7qQkc + 9 RsKChFPWp4DluOlSykaNruH
ArUgHShFNF2NT1qxGO3NBmyxGcE1N2560CFzjrUysO9JAPDDjFOVuKoQuSRTWouBkazbCa3cd8cV
wF7IISQccHBzUSWpV9C3o1x5b5GAjdQD1rs9DjC3kckbEhqKfxIzn8LOupRXqnkPccBSkUAzraK8
87 wooA5rxMSI3HqK8B8bQl9Q8sffY5b / AAraXwkUviNrw9pH2W1ViMMRTdRjw4HpWNtDti9TPc4P
FQs2M5qdyyMHLcfjV63HTAoBGtap0wK0YxigpsuRDtVhVYd6GQydVwwIqdRnqKCR23I5pCMUW6gD
YNKuetAEise9KTxQBWuFyhrznxNZkXjFeN3I + tTIZg2OqmzmxNF0PO3vXp / g2 + hukVl4zyPanTXv
JmVR + 60 dpThXpnlPceopWFAbnV0V553hSGgRynjC5FujOey14Ssp1HxNmTnc + a3kvcIpv37HoEYQ
QmMdVHSsnVbYJF5jVk0dsNzlruVIsl2wKxbjWrVHILjg1CRbZJb + ILHPzyhfStODWLQgFJFYd + el
UJM27HUIXxhga1Y5lLVLKLkMnoauxnPPrSEx7ShF + Y / n2qrc6xBbhizDAqkK1zJuvG9nbg8ZA681
ly / Ei052RO3uKAsZlx8QGd8xxvt9Aa1NH8dK7AXMcip64zigdkdrZX8F7EJLdwwNXMkrz1qRMRly
CK4TxmpidWI49felPYSOMmi80NIoOV6qRzXYeA5SskYPfirpfEjGr8LPWVHyD6U4CvQPL3ZItOYc
UDOoNFeed4Uhpks4H4iE / Z5MeleMeGULeLgjds10S + BGdL + Jc9OSBU2Huc5Nc74yvUtrcDBrJnZF
63 PJdXvLy / lKWw46bvQVz82jXhkLO5Y + 9 ZlsYthcRnbIjY9R3q3awTRkEM3WmJI6C0ea3dGRsr1x
XY6TqW9FLHnjrUs0izpLK5DDjofSta3ckH09KRUkZuuTvFGdvPauE1Y3U6Mqbssf / rUxHPTaJPK2
ZmJPbBqzY6DCZh5xJC9s9aBJHU6dpemJjfEmfetJtI0 + VPkUr / unFOxdiextHs33W07YHQHk11mk
Xb3KbZ1xIvcd6LEyWho4Nct41sTPYb16ipexCPPZN + wYGCvH1rrPAEJmvkPoc1VL4kZVvgZ6yFwK
cBXoHkkqinFaVyzo80GuE7WJRQSziPiGdthK5HQV4x4J / wBI8WPIewNdEvgRNL42emO / yj1UHNef
eNpRczbC + I17DvWT2OqJxc0sMK4TCisy41q0hfEkqj8aixdwTXNOlwvmqD9anS9tXH7uVG + hosO4
/ w C 0 o O h r R 0 + 6 G 4 Y N I E z s N E u C x A P N d j Z r u A 4 x x U m j I N S j U R k s O l c b q F y k b n j F A 1 s Y G o a s s a k n C q O 5
rl7rxhGm7yBnBxuJq0rkSlYpw + NLlsfd5P8AerVsvHEqSBHwPVgcgVpyMyVXU3rXxcHYETAk + hru
/ D W t i 6 Z S T y O K z Z q n d H a x v v U G q 2 r Q + d Y y q R 2 4 q W I 8 d v b r 7 L q D x y D A z X p v w 6 F v I x e P G S M 0 6 X x o y r / A
zviKFHNegeX1J41zUhXioGbuaSuM6wpCaBHG / EcA6HN / exxXjXw2jL67cv8A3Qa6H8CFR + NnoWpO
I4XI44rxLxrqjQzSEsQM1gdSPM9U1uR1YbmWIdXHf2rmpIb67YS28UrRlsLI3c / jW0VZGUpO5pW1
jfLNOjahawzwReYI5cjzMkDavHJ5 / SrVv9uhtPtVxCPLBwzxnlT9KGghLU3tKvvPjHzbl7EGuisJ
GRxWLOg7nRXJEbDjmvSNK + aFSfSoZr0KutRkphc4NcRrdkVjL9aVio7Hk3iqS8ubhrWzUlsZY9kG
cZNc5D4aee5MclzJIFTzHAO0MfatqSOWu7bFS1srDUZEis0vIZoUxPvfcC + 4 / dx2xjr712XiTwXb
WmlQ6hol3cRhoFd4rlg3zY5wR0GelavQwjq7GD4etdVvSnk2wAB + 9 v8A8mvcfA2kXiRo0 / UdcDis
ZnTTulqeoWqbUAJqWUb42X1FZlnjfjSwlGrr5S / eNdD4RkvLAAQ4yRyaUZcruVKl7TQ9I0G + mnzH
ckFwM8VuIK7ac3KF2eXiKapz5UWYxipNtMyNejNch0jSar3cjR27uoyQCRVRWom9DxTx54gu5fMi
lbKdMVjfCZPNlv5v9rFbVHpYqjGzbOn8SzFI9o715L4u0r7arYzk + lYdTqSujy7U / C0u4vHk + WwO
xuh9q3J9dgvbdVukMV1EwbDDgn04rZMwlHoZ + orZ6hfQ3RWVnQYCgZAq + 8 U0ln5NtBsV2yxYcfgK
JtW0CnB31LlroVwJ1nQLGDjeP7w + lb0dsFxjrWB0tHS6NuWPJ6A16ToUm63T3Gallr4S7cxiTjrX
PaxaF7dlVeSMUhxZ5jd + H7qCa4eF3DSE5x3zXN3Wk6jbyeaiFWUY6ZyPStYS5SalPmVipFbX0E4c
W0alvmPHJrag0rVvEE6LdljGpG2NRtQD + tW5XMI0uU9M8NeFo9PiQhecDIIrtrOMIoG3H4VlJm9t
C6CB06VPGM1IHLeItGS6uw + ORT7e3jsbQvj7gzUNam0JaWE + HN7NqOqX80n3FO1RXo8YzXdS + BHk
4 z + KyzGPapcU2YIv7qQtiuaxvcaWqG4O6FwfSrS1JbPnrxoxkv7qIfejcitj4V2f2exumI + 8 + aKn
xHTT + G5d8Txlm4rjLxMsQwzWT3OiK0Mm6sEkVsAcjFc1d + FEmlGwEDPQVopaEuOpr6f4ZWNAu3tW
vHpAj5ZQcUFIWaDjGMVUMQ3cVDBmvbhY7QAV2nh + T / R1yeKhlrY31 + b61FcQK6nIoJMi401WblRi
qr6PCw5UYq9y + YgOgWzNkRrx3xWjp + nx2v3FQcelAbmko9anQ4GBUNisPHWr1qMrQhS2K11HvmYV
hamcxSRZ5xRIqluS / DKAQQXZxyXrvo2FdlL4EeZjH + / Z b j N S Z p s w L N B r E 1 G t 7 V E 4 O D V I l n h / j 6 1 F
j4lmeTGyUbq6LwdEqWbeX0YbhSqfEddP4Bddj4JIrhL5d8h7VjI6oLQqKNzelWre3yc4 / ClFjaL6
wqBxxUUxwCKu5BmXRA6c + 9 ZjP83FSBoQuPs4BrsNBlUW659KmRrDY6G1lyQtW3Hy0lqQ1qVJnAbm
oy3b9KYJCqRj3o4zRctIlhjLHmpSuOBRbQOpLGpPFaES7UqkZzKN1KsEc87 / AHUUmvPLTVGv72aQ
k7WJwKmRrQ3ud74Ltilgz4 ++ 2 a6iNDXdS0gjyMU71my7GpqTbxSbMki3SViajTTHqkSeR / GeyZmg
nQHkEE1S + F + oPPavBL96I4 / Cia1udVF + 4 dVrkW + Fq8 + v4tjMDWUkdVJ6WM0cNV + F + MVmjUcZgqnP
1 qpNNnkcVRLiZtxIS1UzzIF7mghlxUZpVQdq6nTVdAoAOKzkbQWhvwM6gMM1twOJYx3NOJE11Kt1
H1 / pVVlwBkk + 9 NocXoOQ45FPj + fkUJFF2NSB700v / hTEty5ZpkjvVyUgcCq6GM9zC14 / 8 Se6GcZQ
1574 Xs5WkI2HBPHFQ1dm1KSSZ7Rotn9l0 + KPHIHNacae1dy0Vjxaj5ptlhVp + 2 s2CJ9ppCKzuWNx
zSFc1SYrHNeNdIGpaYw25ZeRXmvheyk0jVpEdcLJ0q3ZxNKTa0O3vQHg / DNcHrsJDmsmjspnNzNt
fFIJ24GazOhC + azDmgZIOOKBsp3J2qSaZodubq58yQ4QAnmhGT3NO18pb7BORmu205LfYpyKVkWp
Oxr5gKYWoIZWgfGfloFq1qTPLubnGO1RPtxg4P0oBAkY / hBz6VNDDkZ6AU0W2WSdqkdKr9ZOaGSj
VtcLHmnOcgmmYvcz7mBLy3MbdD1q9ouiRK6bUAVeelOC1InPlidSsWMDFOCEdq3uefykqrinYqGy
rFvApMVka2DAowKAsMkRXQqwyDXn / iWyitNQ3qPl6itIvRoF8RXinW4tQ6HI6GuW8SIVBPalc6qe
5 x9x97r3qruwTjrWZ0ksZ9TUmcDNAmZ9 / wAoao63rR0 + w22MLPtAzt6mghmfofiB76LdJBJBIp5D
d / oa7bSdWLIPnpDi9TM8TeKdas51XTbIyxd3J / pXS + E / EFxqNoFu7do5OmD60maHWrnZyDRkn / 69
MlEyOR0xntVoNx + FUgYjPxg4FLCuWDZyKQr2RoRnP0qO + nEFpJITgAUzLqZnhu6 + 0 rknOTXpOmwJ
Fbrt5yMmnHYyr6Oxb2ijaKLnPYMClwKQWK3n0hn + lachHOJ9pNNN0apQFzsY10a4v4hXQh0xpieQ
MA1XLZNjhK80cT8OdV + 3 Wl3A7ZZJCw + hrR1qLcjZ / CsbnfHRnFXseHJArOYYbrUs1uPhYbuatqFP
ByfSkMq3UIINYkto + 87 Tx6GkSxfsDbflGD7CtTw / pk4nzITtPIFMFudsukh4Rxz71paTpKwP5jcn
0 qTRy0NORMDgVCqewoJTJgAoxjntTiTu7fWmFxAcnn1q3EPl + X8KZMi4gKqB1Peob / Tv7Us5bfeU
yOoq4R5nYxqT5I8xieH9J1DTbvyJELRg8ODwa9Ms5mSFV9BWiptbnNVrKdmif7Q1KLg96XIZc5Is
pNL5pqeUrmMtZs0jzV08phchaY00zH1p2ZNxjS1g + LdJOt6U9ssmxjyGp2urDjLlaZzng / wUPDqz
TSTmWeTrjpVjVk3Rvjr2rnqQ5dDvo1XUd2cTqSNk9OKxXGCeKxZ1DAxHTr2q5C / y8GokUhsz54qu
uCxzSQjQ0 + FZblR2ro4bZYiMVQ0dBb7Qi5x0qzuG5QOh71LYErDufpSeWrHnimIXbjkUjLkH1Hem
gGxryc + tXI19KYmWegq9YLiLJ7mtqS945cS7QsWehqxA9dEjz4krPSxyZqbFFhGxUm6smjRM55Lk
HvSvNxXTY57kLT + 9 MNwKdhXGm5FIbkU7Bca1wMEVhaiuQcVhXWiZ14R6tHGanGBI2OtYkqEHjgVy
s9ErEeo6UBsHipKEZs5qpPdRxcbhx70NCSuybTNWihc5brW9Fq6vjMnFSdEIdDRi8RRKygZbHFbu
m6nb3RA3gMegNJhOm0jbXGOoxTuCc1Rz3FyoGKawz9KaAVcZqeMgCmIkB4FaUTbYwB6V00Fuzixb
0 SFMuDU8Mlbs4UPeXHeiOXkUrDuXYnyKk3cVk0ap6HMxxketSMhrcwRC0dMMZFMQ3yzSeVQAeUaz
9 Vj8uPd271nVV4m + GdpnHX67pCeKyLtBtNcR6xlk9RVeWTb3qRnO6trgttyIfm71z7ai8j7 / AJmN
DNqUVa5Yi1AnjynHuBV + 11 YJhWWXcP8AZNSzqgmaEerSsf3NtIQP4mGKtRavdRgMIpVI9KjU0a7n
R6T43uYQI7qN2Tpkqciu503VVuQGAYZHQjFVc4alPlZrpKGAznpTwxOc9 + lWjIlUACnM4XApiLNk
nmvnsK0NvpXZRVonmYqV52GsmanhXitTmFkSiJTSAvwrxUxXIrJ7miOfjf1pzNWxkRlqYWpgJupu
6 gQbuahvIxPA6eo4pNXVioS5WmefakGhndH4INZs5DJXA10PaTurmLO21uKpSZqGMoXGnRzBiyjd
9 Kx5rcQS428fSkjanLoaOliHGZFB56VswW + mtPufcBsGOAfmxz + tFkd8HpoaUx09FAtFY8DO71qb
Sms / Nb7RbecG6AEjFLS5c78t + p0djpVs9wsyQiJAdyr1rW + zqjErzSe559Sbk9S3C + MA1bjbgE1S
MSXzMVG0vNUI2tPKrAuCMnrVzNd0PhR49W / O2xrHmp4TxVMzQshpIzzQBehqesnuaI5VGzT2bitz
FEbNTC1ADS1JupgG6l3UAc14s04yR / aYRll + 8 BXCtLncDXFWjys9TCz5oW7GddH5qqNzWDOgQnC8
VSuo1kHzAGkPYopEY2 + RWxV23Vzj5G / Kg3jWaNazhZuqNXS6TaKhB2c0jR1nJWOlhOxRxU4YkCgx
Y0OQatQyDbyaaFYe8uF4NY3iC9ltbVGj43NTIL3h7WzMihjzXVQXYYDdW9Cf2WcOJpfaRZ3g9KsQ
mupnCLIabGeaAL0LcVY3cVmzRHIxtUhetzEjZqjLUAIWpN1ArhupwagAfDKQ3Q1594v0c2bm6tx +
5 Y8j + 6 ayrR5onThp8s7dzkZjuqAAmuBnqC7c0iwgtzSA0rWzjfGRW3ZadDu4AoNYo2rfS4v7orSh
05 UA2r0pDbsTm29KRottBNyJ0wpJ9KhD7f6U0ikNWffIFBz60zVUW52ow4UcUN6EPcx44WsbgOmd
ua7TT5Bd24KHnFKnLlZFSN4koluLdueRWvp14swweG9DXoxldHlTjYtzGoo25qzEvwtUxas2jRPQ
5 CNqkLVsYoYzUzdQA3dSFqBBmnqaBhuqhriCXTpVIzxUz + Fl03aSPI9QTypW2 / dz0qKNw3SvOPZR
Mqin8VLKRcs3O4Cuk0w / MDjt1NBtHY6O2IIHY1pxgFaETIRwMkjtVSUEk4570MlFW5bap6dKzWm8
1 tqH8aY + hp2FvGoGayNevVt7 / ap4xzUvYjqTLtvLPcvJxSaVcyWsxTnFZlnT2t15xHmCtOBYwQy4
B9q7cPO + jPPxFO2qLEj5HWo42 + aus4HpoX4W4FTF + KlotbHII9SFuK0MUNZqiLUDE3UbqBBupwag
Bc1DefPbyD / ZND2KjujyPWlKzuPesRZjHJXms9lMuw3StjnmphKDSLTJ7OfE3JrpbO4GQc9qlnRA
3 LO82k5NbFvdADkjBoCSHyXIIIzgVQvdRigT7wzjgUzO1jHknlvG7qnp61etYFQDIpCZoqVijzXn
3 iC8EmsOuaCGb / heR / s0ijkVv6fbxy3QMg5xmsnuX0Ldzut3 + UYTPWk + 2 GJSe + M1pFtamcldalmx
1 eO4XaThhWnC + TXqR2PHqL3maUJ4qRjxSEjj42qXdxVmaGs1MJoATfSbqBAG5p6mgAzTJTmNvpQU
tzzHXY83D / U1zF5FhjgV5r3Pa6FMsV5HWnLe7RhqBRdmTwagN2d2K2rPU1C5LAnPrUs6Iysbdrq6
f3gK0BrUKj / WClY05iM6xLOcQAj3NT29uznfKSzHuadzNu7NSBFjHNSm5VO9IRnajqoWMhTzXFtA
bvUfMduSeg702Qz0rS7FbTToQFwzjJqaGTFyfK5PQViyzUuFmuIdgGABya5u / vTaN5cnUHFUmLoZ
zyskwlgJweSK6zQdUEwVJeGr0aUrxPLxEfe0OrhPAqVjxWhznGRtUwatDK4jNxURbmkAm6jNABup
6 tQAFqhupNtu59qUnZFwV5JHnWsHdIx96w5lz15rzT2uhRmt85xWbcxMnUGmZlB0bdxmrNvFIcfM
350 mWjbs7YkDJY / jW5ZWW4jikWkdNp9mqYJFaJdEHHakUULu / VB1rLn1Ld / FgetMGYd / qWSQmSa0
/ A e m S 3 2 p f a 7 p i L e L k g 9 z 6 U m Q t z 0 W 7 u Q 2 c Z x 0 A 9 B V z R 7 c A e a 6 j 2 r P q X 0 L 9 9 K R a t 5 A 6 D k 1 w O o K Z 5 2 a
YfMORTYRLujiGWEq6 / NWza2yKQVHNdOHerRy4laJo6TTnbbtb8KuM3Fdh5z3OJjbmpt3FaMxAtUZ
agBN1GaQBzTwaAAms3VbjERUGsa07RsdeFpuUuY4jUjljWTKK4j02RE4IpJYFk6imQkVl0xWarsO
mAEcUi0bNnZBR0rWtoguMCkUi21wI161mXuocEKaYXMS4u + pY / hVCSWSY4HT0pEmlouiSahdpEBl
mOceleiwWcNjClvHgJH97Hc1EmVFFi3Czy7mwIl / WtJbjP7uLgd / apQ2VNVvtsBhiPzdK5S4nAuR
nqOCaTGi9pcytPlU + XpmumtWII44rah8ZjiNIXRuWeNvvViQ / LXpJWPJbu7nCRvVkNxVsxBmqJmo
EPiXca0YLMuOlJsuKuPlsSi5IrNuG8s4HWs5VEkbwoOTKsk + FJY4rC1K53k1xTk5O7PSpwVNWRzt
4 cms + WpKICtSLTETQj5q0YeBSGiys23pUguGxQMq3E59ayrm4x3yaAKiRtO2WPHcmhruKFxFajzZ
ScA44qRHoXhuMaLpxaUg6hcDLMf4F9KlhuDeXGASIl + 8 azZslYma68y48m1 + 7 nFW5rtbRNhb5z1p
iMKbUg0zuW4A4rPgb7VdKXOMmpA7HRbMS7nUYiUda0lkQOBngVrS + JGdbWLRt2bAx5BqeQ / LXpnj
PQ4GJ + ashuK0MhWaoWcA0AaOmASMK7jRNPWYBmHyiuepO2x10qfcv6vYxCzYqoGK4HVYVTJrmb5l
c6oaM5TUJ8EgGsG4kLNUHT0M64OaqMMikSRsuKbnFMRLG3zVehOaGNE445NNlnVFpDMu6uie9Vo1
8 z5mOAOST2pDK91cNN + 5 tsrH3PrW54a06KxT7fdrlh / q1Pc + tJ6IUdZGvHPLezMcnBOWbsPap5r3
ylFtbdT1xUWNWzU0 / Zbwlgfmx8zGsHWtRHmMqE59aAMyNifvHPc1f0gtPdqkY5JosJHeNci2tktY
2021-10-04 22:29:15 +02:00
euPnNY + oXWZEVJNrZ9aun8SIq / CzodHuriIokhDIR1ronbKZr0o6o8ipoz //2Q==`,XS=`
2021-03-10 00:32:35 +01:00
/ 9 j / 4 A A Q S k Z J R g A B A Q A A A Q A B A A D / 2 w B D A A s I C A o I B w s K C Q o N D A s N E R w S E Q 8 P E S I Z G h Q c K S Q r K i g k
JyctMkA3LTA9MCcnOEw5PUNFSElIKzZPVU5GVEBHSEX / 2 wBDAQwNDREPESESEiFFLicuRUVFRUVF
RUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUX / wAARCASwBLADASIA
AhEBAxEB / 8 QAGwABAAIDAQEAAAAAAAAAAAAAAAEDAgQFBgf / xABDEAEAAgECBAMECQIDBgUFAQAA
AQIDBBEFEiExE0FRBiJhcRQjMkJSgZGhsWLBJDNyFSVTY3OSNEPR4fAHFjWCokT / xAAYAQEAAwEA
AAAAAAAAAAAAAAAAAQIDBP / EACARAQEBAQADAQEBAQEBAAAAAAABAhEDITFBEjJRIhP / 2 gAMAwEA
AhEDEQA / APqYAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAKNTq8OkxzfNkisQC8eb1XtRNbzXT4q7eU2nu0MntRq / D8StMccvW29ZmdvgjsTyvZjxOLj
+ s8WLxn8TFPXs6Oj9oct7c14rkxz22nrB2I49KOdTjelmszfmpMeUxv / AA28OqwZ4icWWtt / SUi4
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAmdo3nsPNe0Pt
Fh09Z0 + DNWL7 + 9 O / 7 A3eJcZppsV5raI27esvH6jX5ddM25p79Ilo59VbUZOe2Tm / PeGvfPfT2iKR
PLv1 + DO678XmW / a97U6TtOyzTbTF538 / T9WjTNecm9a7126tqk3rSYxY5ta1plRZqZNXGjyZcPXl
mZmsx + qjBrsuO16xM7eXRt04JrdTltk5OWJnfaWf0a2lty5MdZnfzSn + WOHiOutFpjHa9e8bQ2fp
+ alYy462pk7zXbuxjPesbRS0f6ZZV1ET1tErzXFLHo + A + 1 ddZf6NrI8PJHa1vN6iJi0bxMTHwfOa
zhzd61v1846utwniM6DUdb3nBaNrVmd9vjC / ZVePYirBqMWppz4rxaPgtEAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAItaK1m09ojcHnvarjM8P0vh49 + a / eY8ng9D
h1fGM1rxjtGPfvbzdbjuTJxHX48cTPNltM / KsS9Dw7S49Jp6UpHaGe2vjz1y9J7LYK13vHWe7bj2
ex1tvM80ekuxW3RnW3Vm6P5jRx8H0 + OYmMcb + bapo8GKPdpC6bQwtdHU8JpWkdJ / JweL6e23iU67
d4dubSqyVi9Zi0bwIs68XGp36TtEq7ZJmZmevzdbifCKWtbJinkt6eTgZPFw32t + sRurbWVzxs1y
Rv6T8V1NZNPtfq0seTm + Kevr + SZuxXjvaPiV8N4viycto9HseG6 + uu08W6Rkj7UPmFck1tE1nlmP
Ld3eA8V8HVVi1pjq6Ma / pnqce / ERMTETHaUrKgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAADW19 + TQ5p / p2bLS4v04Zmt5VjeQeJ4bjnLqsupv + Ka1 + ERLv4reTmcNxcuC
vy3l0qdI2hlr66sT02ot0ZV7qqrInruzrVZLGSZ37JjqgYTG0K5lbaFVhDT1Ub456RPweY4hixWi
eSdpjvD1eWejz3FNHWYtkpvFo9EIseb3tS3SerOms22rfpPqZKzvvHSYUz70TExG6Gdbs2rljeJ /
Mx5L0vEzPaelnOi98c9J2bFNTFpit47 + a + PVUvx9T9nOIfT + GV5p3yY / ds67wvsXqpxau + G09Lx +
r3TqrEAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAADV4ljnLw3U0jvO
O0fs2lWqyUw6XLkyfYrWZkHldBEV09eveG3Fq1mI3jd4vPrOIaid8G9MP3Y38k6fNrt / rMk9Ou8s
tfXXn49rGWInuy8SO / k5Gl1E3rG / fzbOe94wTy99mbRvTrMOOvNfJWsesywniukrG / jU6fF43WYN
TmtEeJtEQ06aSmK2 + bNtEd + qfSO17unF9Hmvy1y13XWyVmN4tExLxVK8PmNq5NrT58zawam + m / yc
0 Xj8NpRYSvQZ7xEOdqI3rPozxayNRXe0ct / ON03jmrKB5nV4q1yTO20Obmv4c + cx8HoeI6WZpNoj
q83niYmYscU0r8aJ6T1n49zeJ + Meqm1drb9J + Kd5p136StGVem9l9TbHxLDFp7W7 + sS + q1nesT6w
+ PcAzVjiGHftzQ + v4f8AJpv6On8jH9ZgIAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAABp8VrW / C9TW0ztOO3b5Nxp8VmI4bn37TWYB8f1HFtTfUfR9FWJmsdZ9I7MtJxDX5s
d8ta1y0xzteaR2277rcuhycP12SceLxMeWNpjttHwlu8I0mfQ1y + D7k5YmJmY36T36Ka43z / AF1t
cI1ds + qxVj7 / AEej19PCw9HJ4NoK4OIU5Y35YmZdzVTGebVZabx5jJS + Tmns81rNLm1Wrzc9rVw4
Yibbem72mXTTS0w0M3BvEta1bWrM95ie5EanY87wXgNOL6XPfxraXLhra / W28bR / dzYzarBqJxRe
bzE7Rt5vWU9n8mPHOGmS0Ypnea1naJb + k9ncNLR7u2y / WcxXO4TOoyUrN6zD0FaW5Y3hu49FiwUi
KxCvLMR0hlW0jn6ukWw3iXjOJzbDlneOj3GaN6zDzfFOH + LE7SRGo83XNSZ2lbG2 / WfdlvaT2cy6
rNFInlrv1mfJ37cK4PwTTxOoidRm2 + / 2 / K F u y M p 4 7 X B 4 L i v X i u n r H 2 b 2 i H 2 q n 2 K / J 8 x 4 f G D N x T S Z
9 Nh8OviRvTyfT6xtWI + DeXs9MNZubypASqAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAOZx6 / LoOWPvWiHTcf2hiZ0e8fc2mf1E5 + vP / AEeuSd7RC2uKtI6QjHfeINTfwtPf
Jvty9WPfbt / lucP03gxfJf7d / wBoReYpm97zaNeLb4Ims9Nt94auDjem1Wo5PFi1onylS + 1 o7l8V
bxvtupjDMdNkYtXS1 + Stt + m63xImEJ4xjHER2ZxMUjeUTO3VRmydBbjLJqPi08mbeVOXJPq1sl5Q
Vbkz9 + rRy35rxHqzmZlVEe / Ez5LRlW5iyfR6zffaIjq1OSNZps2a21rZInafSPJhxGMl9LStLRWM
lorM / A4dkrWbYfLZC2W / 7 K6eubX6b4RzT + W76K8b7G6X62cu3Sten59nsm3j + OXz3 / 0 ANGIAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA0OIYfpOHPijvNNo + fdvtXJO18k /
/ O w P F Y b z 2 l s 3 j x 8 V q W 6 x M d W P E d P 9 D 4 l k x / d t 7 9 f l L L H b k x T P w Y 6 n t 2 5 1 2 O R T R z E 2 x 4 / d p E 7 c v k m e
E4IrW3hRMxO8THRtU1FKWtvtvK2upx22rzRCtXkqzh2jtF7ZbT122b01ndnpuWuP3Z3 + Ky20qDVv
fauzVy3mejZzNK8dVjqi87KLRLYtXruqvXzkQp7Qoid88R6rcl + WGlW0 / Sa22mfhCZOq2x082ix6
jkm822pO8VrPdr4dNObVeDo8XW3uzMbzK + mvxT7szE27cvnu9j7PcNjSaXx8mOIzZevbrEeic5tN
+ SZnpt8J4fHD9HXHO3PPW0x / DeBtJxx29vaAJQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAKNRim9Z5e89Nl4DzXtVh5babURHrSf7f3ec1 + qnDorWrvvt5Pccb0n0zhmWk
Rvevv1 + cPE2rGTFNZU26PFfxwa5dVkjelI2772nZnX6bbrEUq3o0d678u8wmuDL2ittvVjXdneeK
cGv4jpJ6U56 + kS7 + j118 + GLXpakzHaWlp9NNY3tv + bbiYiNoQy1y30uyZJlrWmZnuym6q1iIJnop
yW2Te8bdWnnypQqzZOadokiIpSZntWN5lrxki19vNRxrUeBwnNNd + fJEY6 / OejXLn3Xe / wDp9wyn
E8uo4lqqxblv7lJ26T6vpD5X7G8QycKzeBMbzMRM1 / FH / wA / h9QwZ6ajDXLitvWzRgsAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAeL45w + dDrZvWv1OWd4 + E + j2jX
12 jx67TWw5Y6T2nzifU + rZ1y9eHwzDYxxEy18 + DJodXfT5o96vafWPVbjyxDn1OOzHudbM0rt2UW
iI69mVtRXZq5tREb9VUoy2iIlRbJ0UX1VZ6btTLrI7V6yk62M2oisT1c7JmtkttVMUyZp6x0beDS
RWOvdKijDimvWd3G9pNRMfRcNfvZOb9Hpb0itJeP47k / 3 hgjaZnbaP1XxWW3T0movbNS0W645nbf
0 nrMPpXs3xamoxdJiLbe / X1n8Uf3fKsOTw4jbaXo + EarJhtGTHMxeJ6xH7Sti9Zaj6x3HM4NxXFx
DS1mtoi8dJrv2l011QAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AGjxLhODieOIye7kr9m8d4eM4to9RwjPXFa0ZIvG9bR0fQXmPbDFvTTZPOJmEWS / V8bs9R43NxLL
G8eFbePg1bajU5 / s0l1ceKLx1hbjwRE9mOpx0y2uRTSZsm3PMw2aaKtIjo6kYo9EXpET0hVLXxYK
xC6MZvyx1lFs0RHfaPiCnU12pLyHGNDbUajBekWma2npWN3p8 + opa20e9LSyZLxExTlpM + vdOdcZ
a9tPS8MyUvFrzWlI6727u1pYxYrbVmb7x + TQx6au3Nqcl7 / 0 rcmW9axGnwZJj1novmxnZXV0fFp4
ZxLBPgTGK8xzXr5fOH0bFlpmxVyY7Rato3iYfNuG2x56Wrqa8s2jz + 7 Lu8O12bS6jkwzN6THNNI6
tvrN68Y4rxlx1vHa0bskAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAA4XtTTm0OKfTJ / aXdcL2pyRGjwU362yb7fkJz9eTxxyZJjyltRXzUZK7TFtl9Lbwy06YzrHwa +
fJFd / wCVt8m0bQ0eS2qzcm + 1 K / an + zNZFL5M1pjFXeI72ky48eGnPkvNp27 + TPU6nHpMfLXaIjpE
erk5dRMxOfN1mPeisfshW1ne1a1577Y6x5R3U0zze31FOWI6ze0byU098kRlzbxM9qrMlPDpyRMR
Md5Vt / Ihp5898mWZm1pjftE91uCt7fCI7dWeHDEW3t723l6rslqxWZnasR + SYhFbzhnfxJ2jyeq9
lcGXWZcmW0zWKxHLaI7794eJx5fpfEKabT8t8l5isddo3l9S4VjrwrRUwzSJt3tav3pdOL6Y6dXD
j8HFWm + / K s U 4 N R X P v t W a z H q u W V A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAa + fXYNP9u8b + kdZBsDkZOO135cWOZn4y5Wu4xqctbe9y19Kp4njt6vi + PDm8DFMWybbzPlV
5 PiGtz67UxbNbeKTtWIjaIXYpnwuaftT5tXJT3vmi1pMsrU5qIrG1V1a + 5 DCa7b9GFbRr5J6Wnbt
Cu + Wmk0m8956z8ZWZNorbfzcbX5rZslazPux3hUt41NTntktObJ13 + zX1bek01r4 / HzVm0bxPXy /
+ bNfDgjVa2uOY92kdfg6ufJOKvLXtttVVSqbcta2vM7zXtHpLQy5ZtMd + vWd + 7 Zy3mdJHXra3f0c
vUarw7zFY5rT2hH1Lavnrgx81p3U49Pk4nE5L35MO / StfNRXR5tXnrS8W67WvfyiPSPi7uLHFK1p
jrtSsbR5Lc4RzsXBaYreP4l45esRD2HD9fnw6evvWvO3Tfr0aGk0U55ra0TFInv6uzgrXFXlx0i0
77 RPlC83Yj + JW7oddqr6vHzTTw9 / f6dod + L1t9m0T8pcbFSmPHER3892W0zPuz + jSbVvidkcqmfP
Sel7bekrI4n4dZnPWIrHeYnZee2Wpy8dEaml4npNZblw5qzb8M9JbYgAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAABEzFYmZnaI7yCXL1XGa0jJXT0571nbee27DiXEprp8nhbxG20W8
5 cbD0ikfnKO + urTPvjoZdXqctdsmTaPSvRpWmsdZ6yztfaGplvv3lWW1tyRlz1x0vkn7Vo5atTNe
Y0 + 1 o79V2KsZsvX7Ne5mwxnyTNvsx2iGneM / rCdRSuOsTasTt5kRFtpjqmOH4t4nk7estiMNa97R
Hwhna0iuKTEdmGWa4672nZtRele1N59Zlq6vLOSsYorEc07qcW65euzRvtXvPZy52naZ7ujr6fXV
rWdukREK8 + njHgmZmPc67bq6ivVWhxxgxZLztNrT1mZ / SP4VZs0zaOvfp84WUtNsXLvtv3699 + rU
z7 + Jtt5qURqMnPpctaR1rMSw4ZoK57eNk6xHaJRh97Ltt7lo5Z + L1HAPZvVauZ2nFTSzMTzeJEz8
to6xPfvsZntPZ9rXxabmxzefdrv0j1dXh / BcmstW1qxTHHasR3 + b0GPhGl + kWmd64dNEVjf73T7X
y8vy + Ddx6O3iRakxTH5RXrMw1 / lX + 3 Itw2MFIraN48qRHdZi0cUjmmPen9noox1iO0fNzdXEYrTt
stcmd9aX0bJ + HePmiKTitO8TMLZ1cVjrMfqpz6ys4pjfrPRWZ9rXXptUit6zO + 23 VyaRHEc05L1 /
w9J9ys / en1ljqdVbwYw452tlnl3jyjzbmmiMeKtYjpEbLeTXPUU8ee / + qjJpsV5rbkrFqzE1tEbT
DpYNbW21Mnu29fKWna0KbqTdjXXjld0cvQ63ltGHNPSfs2n + HUbS9c2s2UASqAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAOVxPWe99HpP8ArmP4b + r1EabT3yT3iOkesvMVtN7za07zad5l
XV5GmM9vVfEstvDx0jtaVVMlq + UJ18b5cMRvPeSuK87bUt + i2Z3PtG7zXpjkzXt6R + TXyTMzvM7t
ydHqZ + zhv1 + Cv / ZuqvPTHMfOYaTMil1a1K2vHSLTELq2v + KWzThGo84rH5rq8JzedqR + ZeI7WnOS
34 pYTafWXR / 2 Pln / AMyrKOCWnvmiPyR6O1y9585lhWJvl557Q6eo4T4dYiMvW3b3UanhldHpJtGX
e09unmjsT7eb1l4trI2t0hsZfrdNO0bzy + nzU20 / + NmkzO9esz + TZxWis9dttvPv + Tn21jjaW8zn
26 bTG3mp1M / Wzv3t0jyWXiKZJmsTERaZhXXDbNl8WaztWenxZLstPp5pau8frDtVrNMM5cfTfpMf
3 aunxxbes9d / R09Dp8ebJi09ptFr3jtt2WyrW9wy1Jx132mK + Xq9PotT0iIU19ntLtExa3T47T + q
6 nBaYvsZstZ + cT / LeMnUi0TXffo1s2m8Ws2 / OIMWk5Jib5L328rS2t94Sh5TV4ppklpW6PT6rh + P
NbebTHyas8E081mZy5P2W6OFhjxNTE / hr / LoRO0Kvo9dPqctKzMxEx1la5t3tdnjnMs4noievcrO
yZjeFF1OSnNV0OG62cn1GWffj7Mz5w05joovzY7xes7TE7w0xrjPeex6Ua + j1UarBFu1o6Wj0lsN
3 JfQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACrU5o0 + nvlt92P3BxuM6nxNRGCs + 7 Tv8
2 hToxm1r3m9utrTvMsonqyt7XTmcja0u3O6FMfi5t / u0 / lzdJM81p9O3zdvHTwsUR5 + bfPqOfX1h
dqV + 3 O7bs1 + T31oqmI3TEM4rvCdkDGIIhlFd2daboS0NXG2bD6bufxXU1vlmu / u4us / N0 + L1tTSx
kr9qk7w89j1FNZMV3jxLzvaJ8mer + LSOZqK2xZotbvljfr / 89 U453rXt9lse081xZtNjx7TGKu0t
DHlrevSevaN5Y6 + tJ8c7VRNMt63n3ub + 6 / R54rERMztDYy4a5omclYmfxKcenrjtHLvtPrCnVmdb
eFe3JXmjy6eS / DrMuLVYsta9Mdt ++ 6 qLxO + 0 dEc8UmInr18iUfReHcXrqccb9Z27Q61Lb13eJ9nc
1 Z35rTvE9avY4bTkpG8xEfB05vYxqybc07R281naGMREdoT5JQqy9mply7Q3bV3iXG1eXw7TWSka
c258t7 + tpT5 / BjT7MfHqndz12Z + M4lMMKyziUJJiN1WSu9fku23RaOgKNJqbaTU1t9yelo + D0cTE
xEx1iXmM1Nt3W4PqvFweDaffx9vjDbGvxz + TP66QDRiAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAOJxzU73rp6z296zsZMkYsdr2naKxvLyObNOfNfJbvad1dXkaeOdpvsc2yuZVzfbfqybutwu
s5s8R92J3dvJb3tnO4HSMegtmt3nfZvYp8SZl0z45NfSK7onH1bNcfRFqnUKJr0Y7dVtq7prjEsK
0 XVpEM6028mW20IHK41aPo3J6zs4ODhdcvPnvExFevNXpMOrxi / PlrTee7PLX6Pwa09uaNlKtHg9
dM3z5d7ReOu02nu0JzZMfblrv5R5uvrcdImZ26T1mYhxs1Os7RH93PZ7axuafNfLitvbaYU3yZYt
PXs9NwHhui1HBa5LVicsb81onrEuVqNNSuS8Y67dZ6xPZa59Il9uX41vEitImZme3q2Kxbxora0T
Md / ROSa4Ztkj7c9OafL5LuGYubmyX3iu / TfbdSfVnpvZLT / XZK233 + Mbbva1xRXyiPk8pwbH4N6T
adq5a71n0tD1WDL4tPe6Xr0tDpz8YVnJHWEXYxbqlBedoef4tW0XraO09HdyztSZcbUz43C + ee9b
SVMaeOfqq7 + jGckQ1Yz7 + 7 v2RN / WXPXZPjci2 + 2 yyJaVMuy + uSJlA2d + pNoVRbeDcSxyTE + TDDlt
pdRXLTynrHrDOyiyZeVFnY9TjvXJjres71tG8MnJ4Nqt4tp7T1jrV1nRL1x2cvABKAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAHJ49qfD09cNZ97JPX5PPw2uI6j6Vrsl / ux7tfk1mWr7dOM8iLdm
vfebREefRsWldw7SxqNbWbR7lPesrn3Vteo7dYjDpMGCvfbeXQ0uLlxRLRxROfUc34p6fCHYrXlr
EejqrjY8uzCYW7MZjdVKqK9VlaxCYrsnYExBMRMJRPZA8 / xPHtmpP9W2xx76vhWOInvt / C7ike7N
vwzE9kcapGfhlevTaFbFo8RqJ5vy8 / RoW09ek0msxHfp3dzNoLzp4zUmZpMbT8HJyYJi20X2n0lh
ZY1li / RaidBF4w2mK3jrHaFGp1lN + tptPp5IjBkid5mIp16TKu0abBPv33vPlM7z + iPdFNcWXU5I
tkrNce / b1W5db1nTaf3ax9q0fxDW1ebNk2phty1mOu09VOm8W19orEz23j1TwfSeERFuEYMddptW
d43dvBn21eKJ75KbW + cf / JcTgMxXTb3nbljz + TpcPmc2uyZO1KRtVtGVdi0bx07qJnllsRO6rNTe
N4XVamsy8mnvPwc3R2jPwe8TPbdlxXNOPSZfhWWpwO85OFzv57qrODkzeHntSe8Sn6Rv0a3EZ218
8 nXekfr1a0ZLVnqx19dWb6demXybOO7lYMvNMdW9S / VVLo0us7tPHdtUtEwJiZU3jq2Jhham8CVG
PNODNTJXvWd3qcWSubFXJWd4tG8PK3pPd1OB6veLaa89Y61 / u2xfxh5c / rsgNHOAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAANLimq + i6O0xPv392rdeZ4rq / pOqnlnelOkIt5F8Z7Wj27I2I6sb25YY
V1ImY3dbQ08LRc23vZp2j5OJG + XJWle9p2h6HHtbJXFT7OOIpX + 7 TxT31j5rycdTh + Dpz + XaG / sw
w18PHWseULN2trBE9UcrJKBhFU7JAQi0dEomegNDUYovM7x3jb5tO1ZvpbaTLtzRExWfWPJ08kbT
Ex5NXWYYyV5omYtHWJieyeDzuizfRs19Jn6TM7Ru1uMcJxZqTkw + 5 f4ebqa7SV1MR4tdrx2vEfy1
axqsNOTLjnLXytVXi3Xj8 + nmsxTLM16d5npPyUzpekTtSK + U7vS6vQ / SYmK1vWPS1HOn2dvvvvE /
tDO5XlcO + LbfHSd / W3o6 / BdDOXPTnj3Kz38rS6Wm4FNrRyRzTH3p6RH / AKvR8L4dXSzE3jmtHn5I
mbfqLV + m4dbLSsZInHjr3iI6zLpYaxS01rHuxHRHiT9mv6s67Vj1aqL6326MrWiYa + / Q 5 4 B x P a G e
XRZpj8MquB4 + Xg8zPnB7SX30to379GxpK1xcHiKz5IS8xr8PLPixH2bftLTy05o6dHYyVjLhy0t1
izjZa3pMVv3iO / qz1G2L + NbSajbNyW7xLsY8kTDz + fJXFqKZN4iZnafi6WHL0iYlStI7OO + 7 axW2
crFl7dW9jvE9ULN + J3ZbdFGOy + AYWpEqN7afNXLj + 1 Wd23KrJVMvCzseh0 + auow1yU7WhY4fCdV4
OadPefcvPuz6S7jol649Tl4AJVAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAV581NPhtkvO0R + 4 NPi2
r8DB4dJ9 + / 7 Q 8 3 P o 2 N T q L a n N b L f v P a P S F D H V 6 6 s Z 5 E T 0 h R k n y W 2 l T t M y o u z 0 c 8 u s x 2 n 7 s 7 v S c K w x
zc1vu / y85p + maJh6Th + SOWeveXR4 / wDLm8v + nX5mUWa9bbrInolmu5jdTNkxYFk2Isr3TuCzeGMz
+ THdEyDDJO9Ja823rt2XWnya946pGvktDXta0ztWu / ybvLE9dkcoOf4GbJPWK1j49VmLh9JtE33v
Mevb9G7WsW8l1ccREISophiJ2jpDYpijbaOjOuOJ8ujOdqxsgVcsUjaETYvbaFFrgu5lVsm0yUtu
ryg43H5m + GIj1XcJzePoL4pnrWGtxmfchr8JvfHS1622if3QljzTTLes + qrNjrkiYtCzPMxnm095
YZJ6boS5teB49Tqscza97VtvWvlv8V / FOF34RrIxTM2xXjelp / eHoeA6XnzReY3ivX / 0 dfivDcfE
9 HbDbaLx1pb0lOs + jO7K8Lis3cN + 0 NKcd9PmthzV5clJ2mF9J9GHHVL108dm1SznYr / Ft0tuhLb8
mNohFbMhLWy0mJ3rPXvDvcO1karBG8 / WV6Wj + 7 kWrvDDBlvpdRGSnbzj1hpjX4z8mOx6UYYstc2O
uSk71tG7Ns5AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACZ2jeXneJ62dVl5KT9VTt8Z9W9xbWclPo + O
fft9qfSHEU1pv48ftYST23ZTDC / p0YtlVuvVjMbM5 + LCZjYGWGdrTPxiHY4ffaf3cjTxz1v6xMS6
Olty2iXVj / Dk8n + ndrkhnGRo1v8AFdW3RCrZ5uiYsqrboncSu508yjmZRYQt50TfowYTbYGVrKrT
uTZjvukQnYhMIGVY2ZxPVWyrHVCWzXpVXkt3TE7Va + W4K7X3jv1auTNy3jdba0RZpamfroQN7Hk3
6 wr1GTaN2OOJiu6Mu98NvgDi8Wy74d / yZ8PiPAiO2zU4nb6qIn1bugjfFE / ASp1ke9u15mbbRDZ1
Mb823kx0Ontn1OOkedoJCvT8I03gaKsz9q / WW + isRWsVjtHRKyrhe0XCfpWL6Vgr9fjjrEfeh5fF
feH0V5Dj3DPoOo + k4a / U5J6xH3ZZ7z3228evytOk7NvFbo0cdols47bSybt7HbddHVqUs2aW3Qnq
xVeu8LILR3SlZw3V / R8nhXn6u0 / pLuPMXjeHT4Zruf6jLPvR9mZ8 / g1xrvpz + TH7HUAaMAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAABRq9VXSYJyW79qx6yvmdo3l5viGs + maqYrO + OnSvx + KLeLZz2te1rZL2v
ed7WneZYWnZl5K72YV1xEyxmeqJljzIEWlVkszvbZp5soN3h2SJz3pP3odCnuWmPRxuERfJrZmtZ
mtY96fR28kbX3dXj / wAuTyf6bmK + 9 YX1s0cNtm3Sd4LFY2K23W1s16StiUJW7bp22RW3RluBuruz
mWEgrmCGWyNkoExKE1QlPmsqRDKeyBjaejWy2W3ttDUyz1QKslvehVqKTNosyyTvELabXptIJpaP
B39Ia2mz + JGpr51jdZefDx2hzuHZObNq58poJaGtjxJ2 + LoaKP8ADRPo5 + T3skx5OhpOmC0fBNQ0
5 yTbn + bt8A0u9raiY6RHLVwY62mI6zMvaaHBGn0mPHt1iN5 + aYVsACBXqMFNTgviyxvW0bSsAeE1
mkvw7V2w5Ote9besJx2er4rw2nEdNNekZa9aW9JeQjnxZLYskTW9Z2mJY7zz26fHrrdpbZsY7NGt
mxjvso1b9NmUwpx33XRO4K7VUTE1nmrvEx1bVo2VWiJE / XY4frY1WPlt0y17x6 / FuPM0m + HJGTHO
1 qu9pNVXVYt46Xj7VfRtnXXL5MfzexsALsgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHM4jxOMFJphmJv529Dq
ZLfjDjPEIx450 + K3v2 + 1 MeUOHSOWFc3nJkmZnf4yujpVlqunOeFpV2nctLCZUXRM7MJtsWlRkv3Q
ky5NmpWt9RnrixVm17TtEQnJabXisRMzPSIew9n + CRoccajURvqLx5 / chfOest642OGcIpoOG2w7
ROW9d72 + LQvXevyejcPUU5M + SvpLeOataraw2a0dLbLqTtK1G3Es4lVWWUSoldFtmcXUbpidgXzK
GEW3TuCUSncnsDFMMLSms9EC6J6FpVzbZE5ALy0809ZbFr9GtfrEoFMzuuwz0Ueey3HbaBLDXe7i
tMOfwWnP9I + NZbuttvhs1uBRtXPb4SDm3iIvf57N7Dbl0VrS5 + XrltEd + Z1Jx7cNms9N4TURRw3T
+ PrcO3WszEvZOD7P6aYiMlvu16S7y1QAIAABxOPcLnUY / pWCv1tI96I + 9 DtgmXl68Biy7 / NtUu3 +
O8HnFa2s0tfd75KR5fFyMWTdhrPHVnX9R0cd21S3Rzsdm1iuqs256wrmGcT0RYSx5d047X02SMmO
esd49YRE9WcdSXhZ2O1p89NRji9J + cei1xMc3wXi + KZj1j1dTTaqmor06WjvWW + ddcu8XK8BZmAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAMMmWmKu952UZ9XFZmuP3revlDTtzWnmvO8q3XGmfHb9ZanV3yxtWeWn7y4es
vPNtDqZJ6Ts5mppvdl / XXRMyfGvSNlu / RVvtOzLfoipLT1VTKbSpvfogRkvtDVyZOhkyvQcA4Dzz
XV6yvTvTHMfvK + c9U3rkW + zvA / D21urr789cdZ8vi9KDb45rejl8Rry6iJ / FV1HP4vXbBTJEfYt1
+ UpiHM295bXsqrO9l8QkZ0lZEqqLeyBZHZLGvZkhIndADKJ3TMoqWQMZ6pjsxll2jsCLSrmU2lFY
36 gieyu0LJk3jbsga0wdqzK20QpyztQGprL / AFMrOE05NLkt6qdVWZxNrSe5o9vWBLiUjnzXn0vL
q555dHt8HOwV928 / 1 z / LpzXxbYccRvzTB + jucOwxh0dI22mY3ltIrHLWIjyjZKyoAAAAACJiJjaY
3 iXleM8InR5J1GniZw2n3oj7s / 8 Ao9Wi9a3rNbRE1mNpifNFnVs65XhcWTdt47bnFuF24dm8TFEz
p7T0 / pn0a + HJux1OOrOux08d1ndqY7tillVkzExLOk7yd4YxGwluViJhE45raL0na0dtlWO0 + bZr
1 TKi + 2 zptZGTamT3b / tLacvJjiY3XaTWdYxZZ6 / dtPm1zrv1z78fPcbwC7EAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABhkyV
xUm152iAZWtFazNp2iGhm1Vss8uP3aevnKrNntqLdelI7VRHRnrX / HRjx / tZREVjZXeybW6KbWZt
pCZ6S08tN7Nmbb7zCrJtyoS5145bSx5mWafelr3tsKmS / o08uXyhlly7RPV2 + AcBnPNdZrK + 53 pS
fP4ytnPVda4y4BwHxOXV6uvu96Unz + MvVxG0bQRG0bR2G0nHLb2gCUDX12LxtFmpHeazt82wT1gH
mMN4tWs + rcr2aEV8DU5sM / cvO3yb + O0csLUTSdrLphRE8tlkZI7Atr2ZMazDJVKTYSCawi7Ksq7z
1 QERvLK3ZGPrKbyCrbdnMcsbeaa18 / RhvvM7oGEwTG0JmYYTIML22a2e28xELM19oURPNO4lOem +
n3ZY5 + prVnMc2GYU4 / L4A0a15cNf6rz / AC6fC6 + NxCPOuOu / 5 tHJTbHj + F5 / l1 + BYumXJMd9o3 / d
MRXYASgAAAAAAABhlxUz4rY8lYtS0bTEvH8R4ffhmo6bzhtPu29Pg9mq1Gnx6rDbFmrzVsizq2df
zXkMWTeIbNL7tbXaHLwzUctvexWn3bmPL8WFnHVL326VZ91MfFVjvvVlz79kLrcf2m7j7bNHH3bl
J2SirLQoy4t1 ++ 7 G0dBC / RanxI8PJPv18 / WG241 + alovSdrV6w6mDNGfFF4 / OPSW2b1zeTPL1aAs
zAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAVZ9RXBTe3WZ7R6iZOpzZq4ac1p + UermZMl89 + a / byj0Ra9815ted59PQ32hlrXXRjH
DpCLX6ML5NlNsm / ZRqstfdXzbsZt06sLZNvNB1Za8RDWyZdo7q8udq5Mu / mIMt4md2lmy7JzZuWJ
dHgfBL8RvGo1MTXTxPSPx / 8 AstJ1XWpIs4BwSdbeNVqq / URPu0n73 / s9hEREbRG0QUpWlYrWIisR
tER5JbSccur2gCUAAAAPM8Sry8Uyz67fwuxbzVPGsE49XGbvF42V4M0TEL33ERnktsxpk3sumK2j
admFdPFZ33VS2Mdui2J3UU6LYlFSsN2O5NkCyJ6K7T1TEsbAsxdpReerKkTFGMxvYEz0rsqtbbpC
b2VT1QEzuwtbaGUxspuJU3neWdKoiu8rq12gCI92YatLcublnzbEz1aOptyZqTuDHLfxN6R0 + t5X
qdJhjBp6UiPLeXl9NSMnEKxHa1 + bb8nrlvxUAAAAAAAAAAABTqtNj1eC2LLXeto / R43VabJw / VTh
ydY + 7 b1h7ho8V4dXiGlmvbJXrS3xRZ1fGv5rzeHN02bEW3cys3xZJx5ImtqztMS3MeTeGFjqlb2O
8 btql3NpbZtYsnSBLeiWfdTjtutid + ghherHS5p0 + f3vsX6T8Fkw181d4lMvEWdnHaGnw / UeNh5L
T7 + PpPxbjdyWcvAAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAo1Oprgr63ntAmTqdRqK4K9etp7Q5d7Wy2m953lNrWyWm953mVd77R0
Za1104xxlN9lV8qnJl2a9s3xUXX2ybsJyRDWtl3YWydEC + 2 VRkzeW6q + T4tbJm + KRdfK1cmWZnlr
vNp7RC / R6HU8SycmCk7ed57Q9ZwvgOn4fEXtHi5 / O9o7fJaZ6z1uRyOEezVstq6jiEbV71xevzer
rWtKxWsRFY6REeSRrJxz22gCUAAAAAANbX6aNVpL0npMRvWfSXlKamsRMVvXm + EvZXjmpaPWHzfL
oNRjzXicfWJ8phfPxFejx72x7xMzK + sXiNoiXlq + Pi6fWV / VfTNqfLJl / WTg9Pji8R70LqvMV1Gq
j / zcv6yz + lanzzZP1lWpelTET6S81Gp1P / Gyf90s412rjtnyfqql6asREdWM9 + jz9eJ6yP8Az7uh
odZqMt458tpB1JvEViI3 / RhzRt13 / R1MNaziiZiJn5K9ZNceKZiIiQcu / WekT + iYrWI3lzdTrs + 8
8 uW0fJzcur1Np / zsn6g79phVaIeetqNR / wAXJ / 3 SwnUaj / i5P + 6 UD0ldonum161h5mNRqP8Ai5P1
lNtRqJjacuT9Qd22WN5aGeZyZd / KHJy59RHbLf8AVq31Gp / 4 uT9ZEvS8Lr / vSs2npzRtL1z53wK +
oza / HW2XJNd99pmX0Rb8VAAAAAAAAAAAAAAcHj / C5yV + l4I9 + v24jzj1cLFk8nu5jeNpeW41wmdL
knU6ev1Vp96sfdn / ANFdTrXG + eq1q5F2LLtbZoY8m8d11bbSydErsYsm + zZrO / zcnBm226uhiyRK
EtrvCrJDOJTeu8A1MWX6Lqq5N / dnpb5O5ExMbx2cPNTeJb / DM / iYPDtPvY + nzhri / jDy5 / W6AuwA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAa2p1UYo5adbz + xbxMlvqJ1OqjDHLXree0ejmzNrWm953tPmTPWbWneZ7yoy5YhjrXXTjH8s75N
mtkyxt0VZM2 / m175N1V03yTKubMLXVXybeYLLX2VXy7eam + b0bOg4VquJW + rry4 / O9uyZOq3UjVm
9 r25axMzPaIdvhns1kzbZddM0p5Y47z8 / R2 + HcF03Doi1a8 + Xzvbv + TotJnjDXkt + K8ODHp8cY8N
IpSO0RCwF2YAAAAAAAAACvUZYw6fJkntWN3k8dfHz2vLucdz8mkjFE9bz1 + UOZosX1UzPm0nqI / W
MYo9FlcPNklfFGeH / NshLGun + Cz6PtHZtVZWlRLS + jxPkRpIn7rdoupHTdA5s6SI + 7 H6Mfo + 32 Y2
+ To3neSIiZ7A0IjPXpXLePlMotGW3272t85datKzHZjbTVnsDj + FG / 2 Y / RlGP4R + jo20u7H6N1Ql
o + H8I / REY957R + jpfReiK6eOYHLtj2tttH6KrY / 6 Y / R2c + kjeJiFVtLG24hxpw7 / AHY / RRkw9O37
O99Hrt1YX0tfOBLjcGp4XF8c + u8fs9c4dcVcGemSI61nd3IneN1orQAAAAAAAAAAAAABFqxes1tE
TE9JiUgPKcX4RbRXnNgiZwWnrH4XPi28PdXpW9JraImsxtMS8pxXhF9DecuGJtgmf + 1 TWW2N / la1
L7N7T5e3Vy6W3hsYcvLbqzbO9jvvCzvDR0 + XeO7crO6FmGSvRThy / RtVXJ92elvk2rRvDUzU7pl4
izsd2J3jeBpcNz + Lg5LT7 + Pp + Xk3W7js5eAAgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAADs0NTrN96Yp6edkW8Wzm6 + LNTq4pvTHO9vOfRoWtt
1 mes95YWvs1s2fZldddOczLPLn2ju0MmebT3YZc2 / mpm3qqllN1drsbZIhr3yzvtHf4AsvlYYseb
V5Yx4KTe0 + UQ6nDvZ3UazbJqd8OKeu33peq0eh0 + hxcmnxxWPOfOfm0mP + steT / ji8N9mKY9suum
L37 + HHaPm9DSlaVitKxWsdohI0Y22gAgAAAAAAAAAABXnyRhw3yT92Nwef4xm8bVzET0rPJH5d12
CvLhho3rN9RWs9Z23n5y6O21YhrVYbdGOCfrrLPJRpv863zVS6FS09SvZj3lVZZRdPSqmnSWdrIE
ebOkK4ldTsgW1WKqd1oMZhEVZyRAImOjGI6rJ7IiATNd46qL02bHkiaxaoNGY2n4ImPgtyV2n0Vo
Gvlx7x2beiyTk08RPevSVUxux00 + Fn2n7N + n5rRFb4AAAAAAAAAAAAAAACLVres1tETWekxKQHlu
L8InR2nPp43wz3j8P / s5dLveWrFqzW0bxPeJeV4xwmdFec + CJnDM9Y / CrY1xv8qvTZ + WYdbDk5oh
5 zHk283U0eo3jaZZ2N5XYjrCnLSJhOK + 8 d1kxvCqzSwZvousrb7k9LfJ3nB1OLeJdLhufx9LEWn3
6e7 LXN9Ofy5 / W4AuxAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAETaKxMzO0Qi9646Ta07RDmZ9VbPbaOlI7Qi3i + c3TPUaqcu9adKfy0722ZXvFa9
XO1OrjrESxt66ZJmcjPUanlidmhkzTZVfLN5VWvsC2b7R3U3yqrZZtO1esz2h2 + F + zWTUcuXXTNM
feKR3n5 + iZLVbqRzNJo9TxHLyaekz62ntD1fDOA6fQbZL7Zc / wCKY6R8odLBgxabFGPDSKUjyiFj
SZkYa3aALKAAAAAAAAAAAAAADQ4pl2pTFH3p3n5Q33E12Tn1eSfKscsLZ + orS00eJqbW + Lfnu1tF
XaJnZsz3WpCfsyp00fWSvmPdVYOmSUDd8kR3InoQosy7JmUX7MdwZ17ro7KKT1XRPRAsrO0rYndr
79 V1ZBaQiJ6JgCSIJASwrO07MpV2nqBlrv1a1o2bf2qtfLXaQUTO0sb05o3jv3ZXhjS20xEphW5h
yeJjjf7UdJWNKLziyRePsz0lux1SgAQAAAAAAAAAAAAAADG9K5KTS8Rato2mJZAPIcU4ZbQZuekT
OC3afT4NXFkmlntc2GmoxWx5K71tG0vHa / RX0GpmlutJ61t6wrY2xr8dXS5uesN + tt4ef0eaa223
2 dnHk3juyreM81OaFGiy / RtZET9jJ7s / 2 bdutd2jqKeic3iNTsd8a2h1H0jTVtP2o6W + bZbOO + gA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABje9cdJt
adohGTLXFTmvO0fy52bJfU23t0pHaqLeL5xdK9Rnvqb + cUjtCi94xxvK3JetKuHrdZvaa1ljb10y
cnIs1Wt3naJc ++ TmVWvMz1YWybfMGdsm3eWek0mo4jm8PT0mfW3lDf4V7P5tdMZdRviwfvZ6 / TaX
DpMMYsFIpWPTzXmf + steT8jn8L4Dp + HxF77Zc / 4 pjpHydYGjC3oAAAAAAAAAAAAAAAAADG9opS1p
7 RG7zszN6WtPe0zLua + 3 Joss / wBOzhzG2OsL5 + IrY09dsSyYRijbHEMvOChb7KjF0yS2LQ169Mso
S24noyrPVXWejNVKbTuw3T3REdQWU6LYlVvsyiUDPfqupPRr79VuOQX1lZEqoZxIMksd0gT2VT0l
bPZVbuCaW8i8bwr32WxbcGnkjaZa9p2ndv5qbw5 + aNugLItF6TEtvTX5sMb969HMpfazc0d9stqe
vVZDdAQAAAAAAAAAAAAAAAADV1 + iprtPOO / 2 u9bektoB4TJTJpNRbHkja1Z6uto8viVht + 0 HDvpG
H6Tjj6zHHvbecONw7Ltfkmeqmo6Ma69DXbbZTkr1mGWO3RneOaGbZRoM30fVzSelMnT83aef1FZ7
x3h1tBqfpGnjmn369LNc3sc3kzy9bQCzIAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAa + q1dNNXr7157VhGp1Xh70x + 9 f9ocy283m1p5rz3mVbrjXHjt91lz
5 c9 + fJ1nyjyhdM8lZlOOIiqrUXikd + kMreunnI5XEdX4dZiZcG + XmtNl / F83PeeWWHDOGanieSKY
q + 5 H2rz2hMzWd1Iqx1yajJXHhrNrW6REeb1nCPZumn2z62Ivl7xTyr / 6 uhwzhGn4Zj2xxzZJ + 1 kn
vLoNJnjHW7TbbsAszAAAAAAAAAAAAAAAAAAAAaPFrbaSK / itEOXt0rDf4xb / ACa / GZacRvaF58Q2
IjasQnzPIhCU92tMbZGzHmotG10C6nZkwpPRmipIllEbMIZIE7solgmJBnCyk9VMM6z1BtVllEqK
z0WRILYlluriWcSDJVbusV27gwInaSWM9ECyZ3hqamnSWxFmOSOaqRx725bNnSZNs9J + OynVY + WZ
YYr7TE + nVaIr0Ais81Yn1hKAAAAAAAAAAAAAAAAAABExvG09peU4nov9n66L0j6q / WPg9Y1OJaON
ZpL0 + 9 HWs / EWzeVz9PbmrEtnyc3h9reHy26TWdnSr2YX6657ijLXpLX0 + onSamL / AHJ6W + Tbv2aW
ekTv16JzeI1Ox6KJiYiY7Slz + E6jxdN4dp3vj6fl5Og2clnKACAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACZ2jeQRMxEbzO0Q08uqtkma4ulfO3r8lefUePMxWf
cjy9WvlzVxV6T1Z61 / x0Y8f7Wc7Ur1lqVy + LqOWJ2hp6rXddon5rOF1tfmz5OkT0qzb8dWbxjp1c
biuuilJ5Z6r + IcQrixzEy8zl1E6rNt1tMztFY81sztU1eRucN4ffi2p5esRM72n0h7rS6XFo8FcO
CkVpX082nwXh3 + z9FWLxHi36328vg6TZyW9ABAAAAAAAAAAAAAAAAAAAAAADj8Unm1tK / hqppHvw
y1k8 / EMk + m0GOPeafiFpCZYwolnXspvHvLa9mF46gmnZmwozRUiUCBKYYsoBLOFbKAX0llEqqyzi
QXRLOJVRLOOwLIljZMEgrlhKyYYTAK5nZPN0RZjugUanHzVlz6xtLq361c + 9 eXItPpXX0dubTU + E
bL2lw2 ++ O1fSW6m / VYAISAAAAAAAAAAAAAAAAAp1GbwcfTreelYEydcuMcRrM / L9nnlsV6wqpi2r
tv133mfWVkRyRtEdGFva7MzkYZNoamWN4bV4mYa9qztKIujhVppxGI8r1mJegeZpknBqKZY + 7 L0t
LRekWrO8TG8Ns / HJ5ZypAWZAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAADS12fp4VJ6z9qVuq1HgUiI + 3 bpDl589cOKZmevqprXPTbx477rDJlrhr1nq4 + s182tMRP
RqaziXiZJrWekNG17ZbxWJ336M5LXRbI3dLTJrs07RMY6fan1dHLrowY + X7MVjt6N3R6Kul0EbWm
s7bz8Z + LnabQX43r7Y53php / mXj + Dnv0f1JO1x / 8 ZxbUzj02O15mfLtD13AvZqnDds + pmMmo26el
XX0Wh0 / D8EYtNjilY7 + s / NstpOOTW7QBKgAAAAAAAAAAAAAAAAAAAAAADG88tLW9I3BwJtz6nNf1
vK / DHVqYJ3pzT5y3MPZeojOWMQylEKpTVjZnDCwkqzYQyRRICATCITAJZQxhMAshnEq4ZQC2srKq
qrIBZCWNZZgwswmFloVyCu0dFcx1WyrtCBhv5NTPHXds2U5o3hIz4ffbPt + KHUcTSW5c9Jme0u2v
VYAKpAAAAAAAAAAAAAAAAYZctcVOa35R6tLrltN795 / YvknNqrfhpPLH92V5isd9mWq6fHjk6rn0
ZxG8KK5Jm / wbVZiYZtqrmkqL023bkxvCiY3lJHNyRG81mHS4Rn5sNsNp64 + 3 yaWaNrzOzHBl + i6q
mT7s9J + S + ay8mex6EIneN47SNXKAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAImYiJme0JafEs3h6fkidrZOn5eaLeJk7eOdm1Hi2vmtPTry / CHmOJcUvmvOPF1n09Pm
6 HF9ZGm01qxO3R5vSY7XwzmzTy47zzTEd7en5Mfvt2 / PURWdo3tvPrPlKymbktFqTtMTvHzbOLDG
f63JXbFX7FdnoODcDprZpq9TjiMMTvSn4vj8l5fxnrk91saPSa7i2hpOfbTVt5x1m0fLydzR6PDo
dPGHBXasd585n1lsRERG0dIF5OOe6tAEqgAAAAAAAAAAAAAAAAAAAAAAADX11 + TRZrf0y2Gjxe22
gtH4piP3TPpXKwxtjhuYo9xq442iIblI2pC1RET2ILd9kxCqRjZmwlCSEohIJAQAAJZISDKGUd2M
MoBnVbVVCyAWVWeSuqyOwIlXZZKue4MJV2WWYT2QKbKL9YlfdRdIo35b7 / Hd3KTzUrPrDh27uxpb
c2mpPwX / ABX9XAKpAAAAAAAAAAAAAACekTIp1eTwtJmv + GkyJn1oafeazbfpMzLR4jq / o8b823zX
6 XNF8ERCvTcNpxLV5LauvPhx9Irv3lhztdtv8TtaWLicXrt03jzjzb2k1nid56ty3s / w + a7Uwzjn
1 raejlarhmbhl / FpbxMO / fzj5p / ixSeXOvTtRfeI280ZI26tfDm3pWe63LaZx7qtGvniJ6tPLvOK
fOa9WzbJvTbza02jl3n5SSljscK1MajSxWZ96nSW88xw / VfQ9XMT9nfa3yemid43jtLeXsce88qQ
EqAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAADia3UTm1l4j7OP3Y / u
7 Vp2rM + kPJW1PhYcmS0 + 9 MzKm / jbwz31weMzbV8UppazPL9q0 / BF4rk1GLDSNqxPWPhCnHmnNrtT
qPKteWPm6U6OdHaZvO + SaRNvhv12Ub / q3FhtrNVj0uKOt56z6R5y9zix1w4qY6RtWsREOJ7L6OKa
S2rvX6zNM7T6Vh3mmZyOfya7eACzIAAAAAAAAAAAAAAAAAAAAAAAAAAczjVvqMVfW / 8 AZ03I41bf
Lp6 / OVs / UVrY47NyOzUxd4bUJpEbb3Z7IiOrKIVSjZhMLJYyhKIgmGUQSDESIEbJEgQmCITEAmGU
IiGUAyhZVhDOoM4Wx2VQtqBKuyyWEgqlhKyyuyBVaGtkbNmvk7A15l1eH2300R6TMORPSXT4ZO + O
8 fFefEX63gEAAAAAAAAAAAAAAAq1WPxdLlp + Kkx + y1Fvsz8gjhaDauGK8sx07y3OE3m1tT6RaP4c
vU6yMNKUx73zT0ilY3l2eF6a + m0kRl / zbzz3 + Ez5M8z26fJruW6wzYq5sV8d43raNpZjRzPPaTmx
5 b6bJ9rHO3zb2WJ8GWPEscY9bgzxH2t62n19GWW0eHOzHU5XbjXZ1x8WTnz2iZ7S2M1IjH2 + LX0V
KTqs8zO9ot0j8nUthi1J3UaOFMTfLFo6xMbS9BwHWTqdHOO8 + / h n l n 5 e T j Y M F o 1 W T H 5 V n e P z X c I m
2 k4zlpPSmXy / hfF5eMfJns69OA2cgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAADG / 2 LfJ874rW845mubliY7bPoto5qzHrDz0 + yePNF41OotaJ7RWNtpV1OtfHqZ715fhu
j8adNpcVfeyzE2 / vLuanhOu1nEctIxTTFa / + ZPbZ3eHcF0vDbTfFE2yzG03t32 + DokynXl9 + leDB
TTYKYccbUpWIhYCzEAAAAAAAAAAAAAAAAAAAAAAAAAAAAcXjE / 4 zDH9M / wAu04XF5 / 3 jj / 0 f3Wz9
RUYmzDWxS2I7FSyjuzY1ZKpRKEygEwiWUIkGIk2QJNhKQhMIhkCYZQxhlAMoZwwZwgWQshVCyATL
CWc9ldpBhZXLOVdpQK7NfJPRdaWvknoDVvPvOnwuel4 + TlXn3nS4VPvXj4QtEV0wAAAAAAAAAAAA
AAAAAVV02CmTxK4qRf8AFFeq0AAAanEsfPpZmO9Ji0NDLfkwdOsulrumiyzHlVzJrz4Ovoy26vB8
cTBa9NffLtMY77Rv8Yegx5ImkKdJoY1HC81Y + 3 OSbVn0mGGkmbY45u6tnrrTOu2xGO0RxCd + nNVj
qKxTV1vH2pjaGtnyzXXYdo96ZmGXEMk15b7 / AGZiVerWPTYckZcNbx5wzc7hGbnxXxzPWk7x8pdF
0 S9jh1OXgAlUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAcPjEf4 / FP9H93ccXjMf4vDP9Mx + 62 fqKrx + S + GvibEFSsqyYwlVK
ZYsmIMoRKYJQIPIEiQ2ATCUQygCGUIhMAyhnDCGUIFkLIV1ZxIMpVWWSrsCuyqyyyq09ECq8tfJK
66 jJ2Bp5J6upwn7dv9Lk5J951uE / av8AJaIrqAAAAAAAAAAAAAAAAAAAAAAq1Mc2myxPnWf4cmtu
XT9fR0tffk0WSe28bfq5Wbamm3326MtunwfK6PCv / AxPraZ / dz9PO97 / AOqf5dHhdZrw7Dv3mOb9
XOxRFM + avpe38mvkPHf / AFWlrKba7Tzt99ZxKkfR7euyNXMTrtPHfa0z + zPiM / UR8Zj + Wbdu8HpN
M2bfzrV13M4dO2pyR61dNvj44 / J / oAWZgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAADj8bj63BPzdhyeNx0wz8ZWz9RWri7Nmv
VrYu0NmqaRZHZlDGGSiwxZSgCEkCBCQSCQBMJRCYgEsoYx3Z17AlMIhlCBnDOGEM4AlhZZKq4KrK
7 LLKrIFN2vdfZReAaObu6 / CO9vk5OePR1uEd7fJeIrqAIAAAAAAAAAAAAAAAAAAAAGtxCk5NFliI
3 mI32 + XVyNTyZOHTee946PQKPoeDffw4777eW / yVs60xv + ZxOnr4Okx1t05KRv8Ao41Z5q3yed5m
XY1szXRZ5jvFJ / hxItP0aOSN9q7yrtr4f2tHFM5 + KT16Yq / vK / iGSbXw4vO14UcPx5MGfNbPG18m
1 oj4THRsTw7VanPXVYpi3gzMcnrvCnG11JOupwuN8 + a3pEQ6jT4divjxWnJExa09pbjbM5HHu90A
JUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAHM41H1GOf6nTc / jEf4Ws + lls / UX45uGekNujTwdm5RNIthKIZKLDFlsiQIShIC
EgCUJ7AmGTGO7IDzZQhMSDJMMYZQgZwzhhDOATuqssmVdgVWVWWyqtCBTeVF19lF + wNLNG7q8I + 9
8 nLyupwnt + S8RXUAQAAAAAAAAAAAAAAAAAAAAAAItWL1mto3iY2lyrcLyUxzix2ia2nvPeK + jrCL
OrTVnxpanhuPPemSs8l6RtE7dJj0ldpNP9GwRSZ3neZmV4cR / Vs4AJQAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAANHi1d9H
M + kt5ra + vPoskfDdOfqK4mn7Q3aNHBPZu0W0RdDOGFWcKLCJZeTGQQlCQSgASBsCYZQxhlAJTAmA
TsmAgGcM4YQyjsgRLC3VnaVcgwsrt3Z2V2QK7tbJ1bN5a9waeWO7p8Knt8nNyebpcK8vkvlFdQBA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAK9RXmwZI + ErEWjesx6wQeZwejeo0cccuW8
elpblJaaRGxVnCuss4ZrMvJEgCAASISCQIBlCYYpieoM0wx8k7gzIRueYM4Z79FcSy3QEsLJmWFp
BjaVVpZWlXMoGNmvkXXlr3kGtknu6XCf7OXkl1OEdl8orqgIAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAHmskcmtzV / rls0U62OXiWX4zErcc9GmkRfWVkSqqziWayxCPIANwBIhIJSxS
CRG6dwZwlhEs4BluMdzfqgZxLLdXuy3AmVdpZTKuZBjaVVpWWV2QlhZRdfZRcGpl7urwfrzfJy8r
rcH61vPyWitdMAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHA4nHLxKZ9awnH2ZcY
jbW459aq8fZpfiI2IZwrqzhmsz3Ebm4JN0AMhCQSIASndiAziWUSriWcAyRujc80DM3RCfIETLCW
UsZEsJYSslXZAwlTddPZTkBp5e7r8Gj6rJPxhx8k9Xa4PG2C8 / FaK10QAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAcfjcbZMFvnDWx9m5x2PqcNvS + zSxT7sNPxH62YZQwqzhRZO6UCB
KUAJTux3SDIRuAncQAmJZRLBMSgZ7iIAZRKd2DICUSlAljLCYWMLIFVukNfI2bNbIDTyT7zu8Ijb
Sz / qcG / 2 nf4T / wCE / wD2WnxWt4AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHL9oL
+ Hw2cm28VvEuPptfgyVj6yIn0no7 / FtJfW8NzYMe3PaPd39d3iMug1WktNc2C9dvPbeP1aZ9xF + v
T471tHu2iflK2HkqWmvaZj5Surqc9Ps5bx + alTHqYHm68S1Vf / NmfnC2vGNTXvyT84Ql6A3cSvHM
sfaxVn5Ssrxyv3sM / lKB1xza8bwT3pePyWV4tpZ + / M f O E j f G r X i G l t 2 z V / P o t r q c N / s 5 a T / + w L R j
FontMSlAlKEgndO6IAZQljDIEgeQljLCzOVdkCu / SGrkbF56NPNeKxMzMRHxENe0 + 89 DwuNtHHzl
5 PJr8NcnLW3Pbf7r1nCZm2gpae8zMrz4i / W6AgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAETETG0xukB4HVaeMHEtRi26RedvkyjBSfX9W77QYvC4xz7dMlYlrU7M929dWJLFc6aPK0q
7 YLxPS0S22FlP6q38Zac0yR92s / KVc3tHfFf8tpbcsLRvB / dR / 8 ALLVnU0r9uL1 + dZI1mnmdvGpv
6 TOy6ym + Oto2tWJ + cJ / tW + KLK5KW + zes / KU7tG + h01p64qx8Y6NXNo6Y + uPJlp8rLf0rfG7MXtHa
0x8 pZxqs9e2a8f8A7Oj7HaTHn0 + f6RWM23LETfr6vRW4PoL99NT8ui7F4 + vEdXXtnt + fVbXjGsr /
AOZE / OsPS29nuH27YrV + VpeV9pdPXhOtw49NG9Mld55 + vXcTPd42I47qo7xSfyWV9oM8d8VJ / VxM
d8l46xWF9cV7en6o / qLfxp2I9ob + eCv / AHMo9op89P8A / wBORGmyT5R + qfo2X8P7n9Q / jTsx7RR5
6 ef + 4 / 8 AuHftg / 8 A6cWcOSO9J / WEbWr3pY7Efzp2Lcfv5YK / 9 zWy8d1E / ZpSv5Oba1 / + Hb9lc + LP
bFt87I7E / wAabWbiurvEx4nL / pjZzc2bJkn372t85ZXx55 / BX85lucC0vPxnTxlnnjm32mOiZqUu
LJ2p4TwnVavNWaYbRTfre0bQ99pcH0bT0xb78vmtiIiNojaErMwAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAHnfarF7umzRHaZrLjYrdIen9ocPi8JyTt1xzF4eUw23rCm3R4r6bMy
wt6kdTaWLdjswmNoZontsCm0K5XWjopnuDC0dGpqG5bs08 / daKV672MjbSaif6oh6Z5f2LtvptRX
0 tEvUN3Jfo8f7cYve0eX4zV7B5z20xc / C8eSPuZIRficfXlcPaG7ino08HWIbePpLF2NuiyOyrHK
3 fZFSwuovHVfaVF4QK5YWTM9UT0EKry6Ps1Tn4zjn8NZn9nOtLseydObiWW34cf918fWfk + PYANn
KAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAq1WKM + ly4p + / W Y e B x T N d 6 z G 0 x O 0
vobw3FcP0bi2em20Tbmj5Srr418V9sa2Z7qKyzi07MXUylhaU7yjqhLCeiq3ddaFNxFYW7NLNG8t
zya + WO6Va9J7FW66mvwidnrXiPY3Ny8RyUn71Jj9Ht3RPjk19HK9pMHj8D1ER3rHN + jqqtTjjNps
uOe16zAifXzfTz7kNyndpYazS9qT0mszDdoxrsi6m8LazMq6zDOsq1ZEyrt1WWlXaUCqyq0rbKbi
Fdp6PReyFd8uqv8ACsfy83aXrPZHHto89 / xX2 / SP / dpj6y8vx6EBq5gAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAB5n2q03LfDqqx39y39npmlxbS / TOG5se29tuavzgWzeV4mtui2
O3RRSY2hdVhqO2MvI36iu9lUsrSrvDHn6spnmSiq5jooyV6tq1VV69RC32byTh43h8otMx + r6I + Z
aK / g8TwX7bXh9Mid4iW + fjl8n1ICWb57xLBOm4zqse20Tbmj8 + qKdnS9q8PhcTw5tumSm0 / OHMxz
0 Za + uzx3sX1t0Zxurr1ZxvspWiZYWZbsbT0QK7KLrZVZJFaqt5vbezNOTg9J / FaZeJns93wCvLwb
T / GJn92uGHldIBowAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAADuAPA67F9H4l
qMW20VvO3yRWW97T4fC4rXJHSMtI / WGhVlue3b473K2KzMML4 + 62 tujG9pnozXaOSOVFMnVbmq1t
trJRW5E7wwvUxTvCyY6CHOt7moxz6Wh9PxTzYaT61h8x1MbZK / OH0zTf + Fxf6I / htj45vL9WgLMn
mvbPFvocGWO9L7fq85p5maw9d7VYvE4JkmPu2if3eW0 + PasdFNOnxfF1Y2hlykRsmY + LJ0MZjZXa
eq2eyi8oQTO0KLdZWzPRjWu6VaqtHR73g0bcI0sf0Q8Nkq93wqNuFaWP + XDTDDytwBowAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAef9q8HNpcGaI60vtPyl56k9Iew49j8ThGe
PwxFv0l4zH2U26fDfTYiyJljvsjf4sm6vJ1hrXjq2MkqLdZEVbgbMx0auGdmzNt6iHN1Ub5af6of
TdPG2nxx6Vj + HzaaTm1 + nx / iyVj930ysbViPRrj45vL9SAuyc7j1efguqj + jd4 / T33rD3HEcPj8O
1 GP8WOY / Z4TTT7sKadHhbcsZnaCJ3TPZk6VdrKbTutmP0U2nqgrGOsr8deiuI2X09EqKM1dt3uuG
f / jdN / 06 / wAPE546S9rwud + Gaaf + XH8NMMPK2wGjAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAABrcRp4nDtRWPPHP8PCYusPoWSvNjtX1iYfPuWaXtX8MzCuvjfw32siu8ptXoxi
0 wy5t4YulReqmazu2skbquURWFInddM7VYRGyL291KFnCcfj8e0le / Lbmn8n0N4b2Ur4nHLWmPsY
5e5 a5 + OXyXugBZmiY3iY9Xz7NjnTa3Ph / BeYj5PoTxftFg8Hjk2iOmWkW / Psrr418V5WrWd2faFc
V2jdnEMXWxntupmN7NiYU27iWML6dVMVnddjgVqMsdHr + CW5uE6f4Rt + 7 yuSsTDv + zWXn0WTHP3L
/ t K + G H l + O 0 A 1 c 4 A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA8Dn93W56 / wDM
t / L3z59qp24jn + OS38lnpr4r7ZxHQ2TEstt3PXUrt27K57rr1VT0BjKnJPRbMqMs7QlV2fYvHvrd
VknyrEfu9m8f7FZI8fVU85iJewbT45NfQBKo817W4eulzxHaZrL0rje09ItwqbfhtBVs3leai8RD
KLw1sduesL606dWFdsZT1jdhNeq6K9DlhCVUU6s4jZnt1YzAhnM71dH2bycmszY / K1d / 0 c6OzY4R
fwuK4p8rTstn6z8k7HrwGzkAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHz3
Vxvr80 / 8 y38voTwGpj / F5 / 8 AqT / JfjTx / WVeyY6FPspc9dZPVXaOq2WEwIUTVRmjo2rNfLHRI3vZ
DJycXtX8dZh7t879nsnhcbwz23tt + r6I2nxyb + gCVBzuPY / E4PqI9K7ui19fTxNBnp60n + Aj5 / pJ
3 jZu1aOnnltMNussdfXbm + l3ZM9URHREdZVXTuT1Nk7boQiOkJw28PU47 / htEp5eivJPLMTCZ9Vv
x7mJ3iJ9UqNHk8XR4b + tIXuhxAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD
weqjbWZ4 / wCZP8vePCaz / wDIaiP + Zb + UX408f0r9lOxWOifJhXWjfyYWllPRXYQxnrCrJHRd3YZI
6 A1NJecHEsN / S0T + 76 bE7xE + r5dk93LW3pL6ZpMni6PDf8VIn9m2fjm8s9rgFmQxvHNS0esbMiew
PnHLyai9fS0w2aNfUTtrs3 + uf5bGPqy068fF227KtSsdFlKqNGMV6myyY6sbdIQI8tlOWOi6Jhhk
j3RD0vA8nicMx9etZmHRcT2Zyb6XNT8N9 / 2 dt0T449T2AJVAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAHhdfG3E9TH9cvdPEcXjk4zqI / q3L8aeP6xr2TsxpLOekMK6mFo6qpXSrm
OqBixvHSVmzC4OfqK7S9 / wAByeLwbTW9K7fo8Fqo6Paeyl + fglI / Da0NcMPK7QC7AAB8313TiOf /
AKk / y2MHWrX4jG3E9R / 1 Lfyv0 / aFNOrHxuU7LI7MMayGTVlHWUXhNe6Z6wIUsb9d1m20q7dkDpez
N9tRqKT5xEvRvKez9 + Xis1 / FSYerb5 + OTyf6AFlAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAB43j9eXjN / jWJ / Z7J5L2mry8Upb8VIF8f6aGOey2eynHvOy7bowrrYSxZSwQJ2YXZ
92 N4BoanrEvVexmTm4blr + HJ / aHltRHSXofYm / 1 Wrp5RaJaYY + X49WA0c4AD51xONuKan / qW / lbp
+ 0 MOLRtxbU / 9 SU4J7KadWPjep2WQrr2WRPRk1TvsndXMpiRCb9FNu0rbTuqvKBscCjfi9PhWZeue
V9n434rafTHL1TfPxy + T / QAszAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHmv
avHtfTZfnV6VxPajHzcNrf8ABeJFs / XnMcr4no18c + 6 vr2YadkY2YM57sEDLyY37Mo7MMnYGlqO0
vQ + xNfqNVb1tEfs87qZ2rL0 / sVX / AHdnt65P7Q0wx8vx6UBo5wAHz / jUbcX1PT78qtO2vaCnJxjP
8 Zif2amnnspp04 + OjWejKJ6MKdmcMmyJn4m5ZHzEVPMwtJv0VZLbQDqezcb8RzT6Y / 7 vUPM + ytZt
n1OTyiIh6Ztn45N / 6 AFlAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABocbxeLw
nUR5xXm / Rvq8 + OMuDJjntaswEeBxT0bNZ6NatZpNqz3rO0rqsdO3PxlaWEMpY + aqWXkryT0ZT2V3
7 A0dVPuy9f7G124NM / iyT / Z4zWT7sw957MYfB4Fp4 / FE2 / WWmGHldcBowAAeM9qKcvFeb8VIly9P
0 nq7ntbTbVYL + tJj93CwT76unR4 / jo0nozhhTsy3Y1sWljM9Ce7HyQIm3RRlttVbaWrnt0Sh6n2U
x8vD8mSfv3 / h3XN4Bi8Lg2nj8Uc36y6TeOPXugCUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAPD8RxeBxXUU26Tbmj8 + quro + 02 Lw + I4ssdslNvzhzazvDPbq8d7GW7Dfqz2VzG
0 s2qd + iu / Zn5Ksk9BVztX1mI8930zh2LwOHabH + HHWP2fNYp4 + vwYvxXiP3fUqxtWIjyjZtj45 / L
faQFmQADzftfj3w6fJ6WmHmsP23rvaqnNwqLfhvEvIYZ + sV038bo0noy36MK9oZQxrdMyrlnMbMZ
QKrS1M07zEestq / RRjr4utwY / wAV4j91p9V18fQdJj8LR4ccfdpEfsuREbREJbuMAAAAAAAAAAAA
BAJAAAAEAJEAJQAJQAJEAJQAJQAJEACUJAQlAJEAJQAJQJAAAEAJEAJBAAAJAABAJEJAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABwvanDzaPFmjvjv8A
tLztJ3h7HjGHx + FainnFeaPnHV4vFbeIU038VbHeGF + kso7Mb9mTdhKnLK3dRm7SIrHhGPxeP6Sv
9 cT / AHfSnz72Zx + J7Q45 / BWZ / Z9BbZ + OXyfQBZQABzeP4 / E4NqI9Ii36S8Ng / wAx9C4jTxOH6ivr
jn + Hz3B / mQi / GvjdCnWNlsdI2V07LIlg6USrt2ZzZXMoFV + zPhGLxeOaavpbm / RVltEN72Yx + Jxm
b7dKUmf7L5 + s9 / HtRA2cqRACRACRACRACUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAACQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAQCQQCRACRACRCQBCQBCQB
ACRACRACRACRACL1i9LVntMbPATTwdRkxT3pea / u + gPE8Xx + DxrPHlaYt + qNfGvjvtXXsi0dOrKk
dEXjZg6VMtbP2bMtXUdpEV0 / Y2nNxbNf8OP + 727 xvsXH + N1U / wBEfy9k3nxyb + gCVQAGOWvNivX1
rMPnGGOXNNfOJ2fSZ6w + dZKeHxDPX8N7R + 6 L8a + L63KdoZ7q6zvEMpnowdKJ6ywmWUyqvIKM0vQ +
x + D6rU55 + 9 aKx + TzWa36vbezmDwODYenW + 95 / Nphj5L6dQBo5wAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAEgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACAEiAAAEoA
AAAAAAAAAAAAAEAkEAkRuAkQbgkQAkQAkQAkQAl5T2nx8nEMOT8dNv0l6pwfarHvpcGWPu32 / WCr
YvK4mOem6b9mGKd4Z3idmFdka0y1c892zfpMtLPaNpEV6D2Kj / Eauf6YeweQ9ieuTVz8K / 3 evbT4
5 NfQBKoAA8FxCvJxrUx / XMvevD8Zry8fz / Haf2RfjTx / 6 RSOnRMyypHu9kXjowrqVSrvPRnZVl6V
kK0775MsUjvadn0nT4ow6bFijtSsVfPuFYvpPGtNTy54mfy6vorXDm8l9pEC7JIgBIgBIgBIgBIg
BIgBIhIAgBIhIAgBIgBIIBIAAhIAhIAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAJAAAAAAAAAAAAAAA
AAAAAAAAABAJQkAEAAAAAAAAAAjc3BIjdG4Mkbo5kcwMjdhzHMDPc3V8xzAs3N1fMjmBZubq + Y5g
Wbm6vmOYFm5ur5jmBZubq + Y5gWbm6vmOYFm5ur5jmBZubq + Y5gWbm6vmTzAz3N2HMnmBlu5ftFTx
OEZJ / DMW / d0t2rxKni8N1FPWkiZ9eS08e7Cy8dGGn6UhZaJljXZGnmc3UT3dPP2cnUT78xCIV6j2
H / 8 A9c / 6 f7vXPI + w8bU1U + vL / d63du5NfUiDcVSIAS8b7RV5eOb / AIqRL2TyXtNX / e2KfXH / AHlF
+ NPH / pr4 + 2 xcxx0hFpY11K7R16KM32ZWz3UaidqSgrc9kcPicWyZJjfw6T + 727 y3sXh2xarN + K0V
h6lvPjj3e0ASqAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAJQAAAAAkQAkQAkAAAAAAAAAAAAAAA
EgAAAAAAAAAAAAAAAAAAAAAgAAABKDcAN0bgkY8xzAyRux5kcwM9zdXNkTcFm6OZXzMeYFvMibKu
ZHMC2bo51U2RuC2bom6rc3BZzom6sBZzI52ADPnOdggFnMc6skFnMc6rc3BbznOp3RzAv50c6nml
HMC / nOf4qOY5wX85zqOc5wbHOc7X5znBsc6edr85zg2ec52vzpi4NjmY5bROG + / b l l V z s N T k 5 d L l
n0pP8BHmMHWNmzt0aum8obm08vVjfrtnxztR0mXHzTvaZdjVRMTLkZo6yiFen9iZ2pqY / wBP93rN
3 kPY + / L f P X 1 r E / u 9 X z N 3 H f q z d O 6 v m T u I Z 7 m 7 H c 3 B n u 8 t 7 T R / v H B P 9 E / y 9 P u 8 3 7 S V 3 1 u m n + i f 5 R f i
/ j / 0 1 M M b 1 h j k r t K z B G 0 b M s m O Z Y 1 1 t O Y a m r 6 U n d 0 L U c 7 X T 7 u 3 r J P q L 8 e r 9 l c P h c F p b z y W m 3 9 v 7 O 0 0
+ FYvA4Zpsc94xxu227jv1IAgAAAAAAAAABKAAAASgASgBIgBIgBIgBIhIAAAAAAAAAAAAAAAAAAC
UACUJAAAAAAAAAAAABIAAAAAAAAAAAAAAAAAAAAg3AEbomQZbo3YzLGbAz3RNlc3YzcFs2YzdVN2
M2Bdzom6nmNwW86JurTAMuY3REJ2BB1ZRVMVBhsbSsiqeUFXLucq3lTygp5TlXcpygp5TlXcpygp
5 TlXcqOUFXKjlXcrGYBXysdlswiYBVMdUTCyY6sZBWxlnMMZgGLGZZSwkDdHMiWO4MuY5mEyjcFn
N1OdVzHMC3nTzqeY5gX85zqOZPMC + Lqdbk20eb / RKOZr8QybaK / XvtH7iZ9aGlp2luzT3fg19NHS
OjbmPcYX67XH1XSZ9XIzRvMuzrK7zLkZYmYnciunb9lZ5dTk + OP + 71 cXeP8AZnJ / ip2nf3J / l6iL
/ F u 5 L 9 b M W Z c z X i 6 Y u I b E W T z K I u y i w L t 3 n u O 2 5 u I 4 a / h x 7 / r L u c z g 8 T i c v F L b f d p E K 6 + N P H / p h h j o
stLGkctUWnoxrrU3j1cnWTzZq1jzl1clo5Zcu8c + txR63iP3Tn6pv4 + g4o5cVI9IiGe7CJ2iE7t3
GyN2O6dwSINwSISAlAAlACRAAlAAlACRACRCQAAAAAAAAAASgASISAAAAAAAAAAAAACQAAAAAAAA
AAAAAASAAAAAAAAAAAAAAAAIAAAQCAJljuljsCJlhMs9mOwMJYys5TkBVsjZdyHICrZPKt5E8oK4
qmKrOVOwMIqyirPY2Bjyp2ZbAI2NmSARsbMgEbI2ZAMdjZICNkbMkSCNmOzJEgx2YyzljMAwlhKy
WEwCuWErJhhMArlhLOWEgxljMpljIImWMyTKJA3N0IBO5vux3NwZbnMx3NwZczT4jf3MdPW27a3a
fJOq1XNP2KdIRfi + J2trSYfcjeF + Wm1OicVeWIiN9kai8xjY12ORqultnI1Ecsujq79XP1FovWYI
rTgeq + j8QrWZ + 3 Mx + r2UXeC0WG2Ti2kiN5mL807eUREvbzbaejefHJv62Iv8WUXa0WTFhVtRdlF2
rz9WUXBtc7jR9dqc2T1ttHyhvZMvJitb0jdq6XHNcNenWVN3028U99WRj6Kb02be3Tq18 / SN2Lpc
3 UdN9nOmZrqKX / DaJ / d0svvTLRzV3jomK6 + Pd1vvWJj0ZczT0mXxNJht60hfFnQ4qu3N1cWTEgs3
Tur5k7gz3N2O5uDM3Y7m4MtxBuCQASIASIASAAAAAAACRCQAAAAAAAAEoSAAAAAAAAAAAlAAlCQA
AAAAAAAAAAASAAAAAAAAAAAAIASgAAAEJAQJQCNkbMgGOyOVnsAw5TlZ7GwMOVPKy2NgY7GzIBGx
skA2AAAAAAAAAAQkBAEghEskAxYzDPZGwK5hjMLJhjMAqmGEwumrCagomFcw2JqqtUFEsLLrV82F
o7gqljKyYYTGwMZRKUSCAQAboJnaN5Bjkneu0d5W4ccViIiOzHFWbTzNumP1Zarr8eeRMbxDW1Mx
NO67NbkhzNVnmInqzaOZrL93JyZeV0M1 ++ 7 S02jvxDWxhxx033tPpC8Z6rrezWjmZyazJG2 / u03h
2 vFibTHoqvamiwVwY + nLGzV0 + SZ1Mx8G0 / 45 tOhzJ5lXMc3UVXRdlF1HP + iYsDPLPPy49 / tz1 + Te
pSIr0ho6ak5Ms5J8o2q6NImOrHV7XX488ypzTtHXo0s9t6zG7c1G1qz6ubeZiZ3UatXJG3yauSO7
cvMTEx5tPLb3prPRMVr0HB8vicNxf0 + 7 + kt + LOJwTJyY / Bnz3tH93X36N58cWvq6LSyiyndMSlC7
mZcymLJiwLosmJVRLKLAtiU7q4lMSCzc3YxJuDMRuAlKAEgAAAlAkAAAAAABKAEgAAAAAJAAAAAA
AAAAAAAEgAAAAAAAAAAAAAkAAAAAAAAEAAAAAAAAAAAAAAAAAAAAAhIAAACAAAASgAAAAAAEAAAA
hGzJAImGMwzQDDZjNVuyNgUTVhNGxysZqDVmiu1G5NN2M4waM0 + DCaN2cbGcQNGaMZq3JxMJxA1J
qx2bU4kU09slorWNwa20z02RXHbJbl26QvtFovbHWkxEdJt5y2MOHlr2U1W3jx + 1 hiw8vSO63lmI
XRTaEWmtY6snRHO1VpmJ + DjavpSZl2s8b7y4HFcnh0n0gha5ebJN55KRM2mdoiPN6fh + kpwXh0Wy
RHj5Otp / s5Ps1p62y31 + em9aTMYt / OfVfxTiPjZ52naI7fBrI5t66xz5 + a1rW7yx0eSL6iZjtEOX
qNbSletom3lENjh2fbHzbbWt3iVozruc + 5 ztWubf4M4ybpQ2Oboyrva0Vjza8WdDR4OkXt3n9ldX
kaePP9VtYqctYhdvt5oivTeCZ2YOxXk6ubqMfV0b9mrljfqlFcq88k7z2U5axeItDa1OPessuC8P
ya7XRWYnwqdbT / ZMilvIu4dpslNdixXja8Y5tt85djZdbDWnGOesRtXFtuw6T27No5Kx2OrKYQlC
ExKJgBnEpiyvdlEgsizKLKollFgWxLKJVRLKJBbEp3VxLKJBnuMWQJEbpBIAAAJAAAABIAAAAAAA
lAJAAAAAAAAAAAAAASAAAAAAAAAAAAAJAAAABAJABAlAAAAAAAAAAAAAAAAAAAAAAAAIAAAAAAAA
AAABAJQAAAAgAABAAI2EoBGyJhkgGPKxmqxAKpownHC + YRMdN5BrTj67R3bOn01o7p01Iv71u89o
b9a7LfBTfS1vWI2jf12VfQPSW8KX2mas + NC2iv6xMNfJpMnLtEbuuxtMRCtzF55NR5rPps1N / ctP
y6uHreE6nXZ4pak48X3rT06fB7fNeI33cbX6mI32R / MWu7XF116aDSRhxbRERs8f499bkyZeeKae
kzE2mdon81 / tfxDLGOunwbzlzbx08oaHBvZHJlx48mrvaa94pu04y617576rNGLRRM0397JEd / lu
9 Dw / S3x4qxffo6mm4NjwUiKY4iI9Ib1dHFY6QIaNabbrYrLfrpJtaK1rMzPZb / s + 05 IpP59OyLeJ
k7eNfRaOc1ue32I7fGXYpi5Y77M8OGMeOKxHSFsU3Y29deZMzirl6dlVvhLatCjJHeYQv1rXnps1
8 k9 / VsW6qLVmZIi1rzitlvFKRvaZ2h6TSaenC9FFY + 3 brM + sqeG8Prp4 + kZ + lvuxPkr1mqm95nfp
DXM459676a2q1dsV7XietvNno78 + CJn1cjX6mOeIm0bR33dfRU5NJjidt9t5afjG / V6JZ7I2QMNh
nyo2BhsMuVG3wAhMSbbQRAMolnE + iuGUSCyJZRKuGUSCyJZK4llEgyZMYTuCUsYSCQASISAAAlCQ
AAAAAAEoASCASAAAAAAAAAAAAlACRACQAAAAAAAAAEgCEoASCAAAAAAAAAAAAAAAAAAAAAAABAAA
AAAAAAAISAIAAAAAAQAAACASgAAAQJAQAAhIDHZhln3do7z0WS18mWsajHjmes7pg3dNi5aRMNqO
yvDHTpPRaigHZhN4hHRlaVN59JY3zRENLUavaO + yq0iNVlitJ6vNcR1MVi0zO0era1 / Ea0rPvbz5
PM5MWp45qvo2GZrhmfrsnpHpHzTCseEcM / 2 vrr8Q1Eb4qzy44nziPN63HpYiIiI7LNHoqabBTFii
IpSNohuVxrKtWMEejPwY9G1FFmHB4mWJn7MdfnIM9JpIx15to5pbUaas / a6rqViI7MxPxqX0UT1r
O3wVzpbR2hviP5i03Y5s6a879FNtHljydhExCv8AMTPJXBnRZbz0iG5ptFjwe / l96zctMVamTJtE
yTMibu1VrdTzRMR0j0ed4lr64MVpm0RERvMz5NvX62uOJ69XhOKX1HH9bHDtFvNYnfJeOy0Z2ojX
6 jjnEq6fRUmccTvN / J9H0eKcOnx45neaxEbubwHgOHg + milI3vP2resu3Wu0JQmITsmISDHZHKz2
JgFc1RMLJhGwK9iIZ7MZgEdgmAEwyiWCdwWRLKJVxKYsC2JTuriWUSDNlEsIlMAySx3SCRCQSIAS
AAACRACQAAAAAAASIASAAAAAAAAAAAAAAACRACRACQASIAAAAAAAAAAAAAAAAAAAAAAAAQCUAAAA
AAAAAAIAAAAAAAAQAAAAAACBICBICAAEJAQJQCJcLjuS2ny6fPG / LWdpd1o8T0X07SXx / e7wCdJx
Wa0jmneHQpxPDMdZmJfNtZm49weZrh0 / j4o7VtSZ2 + Uw0 / 8 A7o49k92vBLc / ntFohFW9PqGXimOI
6 Tu1L8T3eCx6r2t1O3JwvHjifO99v7t / Bwf2l1PXU6rS6eJ8qUm8x + so5TsekzcSjbvs4mt4rzW5
K2mbT0itesy2cHsvbvqtbmyz5xERWP2jd1tJwrTaONsOKtZ8585 + cnDrzmn4Rq + IZObUROHD32n7
Vv8A0ej0uhxaXFGPFSK1j0bkY4jyZRVZVXFGUVWbGwKsk8mObekNrSW3pWf1a2aYjHbm7bNnQ1id
PW0TvuDdhJEbQABMsLW2R0ZTMQrvfbz2YWzVhpanUxEd0dWkW5c8R5uXxDX1w4pnfr5Q19XxKuOJ
2 neXltVqtVxbV / RdJ715 + 1 bypANfiOu1HENV9C0MTfNeesx2rD1PAeBYuE6aKx72W3W9 / WVnBuB4
eF4dqRzZbdb5J72l160WVK02ZxCYhOwI23TsnY2BGxsnYBjsiYZsZBjMMZZSgGEolMsQDdG6NwZ7
piVe6YkFsSziVMWZRILolMSriWUSCyJTuwhMSDMRCQSI3SAlACRCQAAEoAEoASAAAAAAAAACUACR
ACQAAAAAAAAAAAAASAAAAAAAAAAAAAAAAAAACAAAAAAAAAAAAAABAAAAAAAAAAAAACBKAAAAAAAQ
JQAAAhICEbJAYTWJ7wx8KvpC0BV4ceieWGewDHlNmWwCNjZICNhIDmcZredBecdpiY69FXCOLW + i
UiZidukulmxxlx2paN4mNng + K4 + I8Hy2yaTfl37TXetoCPfRxfp1qi3F48ofKMvtvxak8s6LDv61
rZji9rPaLUf5PC + bfttS0q8q3p9W / wBrRMdpUZuKdN99nzvFqPbTVz7nD8OKs + do2 / mW3h4D7Xaq
ZnPrtNpqz35aRaYOHY9Zk4pNt9rR + rl6zi + OnS + WN57Rv1lXp / YrNaYtruL6zNPnGO3hxP6O5w / 2
f0HDuun09Yv55Le9afznqcOvO4tBreMTHu30unnva0bWt8on + 70 nDuE4OHYYx4Kbesz3tPrMuhGO
IjpDOKrK9YVpsyiGUQnYGOyUgI2SlAIEmwMWMs9kTAMJYzDOYRMArmGErZhhMArlHmzmGMwDE3Ts
bAbs4swj5pgFkSziVcM4BZEsolXDKAZwyhjCYBkACQhIAAAAAAAJAAAAAAAAAAAAAAAAAAAShIAA
AAAAAAJAAAAAAAAAAAAAABAJEAAAAAAAAAAAAAAAIEoBKAAAAAAAAAAAAAAABAlAAAAAAAIAAAAA
BAkBAkBAkBAlACEgMZjdjbFW8bWrEx8YWANb6Fp + bfwab + vLDKMFK9qxH5L0bAr8OPRPKz2AY7J2
SbAjYZAI2E7AIEgIEgIEgMdkSy2NgY7MdlmyNoBXsxmFuyNgVTVjNV3KjlBRNTlXTVHKCrlIqt5T
lBhEMohlFerLlBjEMohMVTEARDKCITsAk2AEgAAAkAAAAAAAAAAAAAAAAAAAAAAAASAAAAAAAAD /
2021-10-04 22:29:15 +02:00
2 Q == ` ;async function aQe(e){let t=(s,o="application/octet-stream")=>fetch( ` data : $ { o } ; base64 , $ { s } ` ).then(a=>a.blob()),n,r;switch(e.config.warmup){case"face":n=await t(KS);break;case"body":case"full":n=await t(XS);break;default:n=null}if(n){let s=await createImageBitmap(n);r=await e.detect(s,e.config),s.close()}return r}async function iQe(e){return new Promise(t=>{let n;switch(e.config.warmup){case"face":n="data:image/jpeg;base64,"+KS;break;case"full":case"body":n="data:image/jpeg;base64,"+XS;break;default:n=null}let r;typeof Image!="undefined"?r=new Image:Ee.Image&&(r=new Ee.Image),r.onload=async()=>{let s=Zo(r.naturalWidth,r.naturalHeight);if(!s)Ce("Warmup: Canvas not found"),t({});else{let o=s.getContext("2d");o&&o.drawImage(r,0,0);let a=await e.image(s),i=await e.detect(a.tensor,e.config);t(i)}},n?r.src=n:t(null)})}async function lQe(e){let t=s=>Buffer.from(s,"base64"),n;if(e.config.warmup==="face"&&(n=t(KS)),(e.config.warmup==="body"||e.config.warmup==="full")&&(n=t(XS)),!n)return null;let r;if(typeof void 0!="undefined"){let s=(void 0).decodeJpeg(n),o=s.expandDims(0);e.tf.dispose(s),r=await e.detect(o,e.config),e.tf.dispose(o)}else e.config.debug&&Ce("Warmup tfjs-node not loaded");return r}async function pie(e,t){let n=en();if(e.state="warmup",t&&(e.config=os(e.config,t)),!e.config.warmup||e.config.warmup==="none")return{error:"null"};let r;return new Promise(async s=>{typeof createImageBitmap=="function"?r=await aQe(e):typeof Image!="undefined"||Ee.Canvas!==void 0?r=await iQe(e):r=await lQe(e);let o=en();e.config.debug&&Ce("Warmup",e.config.warmup,Math.round(o-n),"ms"),e.emit("warmup"),s(r)})}var m1,Iv,Cv,ZS,fie=class{constructor(t){nt(this,"version");nt(this,"config");nt(this,"result");nt(this,"state");nt(this,"process");nt(this,"tf");nt(this,"env");nt(this,"draw");nt(this,"models");nt(this,"events");nt(this,"faceTriangulation");nt(this,"faceUVMap");nt(this,"performance");E1(this,m1,void 0);E1(this,Iv,void 0);E1(this,Cv,void 0);nt(this,"gl");nt(this,"analyze",(...t)=>{if(!_1(this,Iv))return;let n=this.tf.engine().state.numTensors,r=_1(this,m1);R1(this,m1,n);let s=n-r;s!==0&&Ce(...t,s)});E1(this,ZS,t=>{if(!_1(this,Cv))return null;if(!t)return"input is not defined";if(this.env.node&&!(t instanceof Vt))return"input must be a tensor";try{this.tf.getBackend()}catch(n){return"backend not loaded"}return null});nt(this,"similarity",lie);nt(this,"distance",qS);nt(this,"match",uie);nt(this,"emit",t=>{var n;this.events&&this.events.dispatchEvent&&((n=this.events)==null||n.dispatchEvent(new Event(t)))});K5(),this.env=Ee,Ec.wasmPath= ` https : //cdn.jsdelivr.net/npm/@tensorflow/tfjs-backend-wasm@${qw}/dist/`,Ec.modelBasePath=this.env.browser?"../models/":"file://models/",Ec.backend=this.env.browser?"humangl":"tensorflow",this.version=qR,Object.defineProperty(this,"version",{value:qR}),this.config=JSON.parse(JSON.stringify(Ec)),Object.seal(this.config),t&&(this.config=os(this.config,t)),this.tf=bh,this.state="idle",R1(this,m1,0),R1(this,Iv,!1),R1(this,Cv,!1),this.performance={backend:0,load:0,image:0,frames:0,cached:0,changed:0,total:0,draw:0},this.events=typeof EventTarget!="undefined"?new EventTarget:void 0,this.models=new wv,this.draw={options:wc,canvas:(n,r)=>eie(n,r),face:(n,r,s)=>GO(n,r,s),body:(n,r,s)=>UO(n,r,s),hand:(n,r,s)=>jO(n,r,s),gesture:(n,r,s)=>WO(n,r,s),object:(n,r,s)=>HO(n,r,s),person:(n,r,s)=>Qae(n,r,s),all:(n,r,s)=>tie(n,r,s)},this.result={face:[],body:[],hand:[],gesture:[],object:[],performance:{},timestamp:0,persons:[]},this.process={tensor:null,canvas:null},this.faceTriangulation=GU,this.faceUVMap=UU,this.gl=fr,this.emit("create")}reset(){let t=this.config.backend;this.config=JSON.parse(JSON.stringify(Ec)),this.config.backend=t}validate(t){return kT(Ec,t||this.config)}image(t){return rx(t,this.config)}async segmentation(t,n){return qae(t,n,this.config)}enhance(t){return a7(t)}async init(){await jS(this,!0),await this.tf.ready(),vU(this.env)}async load(t){this.state="load";let n=en(),r=Object.values(this.models).filter(a=>a).length;t&&(this.config=os(this.config,t)),Ee.initial&&(this.config.debug&&Ce(`version:
2021-04-28 14:58:21 +02:00
/ * *
* @ license
* Copyright 2017 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2018 Google LLC
*
* Use of this source code is governed by an MIT - style
* license that can be found in the LICENSE file or at
* https : //opensource.org/licenses/MIT.
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2018 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
*
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2018 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2019 Google LLC
*
* Use of this source code is governed by an MIT - style
* license that can be found in the LICENSE file or at
* https : //opensource.org/licenses/MIT.
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2019 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
*
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2019 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2020 Google Inc . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2020 Google LLC
*
* Use of this source code is governed by an MIT - style
* license that can be found in the LICENSE file or at
* https : //opensource.org/licenses/MIT.
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2020 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2020 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the License ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an AS IS BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
/ * *
* @ license
* Copyright 2021 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
2021-04-30 17:55:04 +02:00
/ * *
* @ license
* Copyright 2021 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* https : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
2021-08-31 19:00:06 +02:00
/ * *
* @ license
* Copyright 2018 Google LLC . All Rights Reserved .
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* === === === === === === === === === === === === === === === === === === === === === === === === === ==
* /
2021-04-28 14:58:21 +02:00
/** @license See the LICENSE file. */
