Home
last modified time | relevance | path

Searched refs:grad_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc275 const string grad_name = lookup_library->FindGradient(func_name); in AddMissingFunctionDef() local
276 if (!grad_name.empty() && library->FindGradient(func_name).empty()) { in AddMissingFunctionDef()
278 AddMissingFunctionByName(grad_name, lookup_library, library)); in AddMissingFunctionDef()
281 grad_def.set_gradient_func(grad_name); in AddMissingFunctionDef()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1305 const string& grad_name = FindGradientHelper(func_name); in GetAttrImpl() local
1309 if (!grad_name.empty()) { in GetAttrImpl()
1310 return FindHelper(grad_name); in GetAttrImpl()