Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2300 const KernelMappingScheme* mapping_scheme, llvm::IRBuilder<>* builder, in GetStartOffsetAndStepForX() argument
2304 if (mapping_scheme->DilatedX()) { in GetStartOffsetAndStepForX()
2315 void EmitFullElementalTile(const KernelMappingScheme* mapping_scheme, in EmitFullElementalTile() argument
2321 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitFullElementalTile()
2322 int64 num_threads_y = mapping_scheme->GetNumberOfThreadsForDimensionY(); in EmitFullElementalTile()
2323 int64 tile_size_x = mapping_scheme->GetTileSizeForDimensionX(); in EmitFullElementalTile()
2324 int64 tile_size_y = mapping_scheme->GetTileSizeForDimensionY(); in EmitFullElementalTile()
2329 tile_size_x, num_threads_x, mapping_scheme, builder, x, index_ty); in EmitFullElementalTile()
2353 void EmitPartialElementalTile(const KernelMappingScheme* mapping_scheme, in EmitPartialElementalTile() argument
2361 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitPartialElementalTile()
[all …]
Dir_emitter_unnested.h69 explicit KernelCodegenInfo(llvm_ir::KernelMappingScheme* mapping_scheme) in KernelCodegenInfo() argument
70 : mapping_scheme_(mapping_scheme), in KernelCodegenInfo()