@genai-fi/nanogpt 0.9.0 → 0.10.0

package/dist/ops-LuCMAnmM.js

@@ -1,1525 +0,0 @@
-import { E as f, F as c, n as p, G as h, aS as Is, l as D, aT as Ss, aU as Ts, aV as xs, p as Ms, aW as M, w as ls, x as As, b as k, aX as gs, aY as ws, X as qs, ap as Ds, Q as K, aZ as Bs, a_ as Gs, a$ as zs, b0 as Os, b1 as Ls, b2 as Rs, b3 as Ws, b4 as Ks, b5 as vs, a0 as q, b6 as Vs, b7 as Cs, b8 as Ps, b9 as Ys, ba as js, aJ as Fs, bb as Us, t as Zs, bc as Q, bd as Hs, be as Xs, bf as Js, bg as Qs, bh as sn, bi as nn, bj as en, bk as tn, a as g, q as z, o as S, h as rn, c as d, bl as on, J as ss, d as an, a4 as O, z as cn } from "./index-BzFyqcy-.js";
-import { t as w } from "./tensor1d-vML0r3q6.js";
-import { n as un, a as ln, b as pn } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { r as y } from "./reshape-CnIwVG1c.js";
-import { s as ds } from "./split-DK2k5eHf.js";
-import { s as E } from "./sum-CJ0ULhmt.js";
-import { b as ns } from "./broadcast_to-DzlNweb8.js";
-import { s as x } from "./slice-d0Vo9XTN.js";
-import { r as Z } from "./range-CWcz7xFA.js";
-import { t as fn } from "./tensor-CZr4dh61.js";
-import { s as H } from "./stack-DFatutCx.js";
-import { c as mn, z as hn } from "./zeros-Bj5rMYA7.js";
-import { e as $s } from "./axis_util-TbGYJ208.js";
-import { m as es, a as ps, e as os, l as bn } from "./log_sum_exp-DO6z8tSE.js";
-import { c as ts } from "./concat-B912vBbo.js";
-import { m as G } from "./mat_mul-DzjTFx-u.js";
-import { t as rs } from "./tensor2d-D76QGjF3.js";
-import { o as gn } from "./ones-tIJeHlq-.js";
-import { r as Es } from "./relu-BjCh_SYb.js";
-/**
- * @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.
- * =============================================================================
- */
-function dn(e, r, s) {
-  const t = c(e, "x", "bincount"), n = c(r, "weights", "bincount");
-  p(t.dtype === "int32", () => `Error in bincount: input dtype must be int32, but got ${t.dtype}`), p(s >= 0, () => `size must be non-negative, but got ${s}.`), p(n.size === t.size || n.size === 0, () => `Error in bincount: weights must have the same size as input or0-length, but got input shape: ${t.shape}, weights shape: ${n.shape}.`);
-  const o = { x: t, weights: n }, a = { size: s };
-  return h.runKernel(Is, o, a);
-}
-const $n = /* @__PURE__ */ f({ bincount_: dn });
-/**
- * @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.
- * =============================================================================
- */
-function En(e, r, s) {
-  const t = c(r, "a", "where"), n = c(s, "b", "where"), o = c(e, "condition", "where", "bool"), a = D(D(o.shape, t.shape), n.shape), i = ns(o, a), l = ns(t, a), u = ns(n, a), m = {
-    condition: i,
-    t: l,
-    e: u
-  };
-  return h.runKernel(Ss, m);
-}
-const v = /* @__PURE__ */ f({ where_: En });
-/**
- * @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.
- * =============================================================================
- */
-function _n(e, ...r) {
-  const s = r.map((n, o) => c(n, `tensors${o}`, "einsum")), t = { equation: e };
-  return h.runKernel(Ts, s, t);
-}
-const P = /* @__PURE__ */ f({ einsum_: _n });
-/**
- * @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.
- * =============================================================================
- */
-function kn(e, r = null, s = !1) {
-  const n = { x: c(e, "x", "min") }, o = { axis: r, keepDims: s };
-  return h.runKernel(xs, n, o);
-}
-const fs = /* @__PURE__ */ f({ min_: kn });
-/**
- * @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.
- * =============================================================================
- */
-function yn(e, r = "euclidean", s = null, t = !1) {
-  e = c(e, "x", "norm");
-  const n = _s(e, r, s);
-  let o = n.shape;
-  if (t) {
-    const a = Ms(s, e.shape);
-    o = $s(n.shape, a);
-  }
-  return y(n, o);
-}
-function _s(e, r, s = null) {
-  if (e.rank === 0)
-    return M(e);
-  if (e.rank !== 1 && s === null)
-    return _s(y(e, [-1]), r, s);
-  if (e.rank === 1 || typeof s == "number" || Array.isArray(s) && s.length === 1) {
-    if (r === 1)
-      return E(M(e), s);
-    if (r === 1 / 0)
-      return es(M(e), s);
-    if (r === -1 / 0)
-      return fs(M(e), s);
-    if (r === "euclidean" || r === 2)
-      return ls(E(As(M(e), k(2, "int32")), s));
-    throw new Error(`Error in norm: invalid ord value: ${r}`);
-  }
-  if (Array.isArray(s) && s.length === 2) {
-    if (r === 1)
-      return es(E(M(e), s[0]), s[1] - 1);
-    if (r === 1 / 0)
-      return es(E(M(e), s[1]), s[0]);
-    if (r === -1 / 0)
-      return fs(E(M(e), s[1]), s[0]);
-    if (r === "fro" || r === "euclidean")
-      return ls(E(gs(e), s));
-    throw new Error(`Error in norm: invalid ord value: ${r}`);
-  }
-  throw new Error(`Error in norm: invalid axis: ${s}`);
-}
-const ks = /* @__PURE__ */ f({ norm_: yn });
-/**
- * @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.
- * =============================================================================
- */
-function Nn(e, r = 0) {
-  const s = c(e, "x", "expandDims", "string_or_numeric");
-  p(r <= s.rank, () => "Axis must be <= rank of the tensor");
-  const t = { input: s }, n = { dim: r };
-  return h.runKernel(ws, t, n);
-}
-const W = /* @__PURE__ */ f({ expandDims_: Nn });
-/**
- * @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.
- * =============================================================================
- */
-function In(e, r) {
-  const s = c(e, "x", "tile", "string_or_numeric");
-  p(s.rank === r.length, () => `Error in transpose: rank of input ${s.rank} must match length of reps ${r}.`);
-  const t = { x: s }, n = { reps: r };
-  return h.runKernel(qs, t, n);
-}
-const U = /* @__PURE__ */ f({ tile_: In });
-/**
- * @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.
- * =============================================================================
- */
-function Sn(e, r, s, t = "float32") {
-  r == null && (r = e);
-  const n = Ds([e, r], t), o = e <= r ? e : r;
-  for (let i = 0; i < o; ++i)
-    n.set(1, i, i);
-  const a = y(n.toTensor(), [e, r]);
-  if (s == null)
-    return a;
-  if (s.length === 1)
-    return U(W(a, 0), [s[0], 1, 1]);
-  if (s.length === 2)
-    return U(W(W(a, 0), 0), [s[0], s[1], 1, 1]);
-  if (s.length === 3)
-    return U(W(W(W(a, 0), 0), 0), [
-      s[0],
-      s[1],
-      s[2],
-      1,
-      1
-    ]);
-  throw new Error(`eye() currently supports only 1D and 2D batchShapes, but received ${s.length}D.`);
-}
-const Tn = /* @__PURE__ */ f({ eye_: Sn });
-/**
- * @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.
- * =============================================================================
- */
-function xn(e, r) {
-  let s = c(e, "a", "greater", "string_or_numeric"), t = c(r, "b", "greater", "string_or_numeric");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Bs, n);
-}
-const is = /* @__PURE__ */ f({ greater_: xn });
-/**
- * @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.
- * =============================================================================
- */
-function Mn(e, r) {
-  let s = c(e, "a", "greaterEqual", "string_or_numeric"), t = c(r, "b", "greaterEqual", "string_or_numeric");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Gs, n);
-}
-const An = /* @__PURE__ */ f({ greaterEqual_: Mn });
-/**
- * @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.
- * =============================================================================
- */
-function wn(e) {
-  const s = { input: c(e, "input", "imag") };
-  return h.runKernel(zs, s);
-}
-const qn = /* @__PURE__ */ f({ imag_: wn });
-/**
- * @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.
- * =============================================================================
- */
-function Dn(e, r) {
-  let s = c(e, "a", "less", "string_or_numeric"), t = c(r, "b", "less", "string_or_numeric");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Os, n);
-}
-const ms = /* @__PURE__ */ f({ less_: Dn });
-/**
- * @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.
- * =============================================================================
- */
-function Bn(e, r) {
-  let s = c(e, "a", "lessEqual", "string_or_numeric"), t = c(r, "b", "lessEqual", "string_or_numeric");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Ls, n);
-}
-const ys = /* @__PURE__ */ f({ lessEqual_: Bn });
-/**
- * @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.
- * =============================================================================
- */
-function Gn(e) {
-  const s = { x: c(e, "x", "log1p") };
-  return h.runKernel(Rs, s);
-}
-const zn = /* @__PURE__ */ f({ log1p_: Gn });
-/**
- * @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.
- * =============================================================================
- */
-function On(e) {
-  const s = { x: c(e, "x", "neg") };
-  return h.runKernel(Ws, s);
-}
-const V = /* @__PURE__ */ f({ neg_: On });
-/**
- * @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.
- * =============================================================================
- */
-function Ln(e, r) {
-  const s = c(e, "a", "logicalAnd", "bool"), t = c(r, "b", "logicalAnd", "bool");
-  D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Ks, n);
-}
-const Rn = /* @__PURE__ */ f({ logicalAnd_: Ln });
-/**
- * @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.
- * =============================================================================
- */
-function Wn(e, r = null, s = !1) {
-  const n = { x: c(e, "x", "mean") }, o = { axis: r, keepDims: s };
-  return h.runKernel(vs, n, o);
-}
-const Kn = /* @__PURE__ */ f({ mean_: Wn });
-/**
- * @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.
- * =============================================================================
- */
-function vn(e, r) {
-  let s = c(e, "a", "minimum"), t = c(r, "b", "minimum");
-  [s, t] = K(s, t), s.dtype === "bool" && (s = q(s, "int32"), t = q(t, "int32")), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Vs, n);
-}
-const as = /* @__PURE__ */ f({ minimum_: vn });
-/**
- * @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.
- * =============================================================================
- */
-function Vn(e, r) {
-  let s = c(e, "a", "notEqual", "string_or_numeric"), t = c(r, "b", "notEqual", "string_or_numeric");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t };
-  return h.runKernel(Cs, n);
-}
-const Cn = /* @__PURE__ */ f({ notEqual_: Vn });
-/**
- * @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.
- * =============================================================================
- */
-function Pn(e) {
-  const s = { input: c(e, "input", "real") };
-  return h.runKernel(Ps, s);
-}
-const Yn = /* @__PURE__ */ f({ real_: Pn });
-/**
- * @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.
- * =============================================================================
- */
-function jn(e) {
-  const s = { x: c(e, "x", "round") };
-  return h.runKernel(Ys, s);
-}
-const Fn = /* @__PURE__ */ f({ round_: jn });
-/**
- * @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.
- * =============================================================================
- */
-function Un(e, r) {
-  let s = c(e, "a", "squaredDifference"), t = c(r, "b", "squaredDifference");
-  [s, t] = K(s, t), D(s.shape, t.shape);
-  const n = { a: s, b: t }, o = {};
-  return h.runKernel(js, n, o);
-}
-const Zn = /* @__PURE__ */ f({ squaredDifference_: Un });
-/**
- * @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.
- * =============================================================================
- */
-function Hn(e, r) {
-  const s = c(e, "x", "squeeze", "string_or_numeric");
-  return y(s, Fs(s.shape, r).newShape);
-}
-const Xn = /* @__PURE__ */ f({ squeeze_: Hn });
-/**
- * @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.
- * =============================================================================
- */
-function Jn(e, r = 0) {
-  const s = c(e, "x", "unstack", "string_or_numeric");
-  p(r >= -s.shape.length && r < s.shape.length, () => `Axis = ${r} is not in [-${s.shape.length}, ${s.shape.length})`);
-  const t = { value: s }, n = { axis: r };
-  return h.runKernel(Us, t, n);
-}
-const Ns = /* @__PURE__ */ f({ unstack_: Jn });
-/**
- * @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.
- * =============================================================================
- */
-function Qn(e, r, s) {
-  const t = c(e, "x", "transpose");
-  if (r == null && (r = t.shape.map((a, i) => i).reverse()), p(t.rank === r.length, () => `Error in transpose: rank of input ${t.rank} must match length of perm ${r}.`), r.forEach((a) => {
-    p(a >= 0 && a < t.rank, () => `All entries in 'perm' must be between 0 and ${t.rank - 1} but got ${r}`);
-  }), t.rank <= 1)
-    return t.clone();
-  const n = { x: t }, o = { perm: r };
-  return t.dtype === "complex64" ? Zs(() => {
-    let a = Yn(t), i = qn(t);
-    return a = h.runKernel(Q, { x: a }, o), i = h.runKernel(Q, { x: i }, o), s && (i = V(i)), mn(a, i);
-  }) : h.runKernel(Q, n, o);
-}
-const hs = /* @__PURE__ */ f({ transpose_: Qn });
-/**
- * @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.
- * =============================================================================
- */
-function se(e, r, s, t, n = "bilinear", o = 0) {
-  const a = c(e, "image", "cropAndResize"), i = c(r, "boxes", "cropAndResize", "float32"), l = c(s, "boxInd", "cropAndResize", "int32"), u = i.shape[0];
-  p(a.rank === 4, () => `Error in cropAndResize: image must be rank 4,but got rank ${a.rank}.`), p(i.rank === 2 && i.shape[1] === 4, () => `Error in cropAndResize: boxes must be have size [${u},4] but had shape ${i.shape}.`), p(l.rank === 1 && l.shape[0] === u, () => `Error in cropAndResize: boxInd must be have size [${u}] but had shape ${i.shape}.`), p(t.length === 2, () => `Error in cropAndResize: cropSize must be of length 2, but got length ${t.length}.`), p(t[0] >= 1 && t[1] >= 1, () => `cropSize must be atleast [1,1], but was ${t}`), p(n === "bilinear" || n === "nearest", () => `method must be bilinear or nearest, but was ${n}`);
-  const m = { image: a, boxes: i, boxInd: l }, b = { method: n, extrapolationValue: o, cropSize: t };
-  return h.runKernel(Hs, m, b);
-}
-const ne = /* @__PURE__ */ f({ cropAndResize_: se });
-/**
- * @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.
- * =============================================================================
- */
-function ee(e) {
-  const r = c(e, "image", "flipLeftRight", "float32");
-  p(r.rank === 4, () => `Error in flipLeftRight: image must be rank 4,but got rank ${r.rank}.`);
-  const s = { image: r };
-  return h.runKernel(Xs, s, {});
-}
-const te = /* @__PURE__ */ f({ flipLeftRight_: ee });
-/**
- * @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.
- * =============================================================================
- */
-function re(e) {
-  const r = c(e, "image", "grayscaleToRGB"), s = r.rank - 1, t = r.shape[s];
-  p(r.rank >= 2, () => `Error in grayscaleToRGB: images must be at least rank 2, but got rank ${r.rank}.`), p(t === 1, () => `Error in grayscaleToRGB: last dimension of a grayscale image should be size 1, but got size ${t}.`);
-  const n = new Array(r.rank);
-  return n.fill(1, 0, s), n[s] = 3, U(r, n);
-}
-const oe = /* @__PURE__ */ f({ grayscaleToRGB_: re });
-/**
- * @license
- * Copyright 2023 Google LLC.
- * 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.
- * =============================================================================
- */
-function ae(e) {
-  const r = c(e, "image", "RGBToGrayscale"), s = r.rank - 1, t = r.shape[s];
-  p(r.rank >= 2, () => `Error in RGBToGrayscale: images must be at least rank 2, but got rank ${r.rank}.`), p(t === 3, () => `Error in RGBToGrayscale: last dimension of an RGB image should be size 3, but got size ${t}.`);
-  const n = r.dtype, o = q(r, "float32"), a = w([0.2989, 0.587, 0.114]);
-  let i;
-  switch (r.rank) {
-    case 2:
-      i = P("ij,j->i", o, a);
-      break;
-    case 3:
-      i = P("ijk,k->ij", o, a);
-      break;
-    case 4:
-      i = P("ijkl,l->ijk", o, a);
-      break;
-    case 5:
-      i = P("ijklm,m->ijkl", o, a);
-      break;
-    case 6:
-      i = P("ijklmn,n->ijklm", o, a);
-      break;
-    default:
-      throw new Error("Not a valid tensor rank.");
-  }
-  return i = W(i, -1), q(i, n);
-}
-const ie = /* @__PURE__ */ f({ rgbToGrayscale_: ae });
-/**
- * @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.
- * =============================================================================
- */
-function ce(e, r, s = 0, t = 0.5) {
-  const n = c(e, "image", "rotateWithOffset", "float32");
-  p(n.rank === 4, () => `Error in rotateWithOffset: image must be rank 4,but got rank ${n.rank}.`);
-  const o = { image: n }, a = { radians: r, fillValue: s, center: t };
-  return h.runKernel(Js, o, a);
-}
-const ue = /* @__PURE__ */ f({ rotateWithOffset_: ce });
-/**
- * @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.
- * =============================================================================
- */
-function C(e, r, s, t, n, o) {
-  t == null && (t = 0.5), n == null && (n = Number.NEGATIVE_INFINITY), o == null && (o = 0);
-  const a = e.shape[0];
-  return s = Math.min(s, a), p(0 <= t && t <= 1, () => `iouThreshold must be in [0, 1], but was '${t}'`), p(e.rank === 2, () => `boxes must be a 2D tensor, but was of rank '${e.rank}'`), p(e.shape[1] === 4, () => `boxes must have 4 columns, but 2nd dimension was ${e.shape[1]}`), p(r.rank === 1, () => "scores must be a 1D tensor"), p(r.shape[0] === a, () => `scores has incompatible shape with boxes. Expected ${a}, but was ${r.shape[0]}`), p(0 <= o && o <= 1, () => `softNmsSigma must be in [0, 1], but was '${o}'`), { maxOutputSize: s, iouThreshold: t, scoreThreshold: n, softNmsSigma: o };
-}
-/**
- * @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.
- * =============================================================================
- */
-function le(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY) {
-  const o = c(e, "boxes", "nonMaxSuppression", "float32"), a = c(r, "scores", "nonMaxSuppression", "float32"), i = C(o, a, s, t, n);
-  s = i.maxOutputSize, t = i.iouThreshold, n = i.scoreThreshold;
-  const l = { maxOutputSize: s, iouThreshold: t, scoreThreshold: n };
-  return h.runKernel(Qs, { boxes: o, scores: a }, l);
-}
-const pe = /* @__PURE__ */ f({ nonMaxSuppression_: le });
-/**
- * @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.
- * =============================================================================
- */
-async function fe(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY) {
-  const o = c(e, "boxes", "nonMaxSuppressionAsync"), a = c(r, "scores", "nonMaxSuppressionAsync"), i = C(o, a, s, t, n);
-  s = i.maxOutputSize, t = i.iouThreshold, n = i.scoreThreshold;
-  const l = await Promise.all([o.data(), a.data()]), u = l[0], m = l[1], { selectedIndices: b } = un(u, m, s, t, n);
-  return o !== e && o.dispose(), a !== r && a.dispose(), w(b, "int32");
-}
-const me = fe;
-/**
- * @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.
- * =============================================================================
- */
-function he(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = 0) {
-  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = C(a, i, s, t, n, o);
-  s = l.maxOutputSize, t = l.iouThreshold, n = l.scoreThreshold, o = l.softNmsSigma;
-  const u = { boxes: a, scores: i }, m = { maxOutputSize: s, iouThreshold: t, scoreThreshold: n, softNmsSigma: o }, b = h.runKernel(sn, u, m);
-  return { selectedIndices: b[0], selectedScores: b[1] };
-}
-const be = /* @__PURE__ */ f({ nonMaxSuppressionWithScore_: he });
-/**
- * @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.
- * =============================================================================
- */
-async function ge(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = 0) {
-  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = C(a, i, s, t, n, o);
-  s = l.maxOutputSize, t = l.iouThreshold, n = l.scoreThreshold, o = l.softNmsSigma;
-  const u = await Promise.all([a.data(), i.data()]), m = u[0], b = u[1], { selectedIndices: _, selectedScores: $ } = ln(m, b, s, t, n, o);
-  return a !== e && a.dispose(), i !== r && i.dispose(), {
-    selectedIndices: w(_, "int32"),
-    selectedScores: w($)
-  };
-}
-const de = ge;
-/**
- * @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.
- * =============================================================================
- */
-function $e(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = !1) {
-  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = C(
-    a,
-    i,
-    s,
-    t,
-    n,
-    null
-    /* softNmsSigma */
-  ), u = l.maxOutputSize, m = l.iouThreshold, b = l.scoreThreshold, _ = { boxes: a, scores: i }, $ = {
-    maxOutputSize: u,
-    iouThreshold: m,
-    scoreThreshold: b,
-    padToMaxOutputSize: o
-  }, N = h.runKernel(nn, _, $);
-  return { selectedIndices: N[0], validOutputs: N[1] };
-}
-const Ee = /* @__PURE__ */ f({ nonMaxSuppressionPadded_: $e });
-/**
- * @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.
- * =============================================================================
- */
-async function _e(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = !1) {
-  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = C(
-    a,
-    i,
-    s,
-    t,
-    n,
-    null
-    /* softNmsSigma */
-  ), u = l.maxOutputSize, m = l.iouThreshold, b = l.scoreThreshold, [_, $] = await Promise.all([a.data(), i.data()]), { selectedIndices: N, validOutputs: A } = pn(_, $, u, m, b, o);
-  return a !== e && a.dispose(), i !== r && i.dispose(), {
-    selectedIndices: w(N, "int32"),
-    validOutputs: k(A, "int32")
-  };
-}
-const ke = _e;
-/**
- * @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.
- * =============================================================================
- */
-function ye(e, r, s = !1, t = !1) {
-  const n = c(e, "images", "resizeBilinear");
-  p(n.rank === 3 || n.rank === 4, () => `Error in resizeBilinear: x must be rank 3 or 4, but got rank ${n.rank}.`), p(r.length === 2, () => `Error in resizeBilinear: new shape must 2D, but got shape ${r}.`), p(t === !1 || s === !1, () => "Error in resizeBilinear: If halfPixelCenters is true, alignCorners must be false.");
-  let o = n, a = !1;
-  n.rank === 3 && (a = !0, o = y(n, [1, n.shape[0], n.shape[1], n.shape[2]]));
-  const i = { images: o }, l = { alignCorners: s, halfPixelCenters: t, size: r }, u = h.runKernel(en, i, l);
-  return a ? y(u, [u.shape[1], u.shape[2], u.shape[3]]) : u;
-}
-const Ne = /* @__PURE__ */ f({ resizeBilinear_: ye });
-/**
- * @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.
- * =============================================================================
- */
-function Ie(e, r, s = !1, t = !1) {
-  const n = c(e, "images", "resizeNearestNeighbor");
-  p(n.rank === 3 || n.rank === 4, () => `Error in resizeNearestNeighbor: x must be rank 3 or 4, but got rank ${n.rank}.`), p(r.length === 2, () => `Error in resizeNearestNeighbor: new shape must 2D, but got shape ${r}.`), p(n.dtype === "float32" || n.dtype === "int32", () => "`images` must have `int32` or `float32` as dtype"), p(t === !1 || s === !1, () => "Error in resizeNearestNeighbor: If halfPixelCenters is true, alignCorners must be false.");
-  let o = n, a = !1;
-  n.rank === 3 && (a = !0, o = y(n, [1, n.shape[0], n.shape[1], n.shape[2]]));
-  const i = { images: o }, l = { alignCorners: s, halfPixelCenters: t, size: r }, u = h.runKernel(tn, i, l);
-  return a ? y(u, [u.shape[1], u.shape[2], u.shape[3]]) : u;
-}
-const Se = /* @__PURE__ */ f({ resizeNearestNeighbor_: Ie });
-/**
- * @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.
- * =============================================================================
- */
-function Te(e, r = "binary", s = !1, t = 0.5) {
-  const n = c(e, "image", "threshold"), o = 0.2989, a = 0.587, i = 0.114, l = n.shape[0] * n.shape[1];
-  let u = g(w([t]), 255), m, b, _, $;
-  if (p(n.rank === 3, () => `Error in threshold: image must be rank 3,but got rank ${n.rank}.`), p(n.shape[2] === 3 || n.shape[2] === 1, () => `Error in threshold: image color channel must be equal to 3 or 1but got ${n.shape[2]}.`), p(n.dtype === "int32" || n.dtype === "float32", () => `Error in dtype: image dtype must be int32 or float32,but got dtype ${n.dtype}.`), p(r === "otsu" || r === "binary", () => `Method must be binary or otsu, but was ${r}`), n.shape[2] === 3) {
-    [m, b, _] = ds(n, [1, 1, 1], -1);
-    const T = g(m, o), B = g(b, a), R = g(_, i);
-    $ = z(z(T, B), R);
-  } else
-    $ = e;
-  if (r === "otsu") {
-    const T = $n(q(Fn($), "int32"), fn([]), 256);
-    u = xe(T, l);
-  }
-  const N = s ? ys($, u) : is($, u);
-  return q(g(N, 255), "int32");
-}
-function xe(e, r) {
-  let s = w([-1]), t = w([0]), n = w([0]), o, a, i, l, u, m;
-  for (let b = 0; b < e.size - 1; b++) {
-    o = x(e, 0, b + 1), a = x(e, b + 1), u = S(E(o), r), m = S(E(a), r);
-    const _ = E(g(o, Z(0, o.size)));
-    i = S(_, E(o));
-    const $ = rn(a.shape, o.size), N = z(Z(0, a.size), $), A = g(a, N);
-    l = S(E(A), E(a));
-    const T = d(i, l), B = d(i, l), R = g(u, m);
-    n = g(g(R, T), B);
-    const Y = is(n, t);
-    t = v(Y, n, t), s = v(Y, w([b]), s);
-  }
-  return s;
-}
-const Me = /* @__PURE__ */ f({ threshold_: Te });
-/**
- * @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.
- * =============================================================================
- */
-function Ae(e, r, s = "nearest", t = "constant", n = 0, o) {
-  const a = c(e, "image", "transform", "float32"), i = c(r, "transforms", "transform", "float32");
-  p(a.rank === 4, () => `Error in transform: image must be rank 4,but got rank ${a.rank}.`), p(i.rank === 2 && (i.shape[0] === a.shape[0] || i.shape[0] === 1) && i.shape[1] === 8, () => "Error in transform: Input transform should be batch x 8 or 1 x 8"), p(o == null || o.length === 2, () => `Error in transform: outputShape must be [height, width] or null, but got ${o}.`);
-  const l = { image: a, transforms: i }, u = { interpolation: s, fillMode: t, fillValue: n, outputShape: o };
-  return h.runKernel(on, l, u);
-}
-const we = /* @__PURE__ */ f({ transform_: Ae });
-/**
- * @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.
- * =============================================================================
- */
-function qe(e, r, s) {
-  const t = c(e, "a", "bandPart");
-  p(t.rank >= 2, () => `bandPart(): Rank must be at least 2, got ${t.rank}.`);
-  const n = t.shape, [o, a] = t.shape.slice(-2);
-  let i, l;
-  typeof r == "number" ? (p(r % 1 === 0, () => `bandPart(): numLower must be an integer, got ${r}.`), p(r <= o, () => `bandPart(): numLower (${r}) must not be greater than the number of rows (${o}).`), i = c(r < 0 ? o : r, "numLower", "bandPart")) : (p(r.dtype === "int32", () => "bandPart(): numLower's dtype must be an int32."), i = v(ms(r, 0), o, as(r, o))), typeof s == "number" ? (p(s % 1 === 0, () => `bandPart(): numUpper must be an integer, got ${s}.`), p(s <= a, () => `bandPart(): numUpper (${s}) must not be greater than the number of columns (${a}).`), l = c(s < 0 ? a : s, "numUpper", "bandPart")) : (p(s.dtype === "int32", () => "bandPart(): numUpper's dtype must be an int32."), l = v(ms(s, 0), a, as(s, a)));
-  const u = y(Z(0, o, 1, "int32"), [-1, 1]), m = Z(0, a, 1, "int32"), b = d(u, m), _ = Rn(ys(b, i), An(b, V(l))), $ = hn([o, a], t.dtype);
-  return y(H(Ns(y(t, [-1, o, a])).map((N) => v(_, N, $))), n);
-}
-const De = /* @__PURE__ */ f({ bandPart_: qe });
-/**
- * @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.
- * =============================================================================
- */
-function Be(e) {
-  let r;
-  if (Array.isArray(e)) {
-    r = !1, p(e != null && e.length > 0, () => "Gram-Schmidt process: input must not be null, undefined, or empty");
-    const n = e[0].shape[0];
-    for (let o = 1; o < e.length; ++o)
-      p(e[o].shape[0] === n, () => `Gram-Schmidt: Non-unique lengths found in the input vectors: (${e[o].shape[0]} vs. ${n})`);
-  } else
-    r = !0, e = ds(e, e.shape[0], 0).map((n) => Xn(n, [0]));
-  p(e.length <= e[0].shape[0], () => `Gram-Schmidt: Number of vectors (${e.length}) exceeds number of dimensions (${e[0].shape[0]}).`);
-  const s = [], t = e;
-  for (let n = 0; n < e.length; ++n)
-    s.push(h.tidy(() => {
-      let o = t[n];
-      if (n > 0)
-        for (let a = 0; a < n; ++a) {
-          const i = g(E(g(s[a], o)), s[a]);
-          o = d(o, i);
-        }
-      return S(o, ks(o, "euclidean"));
-    }));
-  return r ? H(s, 0) : s;
-}
-const Ge = /* @__PURE__ */ f({ gramSchmidt_: Be });
-/**
- * @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.
- * =============================================================================
- */
-function ze(e, r = !1) {
-  if (p(e.rank >= 2, () => `qr() requires input tensor to have a rank >= 2, but got rank ${e.rank}`), e.rank === 2)
-    return bs(e, r);
-  {
-    const s = e.shape.slice(0, e.shape.length - 2).reduce((l, u) => l * u), t = Ns(y(e, [
-      s,
-      e.shape[e.shape.length - 2],
-      e.shape[e.shape.length - 1]
-    ]), 0), n = [], o = [];
-    t.forEach((l) => {
-      const [u, m] = bs(l, r);
-      n.push(u), o.push(m);
-    });
-    const a = y(H(n, 0), e.shape), i = y(H(o, 0), e.shape);
-    return [a, i];
-  }
-}
-function bs(e, r = !1) {
-  return h.tidy(() => {
-    p(e.shape.length === 2, () => `qr2d() requires a 2D Tensor, but got a ${e.shape.length}D Tensor.`);
-    const s = e.shape[0], t = e.shape[1];
-    let n = Tn(s), o = ss(e);
-    const a = rs([[1]], [1, 1]);
-    let i = ss(a);
-    const l = s >= t ? t : s;
-    for (let u = 0; u < l; ++u) {
-      const m = o, b = i, _ = n;
-      [i, o, n] = h.tidy(() => {
-        const $ = x(o, [u, u], [s - u, 1]), N = ks($), A = x(o, [u, u], [1, 1]), T = v(is(A, 0), rs([[-1]]), rs([[1]])), B = d(A, g(T, N)), R = S($, B);
-        R.shape[0] === 1 ? i = ss(a) : i = ts([
-          a,
-          x(R, [1, 0], [R.shape[0] - 1, R.shape[1]])
-        ], 0);
-        const Y = V(S(G(T, B), N)), j = x(o, [u, 0], [s - u, t]), X = g(Y, i), cs = hs(i);
-        if (u === 0)
-          o = d(j, G(X, G(cs, j)));
-        else {
-          const J = d(j, G(X, G(cs, j)));
-          o = ts([x(o, [0, 0], [u, t]), J], 0);
-        }
-        const us = hs(X), F = x(n, [0, u], [s, n.shape[1] - u]);
-        if (u === 0)
-          n = d(F, G(G(F, i), us));
-        else {
-          const J = d(F, G(G(F, i), us));
-          n = ts([x(n, [0, 0], [s, u]), J], 1);
-        }
-        return [i, o, n];
-      }), an([m, b, _]);
-    }
-    return !r && s > t && (n = x(n, [0, 0], [s, t]), o = x(o, [0, 0], [t, t])), [n, o];
-  });
-}
-const Oe = /* @__PURE__ */ f({ qr_: ze });
-/**
- * @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.
- * =============================================================================
- */
-var I;
-(function(e) {
-  e[e.NONE = 0] = "NONE", e[e.MEAN = 1] = "MEAN", e[e.SUM = 2] = "SUM", e[e.SUM_BY_NONZERO_WEIGHTS = 3] = "SUM_BY_NONZERO_WEIGHTS";
-})(I || (I = {}));
-function Le(e, r, s = I.SUM_BY_NONZERO_WEIGHTS) {
-  const t = c(e, "losses", "computeWeightedLoss");
-  let n = null;
-  r != null && (n = c(r, "weights", "computeWeightedLoss"));
-  const o = n == null ? t : g(t, n);
-  if (s === I.NONE)
-    return o;
-  if (s === I.SUM)
-    return E(o);
-  if (s === I.MEAN) {
-    if (n == null)
-      return Kn(o);
-    {
-      const a = t.size / n.size, i = S(E(o), E(n));
-      return a > 1 ? S(i, k(a)) : i;
-    }
-  }
-  if (s === I.SUM_BY_NONZERO_WEIGHTS) {
-    if (n == null)
-      return S(E(o), k(t.size));
-    {
-      const a = g(n, gn(t.shape)), i = q(E(Cn(a, k(0))), "float32");
-      return S(E(o), i);
-    }
-  }
-  throw Error(`Unknown reduction: ${s}`);
-}
-const L = /* @__PURE__ */ f({ computeWeightedLoss_: Le });
-/**
- * @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.
- * =============================================================================
- */
-function Re(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
-  const n = c(e, "labels", "absoluteDifference"), o = c(r, "predictions", "absoluteDifference");
-  let a = null;
-  s != null && (a = c(s, "weights", "absoluteDifference")), O(n.shape, o.shape, "Error in absoluteDifference: ");
-  const i = M(d(n, o));
-  return L(i, a, t);
-}
-const We = /* @__PURE__ */ f({ absoluteDifference_: Re });
-function Ke(e, r, s, t, n = I.SUM_BY_NONZERO_WEIGHTS) {
-  const o = c(e, "labels", "cosineDistance"), a = c(r, "predictions", "cosineDistance");
-  let i = null;
-  t != null && (i = c(t, "weights", "cosineDistance")), O(o.shape, a.shape, "Error in cosineDistance: ");
-  const l = k(1), u = d(l, E(g(o, a), s, !0));
-  return L(u, i, n);
-}
-const ve = /* @__PURE__ */ f({ cosineDistance_: Ke });
-function Ve(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
-  let n = c(e, "labels", "hingeLoss");
-  const o = c(r, "predictions", "hingeLoss");
-  let a = null;
-  s != null && (a = c(s, "weights", "hingeLoss")), O(n.shape, o.shape, "Error in hingeLoss: ");
-  const i = k(1);
-  n = d(g(k(2), n), i);
-  const l = Es(d(i, g(n, o)));
-  return L(l, a, t);
-}
-const Ce = /* @__PURE__ */ f({ hingeLoss_: Ve });
-/**
- * @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.
- * =============================================================================
- */
-function Pe(e, r, s, t = 1, n = I.SUM_BY_NONZERO_WEIGHTS) {
-  const o = c(e, "labels", "huberLoss"), a = c(r, "predictions", "huberLoss");
-  let i = null;
-  s != null && (i = c(s, "weights", "huberLoss")), O(o.shape, a.shape, "Error in huberLoss: ");
-  const l = k(t), u = M(d(a, o)), m = as(u, l), b = d(u, m), _ = z(g(k(0.5), gs(m)), g(l, b));
-  return L(_, i, n);
-}
-const Ye = /* @__PURE__ */ f({ huberLoss_: Pe });
-/**
- * @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.
- * =============================================================================
- */
-function je(e, r, s, t = 1e-7, n = I.SUM_BY_NONZERO_WEIGHTS) {
-  const o = c(e, "labels", "logLoss"), a = c(r, "predictions", "logLoss");
-  let i = null;
-  s != null && (i = c(s, "weights", "logLoss")), O(o.shape, a.shape, "Error in logLoss: ");
-  const l = k(1), u = k(t), m = V(g(o, ps(z(a, u)))), b = g(d(l, o), ps(z(d(l, a), u))), _ = d(m, b);
-  return L(_, i, n);
-}
-const Fe = /* @__PURE__ */ f({ logLoss_: je });
-/**
- * @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.
- * =============================================================================
- */
-function Ue(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
-  const n = c(e, "labels", "meanSquaredError"), o = c(r, "predictions", "meanSquaredError");
-  let a = null;
-  s != null && (a = c(s, "weights", "meanSquaredError")), O(n.shape, o.shape, "Error in meanSquaredError: ");
-  const i = Zn(n, o);
-  return L(i, a, t);
-}
-const Ze = /* @__PURE__ */ f({ meanSquaredError_: Ue });
-/**
- * @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.
- * =============================================================================
- */
-function He(e, r) {
-  const s = c(e, "labels", "sigmoidCrossEntropyWithLogits"), t = c(r, "logits", "sigmoidCrossEntropyWithLogits");
-  O(s.shape, t.shape, "Error in sigmoidCrossEntropyWithLogits: ");
-  const n = Es(t), o = g(t, s), a = zn(os(V(M(t))));
-  return z(d(n, o), a);
-}
-function Xe(e, r, s, t = 0, n = I.SUM_BY_NONZERO_WEIGHTS) {
-  let o = c(e, "multiClassLabels", "sigmoidCrossEntropy");
-  const a = c(r, "logits", "sigmoidCrossEntropy");
-  let i = null;
-  if (s != null && (i = c(s, "weights", "sigmoidCrossEntropy")), O(o.shape, a.shape, "Error in sigmoidCrossEntropy: "), t > 0) {
-    const u = k(t), m = k(1), b = k(0.5);
-    o = z(g(o, d(m, u)), g(b, u));
-  }
-  const l = He(o, a);
-  return L(l, i, n);
-}
-const Je = /* @__PURE__ */ f({ sigmoidCrossEntropy_: Xe });
-/**
- * @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.
- * =============================================================================
- */
-function Qe(e, r, s = -1) {
-  if (s === -1 && (s = r.rank - 1), s !== r.rank - 1)
-    throw Error(`Softmax cross entropy along a non-last dimension is not yet supported. Labels / logits was rank ${r.rank} and dim was ${s}`);
-  return cn((n, o, a) => {
-    const l = bn(o, [s], !0), u = d(q(o, "float32"), l);
-    a([n, u]);
-    const m = V(g(u, n));
-    return { value: E(m, [s]), gradFunc: ($, N) => {
-      const [A, T] = N, B = $s($.shape, [s]);
-      return [
-        g(y($, B), d(q(A, "float32"), os(T))),
-        g(y($, B), d(os(T), q(A, "float32")))
-      ];
-    } };
-  })(e, r);
-}
-function st(e, r, s, t = 0, n = I.SUM_BY_NONZERO_WEIGHTS) {
-  let o = c(e, "onehotLabels", "softmaxCrossEntropy");
-  const a = c(r, "logits", "softmaxCrossEntropy");
-  let i = null;
-  if (s != null && (i = c(s, "weights", "softmaxCrossEntropy")), O(o.shape, a.shape, "Error in softmaxCrossEntropy: "), t > 0) {
-    const u = k(t), m = k(1), b = k(o.shape[1]);
-    o = z(g(o, d(m, u)), S(u, b));
-  }
-  const l = Qe(o, a);
-  return L(l, i, n);
-}
-const nt = /* @__PURE__ */ f({ softmaxCrossEntropy_: st });
-/**
- * @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.
- * =============================================================================
- */
-const kt = {
-  flipLeftRight: te,
-  grayscaleToRGB: oe,
-  resizeNearestNeighbor: Se,
-  resizeBilinear: Ne,
-  rgbToGrayscale: ie,
-  rotateWithOffset: ue,
-  cropAndResize: ne,
-  nonMaxSuppression: pe,
-  nonMaxSuppressionAsync: me,
-  nonMaxSuppressionWithScore: be,
-  nonMaxSuppressionWithScoreAsync: de,
-  nonMaxSuppressionPadded: Ee,
-  nonMaxSuppressionPaddedAsync: ke,
-  threshold: Me,
-  transform: we
-}, yt = {
-  bandPart: De,
-  gramSchmidt: Ge,
-  qr: Oe
-}, Nt = {
-  absoluteDifference: We,
-  computeWeightedLoss: L,
-  cosineDistance: ve,
-  hingeLoss: Ce,
-  huberLoss: Ye,
-  logLoss: Fe,
-  meanSquaredError: Ze,
-  sigmoidCrossEntropy: Je,
-  softmaxCrossEntropy: nt
-};
-export {
-  Fn as A,
-  Zn as B,
-  Nt as a,
-  Rn as b,
-  ys as c,
-  U as d,
-  is as e,
-  ms as f,
-  An as g,
-  W as h,
-  Tn as i,
-  zn as j,
-  as as k,
-  yt as l,
-  Kn as m,
-  V as n,
-  kt as o,
-  Cn as p,
-  fs as q,
-  ks as r,
-  Xn as s,
-  hs as t,
-  Ns as u,
-  Yn as v,
-  v as w,
-  qn as x,
-  Ne as y,
-  Se as z
-};