| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 68 Value generateParamsArray(gpu::LaunchFuncOp launch_op, OpAdaptor adaptor,
74 gpu::LaunchFuncOp launch_op, OpAdaptor adaptor,
107 gpu::LaunchFuncOp launch_op, OpAdaptor adaptor, OpBuilder &builder) const { in generateParamsArray() argument
108 auto loc = launch_op.getLoc(); in generateParamsArray()
109 auto num_kernel_operands = launch_op.getNumKernelOperands(); in generateParamsArray()
111 loc, launch_op.getOperands().take_back(num_kernel_operands), in generateParamsArray()
154 gpu::LaunchFuncOp launch_op, OpAdaptor adaptor, in matchAndRewrite() argument
156 if (!launch_op.asyncDependencies().empty() || launch_op.asyncToken()) { in matchAndRewrite()
158 launch_op, "Cannot convert with async dependency or result."); in matchAndRewrite()
161 Location loc = launch_op.getLoc(); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 127 void OutlineLaunch(tf_device::LaunchOp launch_op, SymbolTable* symbol_table, in OutlineLaunch() argument
130 getUsedValuesDefinedAbove(launch_op.body(), launch_op.body(), live_ins); in OutlineLaunch()
133 BuildFunction(live_ins.getArrayRef(), launch_op, symbol_table, builder); in OutlineLaunch()
134 launch_op->setAttr( in OutlineLaunch()
138 builder->setInsertionPoint(launch_op); in OutlineLaunch()
140 launch_op.getLoc(), outlined_func.getFunctionType().getResults(), in OutlineLaunch()
141 live_ins.getArrayRef(), launch_op->getAttrs()); in OutlineLaunch()
143 launch_op.replaceAllUsesWith(cluster_func_op); in OutlineLaunch()
144 launch_op.erase(); in OutlineLaunch()
|
| D | device_attribute_to_launch.cc | 41 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
44 op->replaceAllUsesWith(launch_op); in WrapOpInLaunch()
46 launch_op.body().push_back(new Block); in WrapOpInLaunch()
47 builder.setInsertionPointToEnd(&launch_op.GetBody()); in WrapOpInLaunch()
51 MLIRContext* context = launch_op.getContext(); in WrapOpInLaunch()
|
| D | outside_compiled_to_host_launch.cc | 42 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
45 host_op->replaceAllUsesWith(launch_op); in WrapOpInLaunch()
47 launch_op.body().push_back(new Block); in WrapOpInLaunch()
48 builder.setInsertionPointToEnd(&launch_op.GetBody()); in WrapOpInLaunch()
53 MLIRContext* context = launch_op.getContext(); in WrapOpInLaunch()
|
| D | cluster_formation.cc | 101 tf_device::LaunchOp launch_op) { in ReplaceLiveOutExternalUses() argument
102 Region* launch_op_region = &launch_op.body(); in ReplaceLiveOutExternalUses()
103 for (const auto& p : llvm::zip(live_outs, launch_op.getResults())) { in ReplaceLiveOutExternalUses()
167 tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>( in BuildLaunchForCluster() local
172 launch_op.body().takeBody(region); in BuildLaunchForCluster()
176 ReplaceLiveOutExternalUses(live_outs, launch_op); in BuildLaunchForCluster()
|
| D | host_launch_to_outside_compiled.cc | 60 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithOutsideCompilation() local
61 launch_op->getBlock()->getOperations().splice( in HoistOpsAndAnnotateWithOutsideCompilation()
62 launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(), in HoistOpsAndAnnotateWithOutsideCompilation()
|
| D | launch_to_device_attribute.cc | 84 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithDevice() local
85 launch_op->getBlock()->getOperations().splice( in HoistOpsAndAnnotateWithDevice()
86 launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(), in HoistOpsAndAnnotateWithDevice()
|
| D | tpu_extract_outside_compilation.cc | 258 auto launch_op = builder.create<tf_device::LaunchOp>( in CreateLaunchOpForOutsideCluster() local
262 launch_op.body().push_back(new Block); in CreateLaunchOpForOutsideCluster()
263 builder.setInsertionPointToEnd(&launch_op.GetBody()); in CreateLaunchOpForOutsideCluster()
267 return launch_op; in CreateLaunchOpForOutsideCluster()
|
| D | shape_inference.cc | 1942 if (auto launch_op = dyn_cast<tf_device::LaunchOp>(op)) { in InferShapeForNonTFDialectOperation() local
1943 auto terminator = launch_op.GetBody().getTerminator(); in InferShapeForNonTFDialectOperation()
|