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.py165 body_graph = func_graph_module.func_graph_from_py_func(
178 loop_vars = loop_vars + body_graph.external_captures
181 body_graph.outputs.extend(body_graph.internal_captures)
188 body_graph.external_captures[:num_cond_captures])
189 for body_capture in body_graph.external_captures[num_cond_captures:]:
200 body_graph.outputs[first_loop_var_index:first_loop_var_index +
208 _check_num_inputs_outputs(cond_graph, body_graph,
212 list(cond_graph.control_captures) + list(body_graph.control_captures)):
216 util.create_new_tf_function(body_graph),
217 output_shapes=[t.shape for t in body_graph.outputs],
[all …]
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc142 EXPECT_TRUE(params_->body_graph != nullptr); in TEST_F()
165 params_->body_graph, s_, "add1"); in TEST_F()
167 TF_Operation* one = ScalarConst(1, params_->body_graph, s_); in TEST_F()
169 TF_Operation* add2 = Add(add1, one, params_->body_graph, s_, "add2"); in TEST_F()
238 TF_NewWhile(params_->body_graph, inner_inputs, 2, s_); in TEST_F()
251 TF_Operation* one = ScalarConst(1, inner_params.body_graph, s_, "one"); in TEST_F()
253 TF_Operation* two = ScalarConst(2, inner_params.body_graph, s_, "two"); in TEST_F()
257 Add(inner_params.body_inputs[1].oper, one, inner_params.body_graph, s_); in TEST_F()
262 inner_params.body_graph, s_, "add2"); in TEST_F()
272 Add(params_->body_inputs[0], inner_outputs[1], params_->body_graph, s_); in TEST_F()
[all …]
Dc_api.cc2288 if (params.cond_graph == nullptr || params.body_graph == 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()
2324 TF_DeleteGraph(params->body_graph); in FreeWhileResources()
2353 TF_Graph* body_graph = TF_NewGraph(); in TF_NewWhile()
2356 body_graph->parent = g; in TF_NewWhile()
2357 body_graph->parent_inputs = inputs; in TF_NewWhile()
2372 if (!CreateInput(inputs[i], body_graph, StrCat("body_input", i).c_str(), in TF_NewWhile()
2379 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2427 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.h1126 TF_Graph* const body_graph; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc486 std::unique_ptr<Graph> body_graph; in FunctionalizeLoop() local
487 TF_RETURN_IF_ERROR(BuildLoopBody(*graph, frame, &arg_types, &body_graph)); in FunctionalizeLoop()
488 FixupSourceAndSinkEdges(body_graph.get()); in FunctionalizeLoop()
489 TF_RETURN_IF_ERROR(FunctionalizeCond(body_graph.get(), library)); in FunctionalizeLoop()
493 << " body: " << DumpGraphToFile("loop_body", *body_graph); in FunctionalizeLoop()
506 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()
/external/tensorflow/tensorflow/python/kernel_tests/
Dwhile_v2_test.py299 body_graph = while_v2._get_graph(while_op, "body")
301 x_input_t = body_graph.inputs[x_input_index]
330 body_graph = while_v2._get_graph(while_op, "body")
331 y_input_t = body_graph.inputs[idx]
334 output_idx = body_graph.outputs.index(push_back_node.outputs[0])
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc1123 Graph* body_graph = body_fbody->graph; in RewriteHostWhileLoopBody() local
1125 for (Node* n : body_graph->nodes()) { in RewriteHostWhileLoopBody()
1145 Node* ret_node = body_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopBody()
1147 body_graph->AddEdge(key_arg, 0, ret_node, 0); in RewriteHostWhileLoopBody()
1150 TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(body_graph)); in RewriteHostWhileLoopBody()
1155 GraphToFunctionDef(*body_graph, body_host_func_name, &body_replace_fdef)); in RewriteHostWhileLoopBody()
Dencapsulate_xla_computations_pass_test.cc301 std::unique_ptr<Graph> body_graph = MakeBodyGraph(); in TEST() local
303 TF_ASSERT_OK(GraphToFunctionDef(*body_graph, "launch0", flib.add_function())); in TEST()
/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()