mirror of https://github.com/llvm/torch-mlir
emit Torch_AtenAnyDimsOp
parent
3180704b14
commit
2e0f7efe82
|
@ -10120,6 +10120,31 @@ def Torch_AtenAnyDimOp : Torch_Op<"aten.any.dim", [
|
||||||
}];
|
}];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
def Torch_AtenAnyDimsOp : Torch_Op<"aten.any.dims", [
|
||||||
|
AllowsTypeRefinement,
|
||||||
|
HasValueSemantics,
|
||||||
|
ReadOnly
|
||||||
|
]> {
|
||||||
|
let summary = "Generated op for `aten::any.dims : (Tensor, int[]?, bool) -> (Tensor)`";
|
||||||
|
let arguments = (ins
|
||||||
|
AnyTorchTensorType:$self,
|
||||||
|
AnyTorchOptionalListOfTorchIntType:$dim,
|
||||||
|
Torch_BoolType:$keepdim
|
||||||
|
);
|
||||||
|
let results = (outs
|
||||||
|
AnyTorchOptionalTensorType:$result
|
||||||
|
);
|
||||||
|
let hasCustomAssemblyFormat = 1;
|
||||||
|
let extraClassDefinition = [{
|
||||||
|
ParseResult AtenAnyDimsOp::parse(OpAsmParser &parser, OperationState &result) {
|
||||||
|
return parseDefaultTorchOp(parser, result, 3, 1);
|
||||||
|
}
|
||||||
|
void AtenAnyDimsOp::print(OpAsmPrinter &printer) {
|
||||||
|
printDefaultTorchOp(printer, *this, 3, 1);
|
||||||
|
}
|
||||||
|
}];
|
||||||
|
}
|
||||||
|
|
||||||
def Torch_AtenArangeOp : Torch_Op<"aten.arange", [
|
def Torch_AtenArangeOp : Torch_Op<"aten.arange", [
|
||||||
AllowsTypeRefinement,
|
AllowsTypeRefinement,
|
||||||
HasValueSemantics,
|
HasValueSemantics,
|
||||||
|
|
|
@ -785,6 +785,7 @@ def emit_ops(emitter_td: TextEmitter, registry: Registry):
|
||||||
emit("aten::all.dim : (Tensor, int, bool) -> (Tensor)")
|
emit("aten::all.dim : (Tensor, int, bool) -> (Tensor)")
|
||||||
emit("aten::any : (Tensor) -> (Tensor)")
|
emit("aten::any : (Tensor) -> (Tensor)")
|
||||||
emit("aten::any.dim : (Tensor, int, bool) -> (Tensor)")
|
emit("aten::any.dim : (Tensor, int, bool) -> (Tensor)")
|
||||||
|
emit("aten::any.dims : (Tensor, int[]?, bool) -> (Tensor)")
|
||||||
emit("aten::arange : (Scalar, int?, int?, Device?, bool?) -> (Tensor)")
|
emit("aten::arange : (Scalar, int?, int?, Device?, bool?) -> (Tensor)")
|
||||||
emit(
|
emit(
|
||||||
"aten::arange.start : (Scalar, Scalar, int?, int?, Device?, bool?) -> (Tensor)"
|
"aten::arange.start : (Scalar, Scalar, int?, int?, Device?, bool?) -> (Tensor)"
|
||||||
|
|
Loading…
Reference in New Issue