@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
11 lines • 805 B
JavaScript
import { SparseTensor } from '../../../../tensor/sparse/tensor';
import { WASMTensor } from '../../../../tensor/wasm/tensor';
export function divideDenseWASM(a, b, resultShape, alpha) {
const vals = new WASMTensor(a.values.wasmTensor.divide_sparse_dense(a.indices.wasmTensor, b.wasmTensor, new Uint32Array(resultShape), alpha), undefined, a.dtype);
return new SparseTensor(vals, a.indices.copy(), resultShape, a.denseDims);
}
export function divideSparseWASM(a, b, resultShape, alpha) {
const vals = new WASMTensor(a.values.wasmTensor.divide_sparse_sparse(a.indices.wasmTensor, b.indices.wasmTensor, b.values.wasmTensor, new Uint32Array(resultShape), alpha), undefined, a.dtype);
return new SparseTensor(vals, a.indices.copy(), resultShape, a.denseDims);
}
//# sourceMappingURL=wasm.js.map