Home
last modified time | relevance | path

Searched defs:inferredShape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/layers/
H A DShapeLayer.cpp45 …auto inferredShape = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape… in ValidateTensorShapesFromInputs() local
H A DRankLayer.cpp44 const TensorShape inferredShape = TensorShape(Dimensionality::Scalar); in ValidateTensorShapesFromInputs() local
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeUtils.hpp399 const TensorShape inferredShape = TensorShape(outputRank, dimSizes.data()); in ComputeReductionTensorShape() local
/aosp_15_r20/external/armnn/src/armnn/
H A DLayer.cpp423 const TensorShape& inferredShape, in ValidateAndCopyShape()