| 60 | } |
| 61 | |
| 62 | SmallVector<VarNode::LayoutConstraintCallback> get_input_layout_constraint( |
| 63 | const OpDef& def, const SmallVector<TensorPtr>& inputs) { |
| 64 | SmallVector<VarNode::LayoutConstraintCallback> layout_checker(inputs.size()); |
| 65 | layout_checker[0] = [](const TensorLayout& layout) { |
| 66 | return layout.is_contiguous(); |
| 67 | }; |
| 68 | return layout_checker; |
| 69 | } |
| 70 | |
| 71 | OP_TRAIT_REG(Flip, Flip) |
| 72 | .apply_on_var_node(apply_on_var_node) |
nothing calls this directly
no test coverage detected