| 31 | using FDH = FunctionDefHelper; |
| 32 | |
| 33 | std::unique_ptr<Session> NewSession() { |
| 34 | SessionOptions opts; |
| 35 | (*opts.config.mutable_device_count())["CPU"] = 1; |
| 36 | return std::unique_ptr<Session>(NewSession(opts)); |
| 37 | } |
| 38 | |
| 39 | class MathGradTest : public ::testing::Test { |
| 40 | protected: |
no outgoing calls
no test coverage detected