Searched refs:body_graph (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | while_loop_test.cc | 142 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 …]
|
D | c_api.cc | 2169 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(¶ms->body_graph->graph, &parent->graph, in TF_FinishWhileHelper()
|
D | c_api_function_test.cc | 925 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()
|
D | c_api.h | 1077 TF_Graph* const body_graph; member
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_control_flow.cc | 470 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()
|