Searched refs:tensorRank (Results 1 – 2 of 2) sorted by relevance
402 uint32_t tensorRank = nnTensor.dimensionCount; in TransposeDepthwiseTensor() local405 …TF_LITE_ENSURE_STATUS(TransposeDims(m_context, tensorDims, tensorRank, DEPTHWISE_TRANSPOSE_AXISS, … in TransposeDepthwiseTensor()437 uint32_t tensorRank = static_cast<uint32_t>(tensor->dims->size); in ConstructNNTensor() local438 m_dimsUnspecified.assign(tensorRank, UNSPECIFIED_DIMENSION_VALUE); in ConstructNNTensor()446 if (scalarAsTensor && tensorRank == 0) { in ConstructNNTensor()447 … tensorRank = SCALAR_TENSOR_RANK; // Use rank 1, shape {1} nnTensor for TFLite scalar tensors. in ConstructNNTensor()451 if (tensorRank == 0) { in ConstructNNTensor()457 nnTensor.dimensionCount = tensorRank; in ConstructNNTensor()
299 uint32_t tensorRank = static_cast<uint32_t>(tensor->dims->size); in ConvertTensorTypeToNn() local308 if (tensorRank == 0) { in ConvertTensorTypeToNn()309 tensorRank = SCALAR_RANK; in ConvertTensorTypeToNn()314 nnTensor.dimensionCount = tensorRank; in ConvertTensorTypeToNn()