| 199 | |
| 200 | MGB_DYN_TYPE_OBJ_FINAL_IMPL(BatchedMatrixMul); |
| 201 | BatchedMatrixMul::BatchedMatrixMul( |
| 202 | VarNode* a, VarNode* b, const Param& param, const ExecutionPolicy& policy, |
| 203 | const OperatorNodeConfig& config) |
| 204 | : Super{a->owner_graph(), config, "batched_matrix_mul", {a, b}} { |
| 205 | init_megdnn_opr(*this, param); |
| 206 | m_policy = policy; |
| 207 | add_input({a, b}); |
| 208 | output(0)->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE); |
| 209 | } |
| 210 | |
| 211 | SymbolVar BatchedMatrixMul::make( |
| 212 | SymbolVar a, SymbolVar b, const Param& param, const ExecutionPolicy& policy, |