Searched defs:FullyConnectedTest (Results 1 – 5 of 5) sorted by relevance
143 struct FullyConnectedTest struct146 using LayerType = FullyConnectedLayer;147 static const bool isElementWise = false;148 static const bool isConstTensorAsInputSupported = true;150 static TensorShape GetInputShape() { return TensorShape( {2, 5, 1, 1}); } // NCinHW in GetInputShape()151 static TensorShape GetOutputShape() { return TensorShape( {2, 3}); } // NCout in GetOutputShape()152 static TensorShape GetWeightsShape() { return TensorShape( {5, 3}); } // CinCout in GetWeightsShape()154 constexpr static const unsigned int inputSize = 10; // batchIn * heightIn * widthIn * channelIn155 …onstexpr static const unsigned int outputSize = 6; // batchOut * heightOut * widthOut * channelOut157 static IConnectableLayer* AddReceiverLayer(INetwork* network, in AddReceiverLayer()[all …]
34 TEST(FullyConnectedTest, MatrixByVectorMultiplication) { in TEST() argument
29 absl::Status FullyConnectedTest(TestExecutionEnvironment* env) { in FullyConnectedTest() function
169 void FullyConnectedTest(std::vector<armnn::BackendId>& backends, in FullyConnectedTest() function
88 LayerTestResult<T, 2> FullyConnectedTest( in FullyConnectedTest() function