| 594 | } |
| 595 | |
| 596 | void CompiledTransformation::assert_tensor_equal(ValueRef lhs, ValueRef rhs) { |
| 597 | if (!lhs.is<HostValue>()) { |
| 598 | lhs = lhs.numpy(); |
| 599 | } |
| 600 | if (!rhs.is<HostValue>()) { |
| 601 | rhs = rhs.numpy(); |
| 602 | } |
| 603 | trace_assert(m_value_comparator(lhs, rhs), "tensors not equals"); |
| 604 | } |
| 605 | |
| 606 | ValueRef CompiledTransformation::trace_input(size_t id, ValueRef value) { |
| 607 | try { |