Home
last modified time | relevance | path

Searched refs:kRelu6 (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_activation_functions.cc35 if (ac_op->type != OperatorType::kRelu6 && in Run()
92 if (ac_op->type == OperatorType::kRelu6) { in Run()
93 op->fused_activation_function = FusedActivationFunctionType::kRelu6; in Run()
Dremove_trivial_quantized_activation_func.cc47 case OperatorType::kRelu6: in IsTrivialUnfusedActivationFunc()
77 case FusedActivationFunctionType::kRelu6: in IsTrivialFusedActivationFunc()
Dpropagate_activation_function_into_constants.cc34 if (ac_op->type != OperatorType::kRelu6 && in Run()
104 case OperatorType::kRelu6: { in Run()
Dresolve_constant_unary.cc134 case OperatorType::kRelu6: in Run()
319 } else if (unary_op->type == OperatorType::kRelu6 || in Run()
337 case OperatorType::kRelu6: { in Run()
Dunfuse_activation_functions.cc45 case FusedActivationFunctionType::kRelu6: in Run()
Didentify_hardswish.cc44 FusedActivationFunctionType::kRelu6)) { in Run()
Dreorder_elementwise_unary.cc39 case OperatorType::kRelu6: in IsElementwiseOperator()
Dpropagate_fake_quant_num_bits.cc111 case OperatorType::kRelu6: in DoesOpBlockBackwardPropagation()
Dquantize.cc88 OperatorType::kRelu6, in SupportsQuantization()
407 op.type == OperatorType::kRelu6 || op.type == OperatorType::kPRelu || in ChooseQuantizationForOperatorOutput()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc119 CheckSimpleOperator<Relu6Operator>("RELU6", OperatorType::kRelu6); in TEST_F()
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()
Dtypes.cc217 case FusedActivationFunctionType::kRelu6: in Serialize()
234 return FusedActivationFunctionType::kRelu6; in Deserialize()
Dop_version.cc136 {{OperatorType::kRelu6, 1}, "1.5.0"}, in GetMinimumRuntimeVersionForModel()
137 {{OperatorType::kRelu6, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
Dtypes_test.cc204 {FusedActivationFunctionType::kRelu6, in TEST()
Doperator.cc1930 ::tflite::BuiltinOperator_RELU6, OperatorType::kRelu6)); in BuildOperatorList()
/external/tensorflow/tensorflow/core/kernels/neon/
Dtypes.h23 enum class FusedActivationFunctionType { kNone, kRelu6, kRelu1, kRelu }; enumerator
Ddepthwiseconv_float.h561 Ac == FusedActivationFunctionType::kRelu6 ||
674 } else if (Ac == FusedActivationFunctionType::kRelu6) {
695 } else if (Ac == FusedActivationFunctionType::kRelu6) {
710 } else if (Ac == FusedActivationFunctionType::kRelu6) {
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_float_test.cc91 FusedActivationFunctionType::kRelu6})); in TryTestOneDepthwiseConv()
Dcommon.h46 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
Dtypes.h27 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enumerator
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto54 kRelu6 = 3; enumerator
Ddnn.cc98 case ActivationMode::kRelu6: in ActivationModeString()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h343 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
542 Ac == FusedActivationFunctionType::kRelu6 || in FullyConnected()
1297 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1353 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAdd()
1420 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAddFivefold()
1474 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1644 Ac == FusedActivationFunctionType::kRelu6 || in AveragePool()
1735 Ac == FusedActivationFunctionType::kRelu6 || in MaxPool()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h77 kRelu6, enumerator
781 Relu6Operator() : Operator(OperatorType::kRelu6) {}
Ddump_graphviz.cc361 case FusedActivationFunctionType::kRelu6: in GetOpAttributes()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1781 Ac == FusedActivationFunctionType::kRelu6 || in FullyConnected()
2795 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
2822 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
2875 Ac == FusedActivationFunctionType::kRelu6 || in ConvAsGemm()
3506 Ac == FusedActivationFunctionType::kRelu6 || in Add()
3575 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAdd()
3614 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAddFivefold()
3669 Ac == FusedActivationFunctionType::kRelu6 || in Add()
3823 Ac == FusedActivationFunctionType::kRelu6 || in AveragePool()
3914 Ac == FusedActivationFunctionType::kRelu6 || in MaxPool()

12