Home
last modified time | relevance | path

Searched defs:DeviceTensor (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/debug/data_dump/
Ddump_utils.h28 using DeviceTensor = mindspore::device::DeviceAddress; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dde_tensor.h30 class DeviceTensor; variable
Dglobal_context.h38 class DeviceTensor; variable
Ddevice_tensor.cc26 DeviceTensor::DeviceTensor(const TensorShape &shape, const DataType &type) in DeviceTensor() function in mindspore::dataset::DeviceTensor
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Ddevice_tensor_copy_store.h28 using DeviceTensor = mindspore::device::DeviceAddress; variable
Ddevice_tensor_store.h29 using DeviceTensor = mindspore::device::DeviceAddress; variable
Dmemory_swap_node_scheduler.cc147 … std::map<DeviceTensor *, void *> *mem_allocated) { in GenMemOffloadStrategy()
308 std::map<DeviceTensor *, void *> *mem_allocated) const { in MockStrategy()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dabstract_actor.cc191 const std::vector<std::pair<OpDataUniquePtr<DeviceTensor>, size_t>> &output_data_list, in SendOutputData()
Dmemory_manager_actor.cc271 OpContext<DeviceTensor> *, const AID &from_aid) { in FreeMemory()
Dactor_common.h45 using DeviceTensor = mindspore::device::DeviceAddress; variable
Dkernel_actor.cc823 void KernelActor::PreLaunchKernel(OpContext<DeviceTensor> *) { in PreLaunchKernel()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline.cc136 using DeviceTensor = mindspore::device::DeviceAddress; typedef