Home
last modified time | relevance | path

Searched refs:Shape (Results 1 – 25 of 112) sorted by relevance

12345

/aosp12/packages/modules/NeuralNetworks/common/include/
H A DOperations.h47 struct Shape;
60 const Shape& filterShape, const float* biasData, const Shape& biasShape,
65 const Shape& outputShape);
91 const Shape& outputShape);
111 const Shape& outputShape);
115 const Shape& outputShape);
117 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
156 const Shape& filterShape, const float* biasData, const Shape& biasShape,
160 const Shape& outputShape);
164 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
[all …]
H A DOperationsUtils.h47 struct Shape { struct
124 bool SameShape(const Shape& in1, const Shape& in2);
127 bool SetShape(const Shape& in, Shape* out);
273 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
277 T requantize(T value, const Shape& oldShape, const Shape& newShape);
280 bool floorPrepare(const Shape& input, Shape* output);
282 bool depthwiseConvPrepare(const Shape& input, const Shape& filter, const Shape& bias,
288 bool genericActivationPrepare(const Shape& input, Shape* output);
297 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
300 const Shape& valueShape, Shape* outputShape, Shape* hitShape);
[all …]
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DReshape.cpp35 const Shape& outputShape) { in copyData()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
52 const Shape& outputShape);
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
73 const Shape& outputShape);
86 T* outputData, const Shape& outputShape) { in padGeneric()
186 const Shape& outputShape);
189 const Shape& outputShape);
192 const Shape& outputShape);
195 const Shape& outputShape);
[all …]
H A DSimpleMath.cpp33 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,
79 float* outputData, const Shape& outputShape);
83 const Shape& outputShape);
84 template bool meanGeneric<int8_t, int32_t>(int8_t* inputData, const Shape& inputShape,
85 const int32_t* axis, const Shape& axisShape,
[all …]
H A DRNN.cpp46 Shape* outputShape) { in Prepare()
68 const Shape& inputShape = input->shape(); in Prepare()
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()
139 const Shape& auxWeightsShape, const T* recurrentWeightsData, in RNNStep()
140 const Shape& recurrentWeightsShape, const int32_t activation, in RNNStep()
239 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
241 const float* weightsData, const Shape& weightsShape,
[all …]
H A DActivation.cpp53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
71 const Shape& outputShape) { in relu1Float()
81 const Shape& outputShape) { in relu6Float()
90 const Shape& outputShape) { in tanhFloat16()
100 const Shape& outputShape) { in tanhFloat32()
111 const Shape& outputShape) { in logisticFloat()
142 const Shape& outputShape) { in reluQuant8()
148 const Shape& outputShape) { in relu1Quant8()
154 const Shape& outputShape) { in relu6Quant8()
160 const Shape& outputShape) { in tanhQuant8()
[all …]
H A DGroupedConv2D.cpp47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32()
51 const Shape& outputShape) { in groupedConvFloat32()
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
109 const Shape& outputShape) { in groupedConvQuant8()
188 const Shape& outputShape);
197 const Shape& outputShape);
218 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel()
220 Shape biasChannelShape = biasShape; in groupedConvQuant8PerChannel()
285 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape, in groupedConvFloat16()
316 int32_t activation, uint8_t* outputData, const Shape& outputShape);
[all …]
H A DRNN.h29 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,
H A DFullyConnected.cpp56 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
59 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
185 bool validateShapes(const Shape& input, const Shape& weights, const Shape& bias, in validateShapes()
186 Shape* output = nullptr) { in validateShapes()
284 Shape input = context->getInputShape(kInputTensor); in validate()
285 Shape weights = context->getInputShape(kWeightsTensor); in validate()
286 Shape bias = context->getInputShape(kBiasTensor); in validate()
296 Shape input = context->getInputShape(kInputTensor); in prepare()
297 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
298 Shape bias = context->getInputShape(kBiasTensor); in prepare()
[all …]
H A DSplit.cpp29 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()
78 const std::vector<Shape>& outputShapes) { in splitQuant8Signed()
83 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
[all …]
H A DBroadcast.cpp77 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
130 bool addQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in addQuant8()
235 bool mulFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in mulFloat32()
262 bool mulFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in mulFloat16()
269 bool mulQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in mulQuant8()
316 bool subFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in subFloat32()
333 bool subFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in subFloat16()
340 bool subQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in subQuant8()
[all …]
H A DGenerateProposals.cpp553 T castTo(float val, const Shape&) { in castTo() argument
557 uint8_t castTo(float val, const Shape& shape) { in castTo()
562 int8_t castTo(float val, const Shape& shape) { in castTo()
625 Shape scoresOutShape, float* roiOutData, Shape roiOutShape, in boxWithNmsLimitFloat32()
1003 Shape tempRoiShape = anchorsShape; in generateProposalsNhwcFloat32Compute()
1005 Shape tempBBoxDeltasShape = bboxDeltasShape; in generateProposalsNhwcFloat32Compute()
1009 Shape tempImageInfoShape = imageInfoShape; in generateProposalsNhwcFloat32Compute()
1453 const Shape& deltaShape, const float* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat32()
1458 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat32()
1558 const Shape& deltaShape, const _Float16* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat16()
[all …]
H A DMaximumMinimum.cpp35 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()
H A DPow.cpp34 bool evalGeneric(const T* baseData, const Shape& baseShape, const T* exponentData, in evalGeneric()
35 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
61 bool prepare(const Shape& baseShape, const Shape& exponentShape, Shape* output) { in prepare()
69 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData, in eval()
70 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
H A DPooling.cpp105 tflite::PoolParams toTfliteParam(const Shape& output) const { in toTfliteParam()
188 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
198 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
210 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
220 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
230 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
256 T* outputData, const Shape& outputShape) { in averagePool()
269 const Shape& outputShape) { in l2Pool()
282 const Shape& outputShape) { in maxPool()
362 Shape input = context->getInputShape(kInputTensor); in prepare()
[all …]
H A DBidirectionalSequenceRNN.cpp80 Shape removeFirstDim(const Shape& input) { in removeFirstDim()
81 Shape output = input; in removeFirstDim()
123 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
165 Shape bwOutputShape; in executeTyped()
227 Shape fixedTimeInputShape = removeFirstDim(inputShape); in executeTyped()
228 Shape fixedTimeAuxInputShape = auxInputShape; in executeTyped()
367 Shape input = context->getInputShape(kInputTensor); in prepare()
370 Shape fwBias = context->getInputShape(kFwBiasTensor); in prepare()
374 Shape bwBias = context->getInputShape(kBwBiasTensor); in prepare()
377 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
[all …]
H A DQLSTM.cpp179 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
298 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
302 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
353 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
354 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
358 Shape cellStateOutShape = context->getOutputShape(kCellStateOutTensor); in prepare()
369 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
371 const Shape recurrentToInputWeightsShape = in execute()
376 const Shape recurrentToForgetWeightsShape = in execute()
384 const Shape recurrentToOutputWeightsShape = in execute()
[all …]
H A DLSTM.h47 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,
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,
164 const Shape& input_to_output_weights_shape,
[all …]
H A DPow.h26 bool prepare(const Shape& in1, const Shape& in2, Shape* output);
28 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData,
29 const Shape& exponentShape, void* outputData, const Shape& outputShape);
H A DMaximumMinimum.h26 bool prepare(const Shape& in1, const Shape& in2, Shape* output);
28 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2,
29 bool isMinimum, void* output, const Shape& outputShape);
H A DUnidirectionalSequenceLSTM.cpp200 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
229 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
289 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
297 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
300 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
303 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
320 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()
324 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()
377 Shape outputShape = context->getInputShape(kInputTensor); in prepare()
384 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()
[all …]
H A DHeatmapMaxKeypoint.cpp90 const float* boxes, const Shape& boxesShape, in heatmapMaxKeypointFloat32Nhwc()
93 const Shape& outputKeypointShape, float fpAtol, in heatmapMaxKeypointFloat32Nhwc()
168 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32()
174 Shape heatmapShape_nhwc; in heatmapMaxKeypointFloat32()
179 const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape; in heatmapMaxKeypointFloat32()
185 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant()
207 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant()
263 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare()
264 Shape boxesShape = context->getInputShape(kBoxesTensor); in prepare()
283 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()
[all …]
/aosp12/packages/modules/NeuralNetworks/common/
H A DOperationsUtils.cpp114 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape()
126 bool SetShape(const Shape& in, Shape* out) { in SetShape()
339 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape()
420 bool depthToSpacePrepare(const Shape& input, int32_t blockSize, Shape* output) { in depthToSpacePrepare()
460 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare()
478 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare()
479 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare()
528 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare()
555 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare()
669 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output) { in argMinMaxPrepare()
[all …]
/aosp12/packages/apps/Car/Cluster/ClusterOsDouble/src/com/android/car/cluster/view/
H A DLaneView.java154 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getShift()
156 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_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()
166 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getShift()
190 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getOffset()
192 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_LEFT) in getOffset()
196 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getOffset()
203 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getOffset()
[all …]
/aosp12/packages/apps/Car/Cluster/DirectRenderingCluster/src/android/car/cluster/
H A DLaneView.java153 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getShift()
155 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_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()
165 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getShift()
189 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_LEFT) in getOffset()
191 || laneDir.getShape().equals(LaneDirection.Shape.SHARP_LEFT) in getOffset()
195 if (laneDir.getShape().equals(LaneDirection.Shape.NORMAL_RIGHT) in getOffset()
202 if (laneDir.getShape().equals(LaneDirection.Shape.STRAIGHT)) { in getOffset()
[all …]

12345