Home
last modified time | relevance | path

Searched defs:launch_op (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dcluster_formation.cc101 tf_device::LaunchOp launch_op) { in ReplaceLiveOutExternalUses()
167 tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>( in BuildLaunchForCluster() local
H A Ddevice_attribute_to_launch.cc41 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
H A Doutside_compiled_to_host_launch.cc42 auto launch_op = builder.create<tf_device::LaunchOp>( in WrapOpInLaunch() local
H A Dhost_launch_to_outside_compiled.cc60 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithOutsideCompilation() local
H A Dlaunch_to_device_attribute.cc84 Operation* launch_op = launch.getOperation(); in HoistOpsAndAnnotateWithDevice() local
H A Dcluster_outlining.cc127 void OutlineLaunch(tf_device::LaunchOp launch_op, SymbolTable* symbol_table, in OutlineLaunch()
H A Dtpu_extract_outside_compilation.cc258 auto launch_op = builder.create<tf_device::LaunchOp>( in CreateLaunchOpForOutsideCluster() local
H A Dshape_inference.cc1942 if (auto launch_op = dyn_cast<tf_device::LaunchOp>(op)) { in InferShapeForNonTFDialectOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_kernel_to_llvm_pass.cc107 gpu::LaunchFuncOp launch_op, OpAdaptor adaptor, OpBuilder &builder) const { in generateParamsArray()
154 gpu::LaunchFuncOp launch_op, OpAdaptor adaptor, in matchAndRewrite()