Searched refs:param_shmem_buffers (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 149 const std::vector<llvm::Value*>& param_shmem_buffers, int64 tile_size, in EmitTiledCompareLoop() argument
199 b->CreateGEP(param_shmem_buffers[i], in EmitTiledCompareLoop()
210 b->CreateGEP(param_shmem_buffers[operand], in EmitTiledCompareLoop()
225 b->CreateGEP(param_shmem_buffers[operand], in EmitTiledCompareLoop()
277 param_shmem_buffers[i], in EmitTiledCompareLoop()
328 std::vector<llvm::Value*> param_shmem_buffers(values_arrays.size(), nullptr); in EmitSortInPlace() local
336 param_shmem_buffers[i] = llvm_ir::AllocateSharedMemoryTile( in EmitSortInPlace()
366 values_arrays, param_shmem_buffers, tile_size, emit_compare_callback, in EmitSortInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 563 llvm::Value* x_loc, absl::Span<llvm::Value* const> param_shmem_buffers);
575 llvm::Value* x_loc, absl::Span<llvm::Value* const> param_shmem_buffers);
|
| D | ir_emitter_unnested.cc | 4100 llvm::Value* x_loc, absl::Span<llvm::Value* const> param_shmem_buffers) { in EmitTileElementForCopy() argument
4103 Load(GEP(param_shmem_buffers[0], {b_.getInt64(0), x_loc, y_loc}), in EmitTileElementForCopy()
4149 llvm::Value* x_loc, absl::Span<llvm::Value* const> param_shmem_buffers) { in EmitTileElementForFusion() argument
4158 if (llvm::Value* param_tile_buffer = param_shmem_buffers[i]) { in EmitTileElementForFusion()
4843 std::vector<llvm::Value*> param_shmem_buffers(context.operand_shapes.size(), in EmitHlo021Tile() local
4866 param_shmem_buffers[id] = get_shared_memory_buffer( in EmitHlo021Tile()
4870 << llvm_ir::DumpToString(*param_shmem_buffers[id]); in EmitHlo021Tile()
4887 param_shmem_buffers); in EmitHlo021Tile()
4891 param_shmem_buffers); in EmitHlo021Tile()
4922 llvm::Value* shmem_buffer = param_shmem_buffers.at(id); in EmitHlo021Tile()
|