| 217 | } |
| 218 | |
| 219 | ::testing::AssertionResult mgb::__assert_shape_equal( |
| 220 | const TensorShape& v0, const TensorShape& v1) { |
| 221 | if (v0.eq_shape(v1)) |
| 222 | return ::testing::AssertionSuccess() |
| 223 | << v0.to_string() << " == " << v1.to_string(); |
| 224 | else |
| 225 | return ::testing::AssertionFailure() |
| 226 | << v0.to_string() << " != " << v1.to_string(); |
| 227 | } |
| 228 | |
| 229 | #if WIN32 |
| 230 | #include <direct.h> |