Searched refs:TensorDeviceCopyConversionPass (Results 1 – 2 of 2) sorted by relevance
40 struct TensorDeviceCopyConversionPass struct42 TensorDeviceCopyConversionPass> {48 void TensorDeviceCopyConversionPass::runOnOperation() { in runOnOperation()113 return std::make_unique<TensorDeviceCopyConversionPass>(); in CreateTensorDeviceCopyConversionPass()
1505 def TensorDeviceCopyConversionPass : Pass<"tf-tensor-device-copy", "mlir::func::FuncOp"> {