Searched refs:body_output (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 161 TfLiteTensor* body_output = in Prepare() local
163 TF_LITE_ENSURE_EQ(context, body_input->type, body_output->type); in Prepare()
166 TF_LITE_ENSURE(context, !IsDynamicTensor(body_output)); in Prepare()
167 if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { in Prepare()
183 TfLiteTensor* body_output = in Prepare() local
185 TfLiteIntArray* output_size = TfLiteIntArrayCopy(body_output->dims); in Prepare()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 183 std::unique_ptr<Graph>* body_output) { in BuildLoopBody() argument
185 *body_output = absl::make_unique<Graph>(graph.op_registry()); in BuildLoopBody()
186 Graph* output = body_output->get(); in BuildLoopBody()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 414 body_output = self._body_outputs[i]
417 body_output)
499 def true_fn(control_inputs, body_pfor, body_output, stacked): argument
512 output = body_pfor.convert(body_output)
514 output, convert_stacked, _ = body_pfor._convert_helper(body_output)
515 assert convert_stacked == stacked, body_output
524 for i, (body_output, stacked) in enumerate(
527 out_dtype = body_output.dtype
535 lambda: true_fn(control_inp, body_pfor, body_output, stacked),
|