Searched refs:new_wrapped_computation (Results 1 – 1 of 1) sorted by relevance
336 HloComputation* new_wrapped_computation = nullptr; in CloneWithNewOperandsImpl() local338 new_wrapped_computation = in CloneWithNewOperandsImpl()341 if (new_wrapped_computation == nullptr) { in CloneWithNewOperandsImpl()342 new_wrapped_computation = module->AddEmbeddedComputation( in CloneWithNewOperandsImpl()346 opcode(), shape, new_operands, new_wrapped_computation, async_group_id_, in CloneWithNewOperandsImpl()