Searched refs:expandedDimToCollapsedDim (Results 1 – 2 of 2) sorted by relevance
23 llvm::DenseMap<int64_t, int64_t> expandedDimToCollapsedDim; in getExpandedDimToCollapsedDimMap() local30 expandedDimToCollapsedDim[dim] = map.index(); in getExpandedDimToCollapsedDimMap()33 return expandedDimToCollapsedDim; in getExpandedDimToCollapsedDimMap()77 llvm::DenseMap<int64_t, int64_t> &expandedDimToCollapsedDim) { in getExpandedOutputDimFromInputShape() argument81 unsigned sourceDimPos = expandedDimToCollapsedDim[dimIndex]; in getExpandedOutputDimFromInputShape()115 llvm::DenseMap<int64_t, int64_t> expandedDimToCollapsedDim = in getExpandedOutputShapeFromInputShape() local121 expandedDimToCollapsedDim); in getExpandedOutputShapeFromInputShape()
1339 llvm::DenseMap<int64_t, int64_t> expandedDimToCollapsedDim; in getExpandedDimToCollapsedDimMap() local1342 expandedDimToCollapsedDim[dim] = en.index(); in getExpandedDimToCollapsedDimMap()1344 return expandedDimToCollapsedDim; in getExpandedDimToCollapsedDimMap()