Home
last modified time | relevance | path

Searched refs:CreateLaunchToDeviceAttributePass (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbridge.cc134 pm.addNestedPass<func::FuncOp>(TFDevice::CreateLaunchToDeviceAttributePass()); in CreateTPUBridgePipelineImpl()
263 add_pass(TFDevice::CreateLaunchToDeviceAttributePass()); in AddGraphExportLoweringPasses()
Dlaunch_to_device_attribute.cc114 CreateLaunchToDeviceAttributePass() { in CreateLaunchToDeviceAttributePass() function
Dpasses.h405 CreateLaunchToDeviceAttributePass();
Dtf_device_passes.td276 let constructor = "TFDevice::CreateLaunchToDeviceAttributePass()";
/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_mlir_passes.cc328 mlir::TFDevice::CreateLaunchToDeviceAttributePass()); in CreateDTensorMLIRPass()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc2278 mlir::TFDevice::CreateLaunchToDeviceAttributePass()); in CreateTFExecutorToTFPipeline()