@hoff97/tensor-js/dist/lib/model/functional/bce/gpu.d.ts

PyTorch like deep learning inferrence library

import { BinaryOperation } from '../../../ops/gpu/binary/binaryOperation';
import { Dispatcher } from '../../../ops/gpu/dispatcher';
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
export declare class BCEOperation<GPUTensor extends GPUTensorI> extends BinaryOperation<GPUTensor> {
    constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
    getOp(a: string, b: string): string;
}
export declare const defaultBCED: Dispatcher<GPUTensorI, unknown, unknown, BCEOperation<import("../../../tensor/gpu/tensor").GPUTensor<DTypeGpu>>>;