| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_lstmparam.i | 76 const armnn::ConstTensor* m_InputToInputWeights;
77 const armnn::ConstTensor* m_InputToForgetWeights;
78 const armnn::ConstTensor* m_InputToCellWeights;
79 const armnn::ConstTensor* m_InputToOutputWeights;
80 const armnn::ConstTensor* m_RecurrentToInputWeights;
81 const armnn::ConstTensor* m_RecurrentToForgetWeights;
82 const armnn::ConstTensor* m_RecurrentToCellWeights;
83 const armnn::ConstTensor* m_RecurrentToOutputWeights;
84 const armnn::ConstTensor* m_CellToInputWeights;
85 const armnn::ConstTensor* m_CellToForgetWeights;
[all …]
|
| /external/armnn/include/armnn/ |
| D | QuantizedLstmParams.hpp | 33 const ConstTensor* m_InputToInputWeights;
34 const ConstTensor* m_InputToForgetWeights;
35 const ConstTensor* m_InputToCellWeights;
36 const ConstTensor* m_InputToOutputWeights;
38 const ConstTensor* m_RecurrentToInputWeights;
39 const ConstTensor* m_RecurrentToForgetWeights;
40 const ConstTensor* m_RecurrentToCellWeights;
41 const ConstTensor* m_RecurrentToOutputWeights;
43 const ConstTensor* m_InputGateBias;
44 const ConstTensor* m_ForgetGateBias;
[all …]
|
| D | LstmParams.hpp | 40 const ConstTensor* m_InputToInputWeights;
41 const ConstTensor* m_InputToForgetWeights;
42 const ConstTensor* m_InputToCellWeights;
43 const ConstTensor* m_InputToOutputWeights;
44 const ConstTensor* m_RecurrentToInputWeights;
45 const ConstTensor* m_RecurrentToForgetWeights;
46 const ConstTensor* m_RecurrentToCellWeights;
47 const ConstTensor* m_RecurrentToOutputWeights;
48 const ConstTensor* m_CellToInputWeights;
49 const ConstTensor* m_CellToForgetWeights;
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | ConstTensorLayerVisitor.cpp | 126 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
155 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
185 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
189 …ConstTensor biases(TensorInfo(4, biasDimensions.data(), DataType::Float32, 0.0f, 0, true), biasDat…
223 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
227 …ConstTensor biases(TensorInfo(4, biasDimensions.data(), DataType::Float32, 0.0f, 0, true), biasDat…
259 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
288 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
317 ConstTensor weights(TensorInfo(4, dimensions.data(), DataType::Float32, 0.0f, 0, true), data);
321 …ConstTensor biases(TensorInfo(4, biasDimensions.data(), DataType::Float32, 0.0f, 0, true), biasDat…
[all …]
|
| D | TestLayerVisitor.hpp | 23 void CheckConstTensors(const ConstTensor& expected,
24 const ConstTensor& actual);
25 void CheckConstTensors(const ConstTensor& expected,
29 const ConstTensor* expected,
30 const ConstTensor* actual);
32 const ConstTensor* expected,
35 …void CheckOptionalConstTensors(const Optional<ConstTensor>& expected, const Optional<ConstTensor>&…
|
| D | ConstTensorLayerVisitor.hpp | 33 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
74 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
115 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
146 const ConstTensor& mean, in TestBatchNormalizationLayerVisitor()
147 const ConstTensor& variance, in TestBatchNormalizationLayerVisitor()
148 const ConstTensor& beta, in TestBatchNormalizationLayerVisitor()
149 const ConstTensor& gamma, in TestBatchNormalizationLayerVisitor()
161 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
191 ConstTensor m_Mean;
192 ConstTensor m_Variance;
[all …]
|
| D | TestLayerVisitor.cpp | 34 void TestLayerVisitor::CheckConstTensors(const ConstTensor& expected, const ConstTensor& actual) in CheckConstTensors()
52 void TestLayerVisitor::CheckConstTensors(const ConstTensor& expected, const ConstTensorHandle& actu… in CheckConstTensors()
73 const ConstTensor* expected, in CheckConstTensorPtrs()
91 const ConstTensor* expected, in CheckConstTensorPtrs()
92 const ConstTensor* actual) in CheckConstTensorPtrs()
108 void TestLayerVisitor::CheckOptionalConstTensors(const Optional<ConstTensor>& expected, in CheckOptionalConstTensors()
109 const Optional<ConstTensor>& actual) in CheckOptionalConstTensors()
|
| /external/armnn/src/armnnSerializer/ |
| D | ArmnnSchema.fbs | 101 table ConstTensor {
269 input:ConstTensor;
275 weights:ConstTensor;
276 biases:ConstTensor;
388 weights:ConstTensor; // ConstTensors are now passed as inputs.
389 biases:ConstTensor;
555 weights:ConstTensor;
556 biases:ConstTensor;
694 mean:ConstTensor;
695 variance:ConstTensor;
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.h | 31 typename TTypes<T, 4>::ConstTensor image,
32 typename TTypes<float, 2>::ConstTensor boxes,
33 typename TTypes<int32, 1>::ConstTensor box_ind,
42 typename TTypes<float, 4>::ConstTensor grads,
43 typename TTypes<float, 2>::ConstTensor boxes,
44 typename TTypes<int32, 1>::ConstTensor box_ind,
52 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
53 typename TTypes<T, 4>::ConstTensor image,
54 typename TTypes<float, 2>::ConstTensor boxes,
55 typename TTypes<int32, 1>::ConstTensor box_ind,
[all …]
|
| D | scale_and_translate_op.h | 62 typename TTypes<int32, 1>::ConstTensor row_starts,
63 typename TTypes<float, 1>::ConstTensor row_weights,
65 typename TTypes<int32, 1>::ConstTensor col_starts,
66 typename TTypes<float, 1>::ConstTensor col_weights,
67 typename TTypes<T, 4>::ConstTensor input_images,
|
| /external/armnn/src/armnnSerializer/test/ |
| D | LstmSerializationTests.cpp | 24 armnn::LstmInputParams ConstantVector2LstmInputParams(const std::vector<armnn::ConstTensor>& consta… in ConstantVector2LstmInputParams()
92 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
195 armnn::ConstTensor inputToForgetWeights(inputWeightsInfo1, inputToForgetWeightsData);
198 armnn::ConstTensor inputToCellWeights(inputWeightsInfo1, inputToCellWeightsData);
201 armnn::ConstTensor inputToOutputWeights(inputWeightsInfo1, inputToOutputWeightsData);
205 armnn::ConstTensor recurrentToForgetWeights(inputWeightsInfo2, recurrentToForgetWeightsData);
208 armnn::ConstTensor recurrentToCellWeights(inputWeightsInfo2, recurrentToCellWeightsData);
211 armnn::ConstTensor recurrentToOutputWeights(inputWeightsInfo2, recurrentToOutputWeightsData);
215 armnn::ConstTensor cellToForgetWeights(inputWeightsInfo3, cellToForgetWeightsData);
218 armnn::ConstTensor cellToOutputWeights(inputWeightsInfo3, cellToOutputWeightsData);
[all …]
|
| D | SerializerTestUtils.hpp | 22 void CompareConstTensor(const armnn::ConstTensor& tensor1, const armnn::ConstTensor& tensor2);
33 const std::vector<armnn::ConstTensor>& constants,
41 const armnn::ConstTensor* expectedPtr,
42 const armnn::ConstTensor* actualPtr);
63 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
115 const std::vector<armnn::ConstTensor>& constants) in LayerVerifierBaseWithDescriptorAndConstants()
121 const std::vector<armnn::ConstTensor>& constants, in ExecuteStrategy()
147 std::vector<armnn::ConstTensor> m_Constants;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 33 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
49 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
50 typename TTypes<T>::ConstTensor features, in operator()
67 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
83 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
84 typename TTypes<T>::ConstTensor features, in operator()
109 typename TTypes<T>::ConstTensor features;
130 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
131 typename TTypes<T>::ConstTensor features, T alpha, in operator()
145 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
[all …]
|
| D | dilation_ops.h | 29 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
30 typename TTypes<T, 3>::ConstTensor filter, int stride_rows,
41 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
42 typename TTypes<T, 3>::ConstTensor filter,
43 typename TTypes<T, 4>::ConstTensor out_backprop,
55 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
56 typename TTypes<T, 3>::ConstTensor filter,
57 typename TTypes<T, 4>::ConstTensor out_backprop,
|
| D | relu_op.cc | 93 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
99 typename TTypes<T>::ConstTensor features, \
105 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
132 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
133 typename TTypes<T>::ConstTensor features, \
139 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
145 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
146 typename TTypes<T>::ConstTensor features, \
156 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
157 typename TTypes<T>::ConstTensor features, T alpha, \
[all …]
|
| D | betainc_op.h | 29 void operator()(const Device& d, typename TTypes<T, NDIM>::ConstTensor a, in operator()
30 typename TTypes<T, NDIM>::ConstTensor b, in operator()
31 typename TTypes<T, NDIM>::ConstTensor x, in operator()
36 void BCast(const Device& d, typename TTypes<T, NDIM>::ConstTensor a, in BCast()
38 typename TTypes<T, NDIM>::ConstTensor b, in BCast()
40 typename TTypes<T, NDIM>::ConstTensor x, in BCast()
|
| D | conv_2d.h | 63 typename TTypes<T, 4>::ConstTensor input, in operator()
64 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
82 typename TTypes<T, 4>::ConstTensor input, in operator()
83 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
111 typename TTypes<Eigen::half, 4>::ConstTensor input,
112 typename TTypes<Eigen::half, 4>::ConstTensor filter,
150 typename TTypes<Eigen::half, 4>::ConstTensor input,
151 typename TTypes<Eigen::half, 4>::ConstTensor filter,
200 typename TTypes<T, 4>::ConstTensor filter,
201 typename TTypes<T, 4>::ConstTensor output_backward,
[all …]
|
| D | conv_3d.h | 45 typename TTypes<T, 5>::ConstTensor input,
46 typename TTypes<T, 5>::ConstTensor filter, int stride_planes,
58 typename TTypes<T, 5>::ConstTensor filter,
59 typename TTypes<T, 5>::ConstTensor output_backward,
75 typename TTypes<T, 5>::ConstTensor input,
76 typename TTypes<T, 5>::ConstTensor output_backward,
|
| /external/armnn/src/armnn/layers/ |
| D | QLstmLayer.cpp | 309 std::vector<ConstTensor> constTensors; in ExecuteStrategy()
343 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
348 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
353 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
358 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
364 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
370 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
376 constTensors.emplace_back(ConstTensor(managedForgetGateBias.GetTensorInfo(), in ExecuteStrategy()
381 constTensors.emplace_back(ConstTensor(managedCellBias.GetTensorInfo(), in ExecuteStrategy()
386 constTensors.emplace_back(ConstTensor(managedOutputGateBias.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | LstmLayer.cpp | 306 std::vector<ConstTensor> constTensors; in ExecuteStrategy()
343 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
348 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
353 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
358 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
364 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
370 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
376 constTensors.emplace_back(ConstTensor(managedForgetGateBias.GetTensorInfo(), in ExecuteStrategy()
381 constTensors.emplace_back(ConstTensor(managedCellBias.GetTensorInfo(), in ExecuteStrategy()
386 constTensors.emplace_back(ConstTensor(managedOutputGateBias.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | UnidirectionalSequenceLstmLayer.cpp | 313 std::vector<ConstTensor> constTensors; in ExecuteStrategy()
350 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
355 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
360 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
365 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
371 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
377 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
383 constTensors.emplace_back(ConstTensor(managedForgetGateBias.GetTensorInfo(), in ExecuteStrategy()
388 constTensors.emplace_back(ConstTensor(managedCellBias.GetTensorInfo(), in ExecuteStrategy()
393 constTensors.emplace_back(ConstTensor(managedOutputGateBias.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | QuantizedLstmLayer.cpp | 175 std::vector<ConstTensor> constTensors; in ExecuteStrategy()
195 constTensors.emplace_back(ConstTensor(managedInputToInputWeights.GetTensorInfo(), in ExecuteStrategy()
201 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
207 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
213 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
220 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
227 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
234 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
241 constTensors.emplace_back(ConstTensor( in ExecuteStrategy()
249 constTensors.emplace_back(ConstTensor(managedInputGateBias.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | QuantizedLstmEndToEndTestImpl.cpp | 63 … armnn::ConstTensor inputToInputWeightsTensor(inputWeightsInfo, inputToInputWeightsVector.data()); in CreateQuantizedLstmNetwork()
66 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in CreateQuantizedLstmNetwork()
69 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in CreateQuantizedLstmNetwork()
72 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in CreateQuantizedLstmNetwork()
76 …armnn::ConstTensor recurrentToInputWeightsTensor(recurrentWeightsInfo, recurrentToInputWeightsTens… in CreateQuantizedLstmNetwork()
80 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork()
85 …armnn::ConstTensor recurrentToCellWeightsTensor(recurrentWeightsInfo, recurrentToCellWeightsTensor… in CreateQuantizedLstmNetwork()
89 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork()
93 armnn::ConstTensor inputGateBiasTensor(biasInfo, inputGateBiasTensorVector.data()); in CreateQuantizedLstmNetwork()
96 armnn::ConstTensor forgetGateBiasTensor(biasInfo, forgetGateBiasTensorVector.data()); in CreateQuantizedLstmNetwork()
[all …]
|
| D | QLstmEndToEndTestImpl.cpp | 112 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in QLstmEndToEnd()
113 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in QLstmEndToEnd()
114 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in QLstmEndToEnd()
123 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd()
125 armnn::ConstTensor recurrentToCellWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd()
127 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd()
134 armnn::ConstTensor forgetGateBiasTensor(biasInfo, forgetGateBiasTensorVector.data()); in QLstmEndToEnd()
135 armnn::ConstTensor cellBiasTensor(biasInfo, cellBiasTensorVector.data()); in QLstmEndToEnd()
136 armnn::ConstTensor outputGateBiasTensor(biasInfo, outputGateBiasTensorVector.data()); in QLstmEndToEnd()
143 …armnn::ConstTensor forgetLayerNormWeights(layerNormWeightsInfo, forgetLayerNormWeightsVector.data(… in QLstmEndToEnd()
[all …]
|
| /external/armnn/python/pyarmnn/test/ |
| D | test_const_tensor.py | 30 ann.ConstTensor(tensor_info, data)
50 ann.ConstTensor(tensor_info, data)
66 tensor = ann.ConstTensor(ann.TensorInfo(ann.TensorShape((2, 2, 3, 3)), dt, 0.0, 0, True), data)
77 copied_tensor = ann.ConstTensor(tensor)
97 copied_tensor = ann.ConstTensor(tensor)
116 ann.ConstTensor('something', 'something')
131 ann.ConstTensor(tensor_info, data)
148 ann.ConstTensor(tensor_info, data)
167 tensor = ann.ConstTensor(tensor_info, data)
168 copied_tensor = ann.ConstTensor(tensor)
[all …]
|