Home
last modified time | relevance | path

Searched refs:CreateTPUMergeVariablesWithExecutePass (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_mlir_passes.cc309 pm->addPass(mlir::TFTPU::CreateTPUMergeVariablesWithExecutePass()); in CreateDTensorMLIRPass()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dbridge.cc186 pm.addPass(CreateTPUMergeVariablesWithExecutePass()); in CreateTPUBridgePipelineImpl()
H A Dpasses.h480 CreateTPUMergeVariablesWithExecutePass();
H A Dtpu_merge_variables_with_execute.cc600 CreateTPUMergeVariablesWithExecutePass() { in CreateTPUMergeVariablesWithExecutePass() function
H A Dtf_passes.td1620 let constructor = "TFTPU::CreateTPUMergeVariablesWithExecutePass()";