Searched defs:weightShape (Results 1 – 6 of 6) 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/pytorch/android/pytorch_android/src/androidTest/java/org/pytorch/ |
| D | PytorchTestBase.java | 366 long[] weightShape = new long[] {3, 3, 1, 1}; in testChannelsLastConv2d() local
442 long[] weightShape = new long[] {3, 3, 1, 1, 1}; in testChannelsLastConv3d() local
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 1209 TensorShape weightShape({1, 4, 4, 2}); in CreateDepthwiseConvolution2dWorkloadTest() local
|