Searched defs:tpu_cluster (Results 1 – 3 of 3) sorted by relevance
199 tf_device::ClusterOp tpu_cluster, Operation* op) { in GetExternalOperands()265 void MoveOpsToHost(tf_device::ClusterOp tpu_cluster, Block* src, in MoveOpsToHost()357 void DecomposeControlFlow(tf_device::ClusterOp tpu_cluster, in DecomposeControlFlow()423 ModuleOp module, tf_device::ClusterOp tpu_cluster, in CreateParallelExecuteForOutsideCompilation()497 module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
88 auto result = module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
228 void MarkVariantInputsOutputs(tf_device::ClusterOp tpu_cluster) { in MarkVariantInputsOutputs()