Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc141 const auto kDimX = KernelMappingScheme::DimX; variable
3986 constant(j * step_x * vector_size + i), kDimX, b); in UnrollInnerTileLoop()
4027 tile_origin_index.AddOffsetToDim(start_offset_x, kDimX, &b_); in EmitTile()
4036 mapping_scheme.GetDimsInElems()[kDimX] % tile_size_x == 0 && in EmitTile()
4372 llvm::Value* x_dim_size = index.GetConstantWithIndexType(dims_in_elem[kDimX]); in GetUntransposedOutputLinearAddress()
4374 return b->CreateAdd(x_block_offset, index[kDimX]); in GetUntransposedOutputLinearAddress()
4403 .AddOffsetToDim(start_offset_x, kDimX, &b_); in EmitEpilogueForReduction()
4437 &b_, start_offset.AddOffsetToDim(constant(j), kDimX, &b_), in EmitEpilogueForReduction()
4550 tiling_kernel_info.output_tile_bounds[kDimX]), in EmitEpilogueForReduction()
5343 bool tile_fit = reduction_dimensions.dimensions[kDimX] % in ComputeReductionCodegenInfo()