Home
last modified time | relevance | path

Searched refs:EncapsulateFuncAndSerialize (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dremote_run_encapsulate.cc73 LogicalResult EncapsulateFuncAndSerialize(func::FuncOp entry_func, in EncapsulateFuncAndSerialize() function
136 if (failed(EncapsulateFuncAndSerialize(callee, &txt_module))) { in runOnOperation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_extract_outside_compilation.cc105 void EncapsulateFuncAndSerialize(func::FuncOp func, in EncapsulateFuncAndSerialize() function
534 EncapsulateFuncAndSerialize(shape_op, &serialized_func_module); in MoveOpsToHost()
H A Dtpu_rewrite_pass.cc87 LogicalResult EncapsulateFuncAndSerialize(func::FuncOp entry_func, in EncapsulateFuncAndSerialize() function
369 if (failed(EncapsulateFuncAndSerialize(func, &txt_module))) return nullptr; in BuildCompileOp()