Home
last modified time | relevance | path

Searched refs:body_parameter (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dparams_test.cc393 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/
Dbfloat16_propagation.cc290 auto* body_parameter = in AllUsersConsumeBF16() local
293 if (OutputTypeAfterChange(body_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3786 const HloInstruction* body_parameter = body->parameter_instruction(0); in CheckWhileBuffersShareAllocation() local
3793 xla_while, body_parameter, index, buffer_assignment)); in CheckWhileBuffersShareAllocation()