| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | geqrf_cpu_kernel.h | 68 size_t batch_num; variable
|
| D | lu_solve_cpu_kernel.cc | 105 size_t batch_num = output_element_num / b_stride; in LaunchKernel() local
|
| D | lu_unpack_cpu_kernel.cc | 207 int64_t batch_num = in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/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/mindspore-src/source/tests/st/mem_reuse/ |
| D | resnet_cifar_normal.py | 138 batch_num = dataset.get_dataset_size() variable
|
| D | resnet_cifar_memreuse.py | 138 batch_num = dataset.get_dataset_size() variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | vsl_compress_tensorrt.cc | 88 int batch_num = inputs[0].d[0]->getConstantValue(); in getOutputDimensions() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/perf/ |
| D | profiling.h | 98 int32_t batch_num; member
|
| D | profiling.cc | 101 void Tracing::Record(const int32_t type, const int32_t extra_info, const int32_t batch_num, const i… in Record()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/core/ |
| D | config.py | 684 def set_sending_batches(batch_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | lu_unpack_impl.cu | 50 …rror_t TransposeEyeMatrix(S *lu_pivots_ptr, T *pivots_ptr, S *final_order, const int64_t batch_num, in TransposeEyeMatrix()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | lu_solve.cc | 157 int64_t batch_num = ctx.Output(0)->NumElements() / b_stride; in LuSolveCompute() local
|
| D | lu_unpack.cc | 158 int64_t batch_num = ctx.Input(0)->NumElements() / Lu_data_stride; in LuUnpackCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | non_max_suppression_fp32.c | 112 int batch_num = score_dims[Index0]; in NonMaxSuppressionSelecte() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | gptq_quantizer.cc | 155 …::AddBatch(const lite::Tensor &tensor, float *hessian_data, int deep, int batch_num, bool transpos… in AddBatch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | matrix_triangular_solve_gpu_kernel.cc | 240 const int64_t batch_num = output_batch_num_; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 68 int64_t batch_num = 0; in operator ()() local
|
| D | data_queue_op.cc | 964 uint32_t batch_num = 0; in WorkerEntry() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | embedding_cache_prefetch_actor.cc | 1044 const size_t batch_num = max_batch_size / embeddings_dim; in PushEmbeddingsToRemote() local
|