Searched defs:inShape (Results 1 – 4 of 4) sorted by relevance
80 mlir::Value inShape = hlfir::genShape(loc, builder, array); genResultShape() local
54 auto inShape = toTransform.getType(); in applyElementWise() local
152 shardShape(const InShape & inShape,const MeshShape & meshShape,const SplitAxes & splitAxes,OutShape & outShape) shardShape() argument
1177 llvm::ArrayRef<int64_t> inShape = arrayTy.getShape(); verify() local [all...]