Searched refs:GetNumThreadsFor (Results 1 – 2 of 2) sorted by relevance
108 int64_t GetNumThreadsFor(int d) const { return num_threads_.at(d); } in GetNumThreadsFor() function112 return GetNumThreadsFor(0) * GetNumThreadsFor(1) * GetNumThreadsFor(2); in GetNumThreadsPerBlock()
3395 : tiling_scheme.GetNumThreadsFor(kDimX); in EmitXTileLoop()3435 llvm::Value* num_threads_y = constant(tiling_scheme.GetNumThreadsFor(kDimY)); in EmitTile()3622 int64_t num_threads_x = tiling_scheme.GetNumThreadsFor(kDimX); in GenerateReductionCodegenState()3642 CHECK_EQ(num_threads_x, tiling_scheme.GetNumThreadsFor(kDimY)); in GenerateReductionCodegenState()3938 constant(tiling_scheme.GetNumThreadsFor(kDimX) / WarpSize())); in EmitReductionOutputForRowReduction()4093 constant(tiling_scheme.GetNumThreadsFor(kDimX)); in EmitThreadIdInfo()