Searched refs:AddOffsetToDim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 95 Index AddOffsetToDim(llvm::Value* addend, int64_t dim, in AddOffsetToDim() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3391 IrArray::Index source_idx_x_base = source_idx.AddOffsetToDim(y_loc, kDimY, b); in EmitXTileLoop()
3409 IrArray::Index source_idx_x = source_idx_x_base.AddOffsetToDim( in EmitXTileLoop()
3441 tile_origin_index.AddOffsetToDim(start_offset_x, kDimX, &b_); in EmitTile()
3741 return tiling_kernel_info.tile_origin.AddOffsetToDim(y_loc, kDimY, &b_) in GetOutputAddressForReduction()
3742 .AddOffsetToDim(start_offset_x, kDimX, &b_); in GetOutputAddressForReduction()
3755 start_offset.AddOffsetToDim(constant(partial_result_idx), kDimX, &b_); in GetOutputAddressForReduction()
4188 IrArray::Index tile_index = tile_origin.AddOffsetToDim( in EmitTilingKernel()
|