Searched refs:GradientFunctionFactory (Results 1 – 2 of 2) sorted by relevance
74 using GradientFunctionFactory = variable81 GradientFunctionFactory gradient_function_factory);86 absl::flat_hash_map<string, GradientFunctionFactory> registry_;
55 const string& op_name, GradientFunctionFactory gradient_function_factory) { in Register()