Searched defs:compile_launch (Results 1 – 2 of 2) sorted by relevance
126 tf_device::LaunchOp compile_launch, in BuildGetLayout()141 tf_device::LaunchOp compile_launch, in BuildCopyWithLayout()152 tf_device::LaunchOp compile_launch) { in HandleInput()167 tf_device::LaunchOp execute_launch, tf_device::LaunchOp compile_launch, in HandleReplicatedInputs()204 tf_device::LaunchOp compile_launch, in HandleCompileAndExecutes()268 auto compile_launch = in runOnFunction() local
101 tf_device::LaunchOp compile_launch) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()361 auto compile_launch = llvm::dyn_cast<tf_device::LaunchOp>(compile); in HandleReplicateOp() local