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