| 1819 | } |
| 1820 | |
| 1821 | Status GetOpGradientCreator(const string& op, Creator* creator) { |
| 1822 | auto fac = GetOpGradFactory(); |
| 1823 | auto iter = fac->find(op); |
| 1824 | if (iter == fac->end()) { |
| 1825 | return errors::NotFound("No gradient defined for op: ", op); |
| 1826 | } |
| 1827 | *creator = iter->second; |
| 1828 | return Status::OK(); |
| 1829 | } |
| 1830 | |
| 1831 | } // end namespace gradient |
| 1832 |
no test coverage detected