| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | cast_tests.cc | 76 for (auto tensor : inputs) { in TEST_F() local
79 for (auto tensor : outputs) { in TEST_F() local
88 for (auto tensor : inputs) { in TEST_F() local
91 for (auto tensor : outputs) { in TEST_F() local
107 for (auto tensor : inputs) { in TEST_F() local
110 for (auto tensor : outputs) { in TEST_F() local
120 for (auto tensor : inputs) { in TEST_F() local
123 for (auto tensor : outputs) { in TEST_F() local
139 for (auto tensor : inputs) { in TEST_F() local
143 for (auto tensor : outputs) { in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/c_api/ |
| D | tensor_c.cc | 73 void OH_AI_TensorDestroy(OH_AI_TensorHandle *tensor) { in OH_AI_TensorDestroy()
83 OH_AI_TensorHandle OH_AI_TensorClone(OH_AI_TensorHandle tensor) { in OH_AI_TensorClone()
99 void OH_AI_TensorSetName(OH_AI_TensorHandle tensor, const char *name) { in OH_AI_TensorSetName()
108 const char *OH_AI_TensorGetName(const OH_AI_TensorHandle tensor) { in OH_AI_TensorGetName()
117 void OH_AI_TensorSetDataType(OH_AI_TensorHandle tensor, OH_AI_DataType type) { in OH_AI_TensorSetDataType()
126 OH_AI_DataType OH_AI_TensorGetDataType(const OH_AI_TensorHandle tensor) { in OH_AI_TensorGetDataType()
136 void OH_AI_TensorSetShape(OH_AI_TensorHandle tensor, const int64_t *shape, size_t shape_num) { in OH_AI_TensorSetShape()
149 const int64_t *OH_AI_TensorGetShape(const OH_AI_TensorHandle tensor, size_t *shape_num) { in OH_AI_TensorGetShape()
159 void OH_AI_TensorSetFormat(OH_AI_TensorHandle tensor, OH_AI_Format format) { in OH_AI_TensorSetFormat()
168 OH_AI_Format OH_AI_TensorGetFormat(const OH_AI_TensorHandle tensor) { in OH_AI_TensorGetFormat()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/communication/ |
| D | comm_func.py | 46 def _check_split_sizes_sequence(tensor, sequence): argument
58 def _check_compute_split_count(tensor, output_split_sizes, input_split_sizes, group): argument
135 def all_reduce(tensor, op=ReduceOp.SUM, group=GlobalComm.WORLD_COMM_GROUP): argument
192 def all_gather_into_tensor(tensor, group=GlobalComm.WORLD_COMM_GROUP): argument
255 def reduce_scatter_tensor(tensor, op=ReduceOp.SUM, group=GlobalComm.WORLD_COMM_GROUP): argument
319 def reduce(tensor, dst, op=ReduceOp.SUM, group=GlobalComm.WORLD_COMM_GROUP): argument
431 def __init__(self, op, tensor, peer, group=None, tag=0, *, recv_dtype=None): argument
439 def __new__(cls, op, tensor, peer, group=None, tag=0, recv_dtype=None): argument
564 def scatter_tensor(tensor, src=0, group=GlobalComm.WORLD_COMM_GROUP): argument
630 def gather_into_tensor(tensor, dst=0, group=GlobalComm.WORLD_COMM_GROUP): argument
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/ |
| D | tensorlist.h | 128 for (auto tensor : tensors_) { in set_init_ref_count() local
137 for (auto tensor : tensors_) { in set_ref_count() local
146 for (auto tensor : tensors_) { in ResetRefCount() local
155 for (auto tensor : tensors_) { in IncRefCount() local
167 for (auto tensor : tensors_) { in DecRefCount() local
176 for (auto tensor : tensors_) { in set_allocator() local
185 for (auto tensor : tensors_) { in set_own_data() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/ |
| D | tensor.cc | 49 TensorPtr tensor = nullptr; in MSNewTensor() local
66 TensorPtr tensor = nullptr; in MSNewTensorFromFile() local
103 TensorPtr tensor = nullptr; in MSNewTensorWithSrcType() local
119 TensorPtr tensor = nullptr; in MSNewTensorScalarFloat32() local
136 TensorPtr tensor = nullptr; in MSNewTensorScalarInt32() local
148 void *MSTensorGetData(ResMgrHandle res_mgr, ConstTensorHandle tensor) { in MSTensorGetData()
161 STATUS MSTensorSetDataType(ResMgrHandle res_mgr, TensorHandle tensor, DataTypeC type) { in MSTensorSetDataType()
175 DataTypeC MSTensorGetDataType(ResMgrHandle res_mgr, ConstTensorHandle tensor, STATUS *error) { in MSTensorGetDataType()
195 size_t MSTensorGetDataSize(ResMgrHandle res_mgr, ConstTensorHandle tensor, STATUS *error) { in MSTensorGetDataSize()
216 size_t MSTensorGetElementNum(ResMgrHandle res_mgr, ConstTensorHandle tensor, STATUS *error) { in MSTensorGetElementNum()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/minddata/wrapper/ |
| D | album_op_android.cc | 180 … AlbumOp::LoadImageTensor(const std::string &image_file_path, int32_t col_num, TensorPtr *tensor) { in LoadImageTensor()
260 …lbumOp::LoadStringArrayTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor) { in LoadStringArrayTensor()
268 Status AlbumOp::LoadStringTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor… in LoadStringTensor()
278 …s AlbumOp::LoadIntArrayTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor) { in LoadIntArrayTensor()
305 …AlbumOp::LoadFloatArrayTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor) { in LoadFloatArrayTensor()
332 Status AlbumOp::LoadIDTensor(const std::string &file, int32_t col_num, TensorPtr *tensor) { in LoadIDTensor()
344 Status AlbumOp::LoadEmptyTensor(int32_t col_num, TensorPtr *tensor) { in LoadEmptyTensor()
354 Status AlbumOp::LoadFloatTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor)… in LoadFloatTensor()
368 Status AlbumOp::LoadIntTensor(const nlohmann::json &json_obj, int32_t col_num, TensorPtr *tensor) { in LoadIntTensor()
387 TensorPtr tensor; in LoadIntTensorRowByIndex() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/ |
| D | tensor_c_utils.c | 120 int GetBatch(const TensorC *tensor) { in GetBatch()
152 int GetHeight(const TensorC *tensor) { in GetHeight()
180 int GetWidth(const TensorC *tensor) { in GetWidth()
211 int GetChannel(const TensorC *tensor) { in GetChannel()
244 void NnaclSetBatch(TensorC *tensor, int batch) { in NnaclSetBatch()
281 void SetHeight(TensorC *tensor, int height) { in SetHeight()
313 void SetWidth(TensorC *tensor, int width) { in SetWidth()
348 void SetChannel(TensorC *tensor, int channel) { in SetChannel()
385 int GetSize(const TensorC *tensor) { in GetSize()
391 int GetElementNum(const TensorC *tensor) { in GetElementNum()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/session/ |
| D | dynamic_session.cc | 44 for (auto tensor : tensors_) { in CompileGraph() local
81 bool DynamicSession::CheckTensorIsVar(Tensor *tensor) { in CheckTensorIsVar()
89 for (auto tensor : kernel->in_tensors()) { in PreProcess() local
119 for (auto tensor : kernel->in_tensors()) { in PrepareInOuts() local
128 for (auto tensor : kernel->out_tensors()) { in PrepareInOuts() local
136 for (auto tensor : kernel->in_tensors()) { in ResetInOuts() local
143 for (auto tensor : kernel->out_tensors()) { in ResetInOuts() local
186 for (auto tensor : kernel->in_tensors()) { in RunKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_tensor.cc | 54 std::string MindIR_Tensor_GetName(ConstTensorPtr tensor) { in MindIR_Tensor_GetName()
67 void MindIR_Tensor_SetName(TensorPtr *tensor, const std::string &name) { in MindIR_Tensor_SetName()
94 DataType MindIR_Tensor_GetDataType(ConstTensorPtr tensor) { in MindIR_Tensor_GetDataType()
109 void MindIR_Tensor_SetDataType(TensorPtr *tensor, DataType data_type) { in MindIR_Tensor_SetDataType()
143 std::vector<int32_t> MindIR_Tensor_GetDims(ConstTensorPtr tensor) { in MindIR_Tensor_GetDims()
163 void MindIR_Tensor_SetDims(TensorPtr *tensor, const std::vector<int32_t> &dims) { in MindIR_Tensor_SetDims()
190 Format MindIR_Tensor_GetFormat(ConstTensorPtr tensor) { in MindIR_Tensor_GetFormat()
205 void MindIR_Tensor_SetFormat(TensorPtr *tensor, Format format) { in MindIR_Tensor_SetFormat()
238 std::vector<uint8_t> MindIR_Tensor_GetData(ConstTensorPtr tensor) { in MindIR_Tensor_GetData()
258 void MindIR_Tensor_SetData(TensorPtr *tensor, const std::vector<uint8_t> &data) { in MindIR_Tensor_SetData()
[all …]
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/cxx_api/ |
| D | types_test.cc | 29 MSTensor tensor; in TEST_F() local
41 MSTensor tensor(tensor_name, data_type, {}, nullptr, 0); in TEST_F() local
51 MSTensor tensor(nullptr); in TEST_F() local
57 MSTensor tensor("", DataType::kNumberTypeInt32, {4}, data.data(), data.size() * sizeof(int32_t)); in TEST_F() local
67 MSTensor tensor("", DataType::kNumberTypeInt32, {4}, data.data(), data.size() * sizeof(int32_t)); in TEST_F() local
78 MSTensor tensor("", DataType::kNumberTypeInt32, {4}, data.data(), data.size() * sizeof(int32_t)); in TEST_F() local
91 MSTensor tensor("", DataType::kNumberTypeInt32, {4}, data.data(), data.size() * sizeof(int32_t)); in TEST_F() local
104 MSTensor tensor("", DataType::kNumberTypeInt32, {4}, data.data(), data.size() * sizeof(int32_t)); in TEST_F() local
118 MSTensor *tensor = in TEST_F() local
132 MSTensor *tensor = in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/ |
| D | tensor_pybind.cc | 94 ….def("set_tensor_name", [](MSTensor &tensor, const std::string &name) { tensor.SetTensorName(name)… in TensorPyBind()
106 .def("is_null", [](const MSTensorPtr &tensor) { return tensor == nullptr; }) in TensorPyBind()
108 [](const MSTensorPtr &tensor) { in TensorPyBind()
116 …[](const MSTensorPtr &tensor, const py::array &input) { return SetTensorNumpyData(tensor, input); … in TensorPyBind()
117 .def("get_data_to_numpy", [](const MSTensorPtr &tensor) -> py::array { in TensorPyBind()
140 …auto tensor = mindspore::MSTensor::CreateTensor("", data_type, shape, nullptr, 0, device_type, dev… in create_tensor() local
150 MSTensorPtr create_tensor_by_tensor(const MSTensor &tensor, const std::string &device_type, int dev… in create_tensor_by_tensor()
241 auto &tensor = *tensor_ptr; in SetTensorNumpyData() local
278 py::buffer_info GetPyBufferInfo(const MSTensorPtr &tensor) { in GetPyBufferInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/ |
| D | nnapi_subgraph.cc | 42 std::for_each(op->inputs().begin(), op->inputs().end(), [&](const MSTensor &tensor) { in Init()
47 std::for_each(op->outputs().begin(), op->outputs().end(), [&](const MSTensor &tensor) { in Init()
87 auto tensor = all_tensors_.at(idx); in CreateNNAPIModel() local
145 auto tensor = inputs_.at(idx); in PreProcess() local
154 auto tensor = outputs_.at(idx); in PreProcess() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/ |
| D | opencl_subgraph.cc | 309 for (const auto tensor : in_tensors()) { in RunPass() local
313 for (const auto tensor : out_tensors()) { in RunPass() local
393 for (auto *tensor : node->in_tensors()) { in GetInOutNodes() local
399 for (auto *tensor : node->out_tensors()) { in GetInOutNodes() local
411 for (const auto tensor : in_tensors()) { in Prepare() local
415 for (const auto tensor : out_tensors()) { in Prepare() local
429 for (const auto tensor : node->out_tensors()) { in Prepare() local
466 for (const auto &tensor : in_convert_tensors_) { in UnInit() local
470 for (const auto &tensor : out_convert_tensors_) { in UnInit() local
515 for (auto &tensor : in_tensors()) { in Execute() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/ |
| D | compile_result.cc | 58 inline std::string DumpTensor(const InferTensor *tensor, int indent = 0) { in DumpTensor()
121 void CompileNode::AppendInputTensor(InferTensor *tensor) { this->inputs_.emplace_back(tensor); } in AppendInputTensor()
123 void CompileNode::AppendOutputTensor(InferTensor *tensor) { this->outputs_.emplace_back(tensor); } in AppendOutputTensor()
224 StatusCode CompileResult::AppendTensor(InferTensor *tensor) { in AppendTensor()
236 StatusCode CompileResult::AppendInputTensor(InferTensor *tensor, bool is_borrow) { in AppendInputTensor()
251 StatusCode CompileResult::AppendOutputTensor(InferTensor *tensor, bool is_borrow) { in AppendOutputTensor()
269 …sCode CompileResult::AppendNodeInputTensor(const CompileNodePtr &compile_node, InferTensor *tensor, in AppendNodeInputTensor()
300 …Code CompileResult::AppendNodeOutputTensor(const CompileNodePtr &compile_node, InferTensor *tensor, in AppendNodeOutputTensor()
351 for (auto &tensor : tensors_) { in Dump() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/js_api/ |
| D | mstensor_napi.cc | 88 napi_value MSTensorNapi::NewInstance(napi_env env, mindspore::MSTensor tensor) { in NewInstance()
157 MSTensorNapi *tensor = nullptr; in GetName() local
186 MSTensorNapi *tensor = nullptr; in GetShape() local
223 MSTensorNapi *tensor = nullptr; in GetElementNum() local
252 MSTensorNapi *tensor = nullptr; in GetDtype() local
281 MSTensorNapi *tensor = nullptr; in GetFormat() local
310 MSTensorNapi *tensor = nullptr; in GetDataSize() local
340 MSTensorNapi *tensor = nullptr; in GetDataBuffer() local
380 MSTensorNapi *tensor = nullptr; in SetData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | op_api_cache.cc | 25 void GatherInfo(mindspore::kernel::KernelTensor *tensor) { in GatherInfo()
108 auto tensor = tensor_and_trans.first; in GatherInfo() local
116 for (auto tensor : tensor_list) { in GatherInfo() local
121 void GatherInfo(const mindspore::tensor::BaseTensorPtr &tensor) { in GatherInfo()
165 void GatherInfo(const std::optional<tensor::BaseTensorPtr> &tensor) { in GatherInfo()
174 for (const auto &tensor : tensors) { in GatherInfo() local
179 void GatherInfo(const mindspore::tensor::TensorPtr &tensor) { GatherInfo(tensor->cast<tensor::BaseT… in GatherInfo()
181 void GatherInfo(const std::optional<tensor::TensorPtr> &tensor) { in GatherInfo()
190 for (const auto &tensor : tensors) { in GatherInfo() local
263 void RefreshAddr(mindspore::kernel::KernelTensor *tensor) { in RefreshAddr()
|
| /third_party/mindspore/test/unittest/ |
| D | mindspore_unit_test.cpp | 305 OH_AI_TensorHandle tensor = outputs.handle_list[i]; variable
366 OH_AI_TensorHandle tensor = OH_AI_ModelGetInputByTensorName(model, "data"); variable
428 OH_AI_TensorHandle tensor = OH_AI_ModelGetOutputByTensorName(model, "prob"); variable
613 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("fc3.bias", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, crea… variable
673 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("fc3.bias", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, crea… variable
813 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("data", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, createSh… variable
842 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("data", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, createSh… variable
865 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("data", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, createSh… variable
891 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("data", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, createSh… variable
915 …OH_AI_TensorHandle tensor = OH_AI_TensorCreate("data", OH_AI_DATATYPE_NUMBERTYPE_FLOAT32, createSh… variable
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/ |
| D | execution_plan.cc | 30 for (auto tensor : inputs_) { in ~ExecutionPlan() local
33 for (auto tensor : outputs_) { in ~ExecutionPlan() local
80 for (auto tensor : kernel_->in_tensors()) { in CalcTensorRefCount() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/utils/ |
| D | coder_utils.cc | 28 bool CheckConstantTensor(const Tensor *const tensor) { in CheckConstantTensor()
33 void TensorDataToFile(const lite::Tensor *tensor, std::ofstream &ofs) { in TensorDataToFile()
54 void PrintTensorData(const lite::Tensor *tensor, std::ofstream &ofs) { in PrintTensorData()
86 for (const auto &tensor : tensors) { in TensorsToString() local
171 std::string GetTensorAddr(lite::Tensor *tensor, bool is_const, DynamicMemManager *dynamic_mem_manag… in GetTensorAddr()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | hook_py.cc | 23 namespace tensor { namespace
26 AutoGradMetaDataWeakPtr BuildAutoGradMeta(const tensor::Tensor &tensor) { in BuildAutoGradMeta()
41 uint64_t RegisterHook::RegisterTensorBackwardHook(const Tensor &tensor, const py::function &hook) { in RegisterTensorBackwardHook()
|
| D | tensor_py.cc | 34 namespace tensor { namespace
405 static py::buffer_info GetPyBufferInfo(const Tensor &tensor) { in GetPyBufferInfo()
412 py::tuple TensorPy::GetPyTupleShape(const Tensor &tensor) { in GetPyTupleShape()
421 py::tuple TensorPy::GetPyTupleStrides(const Tensor &tensor) { in GetPyTupleStrides()
431 py::int_ TensorPy::GetPyItemSize(const Tensor &tensor) { return tensor.data().itemsize(); } in GetPyItemSize()
433 py::int_ TensorPy::GetPyNBytes(const Tensor &tensor) { return tensor.data().nbytes(); } in GetPyNBytes()
460 void TensorPy::FlushFromCache(const Tensor &tensor) { in FlushFromCache()
494 py::bytes TensorPy::GetBytes(const Tensor &tensor) { in GetBytes()
544 tensor::TensorPtr tensor = std::make_shared<tensor::Tensor>(data_type, shape); in ConvertBytesToTensor() local
551 py::array TensorPy::SyncAsNumpy(const Tensor &tensor) { in SyncAsNumpy()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | miniaturization_pass.cc | 40 auto tensor = value->cast<tensor::TensorPtr>(); in GetTensorFromNode() local
47 bool MiniaturizationPass::NeedCompress(const tensor::TensorPtr &tensor) { in NeedCompress()
69 static inline ValuePtr GetFirstVal(const tensor::TensorPtr &tensor) { in GetFirstVal()
103 auto tensor = GetTensorFromNode(inputs[i]); in ProcessOneCNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/ |
| D | ascend_native_session.cc | 92 for (auto tensor : kernel->in_tensors()) { in InitializeTensorRefrenceCnt() local
108 for (auto &tensor : kernel->in_tensors()) { in AllocTensors() local
120 for (auto &tensor : kernel->out_tensors()) { in AllocTensors() local
127 for (auto &tensor : kernel->in_tensors()) { in AllocTensors() local
147 for (auto &tensor : kernel->in_tensors()) { in AllocTensors() local
156 for (auto &tensor : kernel->out_tensors()) { in AllocTensors() local
402 mindspore::tensor::Tensor tensor(type_id, shape64, ref_tensor_data); in LiteTensorToTensor() local
464 [](infer::abstract::Tensor *tensor) { return tensor->tensor_name(); }); in GetOutputNames()
471 [](infer::abstract::Tensor *tensor) { return tensor->tensor_name(); }); in GetInputNames()
476 …::find_if(lite_outputs.begin(), lite_outputs.end(), [tensorName](infer::abstract::Tensor *tensor) { in GetOutputByTensorName()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | debug_info_manager.cc | 64 const mindspore::schema::Tensor *tensor) { in AddQuantParamExtend()
120 auto tensor = iter->second.tensor_data; in FreeBuffer() local
130 auto tensor = info.tensor_data; in FreeBuffer() local
229 …nager::SetOriginStaticInfo(QuantDebugInfo *quant_debug_info, const mindspore::lite::Tensor &tensor, in SetOriginStaticInfo()
265 … QuantDebugInfo *quant_debug_info, const mindspore::lite::Tensor &tensor, in SetQuantStaticInfo()
439 … const mindspore::lite::Tensor *tensor, mindspore::lite::Tensor *new_tensor) { in GetConstTensor()
466 auto tensor = inputs.at(i); in GetOriginBeforeCallBack() local
504 auto tensor = inputs.at(i); in GetQuantBeforeCallBack() local
562 auto tensor = outputs.at(i); in GetAfterCallBack() local
584 auto tensor = outputs.at(i); in GetAfterCallBack() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | runtime_allocator.cc | 51 void RuntimeAllocator::FreeTensorData(lite::Tensor *tensor) { in FreeTensorData()
77 void RuntimeAllocator::SetDataOffset(lite::Tensor *tensor, size_t offset) { in SetDataOffset()
97 void RuntimeAllocator::MallocTensorData(lite::Tensor *tensor) { in MallocTensorData()
|