Home
last modified time | relevance | path

Searched refs:body_graph (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py194 body_graph = func_graph_module.func_graph_from_py_func(
207 loop_vars = loop_vars + body_graph.external_captures
210 body_graph.outputs.extend(body_graph.internal_captures)
217 body_graph.external_captures[:num_cond_captures])
219 cond_graph, body_graph.external_captures[num_cond_captures:])
229 body_graph.outputs[first_loop_var_index:first_loop_var_index +
237 num_original_outputs = len(body_graph.outputs)
242 intermediate_tensors = _get_intermediates(body_graph)
254 with body_graph.as_default():
260 body_graph.outputs.append(appended_tensor_list)
[all …]
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc141 EXPECT_TRUE(params_->body_graph != nullptr); in TEST_F()
164 params_->body_graph, s_, "add1"); in TEST_F()
166 TF_Operation* one = ScalarConst(1, params_->body_graph, s_); in TEST_F()
168 TF_Operation* add2 = Add(add1, one, params_->body_graph, s_, "add2"); in TEST_F()
237 TF_NewWhile(params_->body_graph, inner_inputs, 2, s_); in TEST_F()
250 TF_Operation* one = ScalarConst(1, inner_params.body_graph, s_, "one"); in TEST_F()
252 TF_Operation* two = ScalarConst(2, inner_params.body_graph, s_, "two"); in TEST_F()
256 Add(inner_params.body_inputs[1].oper, one, inner_params.body_graph, s_); in TEST_F()
261 inner_params.body_graph, s_, "add2"); in TEST_F()
271 Add(params_->body_inputs[0], inner_outputs[1], params_->body_graph, s_); in TEST_F()
[all …]
Dc_api.cc1881 if (params.cond_graph == nullptr || params.body_graph == 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()
1917 TF_DeleteGraph(params->body_graph); in FreeWhileResources()
1946 TF_Graph* body_graph = TF_NewGraph(); in TF_NewWhile()
1949 body_graph->parent = g; in TF_NewWhile()
1950 body_graph->parent_inputs = inputs; in TF_NewWhile()
1965 if (!CreateInput(inputs[i], body_graph, StrCat("body_input", i).c_str(), in TF_NewWhile()
1972 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2020 CopyGraph(&params->body_graph->graph, &parent->graph, in TF_FinishWhileHelper()
Dc_api_function_test.cc928 params->body_graph, s_, "add1"); in TEST_F()
930 TF_Operation* one = ScalarConst(1, params->body_graph, s_); in TEST_F()
932 TF_Operation* add2 = Add(add1, one, params->body_graph, s_, "add2"); in TEST_F()
Dc_api.h922 TF_Graph* const body_graph; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc410 std::unique_ptr<Graph> body_graph; in FunctionalizeLoop() local
411 TF_RETURN_IF_ERROR(BuildLoopBody(*graph, frame, &arg_types, &body_graph)); in FunctionalizeLoop()
412 FixupSourceAndSinkEdges(body_graph.get()); in FunctionalizeLoop()
413 TF_RETURN_IF_ERROR(FunctionalizeCond(body_graph.get(), library, node_filter)); in FunctionalizeLoop()
417 << " body: " << DumpGraphToFile("loop_body", *body_graph); in FunctionalizeLoop()
428 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()
/external/tensorflow/tensorflow/python/kernel_tests/
Dwhile_v2_test.py861 body_graph = while_v2._get_graph(while_op, "body", "_body_graph")
862 placeholder = body_graph.inputs[index]
970 body_graph = while_v2._get_graph(while_op, "body", "_body_graph")
972 x_input_t = body_graph.inputs[x_input_index]
1001 body_graph = while_v2._get_graph(while_op, "body", "_body_graph")
1002 y_input_t = body_graph.inputs[idx]
1005 output_idx = body_graph.outputs.index(push_back_node.outputs[0])
1276 body_graph = while_v2._get_graph(forward_while_op, "body", "_body_graph")
1278 o for o in body_graph.get_operations() if o.type == "TensorListPushBack"
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass_test.cc300 std::unique_ptr<Graph> body_graph = MakeBodyGraph(); in TEST() local
302 TF_ASSERT_OK(GraphToFunctionDef(*body_graph, "launch0", flib.add_function())); in TEST()
Dextract_outside_compilation_pass.cc1645 Graph* body_graph = body_fbody->graph; in RewriteHostWhileLoopBody() local
1647 for (Node* n : body_graph->nodes()) { in RewriteHostWhileLoopBody()
1667 Node* ret_node = body_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopBody()
1669 body_graph->AddEdge(key_arg, 0, ret_node, 0); in RewriteHostWhileLoopBody()
1672 TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(body_graph)); in RewriteHostWhileLoopBody()
1676 TF_RETURN_IF_ERROR(GraphToFunctionDef(*body_graph, body_host_func_name, in RewriteHostWhileLoopBody()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_jni.cc282 env, clazz, body_graph_builder, params.body_graph, params.body_inputs, in Java_org_tensorflow_Graph_whileLoop()