| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_runtime.cc | 32 allocator_ = new (std::nothrow) TensorRTAllocator(); in Init()
33 if (allocator_ == nullptr) { in Init()
46 if (allocator_ != nullptr) { in ~TensorRTRuntime()
47 allocator_->ClearDeviceMem(); in ~TensorRTRuntime()
48 delete allocator_; in ~TensorRTRuntime()
49 allocator_ = nullptr; in ~TensorRTRuntime()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | detection_post_process_base_coder.cc | 53 …allocator_->Malloc(kNumberTypeFloat, anchor_tensor->ElementsNum() * sizeof(float), kOfflinePackWei… in Prepare()
64 …allocator_->Malloc(kNumberTypeFloat, anchor_tensor->ElementsNum() * sizeof(float), kOfflinePackWei… in Prepare()
71 …static_cast<float *>(allocator_->Malloc(kNumberTypeFloat, anchor_tensor->Size(), kOfflinePackWeigh… in Prepare()
88 …params_->decoded_boxes_ = allocator_->Malloc(kNumberTypeFloat, num_boxes_ * kMultiply * sizeof(flo… in AllocateBuffer()
90 …params_->nms_candidate_ = allocator_->Malloc(kNumberTypeUInt8, num_boxes_ * sizeof(uint8_t), kWork… in AllocateBuffer()
92 params_->selected_ = allocator_->Malloc(kNumberTypeInt32, num_boxes_ * sizeof(int), kWorkspace); in AllocateBuffer()
94 …params_->single_class_indexes_ = allocator_->Malloc(kNumberTypeInt32, num_boxes_ * sizeof(int), kW… in AllocateBuffer()
99 …allocator_->Malloc(kNumberTypeFloat, (num_boxes_ + params_->max_detections_) * sizeof(float), kWor… in AllocateBuffer()
102 …allocator_->Malloc(kNumberTypeInt32, (num_boxes_ + params_->max_detections_) * sizeof(int), kWorks… in AllocateBuffer()
105 …allocator_->Malloc(kNumberTypeFloat, (num_boxes_ + params_->max_detections_) * sizeof(float), kWor… in AllocateBuffer()
[all …]
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_1x1_int8_coder.cc | 82 std::string src_in = allocator_->GetRuntimeAddr(input_tensor_) + " + batch_index * " + in DoCode()
84 std::string src_out = allocator_->GetRuntimeAddr(output_tensor_) + " + batch_index * " + in DoCode()
159 …packed_weight_ = static_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, kOnlineSize, kOnlinePac… in InitWeightBias()
161 …bias_data_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, kOnlineSize, kOnlinePackW… in InitWeightBias()
164 std::string packed_weight_str = "(int8_t **)&" + allocator_->GetRuntimeAddr(packed_weight_); in InitWeightBias()
165 std::string bias_data_str = "(int32_t **)&" + allocator_->GetRuntimeAddr(bias_data_); in InitWeightBias()
168 filter_zp_str = allocator_->GetRuntimeAddr(filter_zp_ptr_); in InitWeightBias()
204 …filter_zp_ptr_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, output_size, kOffline… in InitFilterPeroc()
212 …left_shift_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, oc_size, kOfflinePackWei… in InitFilterPeroc()
219 …right_shift_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, oc_size, kOfflinePackWe… in InitFilterPeroc()
[all …]
|
| D | convolution_depthwise_int8_coder.cc | 38 …row_buffer_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, row_buffer_size_, k… in InitBuffer()
51 …int8_t *tmp_weight = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, kOnlineSize, k… in InitWeightBias()
59 …packed_weight_ = reinterpret_cast<int16_t *>(allocator_->Malloc(kNumberTypeInt16, kOnlineSize, kOn… in InitWeightBias()
63 code << " " << allocator_->GetRuntimeAddr(packed_weight_) << "[i] = (int16_t)(" in InitWeightBias()
64 << allocator_->GetRuntimeAddr(tmp_weight) << "[i] - " << weight_zp << ");\n"; in InitWeightBias()
68 …bias_data_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, kOnlineSize, kOnline… in InitWeightBias()
|
| D | conv2d_3x3_int8_coder.cc | 57 …static_cast<int16_t *>(allocator_->Malloc(kNumberTypeInt16, transformed_size, kOfflinePackWeight)); in InitWeightBias()
66 …new_bias_addr_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, new_bias_size, kOffli… in InitWeightBias()
91 …tile_buffer_ = static_cast<int16_t *>(allocator_->Malloc(kNumberTypeInt16, tile_buffer_size_, kWor… in InitTmpBuffer()
96 … static_cast<int16_t *>(allocator_->Malloc(kNumberTypeInt16, block_unit_buffer_size_, kWorkspace)); in InitTmpBuffer()
100 …tmp_dst_buffer_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, tmp_dst_buffer_size_… in InitTmpBuffer()
104 tmp_out_ = static_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, tmp_out_size_, kWorkspace)); in InitTmpBuffer()
108 …c8_input_ = static_cast<int16_t *>(allocator_->Malloc(kNumberTypeInt16, c8_input_size_, kWorkspace… in InitTmpBuffer()
|
| D | matmul_base_int8_coder.cc | 35 …pack_a_ptr_ = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, a_pack_ptr_size_, kWo… in InitTmpBuffer()
39 …pack_b_ptr_ = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, kOnlineSize, kOnlineP… in InitTmpBuffer()
41 …pack_b_ptr_ = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, b_pack_ptr_size_, kWo… in InitTmpBuffer()
45 …input_sums_ = reinterpret_cast<int *>(allocator_->Malloc(kNumberTypeInt32, input_sums_size_, kWork… in InitTmpBuffer()
49 …weight_bias_sums_ = reinterpret_cast<int *>(allocator_->Malloc(kNumberTypeInt32, kOnlineSize, kOnl… in InitTmpBuffer()
52 … reinterpret_cast<int *>(allocator_->Malloc(kNumberTypeInt32, weight_bias_sums_size_, kWorkspace)); in InitTmpBuffer()
155 …bias_ptr_ = reinterpret_cast<int *>(allocator_->Malloc(kNumberTypeInt32, kOnlineSize, kOnlinePackW… in InitBias()
213 std::string a_ptr_str = allocator_->GetRuntimeAddr(input_tensor_); in DoCode()
214 std::string c_ptr_str = allocator_->GetRuntimeAddr(output_tensor_); in DoCode()
215 std::string pack_b_ptr_str = allocator_->GetRuntimeAddr(pack_b_ptr_); in DoCode()
[all …]
|
| D | deconvolution_int8_coder.cc | 68 …bias_data_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, size, kOfflinePackWe… in InitBiasWeight()
79 …weight_ptr_ = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, size, kOfflinePackWei… in InitBiasWeight()
90 …reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, size * sizeof(int32_t), kOfflineP… in InitBiasWeight()
104 …input_ptr_ = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, input_ptr_size_, kWork… in InitData()
112 …tmp_buffer_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, tmp_buffer_size_, k… in InitRunBuf()
116 …tmp_output_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, tmp_output_size_, k… in InitRunBuf()
119 …input_sum_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, input_sum_size_, kWo… in InitRunBuf()
|
| /third_party/skia/third_party/externals/tint/src/ |
| D | block_allocator.h | 110 Iterator begin() const { return Iterator(allocator_->objects_.begin()); } in begin()
112 Iterator end() const { return Iterator(allocator_->objects_.end()); } in end()
116 explicit View(BlockAllocator const* allocator) : allocator_(allocator) {} in View()
117 BlockAllocator const* const allocator_; variable
126 return ConstIterator(allocator_->objects_.begin()); in begin()
130 return ConstIterator(allocator_->objects_.end()); in end()
136 : allocator_(allocator) {} in ConstView()
137 BlockAllocator const* const allocator_; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | lstm_fp32_coder.cc | 35 …weight_i_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnl… in InitInputWeightBias()
45 …input_bias_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnlin… in InitInputWeightBias()
62 …weight_h_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnl… in InitStateWeightBias()
71 …reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, weight_h->Size(), kOfflinePackWei… in InitStateWeightBias()
77 …state_bias_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnlin… in InitStateWeightBias()
85 allocator_->GetRuntimeAddr(bias_i) + "+" + std::to_string(4 * lstm_param_->hidden_size_); in InitStateWeightBias()
123 buffer_[0] = reinterpret_cast<float *>(allocator_->Malloc( in MallocRunBuffer()
126 buffer_[1] = reinterpret_cast<float *>(allocator_->Malloc( in MallocRunBuffer()
131 buffer_[2] = reinterpret_cast<float *>(allocator_->Malloc( in MallocRunBuffer()
135 buffer_[3] = reinterpret_cast<float *>(allocator_->Malloc( in MallocRunBuffer()
[all …]
|
| D | matmul_fp32_base_coder.cc | 55 …bias_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnlineP… in InitBiasData()
84 …a_pack_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnlin… in InitBufferA()
86 …a_pack_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, a_pack_ptr_size_, k… in InitBufferA()
98 …b_pack_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnlin… in InitBufferB()
100 …b_pack_ptr_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, b_pack_ptr_size_, k… in InitBufferB()
185 std::string bias_ptr_str = "((float *)(" + allocator_->GetRuntimeAddr(bias_ptr_) + "))"; in DoCode()
195 std::string a_str = allocator_->GetRuntimeAddr(input_tensor_); in DoCode()
196 std::string b_str = allocator_->GetRuntimeAddr(filter_tensor_); in DoCode()
197 std::string c_str = allocator_->GetRuntimeAddr(output_tensor_); in DoCode()
198 std::string a_pack_str = allocator_->GetRuntimeAddr(a_pack_ptr_); in DoCode()
[all …]
|
| D | convolution_winograd_fp32_coder.cc | 90 …trans_input_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, tile_buffer_size_,… in InitTmpBuffer()
92 …gemm_out_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, gemm_out_size_, kWork… in InitTmpBuffer()
94 …tmp_data_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, tmp_data_size_, kWork… in InitTmpBuffer()
96 …col_buffer_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, col_buffer_size_, k… in InitTmpBuffer()
135 …allocator_->Malloc(kNumberTypeFloat32, trans_matrix_data_size * sizeof(float), kOfflinePackWeight)… in InitWeightBias()
159 …new_bias_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, new_bias_ele_size, kO… in InitWeightBias()
268 …code << "\t\tfloat *tmp_buffer_address_list[4] = {" << allocator_->GetRuntimeAddr(trans_input_) <<… in DoCode()
269 … << allocator_->GetRuntimeAddr(gemm_out_) << ", " << allocator_->GetRuntimeAddr(tmp_data_) << ", " in DoCode()
270 << allocator_->GetRuntimeAddr(col_buffer_) << "};\n"; in DoCode()
|
| D | deconv2d_fp32_coder.cc | 32 …packed_output_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, pack_output_size… in InitRunBuf()
40 …tmp_buffer_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, tmp_buffer_size_, k… in InitRunBuf()
48 …packed_input_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, pack_input_size_,… in InitRunBuf()
90 …packed_bias_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOnli… in InitWeightBias()
98 …packed_weight_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, kOnlineSize, kOn… in InitWeightBias()
172 std::string src_in_ptr_str = allocator_->GetRuntimeAddr(input_tensor_); in DoCode()
173 std::string src_out_ptr_str = allocator_->GetRuntimeAddr(output_tensor_); in DoCode()
|
| /third_party/boost/boost/asio/impl/ |
| D | executor.hpp | 55 allocator_(a) in impl()
69 allocator_type alloc(allocator_); in destroy()
93 executor_.dispatch(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in dispatch()
98 executor_.post(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in post()
103 executor_.defer(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in defer()
133 Allocator allocator_; member in boost::asio::executor::impl
137 allocator_type allocator_; member
141 : allocator_(a), in raw_mem()
142 ptr_(allocator_.allocate(1)) in raw_mem()
149 allocator_.deallocate(ptr_, 1); in ~raw_mem()
|
| D | thread_pool.hpp | 58 allocator_ = other.allocator_; in operator =()
81 allocator_ = std::move(other.allocator_); in operator =()
138 typename op::ptr p = { detail::addressof(allocator_), in do_execute()
139 op::ptr::allocate(allocator_), 0 }; in do_execute()
140 p.p = new (p.v) op(BOOST_ASIO_MOVE_CAST(Function)(f), allocator_); in do_execute()
208 typename op::ptr p = { detail::addressof(allocator_), in do_bulk_execute()
209 op::ptr::allocate(allocator_), 0 }; in do_bulk_execute()
210 p.p = new (p.v) op(BOOST_ASIO_MOVE_CAST(Function)(f), allocator_, i); in do_bulk_execute()
|
| /third_party/boost/libs/asio/include/boost/asio/impl/ |
| D | executor.hpp | 55 allocator_(a) in impl()
69 allocator_type alloc(allocator_); in destroy()
93 executor_.dispatch(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in dispatch()
98 executor_.post(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in post()
103 executor_.defer(BOOST_ASIO_MOVE_CAST(function)(f), allocator_); in defer()
133 Allocator allocator_; member in boost::asio::executor::impl
137 allocator_type allocator_; member
141 : allocator_(a), in raw_mem()
142 ptr_(allocator_.allocate(1)) in raw_mem()
149 allocator_.deallocate(ptr_, 1); in ~raw_mem()
|
| D | thread_pool.hpp | 58 allocator_ = other.allocator_; in operator =()
81 allocator_ = std::move(other.allocator_); in operator =()
138 typename op::ptr p = { detail::addressof(allocator_), in do_execute()
139 op::ptr::allocate(allocator_), 0 }; in do_execute()
140 p.p = new (p.v) op(BOOST_ASIO_MOVE_CAST(Function)(f), allocator_); in do_execute()
208 typename op::ptr p = { detail::addressof(allocator_), in do_bulk_execute()
209 op::ptr::allocate(allocator_), 0 }; in do_bulk_execute()
210 p.p = new (p.v) op(BOOST_ASIO_MOVE_CAST(Function)(f), allocator_, i); in do_bulk_execute()
|
| /third_party/boost/libs/asio/include/boost/asio/ |
| D | thread_pool.hpp | 196 allocator_(other.allocator_), in basic_executor_type()
208 allocator_(BOOST_ASIO_MOVE_CAST(Allocator)(other.allocator_)), in basic_executor_type()
255 pool_, allocator_, bits_ & ~blocking_mask); in require()
274 pool_, allocator_, bits_ & ~blocking_mask); in require()
292 pool_, allocator_, (bits_ & ~blocking_mask) | blocking_never); in require()
309 allocator_, bits_ & ~relationship_continuation); in require()
326 allocator_, bits_ | relationship_continuation); in require()
344 pool_, allocator_, bits_); in require()
362 pool_, allocator_, bits_); in require()
529 return allocator_; in query()
[all …]
|
| D | io_context.hpp | 679 allocator_(other.allocator_), in basic_executor_type()
691 allocator_(BOOST_ASIO_MOVE_CAST(Allocator)(other.allocator_)), in basic_executor_type()
737 allocator_, bits_ & ~blocking_never); in require()
754 allocator_, bits_ | blocking_never); in require()
771 allocator_, bits_ & ~relationship_continuation); in require()
788 allocator_, bits_ | relationship_continuation); in require()
806 io_context_, allocator_, bits_); in require()
824 io_context_, allocator_, bits_); in require()
972 return allocator_; in query()
988 return allocator_; in query()
[all …]
|
| D | system_executor.hpp | 44 : allocator_(Allocator()) in basic_system_executor()
69 Relationship, Allocator>(allocator_); in require()
87 Relationship, Allocator>(allocator_); in require()
105 Relationship, Allocator>(allocator_); in require()
123 execution::relationship_t::continuation_t, Allocator>(allocator_); in require()
141 execution::relationship_t::fork_t, Allocator>(allocator_); in require()
265 return allocator_; in query()
281 return allocator_; in query()
412 : allocator_(a) in basic_system_executor()
432 Allocator allocator_; member in boost::asio::basic_system_executor
|
| /third_party/boost/boost/asio/ |
| D | thread_pool.hpp | 196 allocator_(other.allocator_), in basic_executor_type()
208 allocator_(BOOST_ASIO_MOVE_CAST(Allocator)(other.allocator_)), in basic_executor_type()
255 pool_, allocator_, bits_ & ~blocking_mask); in require()
274 pool_, allocator_, bits_ & ~blocking_mask); in require()
292 pool_, allocator_, (bits_ & ~blocking_mask) | blocking_never); in require()
309 allocator_, bits_ & ~relationship_continuation); in require()
326 allocator_, bits_ | relationship_continuation); in require()
344 pool_, allocator_, bits_); in require()
362 pool_, allocator_, bits_); in require()
529 return allocator_; in query()
[all …]
|
| D | io_context.hpp | 679 allocator_(other.allocator_), in basic_executor_type()
691 allocator_(BOOST_ASIO_MOVE_CAST(Allocator)(other.allocator_)), in basic_executor_type()
737 allocator_, bits_ & ~blocking_never); in require()
754 allocator_, bits_ | blocking_never); in require()
771 allocator_, bits_ & ~relationship_continuation); in require()
788 allocator_, bits_ | relationship_continuation); in require()
806 io_context_, allocator_, bits_); in require()
824 io_context_, allocator_, bits_); in require()
972 return allocator_; in query()
988 return allocator_; in query()
[all …]
|
| D | system_executor.hpp | 44 : allocator_(Allocator()) in basic_system_executor()
69 Relationship, Allocator>(allocator_); in require()
87 Relationship, Allocator>(allocator_); in require()
105 Relationship, Allocator>(allocator_); in require()
123 execution::relationship_t::continuation_t, Allocator>(allocator_); in require()
141 execution::relationship_t::fork_t, Allocator>(allocator_); in require()
265 return allocator_; in query()
281 return allocator_; in query()
412 : allocator_(a) in basic_system_executor()
432 Allocator allocator_; member in boost::asio::basic_system_executor
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | fill.cc | 31 auto allocator_ = ocl_runtime_->GetAllocator(); in RunFill() local
40 if (allocator_->GetImageSize(src_data, &img_size) != RET_OK) { in RunFill()
46 cl::Image2D *out_image = allocator_->GetImage(src_data); in RunFill()
56 auto allocator_ = ocl_runtime_->GetAllocator(); in RunShape() local
57 CHECK_NULL_RETURN(allocator_); in RunShape()
69 cl::Image2D *out_image = allocator_->GetImage(src_data); in RunShape()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | tensor.cc | 285 allocator_ = allocator; in MallocData()
301 if (allocator_ == nullptr) { in MallocData()
304 this->data_ = allocator_->Malloc(data_size); in MallocData()
305 allocator_->SetRefCount(this->data_, 1); in MallocData()
316 if (IS_RUNTIME_ALLOCATOR(allocator_)) { in FreeData()
320 if (this->allocator_ != nullptr) { in FreeData()
321 if (allocator_->DecRefCount(this->data_, 1) <= 0) { in FreeData()
322 …allocator_->Free(this->data_); // Due to existing various allocator, here do not set data to null… in FreeData()
324 if (!IS_STATIC_ALLOCATOR(allocator_) || allocator_->RefCount(this->data_) != 0) { in FreeData()
|
| /third_party/skia/third_party/externals/tint/src/reader/spirv/ |
| D | parser_type.cc | 271 BlockAllocator<Type> allocator_; member
413 state->void_ = state->allocator_.Create<spirv::Void>(); in Void()
420 state->bool_ = state->allocator_.Create<spirv::Bool>(); in Bool()
427 state->u32_ = state->allocator_.Create<spirv::U32>(); in U32()
434 state->f32_ = state->allocator_.Create<spirv::F32>(); in F32()
441 state->i32_ = state->allocator_.Create<spirv::I32>(); in I32()
449 return state->allocator_.Create<spirv::Pointer>(el, sc); in Pointer()
456 return state->allocator_.Create<spirv::Reference>(el, sc); in Reference()
462 return state->allocator_.Create<spirv::Vector>(el, size); in Vector()
471 return state->allocator_.Create<spirv::Matrix>(el, columns, rows); in Matrix()
[all …]
|