Searched refs:merged_execute (Results 1 – 1 of 1) sorted by relevance
541 auto merged_execute = builder->create<TF::TPUExecuteAndUpdateVariablesOp>( in MergeForOneTPUExecute() local553 merged_execute.getLoc(), execute_launch.deviceAttr(), in MergeForOneTPUExecute()554 merged_execute.getResultTypes()); in MergeForOneTPUExecute()558 builder->create<tf_device::ReturnOp>(merged_execute.getLoc(), in MergeForOneTPUExecute()559 merged_execute.getResults()); in MergeForOneTPUExecute()561 merged_execute.getOperation()->moveBefore( in MergeForOneTPUExecute()