Searched refs:IsSafelyRemovable (Results 1 – 7 of 7) sorted by relevance
38 if (comp->IsSafelyRemovable(instruction) && in Run()
53 computation->IsSafelyRemovable(instruction) && in RunOnComputation()
228 bool HloComputation::IsSafelyRemovable(const HloInstruction* instruction) { in IsSafelyRemovable() function in xla::HloComputation263 TF_RET_CHECK(IsSafelyRemovable(instruction)) in RemoveInstructionAndUnusedOperands()273 item == root_instruction() || !IsSafelyRemovable(item) || in RemoveInstructionAndUnusedOperands()302 TF_RET_CHECK(ignore_safety_check || IsSafelyRemovable(instruction)) in RemoveInstructionImpl()
219 if (!while_op->parent()->IsSafelyRemovable(while_op)) { in TryRemoveDeadWhileParams()393 if (!while_op->parent()->IsSafelyRemovable(while_op)) { in TryRemoveRepeatedWhileTupleIndices()640 if (!while_op->parent()->IsSafelyRemovable(while_op)) { in TryRemoveWhileLoop()
442 bool IsSafelyRemovable(const HloInstruction* instruction);
74 if (!conditional->parent()->IsSafelyRemovable(conditional) || in TryRemoveConditional()
3037 if (computation->IsSafelyRemovable(instruction) && in SimplifyGraph()