| 36 | } |
| 37 | |
| 38 | bool IValue::ToBool() const { |
| 39 | CHECK_EQ(tag_, Tag::kBool) << "Current value is not a bool."; |
| 40 | return payload_.i.v_bool; |
| 41 | } |
| 42 | |
| 43 | const Tensor& IValue::ToTensor() const { |
| 44 | CHECK_EQ(tag_, Tag::kTensor) << "Current value is not a tensor."; |