Home
last modified time | relevance | path

Searched defs:tensor_name (Results 1 – 25 of 60) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/
Dtensor_load.h39 void MoveTensorCurrentToPrev(const std::string &tensor_name) { in MoveTensorCurrentToPrev()
49 bool TensorExistsInCurrent(const std::string &tensor_name) const { in TensorExistsInCurrent()
54 bool PrevTensorExistsInCurrent(const std::string &tensor_name) const { in PrevTensorExistsInCurrent()
73 bool IsPrevTensor(std::string tensor_name) const { in IsPrevTensor()
90 auto tensor_name = tensor->GetName(); in LoadNewTensor() local
123 std::shared_ptr<TensorData> GetTensor(const std::string &tensor_name) const { in GetTensor()
131 std::shared_ptr<TensorData> GetPrevTensor(const std::string &tensor_name) { in GetPrevTensor()
175 void AppendToCacheEvictQueue(const std::string &tensor_name) { in AppendToCacheEvictQueue()
246 bool DumpTensorToFile(const std::string &filepath, const std::string &tensor_name, size_t slot) { in DumpTensorToFile()
Ddebug_services.cc340 const auto tensor_name = tensor->GetName(); in AddWatchPointsToCheck() local
382 const std::string &tensor_name) { in AddAnalyzedTensorToCache()
457 auto tensor_name = tensor->GetName(); in SetTensorToNotInUse() local
502 …id *DebugServices::PreparePrevTensor(uint64_t *prev_num_elements, const std::string &tensor_name) { in PreparePrevTensor()
547 const auto tensor_name = tensor->GetName(); in CheckWatchpointsForTensor() local
816 void DebugServices::ReadTensorFromNpy(const std::string &tensor_name, const std::string &file_name, in ReadTensorFromNpy()
1887 bool DebugServices::DumpTensorToFile(const std::string &filepath, const std::string &tensor_name, s… in DumpTensorToFile() argument
2290 bool DebugServices::TensorExistsInCurrent(const std::string &tensor_name) { in TensorExistsInCurrent()
2293 void DebugServices::MoveTensorCurrentToPrev(const std::string &tensor_name) { in MoveTensorCurrentToPrev()
2297 void DebugServices::AppendToCacheEvictQueue(const std::string &tensor_name) { in AppendToCacheEvictQueue()
/third_party/mindspore/mindspore-src/source/include/api/
Dmodel.h396 MSTensor Model::GetInputByTensorName(const std::string &tensor_name) { in GetInputByTensorName()
402 MSTensor Model::GetOutputByTensorName(const std::string &tensor_name) { in GetOutputByTensorName()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/
Dembedding_cache_manager.cc150 Status EmbeddingCacheManager::SetDeviceCacheAddr(const std::string &tensor_name, void *device_mem_a… in SetDeviceCacheAddr()
161 int EmbeddingCacheManager::CacheHandle(const std::string &tensor_name, mindspore::MSTensor model_in… in CacheHandle()
Dload_host_cache_model.cc44 std::string tensor_name; in SchemaTensorToMSTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/
Dembedding_cache_manager.cc150 Status EmbeddingCacheManager::SetDeviceCacheAddr(const std::string &tensor_name, void *device_mem_a… in SetDeviceCacheAddr()
161 int EmbeddingCacheManager::CacheHandle(const std::string &tensor_name, mindspore::MSTensor model_in… in CacheHandle()
Dload_host_cache_model.cc44 std::string tensor_name; in SchemaTensorToMSTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_session.h75 mindspore::lite::Tensor *GetInputsByTensorName(const std::string &tensor_name) const override { in GetInputsByTensorName()
86 mindspore::lite::Tensor *GetOutputByTensorName(const std::string &tensor_name) const override { in GetOutputByTensorName()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Dconverter_context.h99 …void UpdateGraphInputTensorShape(const std::string &tensor_name, const std::vector<int64_t> &shape… in UpdateGraphInputTensorShape()
103 std::vector<int64_t> GetGraphInputTensorShape(const std::string &tensor_name) const { in GetGraphInputTensorShape()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Ddebug_info_manager.h63 std::string tensor_name; member
82 std::string tensor_name; member
Dweight_quantizer.cc355 auto tensor_name = parameter->fullname_with_scope(); in InsertAscendDequantNode() local
401 auto tensor_name = parameter->fullname_with_scope(); in InsertDequantNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dtensorsummary_utils.cc43 std::string tensor_name = dataset.GetDatasetName(); in SummaryReceiveData() local
Dtensordump_utils.cc131 std::string tensor_name = dataset.GetDatasetName(); in AsyncSaveDatasetToNpyFile() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/
Dsingle_op_session.cc248 auto tensor_name = FuncGraphUtils::GetTensorName(tensor); in InitInputOutputInfos() local
262 auto tensor_name = FuncGraphUtils::GetTensorName(tensor); in InitInputOutputInfos() local
525 …nsorImplPtr SingleOpInferSession::GetOutputByTensorName(uint32_t, const std::string &tensor_name) { in GetOutputByTensorName()
537 …ensorImplPtr SingleOpInferSession::GetInputByTensorName(uint32_t, const std::string &tensor_name) { in GetInputByTensorName()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/cxx_api/
Dtypes_test.cc39 std::string tensor_name = "Name1"; in TEST_F() local
145 std::string tensor_name = "tensor_name"; in TEST_F() local
162 std::string tensor_name = "tensor_name"; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dmodel_pybind.cc198 …[](Model &model, const std::string &tensor_name) { return model.GetInputByTensorName(tensor_name);… in ModelPyBind()
200 …[](Model &model, const std::string &tensor_name) { return model.GetOutputByTensorName(tensor_name)… in ModelPyBind()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/
Dmodel.cc130 MSTensor Model::GetInputByTensorName(const std::vector<char> &tensor_name) { in GetInputByTensorName()
150 MSTensor Model::GetOutputByTensorName(const std::vector<char> &tensor_name) { in GetOutputByTensorName()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/benchmark/
Dbenchmark_c_api.cc295 int BenchmarkCApi::GetDataTypeByTensorName(const std::string &tensor_name) { in GetDataTypeByTensorName()
305 std::string tensor_name = calib_tensor.first; in CompareOutput() local
Dbenchmark_unified_api.cc226 auto tensor_name = ms_inputs_for_api_[i].Name(); in GenerateInputData() local
379 int BenchmarkUnifiedApi::GetDataTypeByTensorName(const std::string &tensor_name) { in GetDataTypeByTensorName()
573 std::string tensor_name = outputs->at(i).Name(); in CompareOutputForModelPool() local
655 std::string tensor_name = calib_tensor.first; in CompareOutput() local
731 std::string tensor_name = calib_tensor.first; in CompareOutputByCosineDistance() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dnetwork_test.cc61 const char *tensor_name, bool debug) { in runNet()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dtensor_util.h102 std::string tensor_name; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dadjust_print_for_ge.cc99 const auto tensor_name = "print"; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/c_api/
Dmodel_c.cc375 …sorHandle OH_AI_ModelGetInputByTensorName(const OH_AI_ModelHandle model, const char *tensor_name) { in OH_AI_ModelGetInputByTensorName()
394 …orHandle OH_AI_ModelGetOutputByTensorName(const OH_AI_ModelHandle model, const char *tensor_name) { in OH_AI_ModelGetOutputByTensorName()
662 std::vector<std::string> tensor_name; in OH_AI_ExportModel() local
683 std::vector<std::string> tensor_name; in OH_AI_ExportModelBuffer() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dgraph_kernel_splitter_with_py.cc73 std::string tensor_name = output_descs[0][kJsonKeyTensorName]; in IsRecomputeOp() local
106 std::string tensor_name = output_descs[0][kJsonKeyTensorName]; in SetStitchAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/
Ddebugger.cc1356 bool Debugger::DumpTensorToFile(const std::string &filepath, const std::string &tensor_name, size_t… in DumpTensorToFile() argument
1458 string tensor_name = node_name + ':' + "0"; in LoadSingleAnfnode() local
1493 string tensor_name = node_name + ':' + "0"; in LoadSingleParameterMindRT() local
1633 string tensor_name = kernel_name + ':' + std::to_string(j); in LoadGraphOutputs() local
1689 bool Debugger::TensorExistsInCurrent(const std::string &tensor_name) { in TensorExistsInCurrent()

123