| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 71 const TfLiteTensor* cond_output) { in CheckCondOutput() argument
73 TF_LITE_ENSURE_EQ(context, cond_output->type, kTfLiteBool); in CheckCondOutput()
74 if (cond_output->dims->size == 0) { in CheckCondOutput()
79 TF_LITE_ENSURE_EQ(context, cond_output->dims->size, 1); in CheckCondOutput()
80 TF_LITE_ENSURE_EQ(context, cond_output->dims->data[0], 1); in CheckCondOutput()
137 TfLiteTensor* cond_output = in Prepare() local
143 if (IsDynamicTensor(cond_output)) { in Prepare()
146 TF_LITE_ENSURE_STATUS(CheckCondOutput(context, cond_output)); in Prepare()
244 TfLiteTensor* cond_output = in Eval() local
247 TF_LITE_ENSURE_STATUS(CheckCondOutput(context, cond_output)); in Eval()
[all …]
|
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 112 params_->cond_output = {less_than, 0}; in CreateCondGraph()
162 params_->cond_output = {less_than, 0}; in TEST_F()
232 params_->cond_output = {less_than, 0}; in TEST_F()
248 inner_params.cond_output = {inner_less_than, 0}; in TEST_F()
306 params_->cond_output = params_->cond_inputs[0]; in TEST_F()
316 params_->cond_output = inputs_[0]; in TEST_F()
327 params_->cond_output.index = 100; in TEST_F()
422 params_->cond_output = {less_than, 0}; in TEST_F()
|
| D | c_api.cc | 2302 if (params.cond_output.oper == nullptr) { in ValidateInputWhileParams()
2360 TF_Output cond_output = {nullptr, -1}; in TF_NewWhile()
2378 TF_WhileParams params = {ninputs, cond_graph, cond_inputs, cond_output, in TF_NewWhile()
2409 std::vector<tensorflow::Output> cond_output; in TF_FinishWhileHelper() local
2413 scope.impl()->control_deps(), ¶ms->cond_output, in TF_FinishWhileHelper()
2414 /* nreturn_nodes */ 1, &cond_output)); in TF_FinishWhileHelper()
2415 *output = cond_output[0]; in TF_FinishWhileHelper()
|
| D | c_api.h | 1122 TF_Output cond_output; member
|
| D | c_api_function_test.cc | 925 params->cond_output = {less_than, 0}; in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | while_context.cc | 23 OutputTensor cond_output, in WhileContext() argument
29 cond_output_(cond_output), in WhileContext()
|
| D | while_context.h | 38 std::vector<Node*> exit_nodes, OutputTensor cond_output,
45 const OutputTensor& cond_output() const { return cond_output_; } in cond_output() function
|
| D | graph.cc | 795 OutputTensor cond_output, in AddWhileContext() argument
802 cond_output, std::move(body_inputs), in AddWhileContext()
|
| D | graph.h | 646 OutputTensor cond_output,
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.cc | 176 Output* cond_output) { in BuildWhileLoop() argument
198 if (cond_output != nullptr) *cond_output = cond_out; in BuildWhileLoop()
|
| D | while_loop.h | 69 Output* cond_output = nullptr);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 146 std::unique_ptr<Graph>* cond_output) { in BuildLoopCondition() argument
148 *cond_output = absl::make_unique<Graph>(graph.op_registry()); in BuildLoopCondition()
149 Graph* output = cond_output->get(); in BuildLoopCondition()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 278 params.cond_inputs, ¶ms.cond_output, params.ninputs, 1); in Java_org_tensorflow_Graph_whileLoop()
294 params.cond_output = {cond_output_op, in Java_org_tensorflow_Graph_whileLoop()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 70 *output = ToOutput(while_ctx->cond_output()); in AddForwardLoopCounter()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 671 cond_output, _ = self._createCond("cond")
675 sess.run(cond_output, options=run_options, run_metadata=run_metadata)
701 cond_output, cond_op = self._createCond("cond")
711 sess.run(cond_output, options=run_options, run_metadata=run_metadata)
|
| D | control_flow_ops_py_test.py | 1119 cond_output = control_flow_ops.cond(
1121 result = array_ops.identity(cond_output)
1132 cond_output = control_flow_ops.cond(
1134 result = array_ops.identity(cond_output)
|