Home
last modified time | relevance | path

Searched refs:MAX_MALLOC_SIZE (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dbatchnorm_int8.cc60 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, static_cast<size_t>(mean->ElementsNum()) * sizeof(float)); in InitConstTensor()
66 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, static_cast<size_t>(variance->ElementsNum()) * sizeof(float)); in InitConstTensor()
117 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, static_cast<size_t>(mean->ElementsNum()) * sizeof(float)); in InitFusedConstTensor()
123 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, static_cast<size_t>(variance->ElementsNum()) * sizeof(float)); in InitFusedConstTensor()
Dsoftmax_int8.cc128 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, softmax_param_->element_size_ * sizeof(int)); in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dbatchnorm_fp16.cc37 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, mean_fp32->ElementsNum() * sizeof(float16_t)); in InitConstTensor()
38 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, variance_fp32->ElementsNum() * sizeof(float16_t)); in InitConstTensor()
Dsoftmax_fp16.cc66 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, out_plane_size_ * in_plane_size_ * sizeof(float16_t)); in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dbatchnorm_fp32.cc80 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, in_tensors_.at(1)->Size()); in InitConstTensor()
81 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, in_tensors_.at(kNumInput2)->Size()); in InitConstTensor()
Dadder_fp32.cc83 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, pack_weight_size * sizeof(float)); in InitWeightBias()
91 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, oc_block_num * oc_block * sizeof(float)); in InitWeightBias()
Dl2_norm_fp32.cc44 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, shape.size() * sizeof(int)); in MallocTmpBuffer()
50 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, kMaxThreadNum * sizeof(float)); in MallocTmpBuffer()
Dsoftmax_fp32.cc68 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, out_plane_size_ * in_plane_size_ * sizeof(float)); in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/
Dinner_allocator.h71 constexpr int64_t MAX_MALLOC_SIZE = static_cast<size_t>(2000) * 1024 * 1024; variable
Dinfer_manager.cc97 if (output->ElementsNum() >= MAX_MALLOC_SIZE / static_cast<int>(sizeof(int64_t))) { in CheckInfershapeResult()
/third_party/mindspore/mindspore/lite/src/cxx_api/
Dtypes.cc92 if (data_len < 0 || data_len > MAX_MALLOC_SIZE) { in CreateTensor()
185 if (data_len <= 0 || data_len > MAX_MALLOC_SIZE) { in Clone()
/third_party/mindspore/mindspore/lite/src/common/
Dtensor_util.cc110 dst->element_num_ * sizeof(TensorC) > MAX_MALLOC_SIZE) { in TensorList2TensorListC()
/third_party/mindspore/mindspore/lite/src/
Dscheduler.cc604 if (output->ElementsNum() >= MAX_MALLOC_SIZE / static_cast<int>(sizeof(int64_t))) { in InferNodeShape()