@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
22 lines (21 loc) • 978 B
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input } from '../operation';
import { BinaryOperation, BinaryOpInfo, BinaryOpInput } from './binaryOperation';
export interface SubtractInfo extends BinaryOpInfo {
alpha?: number;
beta?: number;
}
export interface SubtractInput extends BinaryOpInput {
alpha: number;
beta: number;
}
export declare class SubtractOperation<GPUTensor extends GPUTensorI> extends BinaryOperation<GPUTensor, SubtractInfo, SubtractInput> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
getOp(a: string, b: string): string;
calc(input: SubtractInput): GPUTensor;
getVariables(): string;
getUniformAttrs(): Input[];
getCompilationInfo(input: SubtractInput): SubtractInfo;
getInputInfoString(input: SubtractInput): string;
}