Searched refs:MAX_MALLOC_SIZE (Results 1 – 13 of 13) sorted by relevance
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
D | batchnorm_int8.cc | 60 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()
|
D | softmax_int8.cc | 128 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, softmax_param_->element_size_ * sizeof(int)); in Run()
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
D | batchnorm_fp16.cc | 37 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()
|
D | softmax_fp16.cc | 66 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/ |
D | batchnorm_fp32.cc | 80 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()
|
D | adder_fp32.cc | 83 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()
|
D | l2_norm_fp32.cc | 44 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, shape.size() * sizeof(int)); in MallocTmpBuffer() 50 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, kMaxThreadNum * sizeof(float)); in MallocTmpBuffer()
|
D | softmax_fp32.cc | 68 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, out_plane_size_ * in_plane_size_ * sizeof(float)); in ReSize()
|
/third_party/mindspore/mindspore/lite/src/runtime/ |
D | inner_allocator.h | 71 constexpr int64_t MAX_MALLOC_SIZE = static_cast<size_t>(2000) * 1024 * 1024; variable
|
D | infer_manager.cc | 97 if (output->ElementsNum() >= MAX_MALLOC_SIZE / static_cast<int>(sizeof(int64_t))) { in CheckInfershapeResult()
|
/third_party/mindspore/mindspore/lite/src/cxx_api/ |
D | types.cc | 92 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/ |
D | tensor_util.cc | 110 dst->element_num_ * sizeof(TensorC) > MAX_MALLOC_SIZE) { in TensorList2TensorListC()
|
/third_party/mindspore/mindspore/lite/src/ |
D | scheduler.cc | 604 if (output->ElementsNum() >= MAX_MALLOC_SIZE / static_cast<int>(sizeof(int64_t))) { in InferNodeShape()
|