| /external/tensorflow/tensorflow/core/graph/ |
| while_context.h | 38 std::vector<Node*> exit_nodes, OutputTensor cond_output,
45 const OutputTensor& cond_output() const { return cond_output_; } function in class:tensorflow::WhileContext
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| while.cc | 71 const TfLiteTensor* cond_output) {
73 TF_LITE_ENSURE_EQ(context, cond_output->type, kTfLiteBool);
74 if (cond_output->dims->size == 0) {
79 TF_LITE_ENSURE_EQ(context, cond_output->dims->size, 1);
80 TF_LITE_ENSURE_EQ(context, cond_output->dims->data[0], 1);
137 TfLiteTensor* cond_output = local
143 if (IsDynamicTensor(cond_output)) {
146 TF_LITE_ENSURE_STATUS(CheckCondOutput(context, cond_output));
244 TfLiteTensor* cond_output = local
247 TF_LITE_ENSURE_STATUS(CheckCondOutput(context, cond_output));
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.h | 1122 TF_Output cond_output; member in struct:TF_WhileParams
[all...] |
