Home
last modified time | relevance | path

Searched refs:GetNumberOfThreadsForDimensionX (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.h145 int64 GetNumberOfThreadsForDimensionX() const { return num_threads_x_; } in GetNumberOfThreadsForDimensionX() function
149 return GetNumberOfThreadsForDimensionX() * in GetThreadsPerBlock()
Dkernel_tiling.cc228 llvm::ConstantInt::get(index_ty, GetNumberOfThreadsForDimensionX()); in EmitThreadYXCoordinate()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2321 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitFullElementalTile()
2361 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitPartialElementalTile()
2593 int64 num_thread = mapping_scheme_->GetNumberOfThreadsForDimensionX(); in GetNumberOfPartialResults()
3121 mapping_scheme->GetNumberOfThreadsForDimensionX() == kWarpSize ? x in EmitKernel()