Searched refs:GradFunc (Results 1 – 4 of 4) sorted by relevance
30 typedef Status (*GradFunc)(const Scope& scope, const Operation& op, typedef41 bool Register(const string& op, GradFunc func);48 Status Lookup(const string& op, GradFunc* func) const;54 std::unordered_map<string, GradFunc> registry_;
27 bool GradOpRegistry::Register(const string& op, GradFunc func) { in Register()32 Status GradOpRegistry::Lookup(const string& op, GradFunc* func) const { in Lookup()
374 ops::GradFunc grad_fn; in IsPrimitiveOpWithNoGrad()383 ops::GradFunc grad_fn; in CallGradFunction()
26 ops::GradFunc grad_fn; in CallGradFunction()