(self)
| 124 | class FunctionTest(test.TestCase, parameterized.TestCase): |
| 125 | |
| 126 | def setUp(self): |
| 127 | super(FunctionTest, self).setUp() |
| 128 | cpus = config.list_physical_devices('CPU') |
| 129 | # Set 4 virtual CPUs |
| 130 | config.set_virtual_device_configuration(cpus[0], [ |
| 131 | context.VirtualDeviceConfiguration(), |
| 132 | context.VirtualDeviceConfiguration(), |
| 133 | context.VirtualDeviceConfiguration(), |
| 134 | context.VirtualDeviceConfiguration() |
| 135 | ]) |
| 136 | |
| 137 | def testBasic(self): |
| 138 | matmul = def_function.function(math_ops.matmul) |
nothing calls this directly
no test coverage detected