Searched defs:execute_launch (Results 1 – 4 of 4) sorted by relevance
140 TF::TPUCopyWithLayoutOp BuildCopyWithLayout(tf_device::LaunchOp execute_launch, in BuildCopyWithLayout()151 TF::TPUExecuteOp execute, tf_device::LaunchOp execute_launch, in HandleInput()167 tf_device::LaunchOp execute_launch, tf_device::LaunchOp compile_launch, in HandleReplicatedInputs()220 auto& execute_launch = std::get<0>(execute_and_input_mapping); in HandleCompileAndExecutes() local279 auto execute_launch = in runOnFunction() local
152 tf_device::LaunchOp execute_launch, in BuildVariableAccessInfo()362 tf_device::LaunchOp execute_launch, in ReplaceParallelExecute()438 void ReplaceExecute(tf_device::LaunchOp execute_launch, in ReplaceExecute()457 tf_device::LaunchOp execute_launch, in MergeForOneTPUExecute()573 for (auto execute_launch : execute_launches) { in runOnOperation() local
121 for (auto execute_launch : execute_launches) { in runOnOperation() local
339 tf_device::LaunchOp execute_launch; in HandleReplicateOp() local