Home
last modified time | relevance | path

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

1234

/frameworks/ml/nn/common/include/
DOperations.h44 struct Shape;
46 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape);
47 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape);
49 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape,
50 const _Float16* filterData, const Shape& filterShape,
51 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
55 _Float16* outputData, const Shape& outputShape);
56 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
57 const Shape& filterShape, const float* biasData, const Shape& biasShape,
62 const Shape& outputShape);
[all …]
DOperationsUtils.h45 struct Shape { struct
76 virtual Shape getInputShape(uint32_t index) const = 0;
81 virtual Shape getOutputShape(uint32_t index) const = 0;
91 virtual Shape getInputShape(uint32_t index) const = 0;
97 virtual Shape getOutputShape(uint32_t index) const = 0;
101 virtual bool setOutputShape(uint32_t index, const Shape& shape) = 0;
136 bool SameShape(const Shape& in1, const Shape& in2);
139 bool SetShape(const Shape& in, Shape* out);
147 uint32_t getNumberOfElements(const Shape& shape);
148 uint32_t getNumberOfElements(const Shape& shape,
[all …]
/frameworks/ml/nn/common/operations/
DReshape.cpp32 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
33 const Shape& outputShape) { in copyData()
41 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
42 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
48 template bool depthToSpaceGeneric<float>(const float* inputData, const Shape& inputShape,
50 const Shape& outputShape);
51 template bool depthToSpaceGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
53 const Shape& outputShape);
54 template bool depthToSpaceGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
56 const Shape& outputShape);
[all …]
DSimpleMath.cpp32 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape) { in floorFloat16()
43 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape) { in floorFloat32()
51 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
52 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
53 const Shape& outputShape) { in meanFloat16()
66 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
67 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
95 template bool meanGeneric<float, float>(float* inputData, const Shape& inputShape,
96 const int32_t* axis, const Shape& axisShape, bool keepDims,
97 float* outputData, const Shape& outputShape);
[all …]
DRNN.h27 struct Shape;
34 Shape* hiddenStateShape, Shape* outputShape);
48 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData,
49 const T* biasData, const T* weightsData, const Shape& weightsShape,
50 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
54 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
55 const Shape& auxInputShape, const T* hiddenStateInputData,
56 const T* biasData, const T* weightsData, const Shape& weightsShape,
57 const T* auxWeightsData, const Shape& auxWeightsShape,
58 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
DActivation.cpp40 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
50 template bool reluFloat<float>(const float* inputData, const Shape& inputShape, float* outputData,
51 const Shape& outputShape, float reluMin, float reluMax);
52 template bool reluFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
53 _Float16* outputData, const Shape& outputShape, float reluMin,
57 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
58 const Shape& outputShape) { in relu1Float()
61 template bool relu1Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
62 const Shape& outputShape);
63 template bool relu1Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
[all …]
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);
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);
DSplit.cpp28 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
30 const std::vector<Shape>& outputShapes) { in splitGeneric()
54 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
56 const std::vector<Shape>& outputShapes) { in splitFloat16()
61 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
63 const std::vector<Shape>& outputShapes) { in splitFloat32()
68 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
70 const std::vector<Shape>& outputShapes) { in splitQuant8()
75 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
77 const std::vector<Shape>& outputShapes) { in splitInt32()
DTopK_V2.h26 bool prepare(const Shape& input, int32_t k, Shape* values, Shape* indices);
28 bool eval(const void* inputData, const Shape& inputShape, const int32_t k, void* valuesData,
29 const Shape& valuesShape, void* indicesData, const Shape& indicesShape);
DPow.cpp32 bool evalGeneric(const T* baseData, const Shape& baseShape, const T* exponentData, in evalGeneric()
33 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
59 bool prepare(const Shape& baseShape, const Shape& exponentShape, Shape* output) { in prepare()
67 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData, in eval()
68 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
DBroadcast.cpp65 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
66 int32_t activation, float* out, const Shape& shapeOut)>;
68 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16()
69 const Shape& shape2, int32_t activation, _Float16* out, in binaryOperationFloat16()
70 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16()
84 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
85 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32()
111 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
112 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16()
117 bool addQuant8(const uint8_t* in1, const Shape& shape1, const uint8_t* in2, const Shape& shape2, in addQuant8()
[all …]
DMaximumMinimum.cpp31 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric()
32 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
56 bool evalQuant8(const uint8_t* aData, const Shape& aShape, const uint8_t* bData, in evalQuant8()
57 const Shape& bShape, bool isMinimum, uint8_t* outputData, in evalQuant8()
58 const Shape& outputShape) { in evalQuant8()
86 bool prepare(const Shape& in1, const Shape& in2, Shape* out) { in prepare()
91 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval()
92 bool isMinimum, void* output, const Shape& outputShape) { in eval()
DTile.h26 bool prepare(const Shape& input, const int32_t* multiples, const Shape& multiplesShape,
27 Shape* output);
29 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples,
30 uint8_t* outputData, const Shape& outputShape);
DBidirectionalSequenceRNN.cpp53 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
68 Shape removeFirstDim(const Shape& input) { in removeFirstDim()
69 Shape output = input; in removeFirstDim()
80 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
83 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
85 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
90 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
92 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
105 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
106 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
[all …]
DTopK_V2.cpp29 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t k, T* valuesData, in evalGeneric() argument
30 const Shape& /*valuesShape*/, int32_t* indicesData, in evalGeneric()
31 const Shape& /*indicesShape*/) { in evalGeneric()
56 bool prepare(const Shape& input, int32_t k, Shape* values, Shape* indices) { in prepare()
67 bool eval(const void* inputData, const Shape& inputShape, const int32_t k, void* valuesData, in eval()
68 const Shape& valuesShape, void* indicesData, const Shape& indicesShape) { in eval()
DLSTM.h44 struct Shape;
51 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape,
52 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 …]
DGenerateProposals.cpp52 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape, in bboxTransformFloat32()
53 const float* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat32()
54 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat32()
55 const float* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat32()
56 float* outputData, const Shape& outputShape) { in bboxTransformFloat32()
102 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16()
103 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat16()
104 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16()
105 const _Float16* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat16()
106 _Float16* outputData, const Shape& outputShape) { in bboxTransformFloat16()
[all …]
DPooling.cpp71 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
95 tflite::PoolParams toTfliteParam(const Shape& output) const { in toTfliteParam()
121 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
122 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
131 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
132 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
144 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
145 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
154 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
155 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
[all …]
DDepthwiseConv2D.cpp28 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape, in depthwiseConvFloat16()
29 const _Float16* filterData, const Shape& filterShape, in depthwiseConvFloat16()
30 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvFloat16()
34 _Float16* outputData, const Shape& outputShape) { in depthwiseConvFloat16()
65 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvFloat32()
66 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvFloat32()
71 const Shape& outputShape) { in depthwiseConvFloat32()
98 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape, in depthwiseConvQuant8()
99 const uint8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8()
100 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvQuant8()
[all …]
DFullyConnected.cpp47 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
48 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
49 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32()
50 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
77 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
78 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
79 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16()
80 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
98 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
99 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
[all …]
DCast.h26 bool prepare(const Shape& input, Shape* output);
28 bool eval(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData,
29 const Shape& outputShape);
DExpandDims.h26 bool prepare(const Shape& input, int32_t axis, Shape* output);
28 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData,
29 const Shape& outputShape);
/frameworks/ml/nn/common/
DOperationsUtils.cpp68 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape()
80 bool SetShape(const Shape& in, Shape* out) { in SetShape()
115 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
123 uint32_t getNumberOfElements(const Shape& shape, in getNumberOfElements()
136 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
140 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
213 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
214 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler()
229 const Shape& outputShape, in CalculateActivationRangeUint8()
312 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape()
[all …]
DUtilsTest.cpp30 Shape shape1; in TEST()
31 Shape shape2; in TEST()
35 Shape expectedOutputShape; in TEST()
38 Shape actualOutputShape; in TEST()
47 Shape shape1; in TEST()
48 Shape shape2; in TEST()
52 Shape actualOutputShape; in TEST()

1234