Searched refs:numDim (Results 1 – 7 of 7) sorted by relevance
166 numDim := numDims[i]168 if numDim < 0 {174 slice := (*[1 << 27]C.int64_t)(unsafe.Pointer(dim))[:numDim:numDim]
286 int32_t ConvertMaskToACLFormat(int32_t mask, int32_t numDim) in ConvertMaskToACLFormat() argument289 for (unsigned int i = 0; i < armnn::numeric_cast<unsigned int>(numDim); ++i) in ConvertMaskToACLFormat()294 reversedMask += (bit << std::max(numDim-(armnn::numeric_cast<int>(i)+1), 0)); in ConvertMaskToACLFormat()
206 int32_t ConvertMaskToACLFormat(int32_t mask, int32_t numDim);
236 unsigned int numDim = shape.GetNumDimensions(); in GetNumElementsAfter() local237 ARMNN_ASSERT(axis <= numDim - 1); in GetNumElementsAfter()239 for (unsigned int i = axis+1; i < numDim; i++) in GetNumElementsAfter()
879 unsigned int numDim = symbol.getArraySizes() in mergeErrorCheck() local884 arraysMatch = (numDim - firstDim) == (unitNumDim - unitFirstDim); in mergeErrorCheck()886 for (unsigned int i = 0; i < (numDim - firstDim) && arraysMatch; i++) { in mergeErrorCheck()
2300 … uint numDim = armnn::numeric_cast<uint>(m_TensorsInfo[node.input(1)].m_tensor->int64_data_size()); in ParseUnsqueeze() local2302 for(uint i = 0; i < numDim; i++) in ParseUnsqueeze()