Searched refs:body_result (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 517 xla::StatusOr<xla::XlaComputation> body_result = in Compile() local
520 OP_REQUIRES_OK(ctx, body_result.status()); in Compile()
521 xla::XlaComputation wrapped_body = std::move(body_result.ValueOrDie()); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 2923 body_result = body(*packed_vars_for_body)
2925 if not nest.is_sequence_or_composite(body_result):
2926 body_result = [body_result]
2939 body_result = nest.map_structure(map_fn, body_result,
2946 nest.assert_same_structure(list(packed_vars_for_body), list(body_result),
2950 original_body_result = body_result
2954 nest.flatten(body_result, expand_composites=True),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2064 const HloInstruction* body_result = body->root_instruction(); in CheckWhileBuffersShareAllocation() local
2072 xla_while, body_result, index, buffer_assignment)); in CheckWhileBuffersShareAllocation()
|