Searched defs:grad_func (Results 1 – 5 of 5) sorted by relevance
261 grad_func=None, argument366 def set_grad_func(self, grad_func): argument624 grad_func=None, argument1130 def _from_definition(fdef, grad_func=None): argument
1175 def expectFunctionsEqual(self, func, grad_func=None, new_func=None): argument
1322 TF_Function* grad_func; in TEST_F() local1366 TF_Function* grad_func; in TEST_F() local1395 TF_Function* grad_func; in TEST_F() local
627 auto grad_func = symbol_table.lookup<FuncOp>(attr.getValue()); in ConvertLibFunction() local
1482 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local