(self, device_option)
| 99 | core.DeviceOption(workspace.GpuDeviceType, 1)])) |
| 100 | @settings(deadline=10000) |
| 101 | def testDirect(self, device_option): |
| 102 | operators = [ |
| 103 | CreateOperator('Direct', 'in', 'hidden'), |
| 104 | CreateOperator('Direct', 'hidden', 'out'), |
| 105 | ] |
| 106 | if device_option: |
| 107 | for op in operators: |
| 108 | op.device_option.CopyFrom(device_option) |
| 109 | desired_grad_operators = [ |
| 110 | CreateOperator('DirectGradient', 'out_grad', 'hidden_grad'), |
| 111 | CreateOperator('DirectGradient', 'hidden_grad', 'in_grad'), |
| 112 | ] |
| 113 | if device_option: |
| 114 | for op in desired_grad_operators: |
| 115 | op.device_option.CopyFrom(device_option) |
| 116 | gradients, _ = GradientRegistry.GetBackwardPass( |
| 117 | operators, {'out': 'out_grad'}) |
| 118 | self.assertOperatorListEqual(gradients, desired_grad_operators) |
| 119 | |
| 120 | def testDirectImplicitGradientSource(self): |
| 121 | operators = [ |
nothing calls this directly
no test coverage detected