| /frameworks/ml/nn/common/operations/ |
| Normalization.cpp | 20 #include "internal/optimized/optimized_ops.h"
27 optimized_ops::L2Normalization<FusedActivationFunctionType::kNone>(
36 optimized_ops::L2Normalization(
47 optimized_ops::LocalResponseNormalization(
|
| Reshape.cpp | 24 #include "internal/optimized/optimized_ops.h"
46 optimized_ops::ResizeBilinear(
57 optimized_ops::DepthToSpace(
64 optimized_ops::DepthToSpace(
81 optimized_ops::SpaceToDepth(
88 optimized_ops::SpaceToDepth(
|
| SimpleMath.cpp | 24 #include "internal/optimized/optimized_ops.h"
35 optimized_ops::Add<FusedActivationFunctionType::activation>( \
41 optimized_ops::BroadcastAdd<FusedActivationFunctionType::activation>( \
99 optimized_ops::Add<FusedActivationFunctionType::activation>( \
110 optimized_ops::BroadcastAdd<FusedActivationFunctionType::activation>( \
138 optimized_ops::Mul<FusedActivationFunctionType::activation>( \
144 optimized_ops::BroadcastMul<FusedActivationFunctionType::activation>( \
183 optimized_ops::BroadcastMul<FusedActivationFunctionType::activation>( \
201 optimized_ops::Floor(inputData, dim, outputData, dim);
209 optimized_ops::Dequantize(inputData, dim
[all...] |
| Pooling.cpp | 20 #include "internal/optimized/optimized_ops.h"
44 optimized_ops::AveragePool<FusedActivationFunctionType::activation>( \
73 optimized_ops::AveragePool<FusedActivationFunctionType::activation>( \
96 optimized_ops::L2Pool<FusedActivationFunctionType::activation>( \
118 optimized_ops::MaxPool<FusedActivationFunctionType::activation>( \
147 optimized_ops::MaxPool<FusedActivationFunctionType::activation>( \
|
| Concatenation.cpp | 20 #include "internal/optimized/optimized_ops.h"
36 optimized_ops::Concatenation<FusedActivationFunctionType::kNone, float>(
55 optimized_ops::Concatenation<FusedActivationFunctionType::kNone, uint8_t>(
|
| FullyConnected.cpp | 20 #include "internal/optimized/optimized_ops.h"
32 optimized_ops::FullyConnected<FusedActivationFunctionType::activation>( \
73 optimized_ops::FullyConnected<FusedActivationFunctionType::activation>( \
|
| Conv2D.cpp | 20 #include "internal/optimized/optimized_ops.h"
75 optimized_ops::Conv<FusedActivationFunctionType::activation>( \
128 optimized_ops::Conv<FusedActivationFunctionType::activation>( \
|
| DepthwiseConv2D.cpp | 49 optimized_ops::DepthwiseConv<FusedActivationFunctionType::activation>( \
96 optimized_ops::DepthwiseConv<FusedActivationFunctionType::activation>( \
|
| Activation.cpp | 20 #include "internal/optimized/optimized_ops.h"
88 optimized_ops::Softmax(inputData, dim, beta,
152 optimized_ops::Logistic(
199 optimized_ops::Softmax(inputData, dim, input_multiplier,
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| depthwiseconv_float.h | 26 namespace optimized_ops { namespace in namespace:android::nn
784 } // namespace optimized_ops
|
| depthwiseconv_uint8.h | 27 namespace optimized_ops { namespace in namespace:android::nn
[all...] |
| optimized_ops.h | 38 namespace optimized_ops { namespace in namespace:android::nn
[all...] |
