Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h129 int64 GetNumThreadsX() const { return num_threads_x_; } in GetNumThreadsX() function
133 return GetNumThreadsX() * GetNumThreadsY(); in GetThreadsPerBlock()
178 mapping_scheme.GetNumThreadsX())); in ReductionCodegenInfo()
Dir_emitter_unnested.cc3955 mapping_scheme.GetTileSizeX() / mapping_scheme.GetNumThreadsX(); in GetStartOffsetX()
4010 int64 num_threads_x = mapping_scheme.GetNumThreadsX(); in EmitTile()
4287 int64 num_threads_x = mapping_scheme.GetNumThreadsX(); in EmitPrologueForReduction()
4392 mapping_scheme.GetNumThreadsX()); in EmitEpilogueForReduction()
4505 constant(mapping_scheme.GetNumThreadsX() / kWarpSize)); in EmitEpilogueForReduction()
4708 mapping_scheme.GetNumThreadsX()); in EmitTilingKernel()