Searched refs:CloneWithReplacementPairs (Results 1 – 3 of 3) sorted by relevance
355 std::unique_ptr<HloComputation> CloneWithReplacementPairs(358 std::unique_ptr<HloComputation> CloneWithReplacementPairs(362 std::unique_ptr<HloComputation> CloneWithReplacementPairs(
411 while_cond->CloneWithReplacementPairs({ in TryRemoveConstantParams()419 while_body->CloneWithReplacementPairs( in TryRemoveConstantParams()703 while_cond->CloneWithReplacementPairs({ in TryFlattenNestedTuples()714 while_body->CloneWithReplacementPairs( in TryFlattenNestedTuples()929 while_cond->CloneWithReplacementPairs({ in TryMergeInductionVariables()946 module->AddEmbeddedComputation(while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()953 temp_new_while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()
892 std::unique_ptr<HloComputation> HloComputation::CloneWithReplacementPairs( in CloneWithReplacementPairs() function in xla::HloComputation902 std::unique_ptr<HloComputation> HloComputation::CloneWithReplacementPairs( in CloneWithReplacementPairs() function in xla::HloComputation914 std::unique_ptr<HloComputation> HloComputation::CloneWithReplacementPairs( in CloneWithReplacementPairs() function in xla::HloComputation