Searched defs:grad_func (Results 1 – 6 of 6) sorted by relevance
234 grad_func=None, argument338 def set_grad_func(self, grad_func): argument592 grad_func=None, argument1096 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
652 auto grad_func = in ConvertLibFunction() local
1279 auto grad_func = module_.lookupSymbol<mlir::FuncOp>(mlir_grad_func_name); in ConvertLibFunction() local
1472 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local