Searched refs:AddOffsetToDim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 102 Index AddOffsetToDim(llvm::Value* addend, int64 dim, in AddOffsetToDim() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2331 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitFullElementalTile()
2332 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitFullElementalTile()
2337 IrArray::Index source_idx_y = source_idx.AddOffsetToDim( in EmitFullElementalTile()
2342 IrArray::Index source_idx_y_x = source_idx_y.AddOffsetToDim( in EmitFullElementalTile()
2370 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitPartialElementalTile()
2371 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitPartialElementalTile()
2374 source_idx.AddOffsetToDim(llvm::ConstantInt::get(index_ty, j * step_x), in EmitPartialElementalTile()
2400 source_idx_x.AddOffsetToDim( in EmitPartialElementalTile()
2985 IrArray::Index tile_index = starting_tile.AddOffsetToDim( in EmitBlock()
|