| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 180 template <FusedActivationFunctionType Ac>
196 template <FusedActivationFunctionType Ac>
261 template <FusedActivationFunctionType Ac>
271 if (Ac == FusedActivationFunctionType::kNone) { in DepthwiseConv()
284 template <FusedActivationFunctionType Ac>
314 template <FusedActivationFunctionType Ac>
344 template <FusedActivationFunctionType Ac>
405 template <FusedActivationFunctionType Ac>
415 static_assert(Ac == FusedActivationFunctionType::kNone || in FullyConnected()
416 Ac == FusedActivationFunctionType::kRelu || in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_trivial_quantized_activation_func.cc | 62 FusedActivationFunctionType activation_function, in IsTrivialFusedActivationFunc()
67 case FusedActivationFunctionType::kNone: in IsTrivialFusedActivationFunc()
69 case FusedActivationFunctionType::kRelu: in IsTrivialFusedActivationFunc()
73 case FusedActivationFunctionType::kRelu1: in IsTrivialFusedActivationFunc()
77 case FusedActivationFunctionType::kRelu6: in IsTrivialFusedActivationFunc()
116 op->fused_activation_function = FusedActivationFunctionType::kNone; in Run()
|
| D | unfuse_activation_functions.cc | 42 case FusedActivationFunctionType::kRelu: in Run()
45 case FusedActivationFunctionType::kRelu6: in Run()
48 case FusedActivationFunctionType::kRelu1: in Run()
62 op->fused_activation_function = FusedActivationFunctionType::kNone; in Run()
|
| D | identify_prelu.cc | 53 add_op->fused_activation_function != FusedActivationFunctionType::kNone) { in Run()
61 FusedActivationFunctionType::kNone) { in Run()
70 mul_op->fused_activation_function != FusedActivationFunctionType::kNone) { in Run()
86 FusedActivationFunctionType::kRelu) { in Run()
96 FusedActivationFunctionType::kNone) { in Run()
|
| D | fuse_activation_functions.cc | 74 if (op->fused_activation_function != FusedActivationFunctionType::kNone) { in Run()
93 op->fused_activation_function = FusedActivationFunctionType::kRelu6; in Run()
95 op->fused_activation_function = FusedActivationFunctionType::kRelu1; in Run()
97 op->fused_activation_function = FusedActivationFunctionType::kRelu; in Run()
|
| D | resolve_constant_binary.cc | 69 FusedActivationFunctionType::kNone); in EvaluateBinaryOperatorOnConstantInputs()
231 FusedActivationFunctionType::kNone) { in Run()
|
| D | fuse_binary_into_following_affine.cc | 215 FusedActivationFunctionType::kNone) { in Run()
|
| D | fuse_binary_into_preceding_affine.cc | 265 FusedActivationFunctionType::kNone) { in Run()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 81 template <FusedActivationFunctionType Ac>
97 template <FusedActivationFunctionType Ac>
162 template <FusedActivationFunctionType Ac>
172 if (Ac == FusedActivationFunctionType::kNone) { in DepthwiseConv()
185 template <FusedActivationFunctionType Ac>
228 template <FusedActivationFunctionType Ac>
246 template <FusedActivationFunctionType Ac>
262 template <FusedActivationFunctionType Ac>
327 template <FusedActivationFunctionType Ac>
339 static_assert(Ac == FusedActivationFunctionType::kNone || in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | types.cc | 211 FusedActivationFunctionType faf_type) { in Serialize()
213 case FusedActivationFunctionType::kNone: in Serialize()
215 case FusedActivationFunctionType::kRelu: in Serialize()
217 case FusedActivationFunctionType::kRelu6: in Serialize()
219 case FusedActivationFunctionType::kRelu1: in Serialize()
226 FusedActivationFunctionType ActivationFunction::Deserialize( in Deserialize()
230 return FusedActivationFunctionType::kNone; in Deserialize()
232 return FusedActivationFunctionType::kRelu; in Deserialize()
234 return FusedActivationFunctionType::kRelu6; in Deserialize()
236 return FusedActivationFunctionType::kRelu1; in Deserialize()
|
| D | types_test.cc | 199 std::pair<FusedActivationFunctionType, ::tflite::ActivationFunctionType>> in TEST()
200 testdata = {{FusedActivationFunctionType::kNone, in TEST()
202 {FusedActivationFunctionType::kRelu, in TEST()
204 {FusedActivationFunctionType::kRelu6, in TEST()
206 {FusedActivationFunctionType::kRelu1, in TEST()
214 static_cast<FusedActivationFunctionType>(10000)), in TEST()
|
| D | types.h | 50 FusedActivationFunctionType faf_type);
51 static FusedActivationFunctionType Deserialize(int activation_function);
|
| D | operator_test.cc | 163 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
231 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
349 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
371 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
387 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
400 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
409 op.fused_activation_function = FusedActivationFunctionType::kRelu6; in TEST_F()
436 op.fused_activation_function = FusedActivationFunctionType::kRelu; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_float_test.cc | 83 FusedActivationFunctionType ac = in TryTestOneDepthwiseConv()
84 RandomElement(std::vector<FusedActivationFunctionType>( in TryTestOneDepthwiseConv()
85 {FusedActivationFunctionType::kNone, in TryTestOneDepthwiseConv()
86 FusedActivationFunctionType::kRelu, in TryTestOneDepthwiseConv()
87 FusedActivationFunctionType::kRelu1, in TryTestOneDepthwiseConv()
88 FusedActivationFunctionType::kRelu6})); in TryTestOneDepthwiseConv()
|
| D | common.h | 53 inline void GetActivationMinMax(FusedActivationFunctionType ac, in GetActivationMinMax()
57 case FusedActivationFunctionType::kNone: in GetActivationMinMax()
61 case FusedActivationFunctionType::kRelu: in GetActivationMinMax()
65 case FusedActivationFunctionType::kRelu1: in GetActivationMinMax()
69 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
82 template <FusedActivationFunctionType Ac>
|
| D | types.h | 25 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enum
706 FusedActivationFunctionType activation_type;
916 FusedActivationFunctionType activation;
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | common.h | 55 inline void GetActivationMinMax(FusedActivationFunctionType ac, in GetActivationMinMax()
59 case FusedActivationFunctionType::kNone: in GetActivationMinMax()
63 case FusedActivationFunctionType::kRelu: in GetActivationMinMax()
67 case FusedActivationFunctionType::kRelu1: in GetActivationMinMax()
71 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
84 template <FusedActivationFunctionType Ac>
|
| D | types.h | 31 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enum
690 FusedActivationFunctionType activation_type;
875 FusedActivationFunctionType activation;
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | depthwiseconv_float.h | 552 template <FusedActivationFunctionType Ac>
559 static_assert(Ac == FusedActivationFunctionType::kNone ||
560 Ac == FusedActivationFunctionType::kRelu ||
561 Ac == FusedActivationFunctionType::kRelu6 ||
562 Ac == FusedActivationFunctionType::kRelu1,
670 if (Ac == FusedActivationFunctionType::kRelu) {
674 } else if (Ac == FusedActivationFunctionType::kRelu6) {
679 } else if (Ac == FusedActivationFunctionType::kRelu1) {
693 if (Ac == FusedActivationFunctionType::kRelu) {
695 } else if (Ac == FusedActivationFunctionType::kRelu6) {
[all …]
|
| D | types.h | 23 enum class FusedActivationFunctionType { kNone, kRelu6, kRelu1, kRelu }; enum
|
| D | neon_depthwise_conv_op.cc | 135 neon::DepthwiseConv<neon::FusedActivationFunctionType::kNone>( in Compute()
|
| /external/tensorflow/tensorflow/lite/toco/runtime/ |
| D | types.h | 28 using tflite::FusedActivationFunctionType;
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 358 case FusedActivationFunctionType::kRelu: in GetOpAttributes()
361 case FusedActivationFunctionType::kRelu6: in GetOpAttributes()
364 case FusedActivationFunctionType::kRelu1: in GetOpAttributes()
|
| D | model.h | 378 FusedActivationFunctionType fused_activation_function;
408 fused_activation_function(FusedActivationFunctionType::kNone) {}
2063 FusedActivationFunctionType fused_activation_function;
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 370 template <FusedActivationFunctionType Ac>
948 template <FusedActivationFunctionType Ac>
1347 template <FusedActivationFunctionType Ac>
1357 static_assert(Ac == FusedActivationFunctionType::kNone || in FullyConnected()
1358 Ac == FusedActivationFunctionType::kRelu || in FullyConnected()
1359 Ac == FusedActivationFunctionType::kRelu6 || in FullyConnected()
1360 Ac == FusedActivationFunctionType::kRelu1, in FullyConnected()
2312 template <FusedActivationFunctionType Ac>
2331 template <FusedActivationFunctionType Ac>
2348 template <FusedActivationFunctionType Ac>
[all …]
|