| /external/pytorch/torch/csrc/lazy/core/ |
| D | ir.h | 35 struct TORCH_API OpKind { struct
36 OpKind() = default;
37 explicit OpKind(c10::Symbol op) : op(op) {} in OpKind() argument
39 bool operator==(const OpKind& rhs) const {
42 bool operator!=(const OpKind& rhs) const {
45 bool operator<(const OpKind& rhs) const {
58 static OpKind Get(const std::string& name); argument
63 inline std::ostream& operator<<(std::ostream& stream, const OpKind& op) { argument
90 Node(OpKind op, size_t num_outputs);
94 OpKind op,
[all …]
|
| D | ir.cpp | 51 OpKind OpKind::Get(const std::string& name) { in Get()
52 return OpKind(c10::Symbol::fromQualString(name)); in Get()
55 hash_t OpKind::hash() const { in hash()
64 Node::Node(OpKind op, size_t num_outputs) in Node()
68 OpKind op, in Node()
93 OpKind op, in Node()
101 Node::Node(OpKind op, OpList operands, size_t num_outputs) in Node()
104 Node::Node(OpKind op, Shape shape, size_t num_outputs) : Node(op, num_outputs) { in Node()
|
| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | graph_helper.cpp | 13 using opkind = dnnl::graph::op::kind; typedef
40 auto o = Operator(node, opkind::Wildcard); in makeWildcardOp()
59 Operator LlgaGraphHelper::makeEltwiseOp(Node* node, opkind kind) { in makeEltwiseOp()
63 Operator LlgaGraphHelper::makeBinaryOp(Node* node, opkind kind) { in makeBinaryOp()
82 return Operator(node, opkind::Convolution) in createOperator()
97 return Operator(node, opkind::Convolution) in createOperator()
111 return Operator(node, opkind::BatchNormInference) in createOperator()
120 return Operator(node, opkind::LayerNorm) in createOperator()
130 return Operator(node, opkind::MatMul) in createOperator()
134 return Operator(node, opkind::MatMul) in createOperator()
[all …]
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ops/ |
| D | generic.h | 11 // specific OpKind and a lowering function. IR nodes carrying
18 OpKind op,
25 OpKind op,
32 OpKind op,
37 Generic(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed);
44 OpKind op,
|
| D | generic.cpp | 7 OpKind op, in Generic()
16 OpKind op, in Generic()
25 OpKind op, in Generic()
31 Generic::Generic(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed) in Generic()
|
| /external/tensorflow/tensorflow/core/transforms/remapper/ |
| D | pass.cc | 103 enum class OpKind { Relu, Relu6, Elu, LeakyRelu, Tanh, Sigmoid }; enum
105 inline std::string GetTfgOpName(OpKind op_kind) { in GetTfgOpName()
107 case OpKind::Relu: in GetTfgOpName()
109 case OpKind::Relu6: in GetTfgOpName()
111 case OpKind::Elu: in GetTfgOpName()
113 case OpKind::LeakyRelu: in GetTfgOpName()
115 case OpKind::Tanh: in GetTfgOpName()
117 case OpKind::Sigmoid: in GetTfgOpName()
228 template <typename BasePatternRewriter, OpKind activation>
307 template <template <OpKind> class PatternT, OpKind... op_kinds,
[all …]
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_node.h | 19 OpKind op,
26 OpKind op,
33 OpKind op,
39 OpKind op,
68 // Note: this OpKind is separate from ltc_ops.h since it would be a circular
71 const OpKind tensor_list_opkind = OpKind::Get("lazy_tensors::tensor_list");
89 static OpKind ClassOpKind() { in ClassOpKind()
|
| D | ts_node.cpp | 40 OpKind op, in TsNode()
55 OpKind op, in TsNode()
64 TsNode::TsNode(OpKind op, OpList operands, size_t num_outputs, hash_t hash_seed) in TsNode()
67 TsNode::TsNode(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed) in TsNode()
|
| D | dynamic_ir.cpp | 27 OpKind{c10::Symbol::fromQualString("aten::size")}, in SizeNode()
52 OpKind{c10::Symbol::fromQualString("aten::add")}, in SizeAdd()
72 OpKind{c10::Symbol::fromQualString("aten::mul")}, in SizeMul()
92 OpKind{c10::Symbol::fromQualString("aten::div")}, in SizeDiv()
|
| /external/swiftshader/third_party/llvm-16.0/configs/common/lib/Target/RISCV/ |
| D | RISCVGenAsmMatcher.inc | 2696 MatchClassKind OpKind;
2698 default: OpKind = InvalidMatchClass; break;
2699 case RISCV::X0: OpKind = MCK_GPRX0; break;
2700 case RISCV::X1: OpKind = MCK_GPRNoX0X2; break;
2701 case RISCV::X2: OpKind = MCK_SP; break;
2702 case RISCV::X3: OpKind = MCK_GPRNoX0X2; break;
2703 case RISCV::X4: OpKind = MCK_GPRNoX0X2; break;
2704 case RISCV::X5: OpKind = MCK_GPRNoX0X2; break;
2705 case RISCV::X6: OpKind = MCK_GPRTC; break;
2706 case RISCV::X7: OpKind = MCK_GPRTC; break;
[all …]
|
| /external/swiftshader/third_party/llvm-16.0/configs/common/lib/Target/PowerPC/ |
| D | PPCGenAsmMatcher.inc | 4300 MatchClassKind OpKind;
4302 default: OpKind = InvalidMatchClass; break;
4303 case PPC::R0: OpKind = MCK_GPRC; break;
4304 case PPC::R1: OpKind = MCK_Reg2; break;
4305 case PPC::R2: OpKind = MCK_Reg2; break;
4306 case PPC::R3: OpKind = MCK_Reg2; break;
4307 case PPC::R4: OpKind = MCK_Reg2; break;
4308 case PPC::R5: OpKind = MCK_Reg2; break;
4309 case PPC::R6: OpKind = MCK_Reg2; break;
4310 case PPC::R7: OpKind = MCK_Reg2; break;
[all …]
|
| /external/swiftshader/third_party/llvm-16.0/configs/common/lib/Target/LoongArch/ |
| D | LoongArchGenAsmMatcher.inc | 1160 MatchClassKind OpKind;
1162 default: OpKind = InvalidMatchClass; break;
1163 case LoongArch::R0: OpKind = MCK_GPR; break;
1164 case LoongArch::R1: OpKind = MCK_GPR; break;
1165 case LoongArch::R2: OpKind = MCK_GPR; break;
1166 case LoongArch::R3: OpKind = MCK_GPR; break;
1167 case LoongArch::R4: OpKind = MCK_GPRT; break;
1168 case LoongArch::R5: OpKind = MCK_GPRT; break;
1169 case LoongArch::R6: OpKind = MCK_GPRT; break;
1170 case LoongArch::R7: OpKind = MCK_GPRT; break;
[all …]
|
| /external/pytorch/torch/csrc/lazy/core/internal_ops/ |
| D | ltc_ops.h | 17 const OpKind& operator*() const {
21 operator OpKind() const { in OpKind() function
26 const OpKind& get() const { in get()
27 c10::call_once(once_, [this]() { op_kind_ = OpKind::Get(name_); }); in get()
32 mutable OpKind op_kind_;
|
| /external/swiftshader/third_party/llvm-10.0/configs/common/lib/Target/PowerPC/ |
| D | PPCGenAsmMatcher.inc | 3691 MatchClassKind OpKind;
3693 default: OpKind = InvalidMatchClass; break;
3694 case PPC::R0: OpKind = MCK_GPRC; break;
3695 case PPC::R1: OpKind = MCK_Reg2; break;
3696 case PPC::R2: OpKind = MCK_Reg2; break;
3697 case PPC::R3: OpKind = MCK_Reg2; break;
3698 case PPC::R4: OpKind = MCK_Reg2; break;
3699 case PPC::R5: OpKind = MCK_Reg2; break;
3700 case PPC::R6: OpKind = MCK_Reg2; break;
3701 case PPC::R7: OpKind = MCK_Reg2; break;
[all …]
|
| /external/swiftshader/third_party/llvm-10.0/configs/common/lib/Target/RISCV/ |
| D | RISCVGenAsmMatcher.inc | 1575 MatchClassKind OpKind;
1577 default: OpKind = InvalidMatchClass; break;
1578 case RISCV::X0: OpKind = MCK_GPRX0; break;
1579 case RISCV::X1: OpKind = MCK_GPRNoX0X2; break;
1580 case RISCV::X2: OpKind = MCK_SP; break;
1581 case RISCV::X3: OpKind = MCK_GPRNoX0X2; break;
1582 case RISCV::X4: OpKind = MCK_GPRNoX0X2; break;
1583 case RISCV::X5: OpKind = MCK_GPRTC; break;
1584 case RISCV::X6: OpKind = MCK_GPRTC; break;
1585 case RISCV::X7: OpKind = MCK_GPRTC; break;
[all …]
|
| /external/pytorch/torch/csrc/lazy/core/ops/ |
| D | arithmetic_ir_ops.cpp | 19 OpKind(at::aten::add), in operator +()
26 OpKind(at::aten::sub), in operator -()
33 OpKind(at::aten::mul), in operator *()
40 OpKind(at::aten::div), in operator /()
|
| /external/llvm/include/llvm/CodeGen/ |
| D | MachineOperand.h | 67 /// OpKind - Specify what kind of operand this is. This discriminates the
69 MachineOperandType OpKind : 8;
185 : OpKind(K), SubReg_TargetFlags(0), ParentMI(nullptr) {} in MachineOperand()
189 MachineOperandType getType() const { return (MachineOperandType)OpKind; } in getType()
230 bool isReg() const { return OpKind == MO_Register; } in isReg()
232 bool isImm() const { return OpKind == MO_Immediate; } in isImm()
234 bool isCImm() const { return OpKind == MO_CImmediate; } in isCImm()
236 bool isFPImm() const { return OpKind == MO_FPImmediate; } in isFPImm()
238 bool isMBB() const { return OpKind == MO_MachineBasicBlock; } in isMBB()
240 bool isFI() const { return OpKind == MO_FrameIndex; } in isFI()
[all …]
|
| /external/pytorch/test/cpp/lazy/ |
| D | test_ir.cpp | 21 static OpKind ClassOpKind() { in ClassOpKind()
22 return OpKind(); in ClassOpKind()
100 OpKind(at::aten::view), in TEST()
105 OpKind(at::aten::view), in TEST()
121 OpKind(at::aten::view), in TEST()
152 OpKind(at::aten::view), in TEST()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/CodeGen/ |
| D | MachineOperand.h | 77 /// OpKind - Specify what kind of operand this is. This discriminates the
79 unsigned OpKind : 8;
201 : OpKind(K), SubReg_TargetFlags(0), ParentMI(nullptr) { in MachineOperand()
217 MachineOperandType getType() const { return (MachineOperandType)OpKind; } in getType()
319 bool isReg() const { return OpKind == MO_Register; } in isReg()
321 bool isImm() const { return OpKind == MO_Immediate; } in isImm()
323 bool isCImm() const { return OpKind == MO_CImmediate; } in isCImm()
325 bool isFPImm() const { return OpKind == MO_FPImmediate; } in isFPImm()
327 bool isMBB() const { return OpKind == MO_MachineBasicBlock; } in isMBB()
329 bool isFI() const { return OpKind == MO_FrameIndex; } in isFI()
[all …]
|
| /external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/CodeGen/ |
| D | MachineOperand.h | 76 /// OpKind - Specify what kind of operand this is. This discriminates the
78 unsigned OpKind : 8;
208 : OpKind(K), SubReg_TargetFlags(0) { in MachineOperand()
224 MachineOperandType getType() const { return (MachineOperandType)OpKind; } in getType()
326 bool isReg() const { return OpKind == MO_Register; } in isReg()
328 bool isImm() const { return OpKind == MO_Immediate; } in isImm()
330 bool isCImm() const { return OpKind == MO_CImmediate; } in isCImm()
332 bool isFPImm() const { return OpKind == MO_FPImmediate; } in isFPImm()
334 bool isMBB() const { return OpKind == MO_MachineBasicBlock; } in isMBB()
336 bool isFI() const { return OpKind == MO_FrameIndex; } in isFI()
[all …]
|
| /external/swiftshader/third_party/llvm-16.0/configs/common/lib/Target/Mips/ |
| D | MipsGenAsmMatcher.inc | 4198 MatchClassKind OpKind;
4200 default: OpKind = InvalidMatchClass; break;
4201 case Mips::ZERO: OpKind = MCK_GPR32ZERO; break;
4202 case Mips::AT: OpKind = MCK_GPR32NONZERO; break;
4203 case Mips::V0: OpKind = MCK_Reg11; break;
4204 case Mips::V1: OpKind = MCK_Reg11; break;
4205 case Mips::A0: OpKind = MCK_GPRMM16MovePPairFirst; break;
4206 case Mips::A1: OpKind = MCK_Reg13; break;
4207 case Mips::A2: OpKind = MCK_Reg13; break;
4208 case Mips::A3: OpKind = MCK_Reg14; break;
[all …]
|
| /external/swiftshader/third_party/llvm-10.0/configs/common/lib/Target/Mips/ |
| D | MipsGenAsmMatcher.inc | 4196 MatchClassKind OpKind;
4198 default: OpKind = InvalidMatchClass; break;
4199 case Mips::ZERO: OpKind = MCK_GPR32ZERO; break;
4200 case Mips::AT: OpKind = MCK_GPR32NONZERO; break;
4201 case Mips::V0: OpKind = MCK_Reg11; break;
4202 case Mips::V1: OpKind = MCK_Reg11; break;
4203 case Mips::A0: OpKind = MCK_GPRMM16MovePPairFirst; break;
4204 case Mips::A1: OpKind = MCK_Reg13; break;
4205 case Mips::A2: OpKind = MCK_Reg13; break;
4206 case Mips::A3: OpKind = MCK_Reg14; break;
[all …]
|
| /external/llvm/utils/TableGen/ |
| D | FastISelEmitter.cpp | 78 class OpKind { class
83 OpKind() : Repr(OK_Invalid) {} in OpKind() function in __anonb44e9ac80311::OperandsSignature::OpKind
85 bool operator<(OpKind RHS) const { return Repr < RHS.Repr; } in operator <()
86 bool operator==(OpKind RHS) const { return Repr == RHS.Repr; } in operator ==()
88 static OpKind getReg() { OpKind K; K.Repr = OK_Reg; return K; } in getReg()
89 static OpKind getFP() { OpKind K; K.Repr = OK_FP; return K; } in getFP()
90 static OpKind getImm(unsigned V) { in getImm()
93 OpKind K; K.Repr = OK_Imm+V; return K; in getImm()
118 SmallVector<OpKind, 3> Operands;
144 Result.Operands.push_back(OpKind::getImm(0)); in getWithoutImmCodes()
[all …]
|
| /external/swiftshader/third_party/subzero/src/ |
| D | IceInst.h | 321 enum OpKind { enum
328 static InstArithmetic *create(Cfg *Func, OpKind Op, Variable *Dest, in create()
333 OpKind getOp() const { return Op; } in getOp()
337 static const char *getOpName(OpKind Op);
346 InstArithmetic(Cfg *Func, OpKind Op, Variable *Dest, Operand *Source1,
349 const OpKind Op;
477 enum OpKind { enum
484 static const char *getCastName(OpKind Kind);
486 static InstCast *create(Cfg *Func, OpKind CastKind, Variable *Dest, in create()
491 OpKind getCastKind() const { return CastKind; } in getCastKind()
[all …]
|
| /external/swiftshader/third_party/llvm-10.0/configs/common/lib/Target/ARM/ |
| D | ARMGenAsmMatcher.inc | 9072 MatchClassKind OpKind;
9074 default: OpKind = InvalidMatchClass; break;
9075 case ARM::R0: OpKind = MCK_Reg11; break;
9076 case ARM::R1: OpKind = MCK_Reg15; break;
9077 case ARM::R2: OpKind = MCK_Reg11; break;
9078 case ARM::R3: OpKind = MCK_Reg15; break;
9079 case ARM::R4: OpKind = MCK_Reg12; break;
9080 case ARM::R5: OpKind = MCK_Reg16; break;
9081 case ARM::R6: OpKind = MCK_Reg12; break;
9082 case ARM::R7: OpKind = MCK_Reg16; break;
[all …]
|