| /third_party/mindspore/mindspore/lite/src/ |
| D | tensorlist.cc | 26 TensorList::TensorList(std::vector<int> shape, std::vector<int> element_shape, Category category) in TensorList() function in mindspore::lite::TensorList
30 TensorList::~TensorList() { in ~TensorList()
32 this->TensorList::FreeData(); in ~TensorList()
37 int TensorList::CopyTensorList(const TensorList &src, bool copy_data) { in CopyTensorList()
60 int TensorList::CopyTensorData(const TensorList &src) { in CopyTensorData()
83 int TensorList::MallocTensorListData(TypeId dtype, const std::vector<std::vector<int> > &tensor_sha… in MallocTensorListData()
120 int TensorList::MallocData(const AllocatorPtr allocator) { in MallocData()
146 void TensorList::FreeData() { in FreeData()
159 int TensorList::FreeTensorListData() { in FreeTensorListData()
174 int TensorList::SetTensor(int index, const Tensor *src_tensor) { in SetTensor()
[all …]
|
| D | tensorlist.h | 58 class TensorList : public Tensor {
60 TensorList() = default;
62 TensorList(std::vector<int> shape, std::vector<int> element_shape, Category category = VAR);
64 ~TensorList() override;
66 TensorList(const TensorList &other) = delete;
68 TensorList &operator=(const TensorList &tl) = delete;
86 int CopyTensorList(const TensorList &src, bool copy_data);
88 int CopyTensorData(const TensorList &src);
|
| D | lite_mindrt.h | 108 void MoveTensorListInputData(TensorList *dst_tensor, TensorList *src_tensor);
109 int CastTensorListInputData(TensorList *dst_tensor, TensorList *src_tensor);
|
| D | lite_mindrt.cc | 113 auto old_tensorlist = reinterpret_cast<TensorList *>(old_tensor); in IsolateInputData()
309 …MoveTensorListInputData(reinterpret_cast<TensorList *>(dst_tensor), reinterpret_cast<TensorList *>… in MoveInputData()
330 …ret = CastTensorListInputData(reinterpret_cast<TensorList *>(dst), reinterpret_cast<TensorList *>(… in CastInputData()
346 reinterpret_cast<TensorList *>(dst_tensor)->tensors_data_type() != in NeedCastData()
347 reinterpret_cast<TensorList *>(src_tensor)->tensors_data_type()) { in NeedCastData()
382 void LiteOpActor::MoveTensorListInputData(TensorList *dst_tensorlist, TensorList *src_tensorlist) { in MoveTensorListInputData()
421 int LiteOpActor::CastTensorListInputData(TensorList *dst_tensorlist, TensorList *src_tensorlist) { in CastTensorListInputData()
714 auto input_tensorlist = reinterpret_cast<TensorList *>(input_tensor); in SetInputShape()
715 auto input_data_tensorlist = reinterpret_cast<TensorList *>(inputs_data_[i]); in SetInputShape()
|
| D | lite_session.cc | 147 auto tensor_list = reinterpret_cast<TensorList *>(dst_tensor); in ConvertTensorsData()
201 dst_tensor = new (std::nothrow) TensorList(shape, std::vector<int>(), src_category); in ConvertTensor()
203 auto tensor_list = reinterpret_cast<TensorList *>(dst_tensor); in ConvertTensor()
|
| D | scheduler.cc | 1595 auto tensor_list = reinterpret_cast<TensorList *>(tensor); in GetFirstFp32Fp16OrInt8Type()
1710 void CopyTensorList(TensorList *dst_tensor, TensorList *src_tensor) { in CopyTensorList()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | infer_manager_test.cc | 47 TensorList *tensorList = new (std::nothrow) TensorList; in TEST_F()
54 TensorList *out = reinterpret_cast<TensorList *>(outputs[0]); in TEST_F()
72 delete reinterpret_cast<TensorList *>(inputs[i]); in TEST_F()
79 delete reinterpret_cast<TensorList *>(outputs[i]); in TEST_F()
88 TensorList *tensorList = new TensorList; in TEST_F()
135 delete reinterpret_cast<TensorList *>(inputs[i]); in TEST_F()
166 TensorList *tensorList = new (std::nothrow) TensorList; in TEST_F()
173 TensorList *out = reinterpret_cast<TensorList *>(outputs[0]); in TEST_F()
187 delete reinterpret_cast<TensorList *>(inputs[i]); in TEST_F()
194 delete reinterpret_cast<TensorList *>(outputs[i]); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | carry_data.cc | 49 ret = MoveTensorListData(reinterpret_cast<lite::TensorList *>(dst_tensor), in MoveData()
50 reinterpret_cast<lite::TensorList *>(src_tensor)); in MoveData()
90 int CarryDataKernel::MoveTensorListData(lite::TensorList *dst_tensorlist, lite::TensorList *src_ten… in MoveTensorListData()
|
| D | carry_data.h | 41 int MoveTensorListData(lite::TensorList *dst_tensorlist, lite::TensorList *src_tensorlist);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensor_array.cc | 59 this->tensor_list_ = std::make_unique<lite::TensorList>(shape, element_shape_v); in Init()
101 lite::TensorList **delta = static_cast<lite::TensorList **>(input_x->data()); in Run()
103 lite::TensorList *tensor_list = *delta; in Run()
|
| D | tensorlist_setitem.h | 41 lite::TensorList *input0_ = nullptr;
43 lite::TensorList *output0_ = nullptr;
|
| D | tensorlist_setitem.cc | 69 input0_ = reinterpret_cast<lite::TensorList *>(in_tensors_[0]); in Run()
71 output0_ = reinterpret_cast<lite::TensorList *>(out_tensors_[0]); in Run()
90 output0_ = reinterpret_cast<lite::TensorList *>(out_tensors_[0]); in Run()
|
| D | tensorlist_stack.h | 49 lite::TensorList *input0_ = nullptr;
|
| D | tensorlist_reserve.cc | 45 auto output = reinterpret_cast<lite::TensorList *>(out_tensors_[0]); in Run()
|
| D | tensorlist_getitem.cc | 39 auto input0 = reinterpret_cast<lite::TensorList *>(in_tensors_.at(0)); in Run()
|
| D | tensor_array.h | 43 std::unique_ptr<lite::TensorList> tensor_list_;
|
| D | tensorlist_fromtensor.cc | 86 auto output0 = reinterpret_cast<lite::TensorList *>(output0_); in Run()
|
| D | tensorlist_stack.cc | 58 input0_ = reinterpret_cast<lite::TensorList *>(in_tensors_.at(0)); in Init()
|
| /third_party/mindspore/mindspore/lite/src/common/ |
| D | tensor_util.h | 38 int TensorList2TensorListC(TensorList *src, TensorListC *dst);
39 int TensorListC2TensorList(const TensorListC *src, TensorList *dst);
|
| D | tensor_util.cc | 101 int TensorList2TensorListC(TensorList *src, TensorListC *dst) { in TensorList2TensorListC()
136 int TensorListC2TensorList(const TensorListC *src, TensorList *dst) { in TensorListC2TensorList()
200 auto *tensor_list = reinterpret_cast<TensorList *>(input); in GenerateInTensorC()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | node_infershape.cc | 358 auto tensor_list = new (std::nothrow) lite::TensorList(data_info.shape_, {}); in GetCNodeTensorListVarInput()
439 …tensor = new (std::nothrow) lite::TensorList(data_info.shape_, std::vector<int>(), tensor_category… in ConvertToLiteTensor()
448 auto tensor_list = reinterpret_cast<lite::TensorList *>(tensor); in ConvertToLiteTensor()
545 auto tensor_list = reinterpret_cast<lite::TensorList *>(tensor); in ConvertTensorListToAbstract()
|
| /third_party/mindspore/third_party/proto/tensorflow/ |
| D | types.proto | 85 // "tensorflow::TensorList" in the variant type registry.
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | vmimpl.h | 41 using TensorList = std::vector<tensor::TensorPtr>; variable
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 110 lite_tensor = std::make_unique<TensorList>(tensor_shape, element_shape); in ConvertTensorList()
117 auto lite_tensor_list = reinterpret_cast<TensorList *>(lite_tensor.get()); in ConvertTensorList()
276 auto tensor_list = reinterpret_cast<TensorList *>(output_tensors[i]); in SetDataType()
|
| /third_party/mindspore/mindspore/lite/src/runtime/ |
| D | infer_manager.cc | 165 auto *tensor_list = reinterpret_cast<TensorList *>(outputs.at(i)); in KernelInferShape()
|