(self, theta, xs, input_batch)
source not stored for this graph (policy: none)
no test coverage detected