Searched defs:compile_launch (Results 1 – 2 of 2) sorted by relevance
150 tf_device::LaunchOp compile_launch, in BuildGetLayout()165 tf_device::LaunchOp compile_launch, in BuildCopyWithLayout()176 tf_device::LaunchOp compile_launch) { in HandleInput()191 tf_device::LaunchOp execute_launch, tf_device::LaunchOp compile_launch, in HandleReplicatedInputs()228 tf_device::LaunchOp compile_launch, in HandleCompileAndExecutes()292 auto compile_launch = in runOnFunction() local
157 tf_device::LaunchOp compile_launch) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()417 auto compile_launch = llvm::dyn_cast<tf_device::LaunchOp>(compile); in HandleReplicateOp() local