Searched refs:bwd_fbody (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 871 std::unique_ptr<FunctionBody> bwd_fbody; in RewriteTensorListWithConstElement() local
873 *bwd_body, AttrSlice(&bwd_body_attr.attr()), fld, &bwd_fbody)); in RewriteTensorListWithConstElement()
876 Node* bwd_arg = bwd_fbody->arg_nodes[bwd_while_dst_input]; in RewriteTensorListWithConstElement()
900 bwd_fbody->graph->NewName(const_input_nodedef.name())); in RewriteTensorListWithConstElement()
901 Node* const_node = bwd_fbody->graph->AddNode(const_input_nodedef, &s); in RewriteTensorListWithConstElement()
906 bwd_fbody->graph->RemoveEdge(e); in RewriteTensorListWithConstElement()
907 bwd_fbody->graph->AddEdge(const_node, 0, dst, dst_input); in RewriteTensorListWithConstElement()
915 GraphToFunctionDef(*bwd_fbody->graph, new_name, &new_fdef)); in RewriteTensorListWithConstElement()
|
| D | tf2xla_util_test.cc | 491 std::unique_ptr<FunctionBody> bwd_fbody; in TEST() local
493 FunctionDefToBodyHelper(*bwd_body, AttrSlice(), &fld, &bwd_fbody)); in TEST()
494 auto node_name_index = bwd_fbody->graph->BuildNodeNameIndex(); in TEST()
|