Searched refs:outDimToInDimMap (Results 1 – 1 of 1) sorted by relevance
1353 DenseMap<int64_t, int64_t> &outDimToInDimMap) { in getExpandedOutputDimSize() argument1360 int64_t inDimIndex = outDimToInDimMap[outDimIndex]; in getExpandedOutputDimSize()1428 DenseMap<int64_t, int64_t> outDimToInDimMap = in getExpandedOutputShape() local1434 reassociation, outDimToInDimMap); in getExpandedOutputShape()