Home
last modified time | relevance | path

Searched refs:TPUMergeVariablesWithExecutePass (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_merge_variables_with_execute.cc63 class TPUMergeVariablesWithExecutePass class
65 TPUMergeVariablesWithExecutePass> {
557 void TPUMergeVariablesWithExecutePass::runOnOperation() { in runOnOperation()
601 return std::make_unique<TPUMergeVariablesWithExecutePass>(); in CreateTPUMergeVariablesWithExecutePass()
H A Dtf_passes.td1589 def TPUMergeVariablesWithExecutePass : Pass<"tf-tpu-merge-variables-with-execute", "ModuleOp"> {