| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Conv2D.cpp | 106 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
109 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
110 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
133 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
134 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
175 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
183 const int filter_width = getSizeOfDimension(filterShape, 2); in needim2colData()
184 const int filter_height = getSizeOfDimension(filterShape, 1); in needim2colData()
192 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
208 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
[all …]
|
| D | DepthwiseConv2D.cpp | 93 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
96 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
97 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
121 [[maybe_unused]] uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
122 [[maybe_unused]] uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
130 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
156 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
164 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc() argument
173 std::vector<float> filterDataFloat32(getNumberOfElements(filterShape)); in depthwiseConvNhwc()
179 depthwiseConvNhwc(inputDataFloat32.data(), inputShape, filterDataFloat32.data(), filterShape, in depthwiseConvNhwc()
[all …]
|
| D | GroupedConv2D.cpp | 42 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
43 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
44 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); \
51 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in groupedConvFloat32() argument
109 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
118 int32_t filterOffset = -filterShape.offset; in groupedConvQuant8()
124 NN_RET_CHECK(GetQuantizedConvolutionMultiplier(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
186 const int8_t* filterData, const Shape& filterShape,
195 const uint8_t* filterData, const Shape& filterShape,
205 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel() argument
[all …]
|
| D | TransposeConv2D.cpp | 70 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
71 int32_t filterWidth = getSizeOfDimension(filterShape, 2); in initialize()
72 int32_t filterHeight = getSizeOfDimension(filterShape, 1); in initialize()
110 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
111 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
123 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in transposeConvNhwc() argument
180 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
200 int32_t filterOffset = -filterShape.offset; in transposeConvNhwc()
206 NN_RET_CHECK(GetQuantizedConvolutionMultiplier(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
274 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc() argument
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 36 const _Float16* filterData, const Shape& filterShape,
43 const Shape& filterShape, const float* biasData, const Shape& biasShape,
50 const uint8_t* filterData, const Shape& filterShape,
57 const int8_t* filterData, const Shape& filterShape,
132 const _Float16* filterData, const Shape& filterShape,
139 const Shape& filterShape, const float* biasData, const Shape& biasShape,
147 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
155 const int8_t* filterData, const Shape& filterShape,
|
| D | OperationsExecutionUtils.h | 117 const Shape& filterShape,
122 const Shape& filterShape,
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 154 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier() argument
158 const double input_product_scale = inputShape.scale * filterShape.scale; in GetQuantizedConvolutionMultiplier()
169 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier() argument
172 const double input_product_scale = inputShape.scale * filterShape.scale; in GetQuantizedConvolutionMultiplier()
|
| D | CpuExecutor.cpp | 1511 Shape filterShape = filter.shape(); in executeOperation() local
1514 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
1515 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
|