@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
16 lines (15 loc) • 984 B
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { UnaryOperation } from './unaryOperation';
export declare class TanOperation<GPUTensor extends GPUTensorI> extends UnaryOperation<GPUTensor> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
operation(input: string): string;
}
export declare class ATanOperation<GPUTensor extends GPUTensorI> extends UnaryOperation<GPUTensor> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
operation(input: string): string;
}
export declare class TanHOperation<GPUTensor extends GPUTensorI> extends UnaryOperation<GPUTensor> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
operation(input: string): string;
}