Searched refs:ComputeReductionTensorShape (Results 1 – 2 of 2) sorted by relevance
352 inline const TensorInfo ComputeReductionTensorShape(const armnn::TensorInfo& input, in ComputeReductionTensorShape() function415 ComputeReductionTensorShape(input, axes, desc.m_KeepDims); \
297 const TensorInfo& reducedTensorInfo = ComputeReductionTensorShape(layerInfo, in ChainReduceLayers()