| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 203 std::vector<size_t> host_shape; in SyncDeviceToHost() local
259 …ptr<LaunchKernel> AscendDeviceAddress::CreateLaunchTransData(const std::vector<size_t> &host_shape, in CreateLaunchTransData()
271 …iceAddress::SyncDeviceToHostAndConvertFormatBasedOnTransData(const std::vector<size_t> &host_shape, in SyncDeviceToHostAndConvertFormatBasedOnTransData()
313 std::vector<size_t> host_shape; in SyncDeviceToHostAndConvertFormat() local
370 std::vector<size_t> host_shape; in SyncHostToDevice() local
435 std::vector<size_t> host_shape; in ConvertFormatAndSyncHostToDevice() local
503 … const ShapeVector &host_shape, TypeId host_type, bool trans_flag) const { in DumpMemToFile() argument
541 … const ShapeVector &host_shape, TypeId host_type, size_t slot, in LoadMemToHost()
|
| D | ascend_launch_transdata.h | 29 std::vector<size_t> host_shape) in AscendLaunchTransData()
|
| /third_party/mindspore/tests/ut/cpp/common/ |
| D | trans_test.cc | 103 std::vector<size_t> host_shape = {43, 120}; in TEST_F() local
115 std::vector<size_t> host_shape = {120}; in TEST_F() local
127 std::vector<int64_t> host_shape = {-1, -1}; in TEST_F() local
139 std::vector<int64_t> host_shape = {-1}; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.h | 66 std::vector<size_t> host_shape; member
228 std::vector<T> host_shape; variable
|
| D | trans.cc | 891 std::vector<size_t> host_shape; in GetRuntimePaddingShape() local
1394 bool TransShapeToNz(const std::vector<size_t> &host_shape, std::vector<size_t> *hw_shape) { in TransShapeToNz()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_device_address.cc | 40 …ess::DumpMemToFile(const std::string &filepath, const std::string &, const ShapeVector &host_shape, in DumpMemToFile()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | device_address.h | 87 …pMemToFile(const std::string &filepath, const std::string &host_fmt, const ShapeVector &host_shape, in DumpMemToFile()
93 … const ShapeVector &host_shape, TypeId host_type, size_t slot, bool keep_prev) const { in LoadMemToHost()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_address.cc | 132 const ShapeVector &host_shape, TypeId host_type, size_t slot, in LoadMemToHost()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | tensor_load.h | 198 … const std::string &host_fmt, const std::vector<int64_t> &host_shape, TypeId host_type, in DumpTensorToFile()
|
| D | debug_services.cc | 1256 … const std::string &host_fmt, const std::vector<int64_t> &host_shape, in DumpTensorToFile()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 183 auto host_shape = host_tensor->shape_c(); in BindAndInitCacheTensor() local
247 std::vector<int64_t> host_shape{static_cast<int64_t>(hashmap_size), 4}; in InitHashMap() local
266 std::vector<int64_t> host_shape{1}; in InitStep() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | switch_actor.cc | 292 ShapeVector host_shape; in GetIndex() local
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 342 auto host_shape = tensor.shape_c(); in FlushFromCache() local
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger.cc | 1154 … const std::string &host_fmt, const std::vector<int64_t> &host_shape, TypeId host_type, in DumpTensorToFile()
|