Home
last modified time | relevance | path

Searched defs:fn_builder (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc246 mlir::OpBuilder fn_builder = mlir::OpBuilder::atBlockBegin(fn_block); in ConditionalSave() local
269 mlir::OpBuilder fn_builder = mlir::OpBuilder::atBlockBegin(fn_block); in ConditionalSave() local
617 mlir::OpBuilder fn_builder = mlir::OpBuilder::atBlockBegin(fn_block); in ExpandRestoreV2OpHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc141 mlir::OpBuilder fn_builder = mlir::OpBuilder::atBlockEnd(fn_block); in CreateSendRecvOpsToTransferProgramKey() local