Home
last modified time | relevance | path

Searched defs:cluster_region (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_host_computation_expansion.cc65 Region* cluster_region = &cluster.body(); in ExpandHeadOutsideCompiledOps() local
Dtpu_extract_head_tail_outside_compilation.cc138 Region* cluster_region = &cluster.body(); in FindOutsideCompiledOpsAtHead() local
236 Region* cluster_region = &cluster.body(); in FindOutsideCompiledOpsAtTailAndClusterResults() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dhandle_cross_cluster_dependencies.cc273 mlir::Region& cluster_region = cluster.body(); in ReplaceCopyToMeshWithVirtualSendRecv() local