Searched refs:targetDims (Results 1 – 3 of 3) sorted by relevance
373 int targetDims; in mergeArray() local377 targetDims = typeDims; in mergeArray()380 targetDims = thisDims; in mergeArray()385 return createArray(targetRoot, targetDims); in mergeArray()387 return createArray(OBJECT, targetDims); in mergeArray()
408 std::vector<int> targetDims; in ComputeReshapeInfo() local414 targetDims.push_back(static_cast<int>(inShape[static_cast<uint>(i)])); in ComputeReshapeInfo()418 targetDims.push_back(val); in ComputeReshapeInfo()422 std::vector<unsigned int> outDims(targetDims.begin(), targetDims.end()); in ComputeReshapeInfo()423 const auto stretchDim = std::find(targetDims.begin(), targetDims.end(), -1); in ComputeReshapeInfo()424 if (stretchDim != targetDims.end()) in ComputeReshapeInfo()426 if (std::find(std::next(stretchDim), targetDims.end(), -1) != targetDims.end()) in ComputeReshapeInfo()430 for(uint i = 0; i < targetDims.size() - 1; ++i) in ComputeReshapeInfo()432 ss << targetDims[i] << ", "; in ComputeReshapeInfo()434 ss << targetDims[targetDims.size() - 1] << " ]"; in ComputeReshapeInfo()[all …]
2671 …const auto targetDims = graph->layers()->Get(layerIndex)->layer_as_ReshapeLayer()->descriptor()->t… in ParseReshape() local2672 std::vector<uint32_t> outputDims(targetDims->begin(), targetDims->begin() + targetDims->size()); in ParseReshape()