Home
last modified time | relevance | path

Searched refs:CreateConstantOpDeviceAssignmentPass (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dconstant_op_device_assignment.cc84 CreateConstantOpDeviceAssignmentPass() { in CreateConstantOpDeviceAssignmentPass() function
H A Dpasses.h253 std::unique_ptr<OperationPass<ModuleOp>> CreateConstantOpDeviceAssignmentPass();
H A Dtf_passes.td2109 let constructor = "TF::CreateConstantOpDeviceAssignmentPass()";
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc1947 pm.addPass(mlir::TF::CreateConstantOpDeviceAssignmentPass()); in AddTfDeviceAssignmentPasses()