Searched refs:kDimX (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 141 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()
|