| /external/armnn/delegate/classic/src/ |
| D | Gather.hpp | 57 auto outputDimensions = outputTensorInfo.GetNumDimensions(); in VisitGatherOperator() local
|
| /external/armnn/src/armnn/layers/ |
| D | PreluLayer.cpp | 54 unsigned int outputDimensions = std::max(inputShapeDimensions, alphaShapeDimensions); in InferOutputShapes() local
|
| /external/android-nn-driver/test/ |
| 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
|
| 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
|
| /external/android-nn-driver/test/1.3/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_mhlo_to_thlo.cc | 122 Value outputDimensions = op.output_dimensions(); in matchAndRewrite() local
|
| D | broadcast_propagation.cc | 49 Value outputDimensions; member
|
| /external/android-nn-driver/ |
| D | ConversionUtils_1_2.hpp | 913 auto outputDimensions = outputInfo.GetNumDimensions(); in ConvertGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.cpp | 363 SmallVector<int64_t> outputDimensions(window.size()); in inferWindowOutputShape() local
2050 SmallVector<int64_t> outputDimensions = in inferConvolutionOpReturnShape() local
|
| /external/armnn/shim/sl/canonical/ |
| D | Converter.cpp | 2073 auto outputDimensions = outputInfo.GetNumDimensions(); in ConvertGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 420 SmallVector<int64_t> outputDimensions(window.size()); in inferWindowOutputShape() local
2345 SmallVector<int64_t> outputDimensions = in inferConvolutionOpReturnShape() local
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 4289 auto outputDimensions = outputTensorInfo.GetNumDimensions(); in ParseGather() local
|