| D | test_shape_analysis.cpp | 302 void assertShapeEqual(c10::SymbolicShape& a, c10::SymbolicShape& e) { in assertShapeEqual() function
308 void assertShapeEqual( in assertShapeEqual() function
315 assertShapeEqual(actual->at(0), symb_expected); in assertShapeEqual()
344 assertShapeEqual(res, {64, 56, 56}); in TEST()
348 assertShapeEqual(res, {64, 56, 56}); in TEST()
352 assertShapeEqual(res, {64, 56, 56}); in TEST()
356 assertShapeEqual(res, {sym_dim, 56, 56}); in TEST()
360 assertShapeEqual(res, {64, 56, 56}); in TEST()
363 assertShapeEqual(res, {sym_dim, 64, sym_dim, sym_dim}); in TEST()
373 assertShapeEqual(res, {sym_dim, 2}); in TEST()
[all …]
|