Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3954 int64 x_num_steps = in GetStartOffsetX() local
3956 return b->CreateMul(thread_id_x, constant(x_num_steps)); in GetStartOffsetX()
3970 bool check_x_tile_bounds, int64 x_num_steps, int64 step_x, in UnrollInnerTileLoop() argument
3980 for (int64 j = 0; j < x_num_steps / vector_size; j++) { in UnrollInnerTileLoop()
4014 int64 x_num_steps = tile_size_x / num_threads_x; in EmitTile() local
4063 return UnrollInnerTileLoop(check_x_tile_bounds, x_num_steps, step_x, in EmitTile()