Searched refs:KernelMappingScheme (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_tiling.cc | 118 KernelMappingScheme::KernelMappingScheme( in KernelMappingScheme() function in xla::llvm_ir::KernelMappingScheme
149 IrArray::Index KernelMappingScheme::GetUnnormalizedIndex( in GetUnnormalizedIndex()
159 IrArray::Index KernelMappingScheme::EmitBlockIndex(llvm::Type* index_ty) { in EmitBlockIndex()
172 IrArray::Index KernelMappingScheme::GetTileIndexForBlockOrigin( in GetTileIndexForBlockOrigin()
186 IrArray::Index KernelMappingScheme::GetElementIndexForTileOrigin( in GetElementIndexForTileOrigin()
199 llvm::GlobalVariable* KernelMappingScheme::GetSharedMemoryBufferForElementType( in GetSharedMemoryBufferForElementType()
218 KernelMappingScheme::EmitThreadYXCoordinate(llvm::Type* index_ty) { in EmitThreadYXCoordinate()
|
| D | kernel_tiling.h | 88 class KernelMappingScheme {
93 KernelMappingScheme() {} in KernelMappingScheme() function
98 KernelMappingScheme(absl::Span<const int64> dims_in_elems, int64 tile_size_y,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 69 explicit KernelCodegenInfo(llvm_ir::KernelMappingScheme* mapping_scheme) in KernelCodegenInfo()
83 llvm_ir::KernelMappingScheme* GetKernelMappingScheme() const { in GetKernelMappingScheme()
92 llvm_ir::KernelMappingScheme* mapping_scheme_;
225 std::tuple<llvm_ir::KernelMappingScheme, bool>
|
| D | ir_emitter_unnested.cc | 94 using llvm_ir::KernelMappingScheme;
2300 const KernelMappingScheme* mapping_scheme, llvm::IRBuilder<>* builder, in GetStartOffsetAndStepForX()
2315 void EmitFullElementalTile(const KernelMappingScheme* mapping_scheme, in EmitFullElementalTile()
2331 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitFullElementalTile()
2332 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitFullElementalTile()
2338 y_indvar, KernelMappingScheme::DimY, builder); in EmitFullElementalTile()
2344 KernelMappingScheme::DimX, builder); in EmitFullElementalTile()
2353 void EmitPartialElementalTile(const KernelMappingScheme* mapping_scheme, in EmitPartialElementalTile()
2370 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitPartialElementalTile()
2371 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitPartialElementalTile()
[all …]
|