| 485 | } |
| 486 | |
| 487 | CompNode OperatorNodeConfig::get_single_comp_node() const { |
| 488 | mgb_assert( |
| 489 | m_comp_node.size() <= 1, |
| 490 | "at most one comp node could be provided, got %zu instead", |
| 491 | m_comp_node.size()); |
| 492 | if (m_comp_node.empty()) |
| 493 | return {}; |
| 494 | return m_comp_node[0]; |
| 495 | } |
| 496 | |
| 497 | OperatorNodeConfig& OperatorNodeConfig::output_dtype(DType dtype) { |
| 498 | m_output_dtype = dtype; |
no test coverage detected