Home
last modified time | relevance | path

Searched refs:maybe_replicate (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc232 auto maybe_replicate = in HandleCompileAndExecutes() local
248 if (maybe_replicate != block_arg.getParentRegion()->getParentOp() || in HandleCompileAndExecutes()
251 maybe_replicate, resource_alias_analysis)) { in HandleCompileAndExecutes()
260 if (maybe_replicate && in HandleCompileAndExecutes()
261 maybe_replicate.body().isAncestor(input_op->getParentRegion())) { in HandleCompileAndExecutes()
Dtpu_space_to_depth_pass.cc482 auto maybe_replicate = in HandleCluster() local
491 if (maybe_replicate == block_arg.getParentRegion()->getParentOp()) { in HandleCluster()
493 maybe_replicate, block_size); in HandleCluster()
502 if (maybe_replicate && in HandleCluster()
503 maybe_replicate.body().isAncestor(input_op->getParentRegion())) { in HandleCluster()