Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc142 const auto kDimY = KernelMappingScheme::DimY; variable
3979 IrArray::Index source_idx_x_base = source_idx.AddOffsetToDim(y_loc, kDimY, b); in UnrollInnerTileLoop()
4368 return index[kDimY]; in GetUntransposedOutputLinearAddress()
4402 return tiling_kernel_info.tile_origin.AddOffsetToDim(y_loc, kDimY, &b_) in EmitEpilogueForReduction()
4552 tiling_kernel_info.output_tile_bounds[kDimY])); in EmitEpilogueForReduction()
4731 for (int i = kDimY; i < kDimTot; ++i) { in EmitTilingKernel()
4746 for (int i = kDimY; i < kDimTot; ++i) { in EmitTilingKernel()