Searched defs:grad_func (Results 1 – 5 of 5) sorted by relevance
236 grad_func=None, argument340 def set_grad_func(self, grad_func): argument595 grad_func=None, argument1099 def _from_definition(fdef, grad_func=None): argument
1179 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
630 auto grad_func = symbol_table.lookup<FuncOp>(attr.getValue()); in ConvertLibFunction() local
1475 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local