Searched defs:func_builder (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 94 mlir::OpBuilder* func_builder) { in GenerateInstructionLocation()
160 mlir::OpBuilder* func_builder, mlir::Location loc, mlir::Operation* op, in CreateTupleFromOpResults()
206 mlir::OpBuilder* func_builder, mlir::Location loc, Value value, in FlattenTupleValue()
223 mlir::OpBuilder* func_builder, mlir::Location loc, in CreateTupleValue()
485 mlir::OpBuilder* func_builder, DynamicShapeHandlingMode mode) { in ImportInstructionImpl()
1550 mlir::OpBuilder* func_builder, DynamicShapeHandlingMode mode) { in ImportInstructionWithLayout()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | initialize_variables_in_session_init.cc | 72 auto func_builder = OpBuilder::atBlockBegin(func.addEntryBlock()); in CreateSessionInitFunc() local
|