Home
last modified time | relevance | path

Searched defs:outputDesc (Results 1 – 7 of 7) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DConv3dTestImpl.cpp337 armnn::TensorInfo outputDesc({ 1, 3, 3, 3, 1 }, ArmnnType); in SimpleConvolution3d3x3x3TestCommon() local
411 armnn::TensorInfo outputDesc({ 1, 1, 3, 3, 1 }, ArmnnType); in Convolution3d2x2x2Strides3x5x5TestCommon() local
469 armnn::TensorInfo outputDesc({ 1, 2, 2, 2, 2 }, ArmnnType); in Convolution3d2x2x2Dilation2x2x2TestCommon() local
543 armnn::TensorInfo outputDesc({ 1, 5, 5, 5, 1 }, ArmnnType); in Convolution3dPaddingSame3x3x3TestCommon() local
625 armnn::TensorInfo outputDesc({ 1, 1, 4, 4, 2 }, armnn::DataType::Float32); in Convolution3dStrideDilationPadding3x3x3TestCommonFloat32() local
697 armnn::TensorInfo outputDesc({ 1, 1, 4, 4, 1 }, armnn::DataType::Float32); in Convolution3d2x2x2Stride3x3x3SmallTestCommonFloat32() local
769 armnn::TensorInfo outputDesc({ 1, 1, 2, 2, 2 }, armnn::DataType::Float16); in Convolution3d2x3x3TestCommonFloat16() local
832 armnn::TensorInfo outputDesc({ 1, 1, 3, 3, 1 }, armnn::DataType::Float16); in Convolution3d2x2x2SmallTestCommonFloat16() local
DLstmTestImpl.cpp2595 armnn::TensorInfo outputDesc({ 2, 4 }, armnn::DataType::Float32); in LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest() local
2614 armnn::TensorInfo outputDesc({ 2, 16 }, armnn::DataType::Float32); in LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest() local
2634 armnn::TensorInfo outputDesc({2, 4}, armnn::DataType::Float32); in LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest() local
2654 armnn::TensorInfo outputDesc({ 2, 3 }, armnn::DataType::Float32); in LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
2676 armnn::TensorInfo outputDesc({2, 4}, datatype); in LstmLayerInt16NoCifgNoPeepholeNoProjectionTest() local
2705 armnn::TensorInfo outputDesc({ 2, 4 }, datatype); in LstmLayerInt16WithCifgWithPeepholeNoProjectionTest() local
2738 armnn::TensorInfo outputDesc({ 2, 16 }, datatype); in LstmLayerInt16NoCifgWithPeepholeWithProjectionTest() local
2769 armnn::TensorInfo outputDesc({2, 4}, datatype); in LstmLayerInt16NoCifgNoPeepholeNoProjectionInt16ConstantTest() local
2795 armnn::TensorInfo outputDesc({2, 4}, armnn::DataType::QAsymmU8); in QuantizedLstmTest() local
2811 armnn::TensorInfo outputDesc({2, 4}, armnn::DataType::QAsymmS8); in QLstmTest() local
[all …]
DConv2dTestImpl.cpp662 armnn::TensorInfo outputDesc({ 1, 3, 4, 1 }, ArmnnType); in SimpleConvolution2d3x3NhwcTestCommon() local
719 armnn::TensorInfo outputDesc({ 1, 3, 3, 1 }, ArmnnType); in SimpleConvolution2d3x3Stride2x2TestCommon() local
813 armnn::TensorInfo outputDesc({ 1, 2, 4, 14 }, ArmnnType); in SimpleConvolution2d3x5TestCommon() local
894 armnn::TensorInfo outputDesc({ 1, 2, 6, 14 }, ArmnnType); in SimpleConvolution2d3x3TestCommon() local
966 armnn::TensorInfo outputDesc({ 1, 1, 8, 6 }, ArmnnType); in Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTestCommon() local
1033 armnn::TensorInfo outputDesc({ 1, 1, 5, 5 }, ArmnnType); in SimpleConvolution2dAsymmetricPaddingTestCommon() local
1531 armnn::TensorInfo outputDesc({ 1, 3, 3, 1 }, armnn::DataType::Float32); in Convolution2d3x3Stride2x2BFloat16Test() local
1638 armnn::TensorInfo outputDesc({1, 3, 3, 1}, armnn::DataType::Float32); in Convolution2d3x3Stride2x2BFloat16SmallValueTest() local
DUnidirectionalSequenceLstmTestImpl.cpp603 armnn::TensorInfo outputDesc({1, 2, 4}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerFloat32TimeMajorSingleBatchTest() local
/external/pytorch/torch/csrc/jit/codegen/fuser/
Dfused_kernel.h59 const std::vector<TensorDesc>& outputDesc() const { in outputDesc() function
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp196 armnn::TensorInfo outputDesc({2, 4}, armnn::DataType::QAsymmU8); in QuantizedLstmEndToEnd() local
/external/mesa3d/src/gallium/drivers/d3d12/
Dd3d12_video_dec.cpp626 CD3DX12_RESOURCE_DESC outputDesc(GetDesc(d3d12OutputArguments.pOutputTexture2D)); in d3d12_video_decoder_end_frame() local
1046 CD3DX12_RESOURCE_DESC outputDesc(GetDesc(*ppRefOnlyOutTexture2D)); in d3d12_video_decoder_prepare_for_decode_frame() local