| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 47 struct Shape;
49 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape);
50 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape);
52 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape,
53 const _Float16* filterData, const Shape& filterShape,
54 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
58 _Float16* outputData, const Shape& outputShape);
59 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
60 const Shape& filterShape, const float* biasData, const Shape& biasShape,
65 const Shape& outputShape);
[all …]
|
| D | OperationsUtils.h | 47 struct Shape { struct
64 virtual Shape getInputShape(uint32_t index) const = 0; argument
69 virtual Shape getOutputShape(uint32_t index) const = 0;
79 virtual Shape getInputShape(uint32_t index) const = 0;
85 virtual Shape getOutputShape(uint32_t index) const = 0;
89 virtual bool setOutputShape(uint32_t index, const Shape& shape) = 0;
124 bool SameShape(const Shape& in1, const Shape& in2);
127 bool SetShape(const Shape& in, Shape* out);
131 uint32_t getNumberOfElements(const Shape& shape);
132 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive,
[all …]
|
| /packages/modules/NeuralNetworks/common/operations/ |
| D | Reshape.cpp | 34 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
35 const Shape& outputShape) { in copyData()
43 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
50 template bool depthToSpaceGeneric<float>(const float* inputData, const Shape& inputShape,
52 const Shape& outputShape);
53 template bool depthToSpaceGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
55 const Shape& outputShape);
56 template bool depthToSpaceGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
58 const Shape& outputShape);
[all …]
|
| D | Activation.cpp | 53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
63 template bool reluFloat<float>(const float* inputData, const Shape& inputShape, float* outputData,
64 const Shape& outputShape, float reluMin, float reluMax);
65 template bool reluFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
66 _Float16* outputData, const Shape& outputShape, float reluMin,
70 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
71 const Shape& outputShape) { in relu1Float()
74 template bool relu1Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
75 const Shape& outputShape);
76 template bool relu1Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
[all …]
|
| D | RNN.cpp | 45 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare()
46 Shape* outputShape) { in Prepare()
68 const Shape& inputShape = input->shape(); in Prepare()
116 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep()
117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep()
118 const T* recurrentWeightsData, const Shape& recurrentWeightsShape, in RNNStep()
122 Shape dummyShape; in RNNStep()
136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep()
137 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep()
138 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep()
[all …]
|
| D | GenerateProposals.cpp | 60 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape, in bboxTransformFloat32()
61 const float* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat32()
62 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat32()
63 const float* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat32()
64 float* outputData, const Shape& outputShape) { in bboxTransformFloat32()
110 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16()
111 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat16()
112 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16()
113 const _Float16* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat16()
114 _Float16* outputData, const Shape& outputShape) { in bboxTransformFloat16()
[all …]
|
| D | GroupedConv2D.cpp | 46 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in groupedConvFloat32()
47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32()
51 const Shape& outputShape) { in groupedConvFloat32()
104 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData, in groupedConvQuant8()
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
109 const Shape& outputShape) { in groupedConvQuant8()
181 template bool groupedConvQuant8<int8_t>(const int8_t* inputData, const Shape& inputShape,
182 const int8_t* filterData, const Shape& filterShape,
183 const int32_t* biasData, const Shape& biasShape,
188 const Shape& outputShape);
[all …]
|
| D | Broadcast.cpp | 77 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
78 int32_t activation, float* out, const Shape& shapeOut)>;
80 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16()
81 const Shape& shape2, int32_t activation, _Float16* out, in binaryOperationFloat16()
82 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16()
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
97 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
124 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16()
130 bool addQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in addQuant8()
[all …]
|
| D | SimpleMath.cpp | 33 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
35 const Shape& outputShape) { in meanFloat16()
48 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
77 template bool meanGeneric<float, float>(float* inputData, const Shape& inputShape,
78 const int32_t* axis, const Shape& axisShape, bool keepDims,
79 float* outputData, const Shape& outputShape);
80 template bool meanGeneric<uint8_t, int32_t>(uint8_t* inputData, const Shape& inputShape,
81 const int32_t* axis, const Shape& axisShape,
[all …]
|
| D | FullyConnected.cpp | 56 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
57 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
58 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32()
59 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
86 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
87 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
88 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16()
89 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
107 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
108 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
[all …]
|
| D | QLSTM.cpp | 179 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
186 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
191 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
197 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
203 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
207 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
213 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
219 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
223 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
237 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
|
| D | Pooling.cpp | 81 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
105 tflite::PoolParams toTfliteParam(const Shape& output) const { in toTfliteParam()
141 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
142 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
151 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
152 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
164 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
165 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
174 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
175 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
[all …]
|
| D | BidirectionalSequenceRNN.cpp | 65 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
80 Shape removeFirstDim(const Shape& input) { in removeFirstDim()
81 Shape output = input; in removeFirstDim()
123 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
126 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
128 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
133 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
135 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
154 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
155 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
[all …]
|
| D | RNN.h | 29 struct Shape;
36 Shape* hiddenStateShape, Shape* outputShape);
50 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData,
51 const T* biasData, const T* weightsData, const Shape& weightsShape,
52 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
56 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
57 const Shape& auxInputShape, const T* hiddenStateInputData,
58 const T* biasData, const T* weightsData, const Shape& weightsShape,
59 const T* auxWeightsData, const Shape& auxWeightsShape,
60 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
|
| D | Conv2D.cpp | 107 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
178 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
194 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
195 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
199 float* outputData, const Shape& outputShape) { in convNhwc()
223 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
224 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
228 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
277 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc()
[all …]
|
| D | Split.cpp | 29 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
31 const std::vector<Shape>& outputShapes) { in splitGeneric()
55 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
57 const std::vector<Shape>& outputShapes) { in splitFloat16()
62 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
64 const std::vector<Shape>& outputShapes) { in splitFloat32()
69 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
71 const std::vector<Shape>& outputShapes) { in splitQuant8()
76 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8Signed()
78 const std::vector<Shape>& outputShapes) { in splitQuant8Signed()
[all …]
|
| D | UnidirectionalSequenceLSTM.cpp | 200 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
208 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
213 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
219 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
225 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
229 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
235 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
241 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
245 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
259 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
|
| D | DepthwiseConv2D.cpp | 96 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
97 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
133 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
134 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
139 const Shape& outputShape) { in depthwiseConvNhwc()
167 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
168 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
169 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc()
173 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
193 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
[all …]
|
| D | LSTM.h | 47 struct Shape;
53 bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* scratchShape,
54 Shape* outputStateShape, Shape* cellStateShape, Shape* outputShape);
108 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
111 const Shape& input_to_output_weights_shape,
116 const Shape& recurrent_to_output_weights_shape,
133 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape,
138 const Shape& input_to_output_weights_shape,
143 const Shape& recurrent_to_output_weights_shape,
161 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
[all …]
|
| D | MaximumMinimum.cpp | 35 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric()
36 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
61 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8()
62 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
90 bool prepare(const Shape& in1, const Shape& in2, Shape* out) { in prepare()
95 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval()
96 bool isMinimum, void* output, const Shape& outputShape) { in eval()
|
| D | HeatmapMaxKeypoint.cpp | 89 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32Nhwc()
90 const float* boxes, const Shape& boxesShape, in heatmapMaxKeypointFloat32Nhwc()
91 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32Nhwc()
93 const Shape& outputKeypointShape, float fpAtol, in heatmapMaxKeypointFloat32Nhwc()
168 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32()
169 const float* boxes, const Shape& boxesShape, bool layout, in heatmapMaxKeypointFloat32()
170 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32()
171 float* outputKeypointData, const Shape& outputKeypointShape, in heatmapMaxKeypointFloat32()
174 Shape heatmapShape_nhwc; in heatmapMaxKeypointFloat32()
179 const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape; in heatmapMaxKeypointFloat32()
[all …]
|
| D | TransposeConv2D.cpp | 75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
126 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc()
129 const Shape& outputShape) { in transposeConvNhwc()
183 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc()
184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
185 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
277 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
278 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
279 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc()
[all …]
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 48 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl()
114 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape()
126 bool SetShape(const Shape& in, Shape* out) { in SetShape()
134 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
142 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive, in getNumberOfElements()
154 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
158 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
163 uint32_t hasKnownRank(const Shape& shape) { in hasKnownRank()
256 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
257 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler()
[all …]
|
| /packages/apps/Car/Cluster/DirectRenderingCluster/src/android/car/cluster/ |
| D | LaneView.java | 153 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getShift()
154 || laneDir.getShape().equals(LaneDirection.Shape.SLIGHT_RIGHT) in getShift()
155 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_RIGHT) in getShift()
156 || laneDir.getShape().equals(LaneDirection.Shape.U_TURN_RIGHT)) { in getShift()
159 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getShift()
160 || laneDir.getShape().equals(LaneDirection.Shape.SLIGHT_LEFT) in getShift()
161 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_LEFT) in getShift()
162 || laneDir.getShape().equals(LaneDirection.Shape.U_TURN_LEFT)) { in getShift()
165 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getShift()
189 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getOffset()
[all …]
|
| /packages/apps/Car/Cluster/ClusterOsDouble/src/com/android/car/cluster/view/ |
| D | LaneView.java | 154 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getShift()
155 || laneDir.getShape().equals(LaneDirection.Shape.SLIGHT_RIGHT) in getShift()
156 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_RIGHT) in getShift()
157 || laneDir.getShape().equals(LaneDirection.Shape.U_TURN_RIGHT)) { in getShift()
160 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getShift()
161 || laneDir.getShape().equals(LaneDirection.Shape.SLIGHT_LEFT) in getShift()
162 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_LEFT) in getShift()
163 || laneDir.getShape().equals(LaneDirection.Shape.U_TURN_LEFT)) { in getShift()
166 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getShift()
190 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getOffset()
[all …]
|