Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc43 Block* lhlo_block, OpBuilder* b) { in ApplySingleResultLhloCode() argument
45 for (auto arg_type : lhlo_block->getArgumentTypes()) { in ApplySingleResultLhloCode()
53 mapping.map(lhlo_block->getArguments(), arg_bufs); in ApplySingleResultLhloCode()
54 for (auto& nested : lhlo_block->without_terminator()) { in ApplySingleResultLhloCode()
67 Block* lhlo_block, OpBuilder* b) { in ConvertToReductionOperator() argument
73 lhlo_block, b)); in ConvertToReductionOperator()