Searched defs:outputDim (Results 1 – 6 of 6) sorted by relevance
74 const auto outputDim = static_cast<unsigned int>(output_shape.size()); in InferOutputShapes() local
47 const unsigned int outputDim = paramsDim - 1 + indicesDim; in InferOutputShapes() local
78 if (int outputDim = operandDimsToOutputDims[i]; outputDim >= 0) { in buildPointOp() local111 if (int outputDim = operandDimsToOutputDims[i]; outputDim >= 0) { in buildTileOp() local
142 unsigned int outputDim = tensorShape.GetNumDimensions() + 1; in ExpandDims() local
2405 unsigned int outputDim = in Validate() local2693 unsigned int outputDim = outputTensorInfo.GetNumDimensions(); in Validate() local2728 …unsigned int outputDim = inputTensorInfo.GetNumDimensions() + indicesTensorInfo.GetNumDimensions(… in Validate() local
1933 …auto outputDim = input.GetNumDimensions() - armnn::numeric_cast<unsigned int>(descriptor.m_Axis.si… in IsMeanSupported() local