#include <FullyConnected.h>

Public Member Functions
	FCTempArena (void)

void	prepare (const Shape &input_shape, const Shape &weights_shape)

Data Fields
bool	prepared

std::vector< int8_t >	input_quantized

std::vector< float >	scaling_factors

std::vector< int32_t >	accum_scratch

Detailed Description

Definition at line 36 of file FullyConnected.h.

Constructor & Destructor Documentation

◆ FCTempArena()

nnfw::cker::FCTempArena::FCTempArena ( void )

inline

Definition at line 39 of file FullyConnected.h.

                    : prepared(false), input_quantized(), scaling_factors(), accum_scratch()
  {
    // DO NOTHING
  }

Member Function Documentation

◆ prepare()

void nnfw::cker::FCTempArena::prepare	(	const Shape &	input_shape,
		const Shape &	weights_shape
	)

inline

Definition at line 44 of file FullyConnected.h.

  {
    auto input_size = input_shape.FlatSize();
    input_quantized.resize(input_size);
 
    assert(weights_shape.DimensionsCount() == 2);
    int batch_size = input_size / weights_shape.Dims(1);
    scaling_factors.resize(batch_size);
    prepared = true;
  }