Searched refs:param_shmem_buffers (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 148 const std::vector<llvm::Value*>& param_shmem_buffers, int64 tile_size, in EmitTiledCompareLoop() argument
198 b->CreateGEP(param_shmem_buffers[i], in EmitTiledCompareLoop()
209 b->CreateGEP(param_shmem_buffers[operand], in EmitTiledCompareLoop()
224 b->CreateGEP(param_shmem_buffers[operand], in EmitTiledCompareLoop()
275 param_shmem_buffers[i], in EmitTiledCompareLoop()
326 std::vector<llvm::Value*> param_shmem_buffers(values_arrays.size(), nullptr); in EmitSortInPlace() local
334 param_shmem_buffers[i] = llvm_ir::AllocateSharedMemoryTile( in EmitSortInPlace()
364 values_arrays, param_shmem_buffers, tile_size, emit_compare_callback, in EmitSortInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3059 std::vector<llvm::Value*> param_shmem_buffers(num_params, nullptr); in EmitKernel() local
3062 param_shmem_buffers[id] = in EmitKernel()
3068 << llvm_ir::DumpToString(*param_shmem_buffers[id]); in EmitKernel()
3110 *unnested_hlo, param_arrays, param_shmem_buffers, reduced_dims, in EmitKernel()
3159 llvm::Value* shmem_buffer = param_shmem_buffers[id]; in EmitKernel()
3172 llvm_ir::TiledParameterInfo tiled_param_info(param_shmem_buffers, y, x); in EmitKernel()
|