Home
last modified time | relevance | path

Searched defs:unit_size (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dscatter_nd_functor_impl.cu21 __global__ void ScatterNdUpdate(const size_t unit_size, const size_t index_depth, const size_t upda… in ScatterNdUpdate()
47 __global__ void ScatterNdAdd(const size_t unit_size, const size_t index_depth, const size_t updates… in ScatterNdAdd()
73 __global__ void ScatterNdSub(const size_t unit_size, const size_t index_depth, const size_t updates… in ScatterNdSub()
99 void CalScatterNdFunctor(enum ScatterNdFunctorType func_type, const size_t &unit_size, const size_t… in CalScatterNdFunctor()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_fp32.cc42 …int unit_size = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ * C6… in InitTmpBuffer() local
44 …int unit_size = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ * C4… in InitTmpBuffer() local
46 int unit_size = in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/
Dtrt_kernel.cc34 size_t unit_size = UnitSizeInBytes(type_id); in Init() local
43 size_t unit_size = UnitSizeInBytes(type_id); in Init() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dscatter_nd_fp32.c21 int DoScatterND(float *output_ptr, const float *update, int *output_unit_offsets, int unit_size, in… in DoScatterND()
Dprior_box_fp32.h29 size_t unit_size = size / thread_num; in PriorBox() local
Ddeconv_winograd_fp32.c444 … const float *bt_buf, float *b_tmp_buf, int unit_size, int w_start, int h_start, in DeConvWgCalWgFp32()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc80 int unit_size = kernel_plane * channel_block * inchannel_block; in TEST_F() local
143 int unit_size = kernel_plane * channel_block * inchannel_block; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dcholesky_trsm_solve_gpu_kernel.h99 size_t unit_size = sizeof(T); in InitSizeLists() local
111 size_t unit_size = sizeof(T); in InitSizeLists() local
Dbroadcast_gpu_kernel.h151 auto unit_size = is_comp_op_ ? sizeof(bool) : sizeof(T); in InitSizeLists() local
Dmatmul_gpu_kernel.h174 size_t unit_size = sizeof(T); in InitSizeLists() local
Dcholesky_solve_gpu_kernel.h237 size_t unit_size = sizeof(T); in InitSizeLists() local
Dupdate_thor_gradient.h157 size_t unit_size = sizeof(T); in InitSizeLists() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_int8.c29 int unit_size; in ConvInt8() local
Dslice_int8.c38 int unit_size = param->size_[7] * sizeof(int8_t); in SliceInt8() local
/third_party/ffmpeg/libavformat/
Drtpenc_vc2hq.c107 uint32_t unit_size; in ff_rtp_send_vc2hq() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_fp16.cc78 int unit_size = in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/
Dprint_gpu_kernel.h115 size_t unit_size = UnitSizeInBytes(type_id); in Init() local
/third_party/ffmpeg/libavcodec/
Dh264_mp4toannexb_bsf.c69 uint16_t unit_size; in h264_extradata_to_annexb() local
Dcbs_mpeg2.c150 size_t unit_size; in cbs_mpeg2_split_fragment() local
/third_party/iowow/src/utils/
Diwarr.c180 iwrc iwulist_init(IWULIST *list, size_t initial_length, size_t unit_size) { in iwulist_init()
195 IWULIST *iwulist_create(size_t initial_length, size_t unit_size) { in iwulist_create()
/third_party/mindspore/mindspore/lite/src/delegate/npu/pass/
Dnpu_pass_utils.cc226 void NPUPassUtils::AssistDataNHWC2NCHW(int *data, size_t unit_size) { in AssistDataNHWC2NCHW()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c177 … const float16_t *bt_buf, float16_t *b_tmp_buf, int unit_size, int w_start, int h_start, in DeConvWgCalWgFp16()
/third_party/mesa3d/src/gallium/drivers/radeon/
Dradeon_vcn_dec.c853 int unit_size = pic->picture_parameter.lr_unit_size[i]; in get_av1_msg() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc423 … size_t unit_size = (max_sum_size / DYNAMIC_MEM_ALLOC_UNIT_SIZE + 1) * DYNAMIC_MEM_ALLOC_UNIT_SIZE; in FetchMemUnitSize() local
/third_party/boost/tools/build/src/engine/
Dfunction.cpp1275 void const * const value, int const unit_size ) in dynamic_array_push_impl()