Searched defs:launch_op (Results 1 – 8 of 8) sorted by relevance
104 tf_device::LaunchOp launch_op) { in ReplaceLiveOutExternalUses()170 tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>( in BuildLaunchForCluster() local
64 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
111 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithDevice() local
186 auto launch_op = builder.create<tf_device::LaunchOp>( in CreateLaunchOpForOutsideCluster() local
696 tf_device::LaunchOp launch_op = AssignDevicesToReplicatedExecute( in Rewrite() local
945 if (auto launch_op = dyn_cast<tf_device::LaunchOp>(op)) { in InferShapeForNonTFDialectOperation() local
97 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in generateParamsArray()145 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in matchAndRewrite()
90 auto launch_op = rewriter.create<mlir::gpu::LaunchOp>( in matchAndRewrite() local