Searched refs:body_result (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 566 xla::StatusOr<xla::XlaComputation> body_result = in Compile() local
569 OP_REQUIRES_OK(ctx, body_result.status()); in Compile()
570 xla::XlaComputation wrapped_body = std::move(body_result.ValueOrDie()); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 2223 body_result = body(*packed_vars_for_body)
2225 if not nest.is_sequence_or_composite(body_result):
2226 body_result = [body_result]
2239 body_result = nest.map_structure(
2240 map_fn, body_result, expand_composites=True)
2247 list(packed_vars_for_body), list(body_result), expand_composites=True)
2250 original_body_result = body_result
2254 nest.flatten(body_result, expand_composites=True),
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 352 Value body_result = body_info.GetValue(result.getResultNumber()); in ResourceAliasAnalysisInfo() local
353 PropagateInputToOutput(body_result, result); in ResourceAliasAnalysisInfo()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2837 TypeRange body_result, in VerifyWhileTypes() argument
2843 {body_result, "body result"}, in VerifyWhileTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3787 const HloInstruction* body_result = body->root_instruction(); in CheckWhileBuffersShareAllocation() local
3795 xla_while, body_result, index, buffer_assignment)); in CheckWhileBuffersShareAllocation()
|