| /packages/modules/NeuralNetworks/common/operations/ |
| D | Conv2D.cpp | 108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
111 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
112 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
136 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
137 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
178 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
186 const int filter_width = getSizeOfDimension(filterShape, 2); in needim2colData()
187 const int filter_height = getSizeOfDimension(filterShape, 1); in needim2colData()
195 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
211 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
[all …]
|
| D | DepthwiseConv2D.cpp | 97 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
100 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
101 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
125 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
126 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
134 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
160 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
168 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc() argument
177 std::vector<float> filterDataFloat32(getNumberOfElements(filterShape)); in depthwiseConvNhwc()
183 depthwiseConvNhwc(inputDataFloat32.data(), inputShape, filterDataFloat32.data(), filterShape, in depthwiseConvNhwc()
[all …]
|
| D | GroupedConv2D.cpp | 38 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
39 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
40 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); \
47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32() argument
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
114 int32_t filterOffset = -filterShape.offset; in groupedConvQuant8()
120 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
182 const int8_t* filterData, const Shape& filterShape,
191 const uint8_t* filterData, const Shape& filterShape,
201 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel() argument
[all …]
|
| D | TransposeConv2D.cpp | 75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
76 int32_t filterWidth = getSizeOfDimension(filterShape, 2); in initialize()
77 int32_t filterHeight = getSizeOfDimension(filterShape, 1); in initialize()
116 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
117 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
204 int32_t filterOffset = -filterShape.offset; in transposeConvNhwc()
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
278 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc() argument
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 53 const _Float16* filterData, const Shape& filterShape,
60 const Shape& filterShape, const float* biasData, const Shape& biasShape,
67 const uint8_t* filterData, const Shape& filterShape,
74 const int8_t* filterData, const Shape& filterShape,
149 const _Float16* filterData, const Shape& filterShape,
156 const Shape& filterShape, const float* biasData, const Shape& biasShape,
164 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
172 const int8_t* filterData, const Shape& filterShape,
|
| D | OperationsUtils.h | 179 __wur bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape,
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 256 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler() argument
260 const double input_product_scale = inputShape.scale * filterShape.scale; in GetQuantizedConvolutionMultipler()
|
| D | CpuExecutor.cpp | 1510 Shape filterShape = filter.shape(); in executeOperation() local
1513 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
1514 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
|