Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_device_assignment.cc97 std::unique_ptr<OperationPass<func::FuncOp>> CreateSimpleTFDeviceAssignmentPass( in CreateSimpleTFDeviceAssignmentPass() function
Dpasses.h184 std::unique_ptr<OperationPass<func::FuncOp>> CreateSimpleTFDeviceAssignmentPass(
Dtf_passes.td2041 let constructor = "TF::CreateSimpleTFDeviceAssignmentPass()";
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc1951 mlir::TF::CreateSimpleTFDeviceAssignmentPass(options.default_device)); in AddTfDeviceAssignmentPasses()