| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | FullyConnected.cpp | 53 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
83 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
104 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
145 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
|
| D | RNN.cpp | 117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep()
138 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep()
|
| D | UnidirectionalSequenceRNN.cpp | 57 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped() local
|
| D | QuantizedLSTM.cpp | 381 Shape weightsShape; in eval() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | OptimizerTests.cpp | 424 const unsigned int* weightsShape, const unsigned int* outputShape, in CreateConvolution2dGraph()
462 const unsigned int weightsShape[] = { 2, 3, 5, 3 }; variable
473 const unsigned int weightsShape[] = { 2, 5, 3, 3 }; variable
481 … const unsigned int* weightsShape, const unsigned int* outputShape, in CreateDepthwiseConvolution2dGraph()
517 const unsigned int weightsShape[] = { 1, 3, 3, 2 }; variable
528 const unsigned int weightsShape[] = { 1, 3, 3, 2 }; variable
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | TransposeConvolution2dTestImpl.cpp | 243 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in SimpleTransposeConvolution2dTest() local
332 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in PaddedTransposeConvolution2dTest() local
423 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in StridedTransposeConvolution2dTest() local
503 TensorShape weightsShape = { 2, 1, 3, 3 }; in MultiChannelTransposeConvolution2dTest() local
|
| H A D | FullyConnectedTestImpl.cpp | 194 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedLargeTestCommon() local
294 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedFloat32Test() local
|
| /aosp_15_r20/external/android-nn-driver/1.0/ |
| H A D | FullyConnected.hpp | 16 const armnn::TensorShape& weightsShape) in FlattenFullyConnectedInput()
39 const armnn::TensorShape& weightsShape, in VerifyFullyConnectedShapes()
|
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | FoldPadTests.cpp | 23 const unsigned int weightsShape[] = {1, 2, 3, 3}; variable
105 const unsigned int weightsShape[] = {1, 2, 3, 3}; variable
646 const unsigned int weightsShape[] = {4, 2, 2, 3}; // CoutHWCin variable
770 const unsigned int weightsShape[] = {1, 2, 2, 12}; // 1HWCout variable
|
| H A D | PermuteDepthwiseConv2dWeightsTests.cpp | 66 unsigned int weightsShape[] = {outputShape[0], outputShape[1], outputShape[2], outputShape[3]}; in PermuteDepthwiseConv2dWeightsTestRunner() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | FullyConnectedEndToEndTestImpl.hpp | 235 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedWithDynamicOrConstantInputsEndToEnd() local
356 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedErrorChecking() local
|
| H A D | TransposeConvolution2dEndToEndTestImpl.hpp | 63 TensorShape weightsShape = MakeTensorShape(batches, channels, hWeights, wWeights, dataLayout); in TransposeConvolution2dEndToEnd() local
|
| H A D | WorkloadDataValidation.cpp | 150 unsigned int weightsShape[] = { 1, 1, inputChannels, outputChannels }; variable
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonDepthwiseConvolutionWorkload.cpp | 105 auto weightsShape = weightsInfo->tensor_shape(); in NeonDepthwiseConvolutionWorkload() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClDepthwiseConvolutionWorkload.cpp | 118 auto weightsShape = weightsInfo->tensor_shape(); in ClDepthwiseConvolutionWorkload() local
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | TransposeConvolution2d.cpp | 20 const TensorShape& weightsShape, in TransposeConvolution2dImpl()
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | ConversionUtils.hpp | 247 const armnn::TensorShape& weightsShape) in FlattenFullyConnectedInput()
270 const armnn::TensorShape& weightsShape, in VerifyFullyConnectedShapes()
|
| H A D | Converter.cpp | 2185 const TensorShape& weightsShape = weights.GetShape(); in ConvertGroupedConv2d() local
|
| /aosp_15_r20/external/armnn/src/armnn/optimizations/ |
| H A D | FuseBatchNorm.hpp | 65 auto weightsShape = weightsTensor.GetInfo().GetShape(); in Run() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadUtils.cpp | 215 auto weightsShape = weightsInfo.GetShape(); in Convert1HWOtoMIHW() local
|
| /aosp_15_r20/external/armnn/src/backends/aclCommon/ |
| H A D | ArmComputeTensorUtils.cpp | 443 const arm_compute::TensorShape& weightsShape, in ComputeDepthwiseConv2dDepthMultiplier()
|
| /aosp_15_r20/external/armnn/src/armnnDeserializer/ |
| H A D | Deserializer.cpp | 1455 auto weightsShape = weightsInfo.GetShape(); in ParseConstant() local
1688 auto weightsShape = weightsInfo.GetShape(); in ParseDepthwiseConvolution2d() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils_1_2.hpp | 1036 const TensorShape& weightsShape = weights.GetShape(); in ConvertGroupedConv2d() local
|