| /third_party/mindspore/tests/st/mix_precision/ |
| D | utils.py | 78 self.batch_index = 0
114 def __getitem__(self, batch_index): argument
115 if batch_index * self.total_batch_size >= len(self):
118 np.random.seed(batch_index + self.random_offset)
146 self.batch_index = 0
150 self.batch_index = 0
153 if self.batch_index * self.total_batch_size < len(self):
154 data = self[self.batch_index]
155 self.batch_index += 1
|
| /third_party/mindspore/tests/st/hcom/ |
| D | hcom_sparsetensor.py | 47 self.batch_index = 0
82 def __getitem__(self, batch_index): argument
83 if batch_index * self.total_batch_size >= len(self):
86 np.random.seed(batch_index + self.random_offset)
114 self.batch_index = 0
118 self.batch_index = 0
121 if self.batch_index * self.total_batch_size < len(self):
122 data = self[self.batch_index]
123 self.batch_index += 1
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | deconv2d_fp32_coder.cc | 175 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in DoCode() local
176 …input_ptr_ = src_in_ptr_str + "+" + std::to_string(batch_index * input_plane_ * conv_param_->input… in DoCode()
177 …output_ptr_ = src_out_ptr_str + "+" + std::to_string(batch_index * output_plane_ * conv_param_->ou… in DoCode()
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 96 self.batch_index = 0
132 def __getitem__(self, batch_index): argument
133 if batch_index * self.total_batch_size >= len(self):
136 np.random.seed(batch_index + self.random_offset)
164 self.batch_index = 0
168 self.batch_index = 0
171 if self.batch_index * self.total_batch_size < len(self):
172 data = self[self.batch_index]
173 self.batch_index += 1
|
| D | multifieldembeddinglookup_parallel.py | 94 self.batch_index = 0
130 def __getitem__(self, batch_index): argument
131 if batch_index * self.total_batch_size >= len(self):
134 np.random.seed(batch_index + self.random_offset)
162 self.batch_index = 0
166 self.batch_index = 0
169 if self.batch_index * self.total_batch_size < len(self):
170 data = self[self.batch_index]
171 self.batch_index += 1
|
| D | parallel_strategy_search.py | 99 self.batch_index = 0
135 def __getitem__(self, batch_index): argument
136 if batch_index * self.total_batch_size >= len(self):
139 np.random.seed(batch_index + self.random_offset)
167 self.batch_index = 0
171 self.batch_index = 0
174 if self.batch_index * self.total_batch_size < len(self):
175 data = self[self.batch_index]
176 self.batch_index += 1
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_fp16.cc | 235 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
236 batch_input_ = input_ptr + batch_index * conv_param_->input_channel_ * input_plane_; in Run()
237 batch_output_ = output_ptr + batch_index * conv_param_->output_channel_ * output_plane_; in Run()
|
| D | convolution_1x1_fp16.cc | 297 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
298 output_ptr_ = output_data + batch_index * matmul_param_->row_ * matmul_param_->col_; in Run()
300 …input_data + batch_index * conv_param_->input_h_ * conv_param_->input_w_ * conv_param_->input_chan… in Run()
|
| D | deconvolution_winograd_fp16.cc | 459 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
460 … nhwc_input_ = input_ptr + batch_index * deconv_param_->input_plane_ * conv_param_->input_channel_; in Run()
461 …nhwc_output_ = output_ptr + batch_index * deconv_param_->output_plane_ * conv_param_->output_chann… in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_1x1_fp32.cc | 241 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
242 output_ptr_ = src_out + batch_index * matmul_param_->row_ * matmul_param_->col_; in Run()
243 …auto tmp_in = src_in + batch_index * conv_param_->input_h_ * conv_param_->input_w_ * conv_param_->… in Run()
|
| D | deconvolution_fp32.cc | 266 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
267 input_ptr_ = src_in + batch_index * input_plane_ * conv_param_->input_channel_; in Run()
268 output_ptr_ = src_out + batch_index * output_plane_ * conv_param_->output_channel_; in Run()
|
| D | deconvolution_winograd_fp32.cc | 494 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
495 nhwc_input_ = src_in + batch_index * deconv_param_->input_plane_ * conv_param_->input_channel_; in Run()
496 …nhwc_output_ = src_out + batch_index * deconv_param_->output_plane_ * conv_param_->output_channel_; in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_int8.cc | 289 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
290 …RowMajor2Row16x4MajorInt8(src_in + batch_index * matmul_param_->row_ * conv_param_->input_channel_… in Run()
292 output_ptr_ = src_out + batch_index * matmul_param_->col_; in Run()
|
| D | convolution_1x1_int8.cc | 560 for (int batch_index = 0; batch_index < conv_param_->input_batch_; batch_index++) { in Run() local
561 …Pre1x1Trans(src_in + batch_index * conv_param_->input_h_ * conv_param_->input_w_ * conv_param_->in… in Run()
562 src_out + batch_index * matmul_param_->row_ * matmul_param_->col_); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | extract_image_patches_impl.cu | 27 const int64_t batch_index = need_batch ? (static_cast<int64_t>(pos) / other_stride) : 0; in ExtractImagePatches() local
29 … need_batch ? (static_cast<int64_t>(pos) - batch_index * other_stride) : static_cast<int64_t>(pos); in ExtractImagePatches()
53 …depth + input_col * col_input_stride + input_row * row_input_stride + batch_index * patch_input_st… in ExtractImagePatches()
|
| /third_party/mesa3d/src/mesa/main/ |
| D | glthread.c | 76 unsigned batch_index = batch - ctx->GLThread.batches; in glthread_unmarshal_batch() local
78 p_atomic_cmpxchg(&ctx->GLThread.LastProgramChangeBatch, batch_index, -1); in glthread_unmarshal_batch()
79 p_atomic_cmpxchg(&ctx->GLThread.LastDListChangeBatchIndex, batch_index, -1); in glthread_unmarshal_batch()
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | dataset.py | 111 batch_index = sample_index[batch_size * counter: batch_size * (counter + 1)]
112 X_batch = X[batch_index]
113 y_batch = y[batch_index]
|
| /third_party/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_perfmon.h | 34 unsigned batch_index; member
|
| D | st_cb_perfmon.c | 100 cntr->batch_index = num_batch_counters; in init_perf_monitor()
304 result.batch[0] = stm->batch_result->batch[cntr->batch_index]; in st_GetPerfMonitorResult()
|
| /third_party/mesa3d/src/virtio/vulkan/ |
| D | vn_queue.c | 95 uint32_t batch_index) in vn_queue_submission_count_batch_semaphores() argument
105 u.submit_batch = &submit->submit_batches[batch_index]; in vn_queue_submission_count_batch_semaphores()
110 u.bind_sparse_batch = &submit->bind_sparse_batches[batch_index]; in vn_queue_submission_count_batch_semaphores()
185 uint32_t batch_index, in vn_queue_submission_filter_batch_wsi_semaphores() argument
198 u.submit_batch = &submit->temp.submit_batches[batch_index]; in vn_queue_submission_filter_batch_wsi_semaphores()
203 u.bind_sparse_batch = &submit->temp.bind_sparse_batches[batch_index]; in vn_queue_submission_filter_batch_wsi_semaphores()
|
| /third_party/mesa3d/src/intel/tools/ |
| D | intel_dump_gpu.c | 216 int batch_index; in dump_execbuffer2() local
322 batch_index = (execbuffer2->flags & I915_EXEC_BATCH_FIRST) ? 0 : in dump_execbuffer2()
324 batch_bo = get_bo(fd, exec_objects[batch_index].handle); in dump_execbuffer2()
|