yum-archive/TaSTT-Whisper

High-performance GPGPU inference of OpenAI's Whisper automatic speech recognition (ASR) model

git clone https://git.yummers.dev/yum-archive/TaSTT-Whisper

KonstantinSource codes8c4603c

master
branchesmasterall refs →
373 B17 linesraw
1#pragma once
2#include "Tensor.h"
3
4namespace DirectCompute
5{
6	// This class reshapes some of the model’s tensor, immediately after they’re loaded.
7	// That feature is used on all AMD GPUs.
8	class Reshaper
9	{
10		CComPtr<ID3D11Buffer> constantBuffer;
11		HRESULT createConstants();
12
13	public:
14		~Reshaper();
15		HRESULT makePanels( Tensor& tensor, eDataType dataType );
16	};
17}