6 #include <boost/test/unit_test.hpp> 19 FullyConnectedFixture()
144 RunTest<1>({ 3 }, { 7 });
158 MatMulUsedInTwoFcFixture()
320 SetupSingleInputSingleOutput({ 1, 1 }, "input",
"output");
326 RunTest<1>({ 3 }, { 32 });
343 MatMulUsedInTwoFcStaggeredFixture()
475 SetupSingleInputSingleOutput({ 1, 1 }, "input",
"output");
481 RunTest<1>({ 2 }, { 75 });
570 SetupSingleInputSingleOutput({ 1, 1 }, "input",
"output");
576 RunTest<1>({ 2 }, { 34 });
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_FIXTURE_TEST_CASE(MatMul, MatMulFixture)
void FullyConnected(const TensorShape &rInputShape, Decoder< float > &rInputDecoder, const TensorShape &rOutputShape, Encoder< float > &rOutputEncoder, Decoder< float > &rWeightDecoder, Decoder< float > &rBiasDecoder, const bool biasEnabled, const unsigned int K, const bool transposeWeights)
Performs a matrix multiplication and optionally adds a bias.
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)