Searched refs:AddOffsetToDim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 89 Index AddOffsetToDim(llvm::Value* addend, int64 dim, in AddOffsetToDim() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3979 IrArray::Index source_idx_x_base = source_idx.AddOffsetToDim(y_loc, kDimY, b); in UnrollInnerTileLoop()
3985 IrArray::Index source_idx_x = source_idx_x_base.AddOffsetToDim( in UnrollInnerTileLoop()
4027 tile_origin_index.AddOffsetToDim(start_offset_x, kDimX, &b_); in EmitTile()
4402 return tiling_kernel_info.tile_origin.AddOffsetToDim(y_loc, kDimY, &b_) in EmitEpilogueForReduction()
4403 .AddOffsetToDim(start_offset_x, kDimX, &b_); in EmitEpilogueForReduction()
4437 &b_, start_offset.AddOffsetToDim(constant(j), kDimX, &b_), in EmitEpilogueForReduction()
4780 IrArray::Index tile_index = tile_origin.AddOffsetToDim( in EmitTilingKernel()
|