Searched defs:ShapeVector (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_device_address.cc | 55 bool CPUDeviceAddress::SyncDeviceToHost(const ShapeVector &, size_t size, TypeId type, void *host_p… in SyncDeviceToHost() argument
101 bool CPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId type, const void *… in SyncHostToDevice()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | shape_utils.h | 21 using ShapeVector = std::vector<int64_t>; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_address.cc | 94 bool GPUDeviceAddress::SyncDeviceToHost(const ShapeVector &, size_t size, TypeId, void *host_ptr) c… in SyncDeviceToHost() argument
98 bool GPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId, const void *host_… in SyncHostToDevice()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 170 std::string ToString(const TypeId, const ShapeVector &, bool use_comma) const override { in ToString() argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 408 bool AscendDeviceAddress::SyncDeviceToDevice(const ShapeVector &, size_t size, TypeId type, const v… in SyncDeviceToDevice()
|