Searched refs:cluster_func_op (Results 1 – 2 of 2) sorted by relevance
116 auto cluster_func_op = builder->create<tf_device::ClusterFuncOp>( in OutlineCluster() local120 cluster_op.replaceAllUsesWith(cluster_func_op); in OutlineCluster()139 auto cluster_func_op = builder->create<tf_device::LaunchFuncOp>( in OutlineLaunch() local143 launch_op.replaceAllUsesWith(cluster_func_op); in OutlineLaunch()
2149 auto cluster_func_op = builder.create<tfrt::jitrt::CallOp>( in OutlineClusterOp() local2153 cluster.replaceAllUsesWith(cluster_func_op); in OutlineClusterOp()