Home
last modified time | relevance | path

Searched defs:inputShape (Results 226 – 250 of 309) sorted by relevance

12345678910>>...13

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSVDF.cpp97 const Shape& inputShape = input->shape(); in Prepare() local
DUnidirectionalSequenceLSTM.cpp90 const Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
DBatchMatmul.cpp85 void transposeRowsColumns(const T* inputData, const Shape& inputShape, T* outputData) { in transposeRowsColumns()
DBidirectionalSequenceLSTM.cpp382 const Shape& inputShape = input_->shape(); in Prepare() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DResize.cpp94 TensorShape inputShape = inputInfo.GetShape(); in Resize() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp66 TensorShape& inputShape = inputInfo.GetShape(); in Execute() local
H A DRefQLstmWorkload.cpp74 const TensorShape& inputShape = inputInfo.GetShape(); in Execute() local
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/customization/processors/
H A DShapeSubstitutionsProcessor.java108 String inputShape = operation.getInput().getShape(); in preprocessAssertNoSubstitutedShapeReferenceInOperation() local
/aosp_15_r20/external/armnn/delegate/test/
H A DPooling3dTestHelper.hpp144 std::vector<int32_t>& inputShape, in Pooling3dTest()
H A DLstmTestHelper.hpp77 std::vector<int32_t> inputShape {batchSize , inputSize}; in CreateLstmTfLiteModel() local
H A DUnidirectionalSequenceLstmTestHelper.hpp87 std::vector<int32_t> inputShape; in CreateUnidirectionalSequenceLstmTfLiteModel() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DTransposeConvolution2dEndToEndTestImpl.hpp61 TensorShape inputShape = MakeTensorShape(batches, channels, hInput, wInput, dataLayout); in TransposeConvolution2dEndToEnd() local
H A DQuantizedLstmEndToEndTestImpl.cpp27 armnn::INetworkPtr CreateQuantizedLstmNetwork(armnn::TensorShape& inputShape, in CreateQuantizedLstmNetwork()
H A DDynamicBackendTests.hpp1446 TensorShape inputShape { 1, 16, 16, 16 }; in CreateReferenceDynamicBackendTestImpl() local
1528 TensorShape inputShape { 1, 16, 16, 16 }; in CreateSampleDynamicBackendTestImpl() local
/aosp_15_r20/out/soong/.intermediates/libcore/core_oj_api_files/gen/22/libcore/ojluni/src/main/java/java/util/stream/
DFindOps.java143 public StreamShape inputShape() { in inputShape() method in FindOps.FindOp
/aosp_15_r20/out/soong/.intermediates/libcore/core_oj_api_files/gen/gensrcs/libcore/ojluni/src/main/java/java/util/stream/
DFindOps.java143 public StreamShape inputShape() { in inputShape() method in FindOps.FindOp
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DConcatLayer.cpp211 auto& inputShape = inputShapes[i]; in InferOutputShapes() local
/aosp_15_r20/libcore/ojluni/src/main/java/java/util/stream/
H A DFindOps.java143 public StreamShape inputShape() { in inputShape() method in FindOps.FindOp
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp1771 TensorShape inputShape; in CreateResizeBilinearWorkloadTest() local
2248 const armnn::TensorShape& inputShape, in CreatePreluWorkloadTest()
2319 const armnn::TensorShape& inputShape, in CreateStackWorkloadTest()
/aosp_15_r20/test/mlts/benchmark/native/
Dmulti_process_test.cpp191 std::vector<int> inputShape; in main() local
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.cpp444 const arm_compute::TensorShape& inputShape) in ComputeDepthwiseConv2dDepthMultiplier()
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_2.hpp1034 const TensorShape& inputShape = inputInfo.GetShape(); in ConvertGroupedConv2d() local
2214 const TensorShape& inputShape = inputInfo.GetShape(); in ConvertResize() local
2234 const TensorShape& inputShape = inputInfo.GetShape(); in ConvertResize() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DLogicalTestImpl.cpp26 const armnn::TensorShape& inputShape, in LogicalUnaryTestHelper()
H A DUnidirectionalSequenceLstmTestImpl.cpp27 const armnn::TensorShape& inputShape, in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
223 const armnn::TensorShape& inputShape, in UnidirectionalSequenceLstmLayerFloat32TestImpl()
412 const armnn::TensorShape& inputShape, in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
H A DTensorImpl.java253 int[] inputShape = computeShapeOf(input); in getInputShapeIfDifferent() local

12345678910>>...13