| /frameworks/ml/nn/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()
135 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
136 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
177 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
185 const int filter_width = getSizeOfDimension(filterShape, 2); in needim2colData()
186 const int filter_height = getSizeOfDimension(filterShape, 1); in needim2colData()
194 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
210 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
[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 | DepthwiseConv2D.cpp | 95 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
98 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
99 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
123 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
124 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
132 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
158 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
166 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc() argument
175 std::vector<float> filterDataFloat32(getNumberOfElements(filterShape)); in depthwiseConvNhwc()
181 depthwiseConvNhwc(inputDataFloat32.data(), inputShape, filterDataFloat32.data(), filterShape, in depthwiseConvNhwc()
[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()
115 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
116 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
126 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
183 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
203 int32_t filterOffset = -filterShape.offset; in transposeConvNhwc()
209 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
277 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc() argument
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 49 const _Float16* filterData, const Shape& filterShape,
56 const Shape& filterShape, const float* biasData, const Shape& biasShape,
63 const uint8_t* filterData, const Shape& filterShape,
70 const int8_t* filterData, const Shape& filterShape,
145 const _Float16* filterData, const Shape& filterShape,
152 const Shape& filterShape, const float* biasData, const Shape& biasShape,
160 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
168 const int8_t* filterData, const Shape& filterShape,
|
| D | OperationsUtils.h | 195 __wur bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape,
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 258 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler() argument
262 const double input_product_scale = inputShape.scale * filterShape.scale; in GetQuantizedConvolutionMultipler()
|
| D | CpuExecutor.cpp | 1566 Shape filterShape = filter.shape(); in executeOperation() local
1569 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
1570 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
|