Searched refs:body_output (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 182 TfLiteTensor* body_output = in Prepare() local
184 TF_LITE_ENSURE_TYPES_EQ(context, body_input->type, body_output->type); in Prepare()
186 TF_LITE_ENSURE(context, !IsDynamicTensor(body_output)); in Prepare()
187 if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { in Prepare()
204 TfLiteTensor* body_output = in Prepare() local
206 TfLiteIntArray* output_size = TfLiteIntArrayCopy(body_output->dims); in Prepare()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 160 std::unique_ptr<Graph>* body_output) { in BuildLoopBody() argument
162 *body_output = absl::make_unique<Graph>(graph.op_registry()); in BuildLoopBody()
163 Graph* output = body_output->get(); in BuildLoopBody()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 527 body_output = self._body_outputs[i]
530 body_output)
610 def true_fn(control_inputs, body_pfor, body_output, stacked): argument
623 output = body_pfor.convert(body_output)
625 output, convert_stacked, _ = body_pfor._convert_helper(body_output)
626 assert convert_stacked == stacked, body_output
634 for i, (body_output,
637 out_dtype = body_output.dtype
645 lambda: true_fn(control_inp, body_pfor, body_output, stacked),
|