Searched refs:bwd_body (Results 1 – 2 of 2) sorted by relevance
489 const FunctionDef* bwd_body = fld.Find("bwd_body_tl_rewrite_0"); in TEST() local490 ASSERT_NE(bwd_body, nullptr); in TEST()493 FunctionDefToBodyHelper(*bwd_body, AttrSlice(), &fld, &bwd_fbody)); in TEST()
865 const FunctionDef* bwd_body = fld->Find(bwd_body_attr.name()); in RewriteTensorListWithConstElement() local866 if (!bwd_body) { in RewriteTensorListWithConstElement()873 *bwd_body, AttrSlice(&bwd_body_attr.attr()), fld, &bwd_fbody)); in RewriteTensorListWithConstElement()