Home
last modified time | relevance | path

Searched refs:cond_graph (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py121 cond_graph = func_graph_module.func_graph_from_py_func(
144 for t in cond_graph.external_captures:
185 with cond_graph.as_default():
186 num_cond_captures = len(cond_graph.external_captures)
187 assert (cond_graph.external_captures ==
190 assert body_capture not in cond_graph.captures
191 cond_graph.capture(body_capture)
208 _check_num_inputs_outputs(cond_graph, body_graph,
212 list(cond_graph.control_captures) + list(body_graph.control_captures)):
215 util.create_new_tf_function(cond_graph),
[all …]
Dcond_v2.py317 def _resolve_grad_inputs(cond_graph, grad_graph): argument
343 assert t.graph == cond_graph
356 assert t.graph == cond_graph.outer_graph
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc108 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_); in CreateCondGraph()
110 LessThan(params_->cond_inputs[0], {one, 0}, params_->cond_graph, s_); in CreateCondGraph()
143 EXPECT_TRUE(params_->cond_graph != nullptr); in TEST_F()
160 params_->cond_graph, s_); in TEST_F()
227 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_); in TEST_F()
230 LessThan(params_->cond_inputs[0], {six, 0}, params_->cond_graph, s_); in TEST_F()
243 TF_Operation* three = ScalarConst(3, inner_params.cond_graph, s_); in TEST_F()
246 inner_params.cond_inputs[0], {three, 0}, inner_params.cond_graph, s_); in TEST_F()
418 TF_Operation* ten = ScalarConst(10, params_->cond_graph, s_); in TEST_F()
420 LessThan(params_->cond_inputs[0], {ten, 0}, params_->cond_graph, s_); in TEST_F()
Dc_api.cc2288 if (params.cond_graph == nullptr || params.body_graph == nullptr || in ValidateConstWhileParams()
2289 params.cond_graph->parent == nullptr || in ValidateConstWhileParams()
2290 params.cond_graph->parent != params.body_graph->parent || in ValidateConstWhileParams()
2291 params.cond_graph->parent_inputs != params.body_graph->parent_inputs || in ValidateConstWhileParams()
2323 TF_DeleteGraph(params->cond_graph); in FreeWhileResources()
2352 TF_Graph* cond_graph = TF_NewGraph(); in TF_NewWhile()
2354 cond_graph->parent = g; in TF_NewWhile()
2355 cond_graph->parent_inputs = inputs; in TF_NewWhile()
2368 if (!CreateInput(inputs[i], cond_graph, StrCat("cond_input", i).c_str(), in TF_NewWhile()
2378 TF_WhileParams params = {ninputs, cond_graph, cond_inputs, cond_output, in TF_NewWhile()
[all …]
Dc_api.h1120 TF_Graph* const cond_graph; member
Dc_api_function_test.cc923 params->cond_inputs[0], params->cond_inputs[1], params->cond_graph, s_); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc481 std::unique_ptr<Graph> cond_graph; in FunctionalizeLoop() local
482 TF_RETURN_IF_ERROR(BuildLoopCondition(*graph, frame, &cond_graph)); in FunctionalizeLoop()
483 FixupSourceAndSinkEdges(cond_graph.get()); in FunctionalizeLoop()
484 TF_RETURN_IF_ERROR(FunctionalizeCond(cond_graph.get(), library)); in FunctionalizeLoop()
492 << DumpGraphToFile("loop_condition", *cond_graph, library) in FunctionalizeLoop()
503 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef)); in FunctionalizeLoop()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc1043 Graph* cond_graph = cond_fbody->graph; in RewriteHostWhileLoopCond() local
1045 for (Node* n : cond_graph->nodes()) { in RewriteHostWhileLoopCond()
1073 Node* recv_pred_node = cond_graph->AddNode(recv_pred_def, &s); in RewriteHostWhileLoopCond()
1075 cond_graph->AddEdge(key_arg, 0, recv_pred_node, 0); in RewriteHostWhileLoopCond()
1083 Node* ret_node = cond_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopCond()
1085 cond_graph->AddEdge(recv_pred_node, 0, ret_node, 0); in RewriteHostWhileLoopCond()
1088 TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(cond_graph)); in RewriteHostWhileLoopCond()
1093 GraphToFunctionDef(*cond_graph, cond_host_func_name, &cond_replace_fdef)); in RewriteHostWhileLoopCond()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_jni.cc277 buildSubgraph(env, clazz, cond_graph_builder, params.cond_graph, in Java_org_tensorflow_Graph_whileLoop()