Searched refs:outlined_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 64 FuncOp outlined_func = in BuildFunction() local
69 outlined_func.setPrivate(); in BuildFunction()
72 Block* outlined_func_block = outlined_func.addEntryBlock(); in BuildFunction()
95 symbol_table->insert(outlined_func); in BuildFunction()
96 return outlined_func; in BuildFunction()
107 FuncOp outlined_func = in OutlineCluster() local
110 builder->getSymbolRefAttr(outlined_func.getName())); in OutlineCluster()
114 cluster_op.getLoc(), outlined_func.getType().getResults(), in OutlineCluster()
|
| D | executor_tpuv1_outline_tpu_island.cc | 108 auto outlined_func = in runOnOperation() local
110 outlined_symbol_table.insert(outlined_func); in runOnOperation()
111 outlined_func.setNested(); in runOnOperation()
117 outlined_func.getBody().takeBody(island_op.body()); in runOnOperation()
129 Block &entry_block = outlined_func.getBody().front(); in runOnOperation()
132 replaceAllUsesInRegionWith(operand, newArg, outlined_func.getBody()); in runOnOperation()
142 kNestedModule, builder.getSymbolRefAttr(outlined_func.getName())), in runOnOperation()
|
| D | region_control_flow_to_functional.cc | 131 auto outlined_func = builder.create<FuncOp>(loc, name, type); in ExtractSingleBlockRegion() local
132 Region& func_region = outlined_func.getBody(); in ExtractSingleBlockRegion()
156 outlined_func.setPrivate(); in ExtractSingleBlockRegion()
160 worklist.push_back(outlined_func); in ExtractSingleBlockRegion()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | while_loop_outline.cc | 169 auto outlined_func = builder.create<FuncOp>(while_op.getLoc(), name, type); in OutlineWhile() local
170 outlined_func.getBody().takeBody(region); in OutlineWhile()
171 Region& func_region = outlined_func.getBody(); in OutlineWhile()
184 Operation* yield_op = outlined_func.getBody().front().getTerminator(); in OutlineWhile()
214 symbol_table.insert(outlined_func); in OutlineWhile()
215 outlined_func.setPrivate(); in OutlineWhile()
216 return outlined_func; in OutlineWhile()
|