Searched refs:FunctionalizeCond (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 285 FunctionalizeCond::FunctionalizeCond(Graph* graph, in FunctionalizeCond() function in tensorflow::functionalize_cond::FunctionalizeCond
292 Conditional(OutputTensor predicate, FunctionalizeCond* parent,
338 FunctionalizeCond* parent_;
377 Conditional::Conditional(OutputTensor predicate, FunctionalizeCond* parent, in Conditional()
919 Status FunctionalizeCond::AddIdentityNode(const Node* replacee, Node* if_node, in AddIdentityNode()
930 StatusOr<Node*> FunctionalizeCond::AddIfNode(const NodeDef& def, in AddIfNode()
951 Status FunctionalizeCond::PropagateUpdatedState(const Node* replacee) { in PropagateUpdatedState()
1003 StatusOr<StateMap::CondId> FunctionalizeCond::JoinCondStatesNonMerge( in JoinCondStatesNonMerge()
1039 StatusOr<StateMap::CondId> FunctionalizeCond::JoinCondStatesMerge( in JoinCondStatesMerge()
1089 StateMap::CondId FunctionalizeCond::StateAlongEdge(const Edge* e) { in StateAlongEdge()
[all …]
|
| D | functionalize_cond.h | 30 Status FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library);
173 class FunctionalizeCond {
200 FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library);
|
| D | functionalize_cond_test.cc | 36 fc_.reset(new functionalize_cond::FunctionalizeCond(graph_.get(), in FunctionalizeCondTest()
60 std::unique_ptr<functionalize_cond::FunctionalizeCond> fc_;
|
| D | functionalize_control_flow.cc | 61 TF_RETURN_IF_ERROR(FunctionalizeCond(graph, library)); in FunctionalizeControlFlow()
|
| D | functionalize_while.cc | 484 TF_RETURN_IF_ERROR(FunctionalizeCond(cond_graph.get(), library)); in FunctionalizeLoop()
489 TF_RETURN_IF_ERROR(FunctionalizeCond(body_graph.get(), library)); in FunctionalizeLoop()
|