| 43 | } |
| 44 | |
| 45 | auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) { |
| 46 | auto&& ds = static_cast<const Dimshuffle&>(def); |
| 47 | OperatorNodeConfig config{ds.make_name()}; |
| 48 | return opr::Dimshuffle::make(inputs[0], ds.pattern, 0UL, config); |
| 49 | } |
| 50 | |
| 51 | std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible( |
| 52 | const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs) { |