Home
last modified time | relevance | path

Searched refs:srcTensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadUtils.hpp46 void CopyTensorContentsGeneric(const ITensorHandle* srcTensor, ITensorHandle* dstTensor, CopyFunc c… in CopyTensorContentsGeneric() argument
52 TensorShape srcStrides = srcTensor->GetStrides(); in CopyTensorContentsGeneric()
53 const TensorShape& srcShape = srcTensor->GetShape(); in CopyTensorContentsGeneric()
54 const auto srcSize = srcTensor->GetStrides()[0] * srcShape[0]; in CopyTensorContentsGeneric()
121 srcDataStart = static_cast<const uint8_t*>(srcTensor->Map()); in CopyTensorContentsGeneric()
184 srcTensor->Unmap(); in CopyTensorContentsGeneric()
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.hpp188 void CopyArmComputeITensorData(const arm_compute::ITensor& srcTensor, T* dstData) in CopyArmComputeITensorData() argument
193 const arm_compute::ITensorInfo& info = *srcTensor.info(); in CopyArmComputeITensorData()
195 const uint8_t* const bufferPtr = srcTensor.buffer(); in CopyArmComputeITensorData()
/aosp_15_r20/external/pytorch/test/cpp/rpc/
H A Dtest_tensorpipe_serialization.cpp78 tensorpipe::Message::Tensor& srcTensor = sendingTpMessage.tensors[i]; in TEST() local
82 srcTensor.buffer.unwrap<tensorpipe::CpuBuffer>().ptr, in TEST()
83 srcTensor.length); in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSProfiler.h160 const OptionalTensorRef srcTensor, const OptionalTensorRef dstTensor) { in getCopyKind()
161 const bool isSrcOnMPS = isStorageOnMPS(srcBuffer, srcTensor); in getCopyKind()
284 const OptionalTensorRef srcTensor,
H A DMPSProfiler.mm364 const OptionalTensorRef srcTensor,
375 copyInfo->srcStrKey = CopyInfo::buildTensorString(srcBuffer, srcTensor, includeBufferId);
377 copyInfo->kind = CopyInfo::getCopyKind(srcBuffer, dstBuffer, srcTensor, dstTensor);
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DScatterGather.mm177 MPSGraphTensor* srcTensor = mpsGraphRankedPlaceHolder(mpsGraph, src);
180 MPSGraphTensor* castSrcTensor = srcTensor;
184 castSrcTensor = [mpsGraph castTensor:srcTensor toType:src_type name:@"cast"];
285 newCachedGraph->srcTensor_ = srcTensor;