Home
last modified time | relevance | path

Searched refs:dw_addr (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dconvolution_fp16_grad_filter.cc91 auto dw_addr = reinterpret_cast<float16_t *>(out_dw->data()); in DoExecute() local
124 ConvDwFilterFp16Grad(x_addr, dy_addr, dw_addr, start, count, conv_param); in DoExecute()
140 float16_t *mat_c = dw_addr + j * nweights / groups; in DoExecute()
154 float16_t *mat_c = dw_addr + j * nweights / groups; in DoExecute()
166 float16_t *mat_c = dw_addr; in DoExecute()
198 auto dw_addr = reinterpret_cast<float16_t *>(out_dw->data()); in Run() local
200 CHECK_NULL_RETURN(dw_addr); in Run()
201 memset(dw_addr, 0, out_dw->Size()); in Run()
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_llvm_tess.c300 LLVMValueRef dw_addr) in lshs_lds_load() argument
308 values[chan] = lshs_lds_load(ctx, type, chan, dw_addr); in lshs_lds_load()
313 dw_addr = LLVMBuildAdd(ctx->ac.builder, dw_addr, LLVMConstInt(ctx->ac.i32, swizzle, 0), ""); in lshs_lds_load()
314 value = ac_lds_load(&ctx->ac, dw_addr); in lshs_lds_load()
326 LLVMValueRef dw_addr, LLVMValueRef value) in lshs_lds_store() argument
328 dw_addr = in lshs_lds_store()
329 LLVMBuildAdd(ctx->ac.builder, dw_addr, LLVMConstInt(ctx->ac.i32, dw_offset_imm, 0), ""); in lshs_lds_store()
331 ac_lds_store(&ctx->ac, dw_addr, value); in lshs_lds_store()
389 LLVMValueRef dw_addr, stride; in si_nir_load_tcs_varyings() local
420 dw_addr = get_tcs_in_current_patch_offset(ctx); in si_nir_load_tcs_varyings()
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dconvolution_grad_filter.cc94 auto dw_addr = reinterpret_cast<float *>(out_dw->MutableData()); in Execute() local
131 ConvDwFilterGrad(x_addr, dy_addr, dw_addr, start, count, conv_param); in Execute()
147 mat_c = dw_addr + j * nweights / groups; in Execute()
161 mat_c = dw_addr + j * nweights / groups; in Execute()
173 mat_c = dw_addr; in Execute()
206 auto dw_addr = reinterpret_cast<float *>(out_dw->MutableData()); in Run() local
207 CHECK_NULL_RETURN(dw_addr); in Run()
208 memset(dw_addr, 0, out_dw->Size()); in Run()
Ddeconvolution_grad_filter.cc83 auto dw_addr = reinterpret_cast<float *>(out_dw->data()); in Execute() local
84 CHECK_NULL_RETURN(dw_addr); in Execute()
106 memset(dw_addr, 0, out_dw->Size()); in Execute()
113 float *mat_c = dw_addr + j * m; in Execute()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dlstm_grad_weight_gpu_kernel.h67 auto dw_addr = GetDeviceAddress<T>(outputs, 0); in Launch() local
79 …kernel_node_, cudaMemsetAsync(dw_addr, 0, outputs[0]->size, reinterpret_cast<cudaStream_t>(stream_… in Launch()
85 … y_addr, workspace_addr, workspace_size_list_[0], dw_desc_, dw_addr, reserved_addr, in Launch()
/third_party/mesa3d/src/amd/llvm/
Dac_llvm_build.h469 LLVMValueRef ac_lds_load(struct ac_llvm_context *ctx, LLVMValueRef dw_addr);
470 void ac_lds_store(struct ac_llvm_context *ctx, LLVMValueRef dw_addr, LLVMValueRef value);
Dac_llvm_build.c2996 LLVMValueRef ac_lds_load(struct ac_llvm_context *ctx, LLVMValueRef dw_addr) in ac_lds_load() argument
2998 return LLVMBuildLoad(ctx->builder, ac_build_gep0(ctx, ctx->lds, dw_addr), ""); in ac_lds_load()
3001 void ac_lds_store(struct ac_llvm_context *ctx, LLVMValueRef dw_addr, LLVMValueRef value) in ac_lds_store() argument
3004 ac_build_indexed_store(ctx, ctx->lds, dw_addr, value); in ac_lds_store()
/third_party/mesa3d/docs/relnotes/
D17.3.7.rst137 - ac/nir: Fix compiler warning about uninitialized dw_addr.