Searched refs:reducedTensorInfo (Results 1 – 2 of 2) sorted by relevance
356 auto reducedTensorInfo = input; in ComputeReductionTensorShape() local357 unsigned int rank = reducedTensorInfo.GetNumDimensions(); in ComputeReductionTensorShape()368 else if (vAxis.size() > reducedTensorInfo.GetNumDimensions()) in ComputeReductionTensorShape()374 …outputRank = reducedTensorInfo.GetNumDimensions() - armnn::numeric_cast<unsigned int>(vAxis.size()… in ComputeReductionTensorShape()385 for (unsigned int i = 0; i < reducedTensorInfo.GetNumDimensions(); ++i) in ComputeReductionTensorShape()389 … dimSizes[outputIndex] = armnn::numeric_cast<unsigned int>(reducedTensorInfo.GetShape()[i]); in ComputeReductionTensorShape()400 reducedTensorInfo.SetShape(inferredShape); in ComputeReductionTensorShape()401 return reducedTensorInfo; in ComputeReductionTensorShape()414 const armnn::TensorInfo& reducedTensorInfo = \422 status = func(inputTensorInfo, reducedTensorInfo, newReduceDescriptor); \[all …]
297 const TensorInfo& reducedTensorInfo = ComputeReductionTensorShape(layerInfo, in ChainReduceLayers() local324 replacementLayer->GetOutputSlot(0).SetTensorInfo(reducedTensorInfo); in ChainReduceLayers()