| /third_party/mindspore/mindspore/lite/src/ |
| D | sub_graph_kernel.cc | 185 tensor->set_allocator(allocator); in Prepare()
192 tensor->set_allocator(context->allocator); in Prepare()
218 tensor->set_allocator(this->Context()->allocator); in Prepare()
222 out->set_allocator(this->Context()->allocator); in Prepare()
|
| D | tensorlist.h | 151 void set_allocator(AllocatorPtr allocator) override { in set_allocator() function
155 tensor->set_allocator(allocator); in set_allocator()
|
| D | lite_mindrt.cc | 120 old_tensor->set_allocator(kernel_->Context()->allocator); in IsolateInputData()
134 new_tensor->set_allocator(old_tensor->allocator()); in IsolateInputData()
137 new_tensor->set_allocator(kernel_->Context()->allocator); in IsolateInputData()
289 dst_tensor->set_allocator(src_tensor->allocator()); in MoveTensorInputData()
387 dst_tensorlist->set_allocator(src_tensorlist->allocator()); in MoveTensorListInputData()
436 dst_tensorlist->set_allocator(src_tensorlist->allocator()); in CastTensorListInputData()
|
| D | mindrt_executor.cc | 164 src_tensor->set_allocator(nullptr); in FreeOutputTensor()
|
| D | tensorlist.cc | 111 tensor_ptr->set_allocator(this->allocator()); in MallocTensorListData()
|
| D | lite_session.cc | 424 new_tensor->set_allocator(src_tensor->allocator()); /* GPU use opencl allocator */ in IsolateOutputTensor()
612 input->set_allocator(this->context_->allocator); in SetAllocatorForDelegateKernels()
616 output->set_allocator(this->context_->allocator); in SetAllocatorForDelegateKernels()
|
| D | tensor.h | 113 void set_allocator(AllocatorPtr allocator) override { allocator_ = allocator; } in set_allocator() function
|
| D | tensor.cc | 66 result->set_allocator(allocator); in CopyTensor()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | opencl_subgraph.cc | 201 tensor->set_allocator(allocator_); in Init()
205 tensor->set_allocator(allocator_); in Init()
307 tensor->set_allocator(allocator_); in Prepare()
311 tensor->set_allocator(allocator_); in Prepare()
326 tensor->set_allocator(allocator_); in Prepare()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensorlist_setitem.cc | 92 output0_->set_allocator(ms_context_->allocator); in Run()
106 output0_->set_allocator(ms_context_->allocator); in Run()
|
| /third_party/mindspore/mindspore/lite/include/ |
| D | ms_tensor.h | 48 virtual void set_allocator(AllocatorPtr allocator) = 0;
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/ |
| D | tensor.h | 45 void set_allocator(AllocatorPtr allocator) override {} in set_allocator() function
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/src/ |
| D | tensor.h | 45 void set_allocator(mindspore::Allocator *allocator) override {} in set_allocator() function
|
| /third_party/mindspore/mindspore/lite/test/st/ |
| D | graph_test.cc | 71 out_tensor->set_allocator(nullptr); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/cxx_api/tensor/ |
| D | tensor_impl.h | 151 lite_tensor_->set_allocator(allocator); in SetAllocator()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_utils.cc | 158 restore_tensor->set_allocator(tensor->allocator()); in CastTensor()
|
| /third_party/boost/boost/property_tree/detail/ |
| D | rapidxml.hpp | 540 void set_allocator(boost_ptree_raw_alloc_func *af, boost_ptree_raw_free_func *ff) in set_allocator() function in boost::property_tree::detail::rapidxml::memory_pool
|
| /third_party/boost/libs/geometry/doc/src/docutils/tools/doxygen_xml2qbk/contrib/rapidxml-1.13/ |
| D | rapidxml.hpp | 552 void set_allocator(alloc_func *af, free_func *ff) in set_allocator() function in rapidxml::memory_pool
|