Searched defs:launch_op (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_formation.cc | 113 tf_device::LaunchOp launch_op) { in ReplaceLiveOutExternalUses() 179 tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>( in BuildLaunchForCluster() local
|
| D | device_attribute_to_launch.cc | 41 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
|
| D | host_launch_to_outside_compiled.cc | 58 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithOutsideCompilation() local
|
| D | launch_to_device_attribute.cc | 83 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithDevice() local
|
| D | outside_compiled_to_host_launch.cc | 43 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
|
| D | cluster_outlining.cc | 125 void OutlineLaunch(tf_device::LaunchOp launch_op, SymbolTable* symbol_table, in OutlineLaunch()
|
| D | tpu_rewrite_pass.cc | 702 tf_device::LaunchOp launch_op = AssignDevicesToReplicatedExecute( in Rewrite() local
|
| D | tpu_extract_outside_compilation.cc | 258 auto launch_op = builder.create<tf_device::LaunchOp>( in CreateLaunchOpForOutsideCluster() local
|
| D | shape_inference.cc | 1157 if (auto launch_op = dyn_cast<tf_device::LaunchOp>(op)) { in InferShapeForNonTFDialectOperation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 102 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in generateParamsArray() 150 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 92 auto launch_op = rewriter.create<mlir::gpu::LaunchOp>( in matchAndRewrite() local
|
| D | ral_legalize_to_llvm.cc | 396 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in generateParamsArray() 444 gpu::LaunchFuncOp launch_op, ArrayRef<Value> operands, in matchAndRewrite()
|