Home
last modified time | relevance | path

Searched refs:KernelCodegenInfo (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h67 class KernelCodegenInfo {
69 explicit KernelCodegenInfo(llvm_ir::KernelMappingScheme* mapping_scheme) in KernelCodegenInfo() function
74 virtual ~KernelCodegenInfo() {} in ~KernelCodegenInfo()
100 std::function<void(HloInstruction* hlo, KernelCodegenInfo* kernel_info)>;
103 std::function<void(HloInstruction* hlo, KernelCodegenInfo* kernel_info)>;
115 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc,
250 KernelCodegenInfo* kernel_info);
252 KernelCodegenInfo* kernel_info, KernelSupportLibrary* ksl,
258 const KernelCodegenInfo* kernel_info,
265 const KernelCodegenInfo* kernel_info,
[all …]
Dir_emitter_unnested.cc2457 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForCopy()
2487 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForFusion()
2521 class ReductionCodegenInfo : public IrEmitterUnnested::KernelCodegenInfo {
2525 : KernelCodegenInfo(mapping_scheme), in ReductionCodegenInfo()
2644 KernelCodegenInfo* kernel_info, GpuElementalIrEmitter* elemental_emitter, in EmitPrologueForOneReduction()
2698 HloInstruction* unnested_hlo, KernelCodegenInfo* kernel_info) { in EmitPrologueForReduction()
2779 HloInstruction* unnested_hlo, KernelCodegenInfo* kernel_info) { in EmitEpilogueForReduction()
2854 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForReduction()
2947 KernelCodegenInfo* kernel_info, in EmitBlock()
3053 KernelCodegenInfo* kernel_info) { in EmitKernel()
[all …]