Home
last modified time | relevance | path

Searched defs:batch_index (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc175 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in DoCode() local
/third_party/mesa3d/src/mesa/state_tracker/
Dst_cb_perfmon.h34 unsigned batch_index; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddeconvolution_int8.cc289 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
Dconvolution_1x1_int8.cc560 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dextract_image_patches_impl.cu27 const int64_t batch_index = need_batch ? (static_cast<int64_t>(pos) / other_stride) : 0; in ExtractImagePatches() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_1x1_fp16.cc297 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
Ddeconvolution_fp16.cc235 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
Ddeconvolution_winograd_fp16.cc459 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
/third_party/mindspore/tests/st/hcom/
Dhcom_sparsetensor.py82 def __getitem__(self, batch_index): argument
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32.cc266 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
Dconvolution_1x1_fp32.cc241 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
Ddeconvolution_winograd_fp32.cc494 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
/third_party/mesa3d/src/mesa/main/
Dglthread.c76 unsigned batch_index = batch - ctx->GLThread.batches; in glthread_unmarshal_batch() local
/third_party/mindspore/tests/st/mix_precision/
Dutils.py114 def __getitem__(self, batch_index): argument
/third_party/mindspore/tests/st/auto_parallel/
Doptimizer_parallel.py132 def __getitem__(self, batch_index): argument
Dparallel_strategy_search.py135 def __getitem__(self, batch_index): argument
Dmultifieldembeddinglookup_parallel.py130 def __getitem__(self, batch_index): argument
/third_party/mesa3d/src/intel/tools/
Dintel_dump_gpu.c216 int batch_index; in dump_execbuffer2() local
/third_party/mesa3d/src/virtio/vulkan/
Dvn_queue.c95 uint32_t batch_index) in vn_queue_submission_count_batch_semaphores()
185 uint32_t batch_index, in vn_queue_submission_filter_batch_wsi_semaphores()