Searched refs:origin_size (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_bucket.cc | 61 auto origin_size = device_address->GetSize(); in AllocateAllReduceAddr() local
62 auto align_size = AlignMemorySize(origin_size); in AllocateAllReduceAddr()
63 size_list.emplace_back(origin_size); in AllocateAllReduceAddr()
67 …ake_shared<kernel::Address>(static_cast<uint8_t *>(device_address->GetMutablePtr()), origin_size)); in AllocateAllReduceAddr()
|
| D | gpu_launch_mul.cc | 53 void GPULaunchMul::CopyHostMemToDevice(size_t origin_size, size_t) { in CopyHostMemToDevice() argument
54 …ager::GetInstance().CopyHostMemToDeviceAsync(input2_addr_, &input2_value_, origin_size, stream_)) { in CopyHostMemToDevice()
|
| D | gpu_launch_mul.h | 40 void CopyHostMemToDevice(size_t origin_size, size_t) override;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_mul.cc | 54 void AscendLaunchMul::CopyHostMemToDevice(size_t origin_size, size_t dst_size) { in CopyHostMemToDevice() argument
55 …auto ret = rtMemcpyAsync(input2_addr_, dst_size, &input2_value_, origin_size, RT_MEMCPY_HOST_TO_DE… in CopyHostMemToDevice()
|
| D | ascend_bucket.cc | 46 auto origin_size = device_address->GetSize(); in AllocateAllReduceAddr() local
47 auto align_size = MemoryManager::GetCommonAlignSize(origin_size); in AllocateAllReduceAddr()
48 origin_size_list.emplace_back(origin_size); in AllocateAllReduceAddr()
|
| D | ascend_launch_mul.h | 41 void CopyHostMemToDevice(size_t origin_size, size_t dst_size) override;
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensorlist_setitem.cc | 55 int TensorListSetItemCPUKernel::IncrementOutputSize(int origin_size) { in IncrementOutputSize() argument
56 int new_tensors_size = origin_size + 1; in IncrementOutputSize()
|
| D | tensorlist_setitem.h | 37 int IncrementOutputSize(int origin_size);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.h | 41 virtual void CopyHostMemToDevice(size_t origin_size, size_t dst_size) = 0;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_fusion.h | 99 size_t origin_size, const AnfNodePtrList &candidates, std::map<AnfNodePtr, int> *origin_indices,
|
| D | parallel_fusion.cc | 474 size_t origin_size, const AnfNodePtrList &candidates, std::map<AnfNodePtr, int> *origin_indices, in DoSearchInSortedCandidates() argument
485 std::vector<bool> origin_candidates_used(origin_size, false); in DoSearchInSortedCandidates()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 324 auto origin_size = GetSliceBeginAndSize(slice_cnode, SliceSizeIndex); in SliceParamDeBroadcast() local
335 …MS_CHECK_TRUE_MSG(origin_size.size() >= origin_axes.size(), RET_ERROR, "origin_size.size() is wron… in SliceParamDeBroadcast()
339 int s = origin_size[i]; in SliceParamDeBroadcast()
|