HomeSort by relevance Sort by last modified time
    Searched refs:new_while_op (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
while_loop_simplifier_test.cc 399 HloInstruction* new_while_op = *std::find_if( local
406 ShapeUtil::Equal(new_while_op->shape(),
408 << ShapeUtil::HumanString(new_while_op->shape());
410 new_while_op->while_body()->root_instruction(),
416 EXPECT_THAT(new_while_op->while_condition()->root_instruction(),
while_loop_simplifier.cc 508 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( local
538 new_while_op->shape().tuple_shapes(gte_idx), new_while_op,

Completed in 70 milliseconds