Searched refs:bwd_body_attr (Results 1 – 1 of 1) sorted by relevance
869 NameAttrList bwd_body_attr; in RewriteTensorListWithConstElement() local870 TF_CHECK_OK(GetNodeAttr(bwd_while->def(), "body", &bwd_body_attr)); in RewriteTensorListWithConstElement()871 const FunctionDef* bwd_body = fld->Find(bwd_body_attr.name()); in RewriteTensorListWithConstElement()874 bwd_body_attr.name(), " for While node ", in RewriteTensorListWithConstElement()879 *bwd_body, AttrSlice(&bwd_body_attr.attr()), fld, &bwd_fbody)); in RewriteTensorListWithConstElement()919 absl::StrCat(bwd_body_attr.name(), "_tl_rewrite_")); in RewriteTensorListWithConstElement()925 bwd_body_attr.set_name(new_name); in RewriteTensorListWithConstElement()927 bwd_while->AddAttr("body", bwd_body_attr); in RewriteTensorListWithConstElement()