Home
last modified time | relevance | path

Searched refs:tensorData (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/armnn/include/armnnTestUtils/
H A DDataLayoutUtils.hpp14 void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNchwToNhwc() argument
20 std::vector<T> tmp(tensorData.size()); in PermuteTensorNchwToNhwc()
21 … armnnUtils::Permute(tensorInfo.GetShape(), nchwToNhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNchwToNhwc()
22 tensorData = tmp; in PermuteTensorNchwToNhwc()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw() argument
32 std::vector<T> tmp(tensorData.size()); in PermuteTensorNhwcToNchw()
33 … armnnUtils::Permute(tensorInfo.GetShape(), nhwcToNchw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNhwcToNchw()
35 tensorData = tmp; in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw() argument
45 std::vector<T> tmp(tensorData.size()); in PermuteTensorNdhwcToNcdhw()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/go/
H A Dtensor.go99 raw := tensorData(t.c)
191 raw := tensorData(t.c)
255 raw := tensorData(t.c)
373 return io.Copy(w, bytes.NewReader(tensorData(t.c)))
376 func tensorData(c *C.TF_Tensor) []byte { func
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dutils.cpp364 std::vector<at::Tensor> tensorData; in wireSerialize() local
378 tensorData = pickler.tensorData(); in wireSerialize()
380 for (const auto i : c10::irange(tensorData.size())) { in wireSerialize()
388 auto writeableTensorData = jit::getWriteableTensorData(tensorData[i]); in wireSerialize()
H A Dtensorpipe_utils.cpp180 const std::vector<torch::Tensor>& tensorDataVec = pickler.tensorData(); in tensorpipeSerialize()
/aosp_15_r20/external/executorch/backends/apple/mps/runtime/
H A DMPSExecutor.mm52 … MPSGraphTensorData* tensorData = [[[MPSGraphTensorData alloc]initWithMTLBuffer:_inputGPUBuffers[i]
55 _inputsArray[i] = tensorData;
59 …MPSGraphTensorData* tensorData = [[[MPSGraphTensorData alloc] initWithMTLBuffer:_outputGPUBuffers[…
62 _outputsArray[i] = tensorData;
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp119 TORCH_INTERNAL_ASSERT(tensor_names.size() == data_pickle.tensorData().size()); in write_archive_current()
123 for (const auto& td : data_pickle.tensorData()) { in write_archive_current()
346 for (const auto& td : data_pickle.tensorData()) { in backport_v5_to_v4()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DTensorTest.cpp172 std::vector<float> tensorData = { 1.0f }; variable
175 armnn::ConstTensor ct(tensorInfo, tensorData);
/aosp_15_r20/external/pytorch/caffe2/serialize/
H A Dinline_container_test.cc448 auto tensorData = std::vector<uint8_t>(tensorDataSizeInBytes, 1); in TEST_P() local
449 auto dataPtr = tensorData.data(); in TEST_P()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/
H A Dexport_data.cpp61 for (const auto& td : data_pickle.tensorData()) { in writeArchive()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpickle.cpp95 pickler.tensorData(), in pickle_save()
H A Dexport_module.cpp547 TORCH_INTERNAL_ASSERT(tensor_names.size() == data_pickle.tensorData().size()); in writeArchive()
549 for (const auto& td : data_pickle.tensorData()) { in writeArchive()
H A Dpickler.h155 const std::vector<at::Tensor>& tensorData() { in tensorData() function