Home
last modified time | relevance | path

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

/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.cc2169 if (params.cond_graph == nullptr || params.body_graph == nullptr || in ValidateConstWhileParams()
2171 params.cond_graph->parent != params.body_graph->parent || in ValidateConstWhileParams()
2172 params.cond_graph->parent_inputs != params.body_graph->parent_inputs || in ValidateConstWhileParams()
2205 TF_DeleteGraph(params->body_graph); in FreeWhileResources()
2234 TF_Graph* body_graph = TF_NewGraph(); in TF_NewWhile()
2237 body_graph->parent = g; in TF_NewWhile()
2238 body_graph->parent_inputs = inputs; in TF_NewWhile()
2253 if (!CreateInput(inputs[i], body_graph, StrCat("body_input", i).c_str(), in TF_NewWhile()
2260 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2308 CopyGraph(&params->body_graph->graph, &parent->graph, in TF_FinishWhileHelper()
Dc_api_function_test.cc925 params->body_graph, s_, "add1"); in TEST_F()
927 TF_Operation* one = ScalarConst(1, params->body_graph, s_); in TEST_F()
929 TF_Operation* add2 = Add(add1, one, params->body_graph, s_, "add2"); in TEST_F()
Dc_api.h1077 TF_Graph* const body_graph; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc470 std::unique_ptr<Graph> body_graph; in FunctionalizeLoop() local
471 TF_RETURN_IF_ERROR(BuildLoopBody(*graph, frame, &arg_types, &body_graph)); in FunctionalizeLoop()
475 << " body: " << dump_graph::DumpGraphToFile("loop_body", *body_graph); in FunctionalizeLoop()
488 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()