Searched refs:OperandKind (Results 1 – 5 of 5) sorted by relevance
407 class OperandKind(Enum): class425 kind: OperandKind, argument452 return (self.kind == OperandKind.SCALAR or453 self.kind == OperandKind.INPUT_TENSOR)456 return (self.kind == OperandKind.INPUT_TENSOR or457 self.kind == OperandKind.OUTPUT_TENSOR)460 return (self.kind == OperandKind.INDEX_ATTR or461 self.kind == OperandKind.UNARY_FN_ATTR or462 self.kind == OperandKind.BINARY_FN_ATTR or463 self.kind == OperandKind.TYPE_FN_ATTR)[all …]
45 if d.kind in [OperandKind.SCALAR, OperandKind.INPUT_TENSOR]48 d for d in all_arg_defs if d.kind == OperandKind.OUTPUT_TENSOR51 d for d in all_arg_defs if d.kind == OperandKind.INDEX_ATTR55 OperandKind.UNARY_FN_ATTR, OperandKind.BINARY_FN_ATTR,56 OperandKind.TYPE_FN_ATTR149 if attr_kind == OperandKind.UNARY_FN_ATTR:153 elif attr_kind == OperandKind.BINARY_FN_ATTR:188 if arg_def.operand_def.kind == OperandKind.SCALAR:
59 def kind(self) -> OperandKind:245 if operand_config.operand_def.kind == OperandKind.SCALAR:313 operand_def.kind == OperandKind.INDEX_ATTR):325 if operand_def.kind == OperandKind.INDEX_ATTR:
82 enum OperandKind { enum in __anon4f8b82ce0111::SystemZOperand91 OperandKind Kind;156 SystemZOperand(OperandKind kind, SMLoc startLoc, SMLoc endLoc) in SystemZOperand()
560 lang.OperandKind.OUTPUT_TENSOR)1551 name: str, kind: lang.OperandKind) -> lang.OperandDef: argument2145 opnd = lang.OperandDef(lang.OperandKind.INPUT_TENSOR, lang.T, dim_sym)