| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 145 cond_graph = func_graph_module.func_graph_from_py_func(
172 for t in cond_graph.external_captures:
214 with cond_graph.as_default():
215 num_cond_captures = len(cond_graph.external_captures)
216 assert (cond_graph.external_captures ==
219 cond_graph, body_graph.external_captures[num_cond_captures:])
250 with cond_graph.as_default():
253 cond_graph.capture(tensor_list)
263 _check_num_inputs_outputs(cond_graph, body_graph,
268 list(cond_graph.control_captures) + list(body_graph.control_captures)):
[all …]
|
| D | cond_v2.py | 399 def _resolve_grad_inputs(cond_graph, grad_graph): argument
425 assert t.graph == cond_graph
445 assert t.graph == cond_graph.outer_graph
|
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 107 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_); in CreateCondGraph()
109 LessThan(params_->cond_inputs[0], {one, 0}, params_->cond_graph, s_); in CreateCondGraph()
142 EXPECT_TRUE(params_->cond_graph != nullptr); in TEST_F()
159 params_->cond_graph, s_); in TEST_F()
226 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_); in TEST_F()
229 LessThan(params_->cond_inputs[0], {six, 0}, params_->cond_graph, s_); in TEST_F()
242 TF_Operation* three = ScalarConst(3, inner_params.cond_graph, s_); in TEST_F()
245 inner_params.cond_inputs[0], {three, 0}, inner_params.cond_graph, s_); in TEST_F()
417 TF_Operation* ten = ScalarConst(10, params_->cond_graph, s_); in TEST_F()
419 LessThan(params_->cond_inputs[0], {ten, 0}, params_->cond_graph, s_); in TEST_F()
|
| D | c_api.cc | 1881 if (params.cond_graph == nullptr || params.body_graph == nullptr || in ValidateConstWhileParams()
1882 params.cond_graph->parent == nullptr || in ValidateConstWhileParams()
1883 params.cond_graph->parent != params.body_graph->parent || in ValidateConstWhileParams()
1884 params.cond_graph->parent_inputs != params.body_graph->parent_inputs || in ValidateConstWhileParams()
1916 TF_DeleteGraph(params->cond_graph); in FreeWhileResources()
1945 TF_Graph* cond_graph = TF_NewGraph(); in TF_NewWhile()
1947 cond_graph->parent = g; in TF_NewWhile()
1948 cond_graph->parent_inputs = inputs; in TF_NewWhile()
1961 if (!CreateInput(inputs[i], cond_graph, StrCat("cond_input", i).c_str(), in TF_NewWhile()
1971 TF_WhileParams params = {ninputs, cond_graph, cond_inputs, cond_output, in TF_NewWhile()
[all …]
|
| D | c_api.h | 916 TF_Graph* const cond_graph; member
|
| D | c_api_function_test.cc | 923 params->cond_inputs[0], params->cond_inputs[1], params->cond_graph, s_); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 405 std::unique_ptr<Graph> cond_graph; in FunctionalizeLoop() local
406 TF_RETURN_IF_ERROR(BuildLoopCondition(*graph, frame, &cond_graph)); in FunctionalizeLoop()
407 FixupSourceAndSinkEdges(cond_graph.get()); in FunctionalizeLoop()
408 TF_RETURN_IF_ERROR(FunctionalizeCond(cond_graph.get(), library, node_filter)); in FunctionalizeLoop()
416 << DumpGraphToFile("loop_condition", *cond_graph, library) in FunctionalizeLoop()
425 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef)); in FunctionalizeLoop()
|
| D | functionalize_control_flow_test.cc | 80 void BuildCondGraph(Graph* cond_graph);
103 void ConditionalTestFixture::BuildCondGraph(Graph* cond_graph) { in BuildCondGraph() argument
131 TF_EXPECT_OK(scope.ToGraph(cond_graph)); in BuildCondGraph()
134 for (Node* n : cond_graph->nodes()) { in BuildCondGraph()
220 Graph cond_graph(OpRegistry::Global()); in RunTest() local
221 BuildCondGraph(&cond_graph); in RunTest()
224 TF_ASSERT_OK(GraphToFunctionDef(cond_graph, "cond_fn", &cond_fdef)); in RunTest()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1568 Graph* cond_graph = cond_fbody->graph; in RewriteHostWhileLoopCond() local
1570 for (Node* n : cond_graph->nodes()) { in RewriteHostWhileLoopCond()
1597 Node* recv_pred_node = cond_graph->AddNode(recv_pred_def, &s); in RewriteHostWhileLoopCond()
1599 cond_graph->AddEdge(key_arg, 0, recv_pred_node, 0); in RewriteHostWhileLoopCond()
1607 Node* ret_node = cond_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopCond()
1609 cond_graph->AddEdge(recv_pred_node, 0, ret_node, 0); in RewriteHostWhileLoopCond()
1612 TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(cond_graph)); in RewriteHostWhileLoopCond()
1616 TF_RETURN_IF_ERROR(GraphToFunctionDef(*cond_graph, cond_host_func_name, in RewriteHostWhileLoopCond()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 277 buildSubgraph(env, clazz, cond_graph_builder, params.cond_graph, in Java_org_tensorflow_Graph_whileLoop()
|