Home
last modified time | relevance | path

Searched defs:outShape (Results 1 – 2 of 2) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp524 resolveBroadcastShape(const ValueShapeRange & operands,SmallVector<int64_t> & outShape) resolveBroadcastShape() argument
574 SmallVector<int64_t> outShape; inferReturnTypeComponents() local
691 llvm::SmallVector<int64_t> outShape; inferReturnTypeComponents() local
716 SmallVector<int64_t> outShape; inferReturnTypeComponents() local
746 SmallVector<int64_t> outShape; inferReturnTypeComponents() local
1354 llvm::SmallVector<int64_t> outShape; NAryInferReturnTypes() local
[all...]
/llvm-project/mlir/lib/Dialect/Mesh/IR/
H A DMeshOps.cpp153 shardShape(const InShape & inShape,const MeshShape & meshShape,const SplitAxes & splitAxes,OutShape & outShape) shardShape() argument