Searched refs:body_parameter (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | params_test.cc | 393 auto body_parameter = Parameter(&builder, 0, while_shape, "body_parameter"); in XLA_TEST_F() local
396 auto add = Add(GetTupleElement(body_parameter, i), in XLA_TEST_F()
401 updates.push_back(GetTupleElement(body_parameter, kParamCount)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 290 auto* body_parameter = in AllUsersConsumeBF16() local
293 if (OutputTypeAfterChange(body_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3786 const HloInstruction* body_parameter = body->parameter_instruction(0); in CheckWhileBuffersShareAllocation() local
3793 xla_while, body_parameter, index, buffer_assignment)); in CheckWhileBuffersShareAllocation()
|