Elementwise.h File Reference

#include "Shape.h"

Go to the source code of this file.

Functions
bool	genericActivationPrepare (const Shape &input, Shape *output)

Function Documentation

genericActivationPrepare()

bool genericActivationPrepare	(	const Shape &	input,
		Shape *	output
	)

Definition at line 21 of file Elementwise.cpp.

{
  ASSERT(getNumberOfDimensions(input) <= 4);
  return SetShape(input, output);
}

References ASSERT, getNumberOfDimensions(), and SetShape().

Referenced by relu6Prepare(), reluPrepare(), and softmaxPrepare().