Home
last modified time | relevance | path

Searched defs:inShape (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/tosaReference/test/
H A DTosaRefLayerSupportTests.cpp239 TensorShape inShape = {1,1,3,4}; variable
265 TensorShape inShape = {1,1,3,4}; variable
292 TensorShape inShape = {1,1,3,4}; variable
312 TensorShape inShape = {1,1,3,4}; variable
335 TensorShape inShape = {3,4}; variable
357 TensorShape inShape = {3,4}; variable
419 TensorShape inShape = {3,2,3}; variable
442 TensorShape inShape = {3,2,3}; variable
554 TensorShape inShape = { 1, 1, 5, 3 }; variable
575 TensorShape inShape = { 1, 1, 5, 3 }; variable
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DElementwiseFunction.cpp34 ElementwiseUnaryFunction<Functor>::ElementwiseUnaryFunction(const TensorShape& inShape, in ElementwiseUnaryFunction()
54 LogicalUnaryFunction<Functor>::LogicalUnaryFunction(const TensorShape& inShape, in LogicalUnaryFunction()
H A DRefLogicalUnaryWorkload.cpp43 const TensorShape& inShape = inputInfo.GetShape(); in Execute() local
H A DBroadcast.cpp33 BroadcastLoop::BroadcastLoop(const TensorShape& inShape, const TensorShape& outShape) in BroadcastLoop()
H A DRefElementwiseUnaryWorkload.cpp53 const TensorShape& inShape = inputInfo.GetShape(); in Execute() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DElementwise.cpp124 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt() local
134 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DTransposeLayer.cpp41 const TensorShape& inShape = inputShapes[0]; in InferOutputShapes() local
H A DPermuteLayer.cpp41 const TensorShape& inShape = inputShapes[0]; in InferOutputShapes() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp444 Shape inShape = from.shape(); in convertToNhwc() local
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp404 const TensorShape& inShape, in ComputeReshapeInfo()