Searched defs:origin_size (Results 1 – 7 of 7) sorted by relevance
53 void GPULaunchMul::CopyHostMemToDevice(size_t origin_size, size_t) { in CopyHostMemToDevice() argument
61 auto origin_size = device_address->GetSize(); in AllocateAllReduceAddr() local
54 void AscendLaunchMul::CopyHostMemToDevice(size_t origin_size, size_t dst_size) { in CopyHostMemToDevice()
46 auto origin_size = device_address->GetSize(); in AllocateAllReduceAddr() local
55 int TensorListSetItemCPUKernel::IncrementOutputSize(int origin_size) { in IncrementOutputSize()
474 size_t origin_size, const AnfNodePtrList &candidates, std::map<AnfNodePtr, int> *origin_indices, in DoSearchInSortedCandidates()
324 auto origin_size = GetSliceBeginAndSize(slice_cnode, SliceSizeIndex); in SliceParamDeBroadcast() local