Searched refs:filterShape (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 29 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
30 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
38 const float* filterData, const Shape& filterShape, in depthwiseConvFloat32() argument
54 filterData, convertShapeToDims(filterShape), in depthwiseConvFloat32()
66 const uint8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8() argument
83 if (!GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, in depthwiseConvQuant8()
94 uint32_t filterOffset = -filterShape.offset; in depthwiseConvQuant8()
99 filterData, convertShapeToDims(filterShape), filterOffset, in depthwiseConvQuant8()
|
| D | Conv2D.cpp | 37 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
38 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
80 const float* filterData, const Shape& filterShape, in convFloat32() argument
98 filterData, convertShapeToDims(filterShape), in convFloat32()
108 const uint8_t* filterData, const Shape& filterShape, in convQuant8() argument
119 int32_t filterOffset = -filterShape.offset; in convQuant8()
128 if (!GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, in convQuant8()
147 filterData, convertShapeToDims(filterShape), filterOffset, in convQuant8()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 64 const float* filterData, const Shape& filterShape,
72 const uint8_t* filterData, const Shape& filterShape,
81 const float* filterData, const Shape& filterShape,
89 const uint8_t* filterData, const Shape& filterShape,
|
| D | OperationsUtils.h | 87 const Shape& filterShape,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 491 Shape filterShape = filter.shape(); in executeOperation() local
494 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
495 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
578 Shape filterShape = filter.shape(); in executeOperation() local
581 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
582 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
|
| D | OperationsUtils.cpp | 111 const Shape& filterShape, in GetQuantizedConvolutionMultipler() argument
115 const float input_product_scale = inputShape.scale * filterShape.scale; in GetQuantizedConvolutionMultipler()
|