Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dregion_control_flow_to_functional.cc90 auto const_builder = OpBuilder::atBlockBegin(&region->front()); in CollectExternValues()
111 auto builder = OpBuilder::atBlockBegin(module.getBody()); in ExtractSingleBlockRegion()
Dresource_op_lifting.cc165 auto builder = OpBuilder::atBlockBegin(block); in SetAllVarIsInitializedToTrue()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtensorflow_abi_knowledge_propagation.cc93 OpBuilder b = OpBuilder::atBlockBegin(&kernel.body().front()); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dinitialize_variables.cc147 auto funcBuilder = OpBuilder::atBlockBegin(func.addEntryBlock()); in CreateSessionInitFunc()
/external/llvm-project/mlir/lib/Dialect/Affine/Transforms/
DAffineLoopNormalize.cpp38 auto builder = OpBuilder::atBlockBegin(op.getBody()); in normalizeAffineParallel()
/external/llvm-project/mlir/include/mlir/IR/
DBuilders.h211 static OpBuilder atBlockBegin(Block *block, Listener *listener = nullptr) {
/external/llvm-project/mlir/lib/Bindings/Python/
DIRModules.h556 static PyInsertionPoint atBlockBegin(PyBlock &block);
DIRModules.cpp1079 PyInsertionPoint PyInsertionPoint::atBlockBegin(PyBlock &block) { in atBlockBegin() function in PyInsertionPoint
2981 .def_static("at_block_begin", &PyInsertionPoint::atBlockBegin, in populateIRSubmodule()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc253 builder_ = OpBuilder::atBlockBegin(func_.addEntryBlock()); in MlirFunctionContext()
/external/llvm-project/mlir/lib/Conversion/PDLToPDLInterp/
DPDLToPDLInterp.cpp670 OpBuilder builder = OpBuilder::atBlockBegin(module.getBody()); in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/
DAsyncToLLVM.cpp305 OpBuilder builder = OpBuilder::atBlockBegin(entryBlock); in setupCoroMachinery()
/external/llvm-project/mlir/lib/Transforms/Utils/
DLoopUtils.cpp1965 OpBuilder innerBuilder = OpBuilder::atBlockBegin(inner.getBody()); in normalizeLoop()
2040 OpBuilder insideLoopBuilder = OpBuilder::atBlockBegin(loops.getBody()); in collapseParallelLoops()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc484 auto region_builder = OpBuilder::atBlockBegin(&fusion.region().front()); in EmitFusionOp()