Home
last modified time | relevance | path

Searched defs:tpu_cluster (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_outside_compilation.cc309 tf_device::ClusterOp tpu_cluster, in GetStaticExternalOperands()
358 tf_device::ClusterOp tpu_cluster, in GetExternalOperands()
591 LogicalResult MoveOpsToHost(tf_device::ClusterOp tpu_cluster, Block* src, in MoveOpsToHost()
647 LogicalResult DecomposeControlFlow(tf_device::ClusterOp tpu_cluster, in DecomposeControlFlow()
723 ModuleOp module, tf_device::ClusterOp tpu_cluster, in CreateParallelExecuteForOutsideCompilation()
801 module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
Dhost_launch_to_outside_compiled.cc72 module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
Doutside_compiled_to_host_launch.cc67 auto result = module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
Dmark_ops_for_outside_compilation.cc272 void MarkVariantInputsOutputs(tf_device::ClusterOp tpu_cluster) { in MarkVariantInputsOutputs()