|
|
|
|
|
|
|
|
| from caffe2.python import core, workspace |
| import unittest |
|
|
| core.GlobalInit(['python']) |
|
|
|
|
| class BlobDeallocationTest(unittest.TestCase): |
| def test(self): |
| net = core.Net('net') |
|
|
| x = net.GivenTensorStringFill([], ['x'], shape=[3], values=['a', 'b', 'c']) |
| y = net.GivenTensorStringFill([], ['y'], shape=[3], values=['d', 'e', 'f']) |
| net.Concat([x, y], ['concated', '_'], axis=0) |
|
|
| workspace.ResetWorkspace() |
| workspace.RunNetOnce(net) |
|
|
| workspace.ResetWorkspace() |
| workspace.RunNetOnce(net) |
| self.assertTrue(True) |
|
|
|
|
| if __name__ == '__main__': |
| unittest.main() |
|
|
