Searched defs:weightShape (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/armnn/layers/ |
| D | FullyConnectedLayer.cpp | 39 const TensorShape weightShape = inputShapes[1]; in InferOutputShapes() local
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.cpp | 44 const TensorShape& weightShape = weightInfo.GetShape(); in ReshapeWeightsForAcl() local
70 const TensorShape& weightShape = weightHandle.GetShape(); in ReorderWeightChannelsForAcl() local
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 654 const TensorShape weightShape{ cOutput, cInput, hInput, wInput }; variable
699 const TensorShape weightShape{ cOutput, cInput, hInput, wInput }; variable
745 const TensorShape weightShape{ cOutput, cInput, hInput, wInput }; variable
|
| D | DynamicBackendTests.hpp | 1448 TensorShape weightShape{ 16, 1, 1, 16 }; in CreateReferenceDynamicBackendTestImpl() local
1530 TensorShape weightShape{ 16, 1, 1, 16 }; in CreateSampleDynamicBackendTestImpl() local
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 1209 TensorShape weightShape({1, 4, 4, 2}); in CreateDepthwiseConvolution2dWorkloadTest() local
|