| /external/armnn/src/backends/reference/workloads/ |
| D | RefElementwiseUnaryWorkload.cpp | 53 const TensorShape& inShape = inputInfo.GetShape(); in Execute() local
72 AbsFunction(inShape, outShape, *input, *output); in Execute()
77 CeilFunction(inShape, outShape, *input, *output); in Execute()
82 ExpFunction(inShape, outShape, *input, *output); in Execute()
87 LogFunction(inShape, outShape, *input, *output); in Execute()
92 NegFunction(inShape, outShape, *input, *output); in Execute()
97 RsqrtFunction(inShape, outShape, *input, *output); in Execute()
102 SinFunction(inShape, outShape, *input, *output); in Execute()
107 SqrtFunction(inShape, outShape, *input, *output); in Execute()
|
| D | Broadcast.cpp | 33 BroadcastLoop::BroadcastLoop(const TensorShape& inShape, const TensorShape& outShape) in BroadcastLoop() argument
44 m_DimData[j].m_Stride1 = (inShape[j] > 1) ? sIn : 0; in BroadcastLoop()
47 sIn *= inShape[j]; in BroadcastLoop()
|
| D | ElementwiseFunction.cpp | 34 ElementwiseUnaryFunction<Functor>::ElementwiseUnaryFunction(const TensorShape& inShape, in ElementwiseUnaryFunction() argument
39 BroadcastLoop(inShape, outShape).Unroll(Functor(), 0, inData, outData); in ElementwiseUnaryFunction()
54 LogicalUnaryFunction<Functor>::LogicalUnaryFunction(const TensorShape& inShape, in LogicalUnaryFunction() argument
59 BroadcastLoop(inShape, outShape).Unroll(Functor(), 0, inData, outData); in LogicalUnaryFunction()
|
| D | ElementwiseFunction.hpp | 34 ElementwiseUnaryFunction(const TensorShape& inShape,
60 LogicalUnaryFunction(const TensorShape& inShape,
|
| D | RefLogicalUnaryWorkload.cpp | 43 const TensorShape& inShape = inputInfo.GetShape(); in Execute() local
55 NotFunction(inShape, outShape, *input, *output); in Execute()
|
| D | Broadcast.hpp | 18 BroadcastLoop(const TensorShape& inShape, const TensorShape& outShape);
|
| /external/armnn/src/backends/tosaReference/test/ |
| D | TosaRefLayerSupportTests.cpp | 239 TensorShape inShape = {1,1,3,4}; variable
241 TensorInfo in(inShape, DataType::Float32);
265 TensorShape inShape = {1,1,3,4}; variable
267 TensorInfo in(inShape, DataType::Float32);
292 TensorShape inShape = {1,1,3,4}; variable
294 TensorInfo in(inShape, DataType::Signed64);
312 TensorShape inShape = {1,1,3,4}; variable
314 TensorInfo in(inShape, DataType::Float32);
335 TensorShape inShape = {3,4}; variable
337 TensorInfo in(inShape, DataType::Float32);
[all …]
|
| /external/armnn/src/armnn/layers/ |
| D | PermuteLayer.cpp | 41 const TensorShape& inShape = inputShapes[0]; in InferOutputShapes() local
42 return std::vector<TensorShape> ({armnnUtils::Permuted(inShape, m_Param.m_DimMappings)}); in InferOutputShapes()
|
| D | TransposeLayer.cpp | 41 const TensorShape& inShape = inputShapes[0]; in InferOutputShapes() local
42 …return std::vector<TensorShape> ({armnnUtils::TransposeTensorShape(inShape, m_Param.m_DimMappings)… in InferOutputShapes()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 404 const TensorShape& inShape, in ComputeReshapeInfo() argument
414 targetDims.push_back(static_cast<int>(inShape[static_cast<uint>(i)])); in ComputeReshapeInfo()
447 outDims[stretchIndex] = inShape.GetNumElements() / targetNumElements; in ComputeReshapeInfo()
|