Searched defs:filterShape (Results 1 - 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| H A D | Conv2D.cpp | 37 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
38 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
80 const float* filterData, const Shape& filterShape,
98 filterData, convertShapeToDims(filterShape),
108 const uint8_t* filterData, const Shape& filterShape,
119 int32_t filterOffset = -filterShape.offset;
128 if (!GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape,
147 filterData, convertShapeToDims(filterShape), filterOffset,
79 convFloat32(const float* inputData, const Shape& inputShape, const float* filterData, const Shape& filterShape, const float* biasData, const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top, int32_t padding_bottom, int32_t stride_width, int32_t stride_height, int32_t activation, float* outputData, const Shape& outputShape) argument
107 convQuant8(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top, int32_t padding_bottom, int32_t stride_width, int32_t stride_height, int32_t activation, uint8_t* outputData, const Shape& outputShape) argument
|
| H A D | DepthwiseConv2D.cpp | 29 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
30 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
38 const float* filterData, const Shape& filterShape,
54 filterData, convertShapeToDims(filterShape),
66 const uint8_t* filterData, const Shape& filterShape,
83 if (!GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape,
94 uint32_t filterOffset = -filterShape.offset;
99 filterData, convertShapeToDims(filterShape), filterOffset,
37 depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, const Shape& filterShape, const float* biasData, const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top, int32_t padding_bottom, int32_t stride_width, int32_t stride_height, int32_t depth_multiplier, int32_t activation, float* outputData, const Shape& outputShape) argument
65 depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top, int32_t padding_bottom, int32_t stride_width, int32_t stride_height, int32_t depth_multiplier, int32_t activation, uint8_t* outputData, const Shape& outputShape) argument
|
| /frameworks/ml/nn/common/ |
| H A D | CpuExecutor.cpp | 491 Shape filterShape = filter.shape(); local
494 int32_t filter_width = getSizeOfDimension(filterShape, 2);
495 int32_t filter_height = getSizeOfDimension(filterShape, 1);
578 Shape filterShape = filter.shape(); local
581 int32_t filter_width = getSizeOfDimension(filterShape, 2);
582 int32_t filter_height = getSizeOfDimension(filterShape, 1);
|
| H A D | OperationsUtils.cpp | 111 const Shape& filterShape,
115 const float input_product_scale = inputShape.scale * filterShape.scale;
110 GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, const Shape& biasShape, const Shape& outputShape, float* multiplier) argument
|
Completed in 57 milliseconds