Searched refs:tpu_cluster (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_extract_outside_compilation.cc | 199 tf_device::ClusterOp tpu_cluster, Operation* op) { in GetExternalOperands() argument
207 tpu_cluster->isAncestor(defining_op) && in GetExternalOperands()
265 void MoveOpsToHost(tf_device::ClusterOp tpu_cluster, Block* src, in MoveOpsToHost() argument
276 auto external_operands = GetExternalOperands(tpu_cluster, &op); in MoveOpsToHost()
357 void DecomposeControlFlow(tf_device::ClusterOp tpu_cluster, in DecomposeControlFlow() argument
360 tpu_cluster.GetBody().walk([&](Operation* op) { in DecomposeControlFlow()
365 MoveOpsToHost(tpu_cluster, &if_op.then_branch().front(), in DecomposeControlFlow()
368 MoveOpsToHost(tpu_cluster, &if_op.else_branch().front(), in DecomposeControlFlow()
396 MoveOpsToHost(tpu_cluster, &while_op.cond().front(), in DecomposeControlFlow()
399 MoveOpsToHost(tpu_cluster, &while_op.body().front(), in DecomposeControlFlow()
[all …]
|
| D | outside_compiled_to_host_launch.cc | 88 auto result = module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation() argument
90 (void)tensorflow::GetHostDeviceOutsideComputation(devices, tpu_cluster, in runOnOperation()
92 tpu_cluster.walk([&](Operation* op) { in runOnOperation()
|
| D | mark_ops_for_outside_compilation.cc | 228 void MarkVariantInputsOutputs(tf_device::ClusterOp tpu_cluster) { in MarkVariantInputsOutputs() argument
230 tpu_cluster.walk([&](Operation* op) { in MarkVariantInputsOutputs()
|