Searched refs:cond_graph (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | while_loop_test.cc | 108 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()
|
D | c_api.cc | 2169 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 …]
|
D | c_api.h | 1071 TF_Graph* const cond_graph; member
|
D | c_api_function_test.cc | 920 params->cond_inputs[0], params->cond_inputs[1], params->cond_graph, s_); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_control_flow.cc | 467 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()
|