Home
last modified time | relevance | path

Searched refs:batch_num (Results 1 – 25 of 36) sorted by relevance

12

/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_apply_adam.py27 … def __init__(self, batch_num, input_channels, output_channels, epoch, lr, weight_decay, epsilon): argument
28 self.batch_num = batch_num
37 input_ = Tensor(np.random.randn(self.batch_num, self.input_channels).astype(np.float32))
41 label_np = np.random.randint(self.output_channels, size=self.batch_num)
42 label_np_onehot = np.zeros(shape=(self.batch_num, self.output_channels)).astype(np.float32)
43 label_np_onehot[np.arange(self.batch_num), label_np] = 1.0
71 …fact = Adam(batch_num=8, input_channels=20, output_channels=5, epoch=5, lr=0.1, weight_decay=0.0, …
/third_party/mindspore/mindspore/train/callback/
D_time_monitor.py60 batch_num = cb_params.batch_num
61 if isinstance(batch_num, int) and batch_num > 0:
62 step_size = cb_params.batch_num
D_loss_monitor.py63 cur_step_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num + 1
/third_party/mindspore/tests/ut/python/dataset/
Dtest_graphdata_distributed.py56 def __init__(self, g, batch_num): argument
58 self.batch_num = batch_num
63 return self.g.graph_info()['edge_num'][0] // self.batch_num
128 batch_num = 2
131 … dataset = ds.GeneratorDataset(source=GNNGraphDataset(g, batch_num), column_names=out_column_names,
132 … sampler=RandomBatchedSampler(edge_num, batch_num), num_parallel_workers=4,
Dtest_graphdata.py172 def __init__(self, g, batch_num): argument
174 self.batch_num = batch_num
179 return self.g.graph_info()['edge_num'][0] // self.batch_num
210 batch_num = 2
213 … dataset = ds.GeneratorDataset(source=GNNGraphDataset(g, batch_num), column_names=out_column_names,
214 … sampler=RandomBatchedSampler(edge_num, batch_num), num_parallel_workers=4)
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pynative/
Dops_test.py36 batch_num, channel, height, width = img.shape
41 col = np.zeros((batch_num, channel, filter_h, filter_w, out_h, out_w)).astype(img.dtype)
49 col = col.transpose(0, 4, 5, 1, 2, 3).reshape(batch_num * out_h * out_w, -1)
68 batch_num, _, x_h, x_w = x.shape
75 out = out.reshape(batch_num, out_h, out_w, -1).transpose(0, 3, 1, 2)
/third_party/mindspore/tests/vm_impl/
Dvm_me.py112 batch_num = x.shape[0]
113 x = x.reshape(batch_num, -1)
125 batch_num = x.shape[0]
126 x = x.reshape(batch_num, -1)
186 batch_num, channel, height, width = input_shape
189 col = col.reshape(batch_num, out_h, out_w, channel, filter_h, filter_w) \
192 img = np.zeros((batch_num,
271 batch_num, _, x_h, x_w = x.shape
278 out = out.reshape((batch_num, out_h, out_w, -1)).transpose(0, 3, 1, 2)
387 batch_num, channel, height, width = img.shape
[all …]
/third_party/mindspore/tests/st/quantization/resnet50_quant/
Dutils.py57 per_step_mseconds = epoch_mseconds / cb_params.batch_num
77 cur_step_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num
81 1, cb_params.batch_num, step_loss,
/third_party/mindspore/tests/ut/python/utils/
Dtest_callback.py108 cb_params.batch_num = 32
124 cb_params.batch_num = 2
146 cb_params.batch_num = 1
211 cb_params.batch_num = 32
240 cb_params.batch_num = 32
271 cb_params.batch_num = 32
383 cb_params.batch_num = 32
/third_party/mindspore/tests/st/quantization/mobilenetv2_quant/
Dutils.py58 per_step_mseconds = epoch_mseconds / cb_params.batch_num
79 cur_step_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num
83 1, cb_params.batch_num, step_loss,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/perf/
Ddevice_queue_tracing.cc26 …d DeviceQueueTracing::Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, in Record() argument
39 …= std::to_string(type) + " " + std::to_string(extra_info) + " " + std::to_string(batch_num) + " " + in Record()
Ddataset_iterator_tracing.cc26 …tasetIteratorTracing::Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, in Record() argument
38 …= std::to_string(type) + " " + std::to_string(extra_info) + " " + std::to_string(batch_num) + " " + in Record()
Ddataset_iterator_tracing.h36 …Status Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, const int32_t…
Ddevice_queue_tracing.h36 …void Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, const int32_t v…
/third_party/mindspore/mindspore/dataset/core/
Dconfig.py470 def set_sending_batches(batch_num): argument
485 if not isinstance(batch_num, int):
487 _config.set_sending_batches(batch_num)
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dnon_max_suppression_fp32.cc112 int NonMaxSuppressionCPUKernel::Run_Selecte(bool simple_out, int box_num, int batch_num, int class_… in Run_Selecte() argument
118 for (auto i = 0; i < batch_num; ++i) { in Run_Selecte()
231 int batch_num = score_dims.at(kBatchIndex); in Run() local
246 auto ret = Run_Selecte(simple_out, box_num, batch_num, class_num, scores_data, box_data); in Run()
Dnon_max_suppression_fp32.h44 …int Run_Selecte(bool simple_out, int box_num, int batch_num, int class_num, float *scores_data, fl…
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Ddavinci_model.h31 …uintptr_t logic_var_base = 0, uint32_t stream_num = 0, uint32_t batch_num = 0, uint32_t event_num …
43 batch_num_(batch_num), in task_info_list_()
/third_party/mindspore/tests/st/tbe_networks/
Dresnet_cifar.py140 batch_num = dataset.get_dataset_size() variable
141 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
143 time_cb = TimeMonitor(data_size=batch_num)
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.cc101 int64_t epoch_num = 0, batch_num = 0, cnt = 0; in operator ()() local
114 … std::make_pair(std::move(table), CBatchInfo(epoch_num, batch_num++, cnt + 1 - epoch_num)))); in operator ()()
117 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()()
124 std::make_pair(std::move(table), CBatchInfo(epoch_num, batch_num++, cnt + 1 - epoch_num)))); in operator ()()
129 batch_num = 0; in operator ()()
133 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()()
/third_party/mindspore/tests/mindspore_test_framework/apps/
Dbert_attention_submodules.py117 self.batch_num = batch_size * num_attention_heads
139 self.batch_num,
310 self.batch_num = batch_size * num_attention_heads
342 self.batch_num,
/third_party/mindspore/tests/st/mem_reuse/
Dresnet_cifar_memreuse.py139 batch_num = dataset.get_dataset_size() variable
140 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
Dresnet_cifar_normal.py139 batch_num = dataset.get_dataset_size() variable
140 config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=10)
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Dcallback.py74 cur_step_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num + 1
/third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/
Dcallbacks.py57 cur_step_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num + 1

12