| /external/tensorflow/tensorflow/c/ |
| while_loop_test.cc | 108 TF_Operation* one = ScalarConst(1, params_->cond_graph, s_);
110 LessThan(params_->cond_inputs[0], {one, 0}, params_->cond_graph, s_);
143 EXPECT_TRUE(params_->cond_graph != nullptr);
160 params_->cond_graph, s_);
227 TF_Operation* six = ScalarConst(6, params_->cond_graph, s_);
230 LessThan(params_->cond_inputs[0], {six, 0}, params_->cond_graph, s_);
243 TF_Operation* three = ScalarConst(3, inner_params.cond_graph, s_);
246 inner_params.cond_inputs[0], {three, 0}, inner_params.cond_graph, s_);
418 TF_Operation* ten = ScalarConst(10, params_->cond_graph, s_);
420 LessThan(params_->cond_inputs[0], {ten, 0}, params_->cond_graph, s_)
[all...] |
| c_api.cc | [all...] |
| c_api.h | 1071 TF_Graph* const cond_graph; member in struct:TF_WhileParams
[all...] |
| c_api_function_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 467 std::unique_ptr<Graph> cond_graph; local
468 TF_RETURN_IF_ERROR(BuildLoopCondition(*graph, frame, &cond_graph));
474 << dump_graph::DumpGraphToFile("loop_condition", *cond_graph, library)
485 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef));
[all...] |
