| /aosp_15_r20/external/android-nn-driver/test/ |
| H A D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
387 void QuantizedLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QuantizedLstmTestImpl()
816 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgNoPeepholeNoProjection() local
1006 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjection() local
1719 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjection() local
1918 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjectionBatch2() local
2120 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2316 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
2455 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QuantizedLstm() local
|
| H A D | UnidirectionalSequenceLstm.hpp | 84 void UnidirectionalSequenceLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in UnidirectionalSequenceLstmTestImpl()
511 hidl_vec<uint32_t> outputDimensions{batchSize, timeSize, outputSize}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
716 hidl_vec<uint32_t> outputDimensions{timeSize, batchSize, outputSize}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
956 hidl_vec<uint32_t> outputDimensions{batchSize, timeSize, outputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1171 hidl_vec<uint32_t> outputDimensions{batchSize, timeSize, outputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1368 hidl_vec<uint32_t> outputDimensions{batchSize, timeSize, outputSize}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| /aosp_15_r20/external/android-nn-driver/test/1.3/ |
| H A D | QLstm.cpp | 62 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
634 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QLstmWithProjection() local
794 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QLstmWithNoProjection() local
955 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in DynamicOutputQLstmWithNoProjection() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 1048 uint32_t outputDimensions[3] = {2, 2, 2}; in argMinMaxTest() local
1107 uint32_t outputDimensions[5] = {2, 2, 2, 2, 2}; in expandDimsTest() local
1135 uint32_t outputDimensions[7] = {2, 2, 2, 2, 2, 2, 2}; in gatherTest() local
1241 uint32_t outputDimensions[2] = {2, 2}; in splitTest() local
1266 uint32_t outputDimensions[8] = {2, 2, 2, 2, 2, 2, 2, 2}; in tileTest() local
1285 uint32_t outputDimensions[4] = {4, 5, 6, 3}; in topkV2Test() local
1682 uint32_t outputDimensions[2] = {4, 6}; in reshapeOpTest() local
1776 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padOpTest() local
1817 uint32_t outputDimensions[4] = {4, 3, 4, 3}; in padV2OpTest() local
4458 uint32_t outputDimensions[3] = {3, 4, 5}; in fillTest() local
[all …]
|
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Gather.hpp | 57 auto outputDimensions = outputTensorInfo.GetNumDimensions(); in VisitGatherOperator() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | PreluLayer.cpp | 54 unsigned int outputDimensions = std::max(inputShapeDimensions, alphaShapeDimensions); in InferOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_mhlo_to_thlo.cc | 122 Value outputDimensions = op.output_dimensions(); in matchAndRewrite() local
|
| H A D | broadcast_propagation.cc | 49 Value outputDimensions; member
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionBuilder.h | 381 const Dimensions* outputDimensions) { in mapOutputToInput()
|
| /aosp_15_r20/external/skia/src/codec/ |
| H A D | SkJpegCodec.cpp | 263 SkISize outputDimensions = SkISize::Make(dinfo.output_width, dinfo.output_height); in onGetScaledDimensions() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils_1_2.hpp | 913 auto outputDimensions = outputInfo.GetNumDimensions(); in ConvertGather() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloOps.cpp | 363 SmallVector<int64_t> outputDimensions(window.size()); in inferWindowOutputShape() local
2050 SmallVector<int64_t> outputDimensions = in inferConvolutionOpReturnShape() local
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | Converter.cpp | 2073 auto outputDimensions = outputInfo.GetNumDimensions(); in ConvertGather() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | hlo_ops.cc | 420 SmallVector<int64_t> outputDimensions(window.size()); in inferWindowOutputShape() local
2345 SmallVector<int64_t> outputDimensions = in inferConvolutionOpReturnShape() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 4289 auto outputDimensions = outputTensorInfo.GetNumDimensions(); in ParseGather() local
|