Searched defs:loopIdx (Results 1 – 7 of 7) sorted by relevance
223 fillShardingOption(Operation * op,ShardingOption & shardingOption,FlatSymbolRefAttr mesh,ArrayRef<MeshAxis> meshAxes,unsigned loopIdx) fillShardingOption() argument 301 for (size_t loopIdx = 0; loopIdx < loopTypes.size(); ++loopIdx) { defaultGetShardingOption() local 335 unsigned loopIdx = *loopIndices->begin(); defaultGetShardingOption() local 406 unsigned loopIdx = dim.getPosition(); getShardingAttribute() local [all...]
133 unsigned loopIdx = 0; in emitLoopNestBody() local
296 unsigned loopIdx = 0; in getExtractSliceParams() local 339 unsigned loopIdx = 0; in getInsertSliceParams() local
188 for (int64_t loopIdx : perm) { in computeOuterDims() local
385 bool isZero = !overflow && isConstantIntValue(numThreads[loopIdx], 0); in calculateTileOffsetsAndSizes() local [all...]
378 // Compute induction variable for `loopIdx`. in createParallelComputeFunction() argument [all...]
781 auto loopIdx = loop.getInductionVar(); matchAndRewrite() local