| /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 | identify_util.h | 27 FusedActivationFunctionType act = FusedActivationFunctionType::kNone);
|
| D | identify_util.cc | 25 FusedActivationFunctionType act) { in IsBinaryOp()
|
| D | resolve_constant_binary.cc | 70 FusedActivationFunctionType::kNone); in EvaluateBinaryOperatorOnConstantInputs()
231 FusedActivationFunctionType::kNone) { in Run()
|
| D | identify_hardswish.cc | 44 FusedActivationFunctionType::kRelu6)) { in Run()
|
| D | remove_trivial_binary.cc | 127 FusedActivationFunctionType::kNone; in Run()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 84 template <FusedActivationFunctionType Ac>
100 template <FusedActivationFunctionType Ac>
165 template <FusedActivationFunctionType Ac>
175 if (Ac == FusedActivationFunctionType::kNone) { in DepthwiseConv()
188 template <FusedActivationFunctionType Ac>
231 template <FusedActivationFunctionType Ac>
249 template <FusedActivationFunctionType Ac>
265 template <FusedActivationFunctionType Ac>
330 template <FusedActivationFunctionType Ac>
342 static_assert(Ac == FusedActivationFunctionType::kNone || in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | types.cc | 219 FusedActivationFunctionType faf_type) { in Serialize()
221 case FusedActivationFunctionType::kNone: in Serialize()
223 case FusedActivationFunctionType::kRelu: in Serialize()
225 case FusedActivationFunctionType::kRelu6: in Serialize()
227 case FusedActivationFunctionType::kRelu1: in Serialize()
234 FusedActivationFunctionType ActivationFunction::Deserialize( in Deserialize()
238 return FusedActivationFunctionType::kNone; in Deserialize()
240 return FusedActivationFunctionType::kRelu; in Deserialize()
242 return FusedActivationFunctionType::kRelu6; in Deserialize()
244 return FusedActivationFunctionType::kRelu1; in Deserialize()
|
| D | types_test.cc | 206 std::pair<FusedActivationFunctionType, ::tflite::ActivationFunctionType>> in TEST()
207 testdata = {{FusedActivationFunctionType::kNone, in TEST()
209 {FusedActivationFunctionType::kRelu, in TEST()
211 {FusedActivationFunctionType::kRelu6, in TEST()
213 {FusedActivationFunctionType::kRelu1, in TEST()
221 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()
462 op.fused_activation_function = FusedActivationFunctionType::kRelu; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_float_test.cc | 86 FusedActivationFunctionType ac = in TryTestOneDepthwiseConv()
87 RandomElement(std::vector<FusedActivationFunctionType>( in TryTestOneDepthwiseConv()
88 {FusedActivationFunctionType::kNone, in TryTestOneDepthwiseConv()
89 FusedActivationFunctionType::kRelu, in TryTestOneDepthwiseConv()
90 FusedActivationFunctionType::kRelu1, in TryTestOneDepthwiseConv()
91 FusedActivationFunctionType::kRelu6})); in TryTestOneDepthwiseConv()
|
| D | common.h | 35 inline void GetActivationMinMax(FusedActivationFunctionType ac, in GetActivationMinMax()
39 case FusedActivationFunctionType::kNone: in GetActivationMinMax()
43 case FusedActivationFunctionType::kRelu: in GetActivationMinMax()
47 case FusedActivationFunctionType::kRelu1: in GetActivationMinMax()
51 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
67 template <FusedActivationFunctionType Ac>
|
| D | types.h | 27 enum class FusedActivationFunctionType : uint8_t { enum
769 FusedActivationFunctionType activation_type;
1038 FusedActivationFunctionType activation;
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | depthwiseconv_float.h | 550 template <FusedActivationFunctionType Ac>
557 static_assert(Ac == FusedActivationFunctionType::kNone ||
558 Ac == FusedActivationFunctionType::kRelu ||
559 Ac == FusedActivationFunctionType::kRelu6 ||
560 Ac == FusedActivationFunctionType::kRelu1,
668 if (Ac == FusedActivationFunctionType::kRelu) {
672 } else if (Ac == FusedActivationFunctionType::kRelu6) {
677 } else if (Ac == FusedActivationFunctionType::kRelu1) {
691 if (Ac == FusedActivationFunctionType::kRelu) {
693 } 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/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 185 template <FusedActivationFunctionType Ac>
201 template <FusedActivationFunctionType Ac>
339 template <FusedActivationFunctionType Ac>
349 if (Ac == FusedActivationFunctionType::kNone) { in DepthwiseConv()
362 template <FusedActivationFunctionType Ac>
618 template <FusedActivationFunctionType Ac>
702 template <FusedActivationFunctionType Ac>
1770 template <FusedActivationFunctionType Ac>
1780 static_assert(Ac == FusedActivationFunctionType::kNone || in FullyConnected()
1781 Ac == FusedActivationFunctionType::kRelu || in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/runtime/ |
| D | types.h | 28 using tflite::FusedActivationFunctionType;
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 361 case FusedActivationFunctionType::kRelu: in GetOpAttributes()
364 case FusedActivationFunctionType::kRelu6: in GetOpAttributes()
367 case FusedActivationFunctionType::kRelu1: in GetOpAttributes()
|
| D | model.h | 395 FusedActivationFunctionType fused_activation_function;
425 fused_activation_function(FusedActivationFunctionType::kNone) {}
2129 FusedActivationFunctionType fused_activation_function;
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 214 enum class FusedActivationFunctionType : uint8_t { kNone, kRelu6, kRelu1, kRelu };
918 FusedActivationFunctionType activation_type;
1137 FusedActivationFunctionType activation;
|