Searched refs:tensorData (Results 1 – 12 of 12) sorted by relevance
/aosp_15_r20/external/armnn/include/armnnTestUtils/ |
H A D | DataLayoutUtils.hpp | 14 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 D | tensor.go | 99 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 D | utils.cpp | 364 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 D | tensorpipe_utils.cpp | 180 const std::vector<torch::Tensor>& tensorDataVec = pickler.tensorData(); in tensorpipeSerialize()
|
/aosp_15_r20/external/executorch/backends/apple/mps/runtime/ |
H A D | MPSExecutor.mm | 52 … 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 D | backport_manager.cpp | 119 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 D | TensorTest.cpp | 172 std::vector<float> tensorData = { 1.0f }; variable 175 armnn::ConstTensor ct(tensorInfo, tensorData);
|
/aosp_15_r20/external/pytorch/caffe2/serialize/ |
H A D | inline_container_test.cc | 448 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 D | export_data.cpp | 61 for (const auto& td : data_pickle.tensorData()) { in writeArchive()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
H A D | pickle.cpp | 95 pickler.tensorData(), in pickle_save()
|
H A D | export_module.cpp | 547 TORCH_INTERNAL_ASSERT(tensor_names.size() == data_pickle.tensorData().size()); in writeArchive() 549 for (const auto& td : data_pickle.tensorData()) { in writeArchive()
|
H A D | pickler.h | 155 const std::vector<at::Tensor>& tensorData() { in tensorData() function
|