Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp564 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ArgMaxOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
589 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,RFFT2dOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
615 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,FFT2dOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
626 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ConcatOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
688 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ValueShapeRange operands,DictionaryAttr attributes,OpaqueProperties properties,RegionRange regions,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
710 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,FullyConnectedOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
741 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,MatMulOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
767 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,PadOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
829 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,SliceOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
854 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,TableOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
870 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,TileOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
922 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ReshapeOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1008 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,TransposeOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1152 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,GatherOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1177 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ResizeOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1216 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,ScatterOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1247 ReduceInferReturnTypes(ShapeAdaptor operandShape,Type inputType,IntegerAttr axis,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) ReduceInferReturnTypes() argument
1353 NAryInferReturnTypes(const ValueShapeRange & operands,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) NAryInferReturnTypes() argument
1451 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,Conv2DOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1513 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,Conv3DOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1585 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,AvgPool2dOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1595 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,MaxPool2dOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1605 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,DepthwiseConv2DOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1680 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,TransposeConv2DOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1744 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,IfOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
1788 inferReturnTypeComponents(MLIRContext * context,::std::optional<Location> location,WhileOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
[all...]
/llvm-project/mlir/test/lib/Dialect/Test/
H A DTestOpDefs.cpp241 inferReturnTypeComponents(MLIRContext * context,std::optional<Location> location,ValueShapeRange operands,DictionaryAttr attributes,OpaqueProperties properties,RegionRange regions,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument
977 inferReturnTypeComponents(MLIRContext * context,std::optional<Location> location,OpWithShapedTypeInferTypeAdaptorInterfaceOp::Adaptor adaptor,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) inferReturnTypeComponents() argument