Searched refs:FunctionalizeCond (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 291 FunctionalizeCond::FunctionalizeCond(Graph* graph, in FunctionalizeCond() function in tensorflow::functionalize_cond::FunctionalizeCond
302 Conditional(OutputTensor predicate, FunctionalizeCond* parent,
348 FunctionalizeCond* parent_;
390 Conditional::Conditional(OutputTensor predicate, FunctionalizeCond* parent, in Conditional()
960 Status FunctionalizeCond::AddIdentityNode(const Node* replacee, Node* if_node, in AddIdentityNode()
977 StatusOr<Node*> FunctionalizeCond::AddIfNode(const NodeDef& def, in AddIfNode()
998 Status FunctionalizeCond::PropagateUpdatedState(const Node* replacee) { in PropagateUpdatedState()
1050 StatusOr<StateMap::CondId> FunctionalizeCond::JoinCondStatesNonMerge( in JoinCondStatesNonMerge()
1086 StatusOr<StateMap::CondId> FunctionalizeCond::JoinCondStatesMerge( in JoinCondStatesMerge()
1136 StateMap::CondId FunctionalizeCond::StateAlongEdge(const Edge* e) { in StateAlongEdge()
[all …]
|
| D | functionalize_cond_test.cc | 43 fc_.reset(new functionalize_cond::FunctionalizeCond( in FunctionalizeCondTest()
67 std::unique_ptr<functionalize_cond::FunctionalizeCond> fc_;
122 TEST(FunctionalizeCond, DuplicateConstNodes) { in TEST() argument
147 auto status = tensorflow::FunctionalizeCond(&graph, &flib_def); in TEST()
|
| D | functionalize_cond.h | 40 Status FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library,
184 class FunctionalizeCond {
209 FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library,
|
| D | functionalize_while.cc | 408 TF_RETURN_IF_ERROR(FunctionalizeCond(cond_graph.get(), library, node_filter)); in FunctionalizeLoop()
413 TF_RETURN_IF_ERROR(FunctionalizeCond(body_graph.get(), library, node_filter)); in FunctionalizeLoop()
|
| D | functionalize_control_flow.cc | 303 TF_RETURN_IF_ERROR(FunctionalizeCond(graph, library, node_filter)); in FunctionalizeControlFlow()
|