Searched refs:SyncDeviceToDevice (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | device_sync.h | 42 …virtual bool SyncDeviceToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *sr… in SyncDeviceToDevice() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.h | 50 bool SyncDeviceToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *src_ptr,
|
| D | ascend_device_address.cc | 408 bool AscendDeviceAddress::SyncDeviceToDevice(const ShapeVector &, size_t size, TypeId type, const v… in SyncDeviceToDevice() function in mindspore::device::ascend::AscendDeviceAddress
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_session.cc | 334 auto status = device_address->SyncDeviceToDevice(trans::GetRuntimePaddingShape(parameter, 0), in TensorNeedSync()
2040 … if (!dst_device_address->SyncDeviceToDevice(trans::GetRuntimePaddingShape(node, output_index), in UpdateOutputTensors()
|