Searched refs:WrapOpInLaunch (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | device_attribute_to_launch.cc | 38 void WrapOpInLaunch(Operation* op, llvm::StringRef device) { in WrapOpInLaunch() function
62 if (!device.getValue().empty()) WrapOpInLaunch(op, device.getValue()); in runOnOperation()
|
| D | outside_compiled_to_host_launch.cc | 39 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch() function
71 WrapOpInLaunch(op, *host_device); in runOnOperation()
|
| D | tpu_colocate_composite_resource_ops.cc | 44 void WrapOpInLaunch(OpBuilder* builder, Location loc, Operation* op, in WrapOpInLaunch() function
104 WrapOpInLaunch(builder, resource_user->getLoc(), resource_user, in ColocateCompositeResourceOpsInReplicate()
|
| D | tpu_rewrite_pass.cc | 310 tf_device::LaunchOp WrapOpInLaunch(OpBuilder* builder, Location loc, in WrapOpInLaunch() function
395 return WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, in BuildCompileOp()
577 WrapOpInLaunch(builder, block.getParent()->getLoc(), execute, device); in AddToParallelExecuteOp()
596 return WrapOpInLaunch(builder, execute_op->getLoc(), execute_op, device); in AssignDevicesToReplicatedExecute()
607 WrapOpInLaunch(builder, compile_op->getLoc(), assert_op, compilation_device); in BuildTPUCompileSucceededAssertOp()
|
| D | tpu_variable_runtime_reformatting.cc | 317 void WrapOpInLaunch(OpBuilder* builder, Location loc, Operation* op, in WrapOpInLaunch() function
413 WrapOpInLaunch(&builder, execute_launch.getLoc(), reformat_op, in HandleReplicateOp()
470 WrapOpInLaunch(&builder, execute_launch.getLoc(), unformat_op, in HandleReplicateOp()
|