| 79 | } |
| 80 | |
| 81 | TensorLayout create_layout(const std::vector<int32_t>& shape, DType dtype) { |
| 82 | TensorShape tshape; |
| 83 | tshape.ndim = shape.size(); |
| 84 | mgb_assert(tshape.ndim <= TensorLayout::MAX_NDIM); |
| 85 | std::copy(shape.begin(), shape.end(), tshape.shape); |
| 86 | return TensorLayout(tshape, dtype); |
| 87 | } |
| 88 | |
| 89 | std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible_remote_send( |
| 90 | const OpDef& def, const SmallVector<LogicalTensorDesc>& input_descs) { |
no test coverage detected