Searched defs:compile_launch (Results 1 – 3 of 3) sorted by relevance
143 tf_device::LaunchOp compile_launch, in BuildGetLayout()158 tf_device::LaunchOp compile_launch, in BuildCopyWithLayout()169 tf_device::LaunchOp compile_launch) { in HandleInput()184 tf_device::LaunchOp execute_launch, tf_device::LaunchOp compile_launch, in HandleReplicatedInputs()221 tf_device::LaunchOp compile_launch, in HandleCompileAndExecutes()285 auto compile_launch = in runOnFunction() local
148 tf_device::LaunchOp compile_launch) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()408 auto compile_launch = llvm::dyn_cast<tf_device::LaunchOp>(compile); in HandleReplicateOp() local
428 auto compile_launch = llvm::dyn_cast_or_null<tf_device::LaunchOp>( in GetTPUCompileOp() local