Home
last modified time | relevance | path

Searched defs:tensorData (Results 1 – 6 of 6) 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()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw()
51 void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNcdhwToNdhwc()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DTensorTest.cpp172 std::vector<float> tensorData = { 1.0f }; variable
/aosp_15_r20/external/tensorflow/tensorflow/go/
H A Dtensor.go376 func tensorData(c *C.TF_Tensor) []byte { func
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpickler.h155 const std::vector<at::Tensor>& tensorData() { in tensorData() function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dutils.cpp364 std::vector<at::Tensor> tensorData; in wireSerialize() local
/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