HomeSort by relevance Sort by last modified time
    Searched defs:while_body (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
while_loop_invariant_code_motion_test.cc 64 HloComputation* while_body = [&]() { local
65 HloComputation::Builder builder(TestName() + ".while_body");
86 while_body, init_value));
97 EXPECT_THAT(transformed_while->while_body()->instructions(),
106 HloComputation* while_body = [&]() { local
107 HloComputation::Builder builder(TestName() + ".while_body");
145 while_body, init_value));
164 EXPECT_THAT(transformed_while->while_body()->instructions(),
168 EXPECT_THAT(transformed_while->while_body()->instructions(),
178 HloComputation* while_body = [&]() local
216 HloComputation* while_body = [&]() { local
252 HloComputation* while_body = [&]() { local
291 HloComputation* while_body = [&]() { local
334 HloComputation* while_body = [&]() { local
383 HloComputation* while_body; local
418 HloComputation* while_body = [&]() { local
    [all...]
while_loop_invariant_code_motion.cc 40 HloComputation* while_body = while_instr->while_body(); local
50 HloInstruction* while_body_param = while_body->parameter_instruction(0);
126 // Populates `gte_set` with the GetTupleElement instructions in `while_body`
128 // iterations. Assumes `while_body` is the body computation of the while loop
130 static void GatherInvariantGTEs(HloComputation* while_body,
133 while_body->root_instruction()->operands();
138 instr->operand(0) == while_body->parameter_instruction(0) &&
152 // while_body(f32[1024, 1024] %param) {
167 HloComputation* while_body = while_instr->while_body() local
    [all...]
while_loop_simplifier_test.cc 130 auto* true_op = while_op->while_body()->AddInstruction(
133 while_op->while_body()->root_instruction()));
146 auto* while_body = while_op->while_body(); local
147 auto* send = while_body->AddInstruction(HloInstruction::CreateSend(
148 while_body->AddInstruction(
151 while_body->AddInstruction(HloInstruction::CreateSendDone(send));
159 auto* while_body = while_op->while_body(); local
160 auto* recv = while_body->AddInstruction
175 auto* while_body = while_op->while_body(); local
425 HloComputation* while_body = [&]() { local
    [all...]
while_loop_simplifier.cc 148 // The while_body computation should have the form
155 auto* while_body = while_op->while_body(); local
156 auto* while_body_root = while_body->root_instruction();
164 auto* while_body_param = while_body->parameter_instruction(0);
203 // where comp is while_op->while_body() or while_op->while_condition().
256 auto* while_body = while_op->while_body(); local
258 while_body->root_instruction()->operand(*indvar_tuple_idx);
301 HloComputation* while_body = while_op->while_body() local
    [all...]
hlo_verifier.cc 286 xla_while->while_body()->ComputeProgramShape().result());
772 auto* while_body = instruction->while_body(); local
776 TF_RET_CHECK(while_body->num_parameters() == 1)
778 << while_body->num_parameters() << ": " << while_body->ToString();
795 auto* body_param = while_body->parameter_instruction(0)
    [all...]
hlo_instruction.cc 2012 HloComputation* HloInstruction::while_body() const { function in class:xla::HloInstruction
    [all...]

Completed in 233 milliseconds