Home
last modified time | relevance | path

Searched refs:GetTensor (Results 1 – 25 of 170) sorted by relevance

1234567

/external/armnn/src/backends/cl/
DClTensorHandle.hpp28 virtual arm_compute::ICLTensor& GetTensor() = 0;
29 virtual arm_compute::ICLTensor const& GetTensor() const = 0;
47 arm_compute::CLTensor& GetTensor() override { return m_Tensor; } in GetTensor() function in armnn::ClTensorHandle
48 arm_compute::CLTensor const& GetTensor() const override { return m_Tensor; } in GetTensor() function in armnn::ClTensorHandle
95 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
100 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
105 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
109 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
114 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
118 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
[all …]
/external/armnn/src/backends/neon/
DNeonTensorHandle.hpp48 arm_compute::ITensor& GetTensor() override { return m_Tensor; } in GetTensor() function in armnn::NeonTensorHandle
49 arm_compute::ITensor const& GetTensor() const override { return m_Tensor; } in GetTensor() function in armnn::NeonTensorHandle
181 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
186 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
190 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
194 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
198 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
203 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
207 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
224 this->GetTensor()); in CopyInFrom()
[all …]
/external/armnn/src/backends/backendsCommon/
DCpuTensorHandle.cpp50 void* CpuTensorHandle::GetTensor<void>() const in GetTensor() function in armnn::CpuTensorHandle
80 ::operator delete(GetTensor<void>()); in operator =()
88 ::operator delete(GetTensor<void>()); in ~ScopedCpuTensorHandle()
93 if (GetTensor<void>() == nullptr) in Allocate()
106 memcpy(memory, GetTensor<void>(), GetTensorInfo().GetNumBytes()); in CopyOutTo()
111 memcpy(GetTensor<void>(), memory, GetTensorInfo().GetNumBytes()); in CopyInFrom()
116 CopyFrom(other.GetTensor<void>(), other.GetTensorInfo().GetNumBytes()); in CopyFrom()
121 ARMNN_ASSERT(GetTensor<void>() == nullptr); in CopyFrom()
127 memcpy(GetTensor<void>(), srcMemory, numBytes); in CopyFrom()
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp113 … m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetTensor<void>()); in Execute()
115 m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetTensor<void>()); in Execute()
117 … m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetTensor<void>()); in Execute()
121 …urrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetTensor<void>()); in Execute()
123 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetTensor<void>()… in Execute()
125 …urrentToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetTensor<void>()); in Execute()
129 m_ForgetGateBiasTensor->GetTensorInfo(), m_ForgetGateBiasTensor->GetTensor<void>()); in Execute()
131 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetTensor<void>()); in Execute()
133 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetTensor<void>()); in Execute()
152 … m_InputLayerNormWeights->GetTensorInfo(), m_InputLayerNormWeights->GetTensor<void>()); in Execute()
[all …]
DRefQLstmWorkload.cpp104 … m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetTensor<void>()); in Execute()
106 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetTensor<void>()); in Execute()
108 … m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetTensor<void>()); in Execute()
111 …urrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetTensor<void>()); in Execute()
113 …_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetTensor<void>()); in Execute()
115 …urrentToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetTensor<void>()); in Execute()
201 … m_InputToInputWeightsTensor->GetTensorInfo(), m_InputToInputWeightsTensor->GetTensor<void>()); in Execute()
203 …ecurrentToInputWeightsTensor->GetTensorInfo(), m_RecurrentToInputWeightsTensor->GetTensor<void>()); in Execute()
211 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetTensor<void>()); in Execute()
214 … m_CellToForgetWeightsTensor->GetTensorInfo(), m_CellToForgetWeightsTensor->GetTensor<void>()); in Execute()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc77 return buffer_map.GetTensor(0); in MakeTensor()
107 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
111 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
125 EXPECT_THAT(GetTensorData<tensorflow::tstring>(buffer_map.GetTensor(0)), in TEST()
129 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
145 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
159 EXPECT_THAT(GetTensorData<tensorflow::tstring>(buffer_map.GetTensor(0)), in TEST()
170 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
174 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
188 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dcc_ops_test.cc53 test::GetTensor(root, m, &out); in TEST()
62 test::GetTensor(root, m, &out); in TEST()
72 test::GetTensor(root, c, &out); in TEST()
83 test::GetTensor(root, l, &out); in TEST()
114 test::GetTensor(root, y_normalized, &out); in TEST()
199 test::GetTensor(root, c1, &out); in TEST()
204 test::GetTensor(root, c2, &out); in TEST()
216 test::GetTensor(root, c1, &out); in TEST()
221 test::GetTensor(root, c2, &out); in TEST()
226 test::GetTensor(root, c3, &out); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/experimental/resource/
Dresource_variable_test.cc51 auto* value = var.GetTensor(); in TEST()
77 auto* value = var.GetTensor(); in TEST()
105 auto* value = var.GetTensor(); in TEST()
117 value = var.GetTensor(); in TEST()
146 auto* value = var.GetTensor(); in TEST()
158 value = var.GetTensor(); in TEST()
/external/armnn/src/backends/cl/workloads/
DClPreluWorkload.cpp35 arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClPreluWorkload()
36 arm_compute::ICLTensor& alpha = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClPreluWorkload()
37 … arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClPreluWorkload()
DClLogicalAndWorkload.cpp40 …compute::ICLTensor& input0 = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClLogicalAndWorkload()
41 …compute::ICLTensor& input1 = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClLogicalAndWorkload()
42 …ompute::ICLTensor& output = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClLogicalAndWorkload()
DClLogicalOrWorkload.cpp40 …compute::ICLTensor& input0 = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClLogicalOrWorkload()
41 …compute::ICLTensor& input1 = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClLogicalOrWorkload()
42 …ompute::ICLTensor& output = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClLogicalOrWorkload()
DClMinimumWorkload.cpp45 arm_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClMinimumWorkload()
46 arm_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClMinimumWorkload()
47 … arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClMinimumWorkload()
DClMaximumWorkload.cpp45 arm_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClMaximumWorkload()
46 arm_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClMaximumWorkload()
47 … arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClMaximumWorkload()
DClGatherWorkload.cpp35 … arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClGatherWorkload()
36 … arm_compute::ICLTensor& indices = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClGatherWorkload()
37 …arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClGatherWorkload()
DClComparisonWorkload.cpp47 arm_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClComparisonWorkload()
48 arm_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClComparisonWorkload()
49 … arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClComparisonWorkload()
DClDivisionFloatWorkload.cpp40 arm_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in ClDivisionFloatWorkload()
41 arm_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in ClDivisionFloatWorkload()
42 … arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in ClDivisionFloatWorkload()
DClAdditionWorkload.cpp27 …_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[0])->GetTensor(); in ClAdditionWorkload()
28 …_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[1])->GetTensor(); in ClAdditionWorkload()
29 …compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor(); in ClAdditionWorkload()
DClSubtractionWorkload.cpp27 …_compute::ICLTensor& input0 = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[0])->GetTensor(); in ClSubtractionWorkload()
28 …_compute::ICLTensor& input1 = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[1])->GetTensor(); in ClSubtractionWorkload()
29 …compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor(); in ClSubtractionWorkload()
/external/armnn/src/backends/neon/workloads/
DNeonLogicalAndWorkload.cpp38 …compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonLogicalAndWorkload()
39 …compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonLogicalAndWorkload()
40 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonLogicalAndWorkload()
DNeonMinimumWorkload.cpp34 …compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonMinimumWorkload()
35 …compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonMinimumWorkload()
36 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonMinimumWorkload()
DNeonMaximumWorkload.cpp33 …compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonMaximumWorkload()
34 …compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonMaximumWorkload()
35 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonMaximumWorkload()
DNeonLogicalOrWorkload.cpp38 …compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonLogicalOrWorkload()
39 …compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonLogicalOrWorkload()
40 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonLogicalOrWorkload()
DNeonPreluWorkload.cpp36 …_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonPreluWorkload()
37 …_compute::ITensor& alpha = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonPreluWorkload()
38 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonPreluWorkload()
DNeonGatherWorkload.cpp33 …ompute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonGatherWorkload()
34 …ompute::ITensor& indices = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonGatherWorkload()
35 …mpute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonGatherWorkload()
DNeonDivisionWorkload.cpp42 …compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); in NeonDivisionWorkload()
43 …compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); in NeonDivisionWorkload()
44 …ompute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); in NeonDivisionWorkload()

1234567