Home
last modified time | relevance | path

Searched refs:ValueShapeRange (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h164 class ValueShapeRange : public ValueRange::RangeBaseT {
168 ValueShapeRange(ValueRange values, ValueShapeMapFn operandShape = nullptr,
172 ValueShapeRange(const std::initializer_list<Value> &values) in ValueShapeRange() function
173 : ValueShapeRange(ValueRange(values)) {} in ValueShapeRange()
175 ValueShapeRange(const ValueShapeRange &) = default;
178 ValueShapeRange &setValueToShapeMapping(ValueShapeMapFn fn) { in setValueToShapeMapping()
183 ValueShapeRange &setOperandShapeMapping(ValueShapeMapFn fn) { in setOperandShapeMapping()
239 MLIRContext *, Optional<Location> location, ValueShapeRange operands,
H A DInferTypeOpInterface.td148 "::mlir::ValueShapeRange":$operands,
/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp43 ValueShapeRange addiRange() { in addiRange()
55 ValueShapeRange range = addiRange(); in TEST_F()
67 ValueShapeRange range = addiRange(); in TEST_F()
88 ValueShapeRange range = addiRange(); in TEST_F()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp344 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
369 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
426 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
458 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
485 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
541 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
556 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
572 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents()
873 static LogicalResult resolveBroadcastShape(const ValueShapeRange &operands, in REDUCE_SHAPE_INFER()
910 const ValueShapeRange &operands, in NAryInferReturnTypes()
[all …]
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp149 ShapeAdaptor ValueShapeRange::getValueAsShape(int index) { in getValueAsShape()
163 ShapeAdaptor ValueShapeRange::getShape(Value val) const { in getShape()
170 ShapeAdaptor ValueShapeRange::getShape(int index) const { in getShape()
178 MLIRContext *, Optional<Location> location, ValueShapeRange operands, in inferReturnTensorTypes()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp212 ValueShapeRange range(op.getOperands(), operandShape); in propagateShapesInRegion()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp1162 MLIRContext *context, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents()