Searched refs:grad_func_name (Results 1 – 7 of 7) sorted by relevance
222 if f.grad_func_name:225 grad_def.gradient_func = f.grad_func_name
1191 self.assertEqual(func.grad_func_name, new_func.grad_func_name)1229 self.assertEqual(Foo.grad_func_name, new_func.grad_func_name)
3240 if f.grad_func_name:3243 grad_def.gradient_func = f.grad_func_name3404 if (function.grad_func_name is not None) and (function.python_grad_func is
345 def grad_func_name(self): member in _DefinedFunction
1693 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions() local1694 if (!grad_func_name.empty()) add_to_func_queue(grad_func_name); in ReachableFunctions()1714 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition() local1715 if (!grad_func_name.empty()) { in ReachableFunctionLibraryDefinition()1718 grad.set_gradient_func(grad_func_name); in ReachableFunctionLibraryDefinition()
1275 auto grad_func_name = func_lib.FindGradient(std::string(func_name)); in ConvertLibFunction() local1276 if (!grad_func_name.empty()) { in ConvertLibFunction()1277 TF_RETURN_IF_ERROR(ConvertLibFunction(grad_func_name)); in ConvertLibFunction()1278 auto mlir_grad_func_name = (*tf_name_to_mlir_name_)[grad_func_name]; in ConvertLibFunction()
502 self.grad_func_name = None