Reshape.h File Reference

#include "Shape.h"
#include <cstdint>

Go to the source code of this file.

Functions
bool	reshapePrepare (const Shape &input, const int32_t *targetDims, const int32_t targetDimsSize, Shape *output)
bool	reshapeGeneric (const void *inputData, const Shape &inputShape, void *outputData, const Shape &outputShape)

Function Documentation

reshapeGeneric()

bool reshapeGeneric	(	const void *	inputData,
		const Shape &	inputShape,
		void *	outputData,
		const Shape &	outputShape
	)

Definition at line 67 of file Reshape.cpp.

{
  size_t count = sizeOfData(inputShape.type, inputShape.dimensions);
  memcpy(outputData, inputData, count);
  return true;
}

References Shape::dimensions, sizeOfData(), and Shape::type.

reshapePrepare()

bool reshapePrepare	(	const Shape &	input,
		const int32_t *	targetDims,
		const int32_t	targetDimsSize,
		Shape *	output
	)

Definition at line 24 of file Reshape.cpp.

{
  // Reshape allows one of the targetDims components to have the
  // special -1 value, meaning it will be calculated automatically based on the
  // input. Here we calculate what that dimension should be so that the number
  // of output elements in the same as the number of input elements.
  int32_t numInputElements = (int32_t)getNumberOfElements(input);
 
  std::vector<uint32_t> outDims(targetDimsSize);
  int32_t numOutputElements = 1;
  int32_t strechDim = -1;
  for (int32_t i = 0; i < targetDimsSize; ++i)
  {
    int32_t value = targetDims[i];
    if (value == -1)
    {
      ASSERT(strechDim == -1);
      strechDim = i;
    }
    else
    {
      numOutputElements *= value;
      outDims[i] = (uint32_t)value;
    }
  }
  if (strechDim != -1)
  {
    int32_t strechValue = numInputElements / numOutputElements;
    outDims[strechDim] = (uint32_t)strechValue;
    numOutputElements *= strechValue;
  }
 
  ASSERT(numInputElements == numOutputElements);
 
  output->type = input.type;
  output->dimensions = outDims;
  output->offset = input.offset;
  output->scale = input.scale;
 
  return true;
}

References ASSERT, and getNumberOfElements().