Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_device_copy_conversion.cc40 struct TensorDeviceCopyConversionPass struct
42 TensorDeviceCopyConversionPass> {
48 void TensorDeviceCopyConversionPass::runOnOperation() { in runOnOperation()
113 return std::make_unique<TensorDeviceCopyConversionPass>(); in CreateTensorDeviceCopyConversionPass()
H A Dtf_passes.td1505 def TensorDeviceCopyConversionPass : Pass<"tf-tensor-device-copy", "mlir::func::FuncOp"> {