Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dexecutor_tpuv1_inline_tpu_island.cc53 Operation *nested_module = symbol_table.lookup(kNestedModule); in runOnOperation() local
54 if (!nested_module) return; in runOnOperation()
81 Block &nested_block = nested_module->getRegion(0).front(); in runOnOperation()
89 nested_module->erase(); in runOnOperation()
H A Dmaterialize_mlir_passthrough_op.cc51 auto nested_module = in runOnOperation() local
53 if (!nested_module) { in runOnOperation()
58 dyn_cast<func::FuncOp>(nested_module->lookupSymbol("main")); in runOnOperation()
H A Dexecutor_tpuv1_outline_tpu_island.cc68 if (Operation *nested_module = symbol_table.lookup(kNestedModule)) { in runOnOperation() local
69 nested_module->emitOpError("unexpected already present outlined module."); in runOnOperation()
H A Dbridge.cc216 OpPassManager &nested_module = pm.nest<ModuleOp>(); in CreateTPUBridgePipelineV1() local
217 CreateTPUBridgePipelineImpl(nested_module); in CreateTPUBridgePipelineV1()