Searched refs:BuildKernelThunkImpl (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.h | 676 StatusOr<std::unique_ptr<Thunk>> BuildKernelThunkImpl(
|
| H A D | ir_emitter_unnested.cc | 1649 BuildKernelThunkImpl(kernel_name, GetThunkInfo(op), slices, &ir_arrays, in EmitLaunchFunc()
2956 StatusOr<std::unique_ptr<Thunk>> IrEmitterUnnested::BuildKernelThunkImpl( in BuildKernelThunkImpl() function in xla::gpu::IrEmitterUnnested
3089 return BuildKernelThunkImpl(name, thunk_info, slices, ir_arrays, in BuildKernelThunk()
3120 return BuildKernelThunkImpl(name, thunk_info, slices, ir_arrays, in BuildKernelThunk()
|