Home
last modified time | relevance | path

Searched defs:biasShape (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/armnn/delegate/test/
H A DConvolution2dTest.cpp27 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesFp32Test() local
81 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesInt8Test() local
124 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesReluUint8Test() local
182 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesRelu6Uint8Test() local
231 std::vector<int32_t> biasShape { 4 }; in Conv2DPerChannelInt8Test() local
H A DConvolution3dTest.cpp43 std::vector<int32_t> biasShape { 1 }; in Conv3DWithBiasesSimpleWithPaddingFp32Test() local
84 std::vector<int32_t> biasShape { 1 }; in Conv3DWithBiasesStridesFp32Test() local
143 std::vector<int32_t> biasShape { 2 }; in Conv3DWithBiasesDilationFp32Test() local
194 std::vector<int32_t> biasShape { 1 }; in Conv3DFp32SmallTest() local
H A DDepthwiseConvolution2dTest.cpp27 std::vector<int32_t> biasShape { 4 }; in DepthwiseConv2dValidReluFp32Test() local
89 std::vector<int32_t> biasShape { 1 } ; in DepthwiseConv2dSameUint8Test() local
136 std::vector<int32_t> biasShape {16} ; in DepthwiseConv2dSameInt8PerChannelTest() local
H A DConvolutionTestHelper.hpp212 const std::vector<int32_t>& biasShape = {}, in ConvolutionTest()
448 const std::vector<int32_t>& biasShape = {}, in Convolution3dTest()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp192 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
221 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
275 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
304 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
331 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
352 const Shape& biasShape, int32_t paddingLeft, int32_t /*paddingRight*/, in convQuant8PerChannelNhwc()
445 const Shape& biasShape, int32_t paddingLeft, int32_t /*paddingRight*/, in convQuant8PerChannelNhwc()
508 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft, in convQuant8PerChannel()
DDepthwiseConv2D.cpp130 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
165 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
241 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
275 const Shape& biasShape, int32_t paddingLeft, int32_t /*paddingRight*/, int32_t paddingTop, in depthwiseConvQuant8PerChannelNhwc()
370 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
393 const Shape& biasShape, int32_t paddingLeft, in depthwiseConvQuant8PerChannel()
DFullyConnected.cpp54 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32()
84 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16()
105 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8()
146 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8()
DTransposeConv2D.cpp180 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
275 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc()
298 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
316 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannelNhwc()
419 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannel()
DGroupedConv2D.cpp109 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
207 const Shape& biasShape, int32_t padding_left, in groupedConvQuant8PerChannel()
292 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16()
DQuantizedLSTM.cpp387 Shape biasShape; in eval() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DWorkloadDataValidation.cpp151 unsigned int biasShape[] = { 1, outputChannels, outputHeight, outputWidth }; variable
655 const TensorShape biasShape { cOutput }; variable
700 const TensorShape biasShape { cOutput }; variable
746 const TensorShape biasShape { cOutput }; variable
H A DFullyConnectedEndToEndTestImpl.hpp242 unsigned int biasShape[] = { outputChannels }; in FullyConnectedWithDynamicOrConstantInputsEndToEnd() local
358 unsigned int biasShape[] = { outputChannels }; in FullyConnectedErrorChecking() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DFullyConnectedTestImpl.cpp200 unsigned int biasShape[] = { outputChannels }; in FullyConnectedLargeTestCommon() local
301 unsigned int biasShape[] = { outputChannels }; in FullyConnectedFloat32Test() local
H A DConv2dTestImpl.cpp1388 unsigned int biasShape[] = {outputChannels}; in CompareConvolution2dTestImpl() local
3065 std::vector<unsigned int> biasShape{ outputChannels }; in CompareDepthwiseConvolution2dTestImpl() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DFullyConnected.cpp32 const TensorShape biasShape{outputSize}; in FullyConnected() local
H A DConv3dImpl.cpp60 const TensorShape biasShape{outChannels}; in Convolve3d() local
H A DConvImpl.cpp117 const TensorShape biasShape{outputChannels}; in Convolve() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp155 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultiplier()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp613 std::vector<ExprHandle> biasShape; in computeValue() local