Searched refs:gradient_function_factory (Results 1 – 2 of 2) sorted by relevance
81 GradientFunctionFactory gradient_function_factory);
55 const string& op_name, GradientFunctionFactory gradient_function_factory) { in Register() argument61 registry_.insert({op_name, gradient_function_factory}); in Register()