Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc108 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_); in CreateCondGraph()
110 LessThan(params_->cond_inputs[0], {one, 0}, params_->cond_graph, s_); in CreateCondGraph()
143 EXPECT_TRUE(params_->cond_graph != nullptr); in TEST_F()
160 params_->cond_graph, s_); in TEST_F()
227 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_); in TEST_F()
230 LessThan(params_->cond_inputs[0], {six, 0}, params_->cond_graph, s_); in TEST_F()
243 TF_Operation* three = ScalarConst(3, inner_params.cond_graph, s_); in TEST_F()
246 inner_params.cond_inputs[0], {three, 0}, inner_params.cond_graph, s_); in TEST_F()
418 TF_Operation* ten = ScalarConst(10, params_->cond_graph, s_); in TEST_F()
420 LessThan(params_->cond_inputs[0], {ten, 0}, params_->cond_graph, s_); in TEST_F()
Dc_api.cc2169 if (params.cond_graph == nullptr || params.body_graph == nullptr || in ValidateConstWhileParams()
2170 params.cond_graph->parent == 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()
2204 TF_DeleteGraph(params->cond_graph); in FreeWhileResources()
2233 TF_Graph* cond_graph = TF_NewGraph(); in TF_NewWhile()
2235 cond_graph->parent = g; in TF_NewWhile()
2236 cond_graph->parent_inputs = inputs; in TF_NewWhile()
2249 if (!CreateInput(inputs[i], cond_graph, StrCat("cond_input", i).c_str(), in TF_NewWhile()
2259 TF_WhileParams params = {ninputs, cond_graph, cond_inputs, cond_output, in TF_NewWhile()
[all …]
Dc_api.h1071 TF_Graph* const cond_graph; member
Dc_api_function_test.cc920 params->cond_inputs[0], params->cond_inputs[1], params->cond_graph, s_); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc467 std::unique_ptr<Graph> cond_graph; in FunctionalizeLoop() local
468 TF_RETURN_IF_ERROR(BuildLoopCondition(*graph, frame, &cond_graph)); in FunctionalizeLoop()
474 << dump_graph::DumpGraphToFile("loop_condition", *cond_graph, library) in FunctionalizeLoop()
485 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef)); in FunctionalizeLoop()