Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_cluster_util.cc41 tf_device::ClusterOp tpu_cluster) { in WalkReachableFromTpuCluster()
50 auto result = module.walk([&](tf_device::ClusterOp tpu_cluster) { in WalkReachableFromTpuCluster()
76 tf_device::ClusterOp tpu_cluster = pair.second; in WalkReachableFromTpuCluster() local
108 std::optional<std::string> host_device) { in WalkReachableFromTpuCluster()
/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()
590 LogicalResult MoveOpsToHost(tf_device::ClusterOp tpu_cluster, Block* src, in MoveOpsToHost()
646 LogicalResult DecomposeControlFlow(tf_device::ClusterOp tpu_cluster, in DecomposeControlFlow()
722 ModuleOp module, tf_device::ClusterOp tpu_cluster, in CreateParallelExecuteForOutsideCompilation()
800 module.walk([&](tf_device::ClusterOp tpu_cluster) { in runOnOperation()
Dtpu_cluster_cleanup_attributes.cc40 auto traverse_op = [&](Operation* op, tf_device::ClusterOp tpu_cluster) { in runOnOperation()
Doutside_compiled_to_host_launch.cc63 std::optional<std::string> host_device) { in runOnOperation()
Dhost_launch_to_outside_compiled.cc70 std::optional<std::string> host_device) { in runOnOperation()
Dmark_ops_for_outside_compilation.cc274 void MarkVariantInputsOutputs(tf_device::ClusterOp tpu_cluster) { in MarkVariantInputsOutputs()