Whisper/ML/Reshaper.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#pragma once
#include "Tensor.h"

namespace DirectCompute
{
	// This class reshapes some of the model’s tensor, immediately after they’re loaded.
	// That feature is used on all AMD GPUs.
	class Reshaper
	{
		CComPtr<ID3D11Buffer> constantBuffer;
		HRESULT createConstants();

	public:
		~Reshaper();
		HRESULT makePanels( Tensor& tensor, eDataType dataType );
	};
}