| 153 | return {out}; |
| 154 | } |
| 155 | SmallVector<VarNode::LayoutConstraintCallback> get_input_layout_constraint( |
| 156 | const OpDef& def, const SmallVector<TensorPtr>& inputs) { |
| 157 | SmallVector<VarNode::LayoutConstraintCallback> layout_checker(inputs.size()); |
| 158 | layout_checker[0] = layout_checker[1] = layout_checker[2] = |
| 159 | [](const TensorLayout& layout) { return layout.is_contiguous(); }; |
| 160 | return layout_checker; |
| 161 | } |
| 162 | |
| 163 | OP_TRAIT_REG(IndexingSetOneHot, IndexingSetOneHot) |
| 164 | .infer_output_attrs_fallible(infer_output_attrs_fallible) |
nothing calls this directly
no test coverage detected