Home
last modified time | relevance | path

Searched defs:batch_offset (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dpack_int8.c198 int batch_offset = b * channel * plane; in PackNHWCToNHWC4Int8() local
220 int batch_offset = b * channel * plane; in PackNHWC4ToNHWCInt8() local
240 int batch_offset = b * channel * plane; in PackNHWCToNHWC8Int8() local
259 int batch_offset = b * channel * plane; in PackNHWC8ToNHWCInt8() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcsr_sparse_matrix_to_sparse_tensor_cpu_kernel.cc153 const int64_t batch_offset = x_batch_pointers_ptr[batch_idx]; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcsr_sparse_matrix_to_sparse_tensor.cc198 const int64_t batch_offset = x_batch_pointers_ptr[batch_idx]; in SpecialCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dnon_max_suppression_fp32.cc119 int batch_offset = i * class_num * box_num; in Run_Selecte() local
Dlstm_mindir_fp32.cc227 int batch_offset = b * dir_mult * lstm_param_->output_size_; in LstmUnidirectional() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dpack_fp16.c193 int batch_offset = b * channel * plane; in PackNHWCToNHWC4Fp16() local
217 int batch_offset = b * channel * plane; in PackNHWCToNHWC8Fp16() local
239 int batch_offset = b * c4 * C4NUM * plane; in PackNHWC4ToNHWCFp16() local
257 int batch_offset = b * channel * plane; in PackNCHWToNHWC4Fp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dnon_max_suppression_fp32.c133 int batch_offset = i * class_num * box_num; in NonMaxSuppressionSelecte() local
Dpack_fp32.c1120 int batch_offset = b * channel * plane; in PackNHWCToNHWC4Fp32() local
1140 int batch_offset = b * channel * plane; in PackNHWCToNHWCXFp32() local
1284 int batch_offset = b * channel * plane; in PackNHWCToNHWC8Fp32() local
1306 int batch_offset = b * c_algin * cx_num * plane; in PackNHWCXToNHWCFp32() local
/third_party/mesa3d/src/intel/tools/
Daub_write.c714 uint64_t batch_offset) in aub_dump_ring_buffer_execlist()
765 uint64_t batch_offset, in aub_dump_ring_buffer_legacy()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dlstm_grad_fp32.c111 int batch_offset = b * dir_mult * lstm_param->hidden_size_; in LstmGradReorderDy() local
/third_party/mesa3d/src/intel/executor/
Dexecutor_main.c689 const int batch_offset = ec.batch_start - ec.bo.batch.addr; in l_execute() local
/third_party/mesa3d/src/gallium/drivers/crocus/
Dcrocus_batch.c432 crocus_command_reloc(struct crocus_batch *batch, uint32_t batch_offset, in crocus_command_reloc()
/third_party/mesa3d/src/intel/common/tests/
Dmi_builder_test.cpp144 uint32_t batch_offset; member in mi_builder_test