Home
last modified time | relevance | path

Searched defs:tensor_list (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/ut/python/ops/
Dtest_list_slice.py38 def construct(self, tensor_list): argument
60 def construct(self, tensor_list): argument
84 def construct(self, tensor_list): argument
96 def construct(self, tensor_list): argument
107 def construct(self, tensor_list): argument
118 def construct(self, tensor_list): argument
128 def construct(self, tensor_list, y): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/debugger/gpu_tests/
Ddump_test_utils.py110 def build_dump_structure(tensor_name_list, tensor_list, net_name, tensor_info_list): argument
153 def build_dump_structure_with_constant(tensor_name_list, tensor_list, net_name, tensor_info_list): argument
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/
DtfReader_op_test.cc72 TensorRow tensor_list; in TEST_F() local
132 TensorRow tensor_list; in TEST_F() local
192 TensorRow tensor_list; in TEST_F() local
252 TensorRow tensor_list; in TEST_F() local
315 TensorRow tensor_list; in TEST_F() local
Dmind_record_op_test.cc109 TensorRow tensor_list; in TEST_F() local
180 TensorRow tensor_list; in TEST_F() local
Dconcatenate_op_test.cc76 TensorRow tensor_list; in TEST_F() local
112 TensorRow tensor_list; in TEST_F() local
Dresize_op_test.cc56 std::vector<std::shared_ptr<Tensor>> tensor_list; in TEST_F() local
Dnormalize_op_test.cc70 std::vector<std::shared_ptr<Tensor>> tensor_list; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/control/
Dtensor_array.cc78 auto tensor_list = this->tensor_list_.get(); in Run() local
110 lite::TensorList *tensor_list = *delta; in Run() local
/third_party/mindspore/mindspore-src/source/tests/st/debugger/
Ddump_test_utils.py77 def build_dump_structure(path, tensor_name_list, tensor_list, net_name, tensor_info_list): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dtensor_util.cc142 auto *tensor_list = reinterpret_cast<TensorList *>(input); in GenerateInTensorC() local
571 auto tensor_list = reinterpret_cast<TensorList *>(tensor); in TensorListDataType() local
583 auto *tensor_list = reinterpret_cast<TensorList *>(tensor); in MallocTensorListDataAccordingToTensorListC() local
594 auto tensor_list = reinterpret_cast<TensorList *>(tensor); in DecodeTensorLsit() local
605 auto tensor_list = reinterpret_cast<TensorList *>(dst_tensor); in CreateTensorList() local
690 auto tensor_list = reinterpret_cast<TensorList *>(tensor); in MallocTensorData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dtensorlist_c_utils.c19 int MallocTensorListData(TensorListC *tensor_list, TypeIdC dtype, const vvector *tensor_shape) { in MallocTensorListData()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/communication/
Dcomm_func.py89 def _check_all_tensors(tensor_list): argument
99 def _check_all_tensors_or_tuple(tensor_list): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils.cc593 ShapeVector ConvertTensorListToShapeVector(const tensor::TensorPtrList &tensor_list, size_t index) { in ConvertTensorListToShapeVector()
614 tensor::CSRTensorPtr TensorListToCSRTensor(const tensor::TensorPtrList &tensor_list) { in TensorListToCSRTensor()
623 tensor::COOTensorPtr TensorListToCOOTensor(const tensor::TensorPtrList &tensor_list) { in TensorListToCOOTensor()
633 const tensor::TensorPtrList &tensor_list) { in TensorListToSparseTensor()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/
Dnpu_map_job.cc130 for (auto &tensor_list : hold_input_list) { in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dlite_tensor_extractor.cc96 auto tensor_list = std::static_pointer_cast<lite::TensorList>(tensor); in ConvertToLiteTensor() local
128 auto tensor_list = std::make_shared<lite::TensorList>(data_info.shape_, std::vector<int>{}); in GetCNodeTensorListVarInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dop_api_convert.h464 inline aclTensorList *ConvertType(const std::vector<tensor::BaseTensorPtr> &tensor_list) { in ConvertType()
486 inline aclTensorList *ConvertType(const std::vector<tensor::TensorPtr> &tensor_list) { in ConvertType()
497 inline aclTensorList *ConvertType(const std::vector<mindspore::kernel::KernelTensor *> &tensor_list in ConvertType()
Dop_api_cache.h101 inline void RefreshAddr(const std::vector<mindspore::kernel::KernelTensor *> &tensor_list) { in RefreshAddr()
Dop_api_cache.cc115 void GatherInfo(const std::vector<mindspore::kernel::KernelTensor *> &tensor_list) { in GatherInfo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dinfer_manager.cc225 auto tensor_list = MallocTensorListDataAccordingToTensorListC(outputs.at(i), tensor_list_c); in KernelInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/providers/nnie/
Dnnie_micro.cc47 int MakeTensorList(TensorC *tensors, int tensor_num, NnieTensors *tensor_list) { in MakeTensorList()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/
Dtensor_load.h114 std::vector<std::shared_ptr<TensorData>> tensor_list; in GetTensor() local
Ddebug_services.cc537std::vector<std::shared_ptr<TensorData>> *const tensor_list, int begin, in CheckWatchpointsForTensor()
651 std::vector<std::shared_ptr<TensorData>> *const tensor_list, in CheckWatchpoints()
1204std::vector<std::shared_ptr<TensorData>> *const tensor_list) { in GetTensorDataInfoAsync()
1690 std::vector<std::shared_ptr<TensorData>> tensor_list; in ReadNeededDumpedTensors() local
1758std::vector<std::shared_ptr<TensorData>> *const tensor_list, in ProcessTensorDataSync()
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dparam_validator.cc69 TypePtr CheckTensorsDTypeSame(const AbstractTensorPtrList &tensor_list, const TypePtrList &accepts, in CheckTensorsDTypeSame()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_utils.h224 static ValueTuplePtr ConvertTensorVectorToTuple(const std::vector<BaseTensorPtr> &tensor_list) { in ConvertTensorVectorToTuple()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/offline_debug/
Ddbg_services.cc218 std::vector<std::shared_ptr<TensorData>> tensor_list; in CheckWatchpoints() local

12