Searched defs:batch_num (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/perf/ |
| D | dataset_iterator_tracing.cc | 26 …tasetIteratorTracing::Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, in Record()
|
| D | device_queue_tracing.cc | 26 …d DeviceQueueTracing::Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, in Record()
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_apply_adam.py | 27 … def __init__(self, batch_num, input_channels, output_channels, epoch, lr, weight_decay, epsilon): argument
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_graphdata_distributed.py | 56 def __init__(self, g, batch_num): argument
|
| D | test_graphdata.py | 172 def __init__(self, g, batch_num): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | non_max_suppression_fp32.cc | 112 int NonMaxSuppressionCPUKernel::Run_Selecte(bool simple_out, int box_num, int batch_num, int class_… in Run_Selecte()
231 int batch_num = score_dims.at(kBatchIndex); in Run() local
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 139 batch_num = dataset.get_dataset_size() variable
|
| D | resnet_cifar_normal.py | 139 batch_num = dataset.get_dataset_size() variable
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet_cifar.py | 140 batch_num = dataset.get_dataset_size() variable
|
| /third_party/mindspore/mindspore/dataset/core/ |
| D | config.py | 470 def set_sending_batches(batch_num): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | device_queue_op.cc | 542 uint32_t batch_num = 0; in WorkerEntry() local
|
| D | batch_op.cc | 101 int64_t epoch_num = 0, batch_num = 0, cnt = 0; in operator ()() local
|