Lines Matching refs:reducedTensorInfo
356 auto reducedTensorInfo = input; in ComputeReductionTensorShape() local
357 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); \
433 inputTensorInfo = reducedTensorInfo; \