Home
last modified time | relevance | path

Searched defs:inputShape (Results 101 – 125 of 309) sorted by relevance

12345678910>>...13

/aosp_15_r20/external/armnn/delegate/test/
H A DFillTestHelper.hpp27 const std::vector<int32_t>& inputShape, in CreateFillTfLiteModel()
113 void FillTest(tflite::BuiltinOperator fillOperatorCode, in FillTest()
H A DElementwiseUnaryTestHelper.hpp84 std::vector<int32_t> inputShape { { 3, 1, 2} }; in ElementwiseUnaryFP32Test() local
114 std::vector<int32_t>& inputShape, in ElementwiseUnaryBoolTest()
H A DPreluTestHelper.hpp26 const std::vector<int32_t>& inputShape, in CreatePreluTfLiteModel()
119 const std::vector<int32_t>& inputShape, in PreluTest()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSlice.cpp45 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric()
78 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
DInstanceNormalization.cpp39 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
82 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
DStridedSlice.cpp47 bool compute(const T* inputData, const Shape& inputShape, const int32_t* beginData, in compute()
99 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
DTile.cpp69 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
90 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval()
DRoiAlign.cpp47 inline bool roiAlignNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiAlignNhwc()
172 inline bool roiAlignQuantNhwc(const T_Input* inputData, const Shape& inputShape, in roiAlignQuantNhwc()
305 inline bool roiAlign(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiAlign()
DUnidirectionalSequenceRNN.cpp37 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
55 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
DReduce.cpp46 const Shape inputShape = context->getInputShape(kInputTensor); in compute() local
66 Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
DMirrorPad.cpp41 const Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
187 const Shape inputShape = context->getInputShape(kInputTensor); in eval() local
DBidirectionalSequenceRNN.cpp36 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
94 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DStridedSlice.cpp63 TensorShape ExtendShape(const TensorShape& inputShape, in ExtendShape()
99 const TensorShape inputShape = ExtendShape(inputInfo.GetShape(), 4); in StridedSlice() local
/aosp_15_r20/external/armnn/src/backends/tosaCommon/test/
H A DTosaTestUtils.hpp21 std::vector<int32_t> inputShape, in VerifyTosaAttribute()
176 std::vector<std::vector<int32_t>> inputShape, in AssertTosaOneToOneMappingBasicBlock()
/aosp_15_r20/external/pytorch/android/pytorch_android/src/androidTest/java/org/pytorch/
H A DPytorchTestBase.java317 long[] inputShape = new long[] {1, 3, 2, 2}; in testChannelsLast() local
350 long[] inputShape = new long[] {1, 3, 2, 2}; in testChannelsLastConv2d() local
410 long[] inputShape = new long[] {1, 3, 2, 2, 2}; in testChannelsLastConv3d() local
/aosp_15_r20/out/soong/.intermediates/libcore/core_oj_api_files/gen/gensrcs/libcore/ojluni/src/main/java/java/util/stream/
DMatchOps.java197 private final StreamShape inputShape; field in MatchOps.MatchOp
223 public StreamShape inputShape() { in inputShape() method in MatchOps.MatchOp
/aosp_15_r20/out/soong/.intermediates/libcore/core_oj_api_files/gen/22/libcore/ojluni/src/main/java/java/util/stream/
DMatchOps.java197 private final StreamShape inputShape; field in MatchOps.MatchOp
223 public StreamShape inputShape() { in inputShape() method in MatchOps.MatchOp
/aosp_15_r20/libcore/ojluni/src/main/java/java/util/stream/
H A DMatchOps.java197 private final StreamShape inputShape; field in MatchOps.MatchOp
223 public StreamShape inputShape() { in inputShape() method in MatchOps.MatchOp
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DSpaceToDepthTestImpl.cpp99 unsigned int inputShape[] = {1, 2, 2, 1}; in SpaceToDepthSimpleTest1() local
134 unsigned int inputShape[] = {1, 2, 2, 2}; in SpaceToDepthSimpleTest2() local
H A DFullyConnectedTestImpl.cpp192 unsigned int inputShape[] = { inputNum, inputChannels, inputHeight, inputWidth }; in FullyConnectedLargeTestCommon() local
292 unsigned int inputShape[] = { inputNum, inputChannels, inputHeight, inputWidth }; in FullyConnectedFloat32Test() local
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp1927 TensorShape inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseFlatten() local
1975 const TensorShape& inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseGather() local
2163 TensorShape inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseGlobalAveragePool() local
2197 TensorShape inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseShape() local
2220 TensorShape inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseReshape() local
2287 TensorShape inputShape = m_TensorsInfo[node.input(0)].m_info->GetShape(); in ParseUnsqueeze() local
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonCreateWorkloadTests.cpp324 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in NeonCreateDepthWiseConvolutionWorkloadTest() local
479 static void NeonCreatePreluWorkloadTest(const armnn::TensorShape& inputShape, in NeonCreatePreluWorkloadTest()
955 static void NeonCreateStackWorkloadTest(const std::initializer_list<unsigned int>& inputShape, in NeonCreateStackWorkloadTest()
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/model/intermediate/
H A DOperationModel.java70 private ShapeModel inputShape; field in OperationModel
168 public void setInputShape(ShapeModel inputShape) { in setInputShape()
/aosp_15_r20/external/armnn/src/armnn/
H A DDescriptors.cpp366 int StridedSliceDescriptor::GetStartForAxis(const TensorShape& inputShape, in GetStartForAxis()
393 int StridedSliceDescriptor::GetStopForAxis(const TensorShape& inputShape, in GetStopForAxis()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp684 auto inputShape = inputTensorInfo.GetShape(); in Validate() local
952 const TensorShape& inputShape = m_Parameters.m_InputShape; in Validate() local
1832 const TensorShape& inputShape = inputTensorInfo.GetShape(); in Validate() local
1913 const TensorShape& inputShape = inputTensorInfo.GetShape(); in Validate() local
3580 const TensorShape& inputShape = inputTensorInfo.GetShape(); in Validate() local
3636 const TensorShape& inputShape = inputInfo.GetShape(); in Validate() local

12345678910>>...13