Home
last modified time | relevance | path

Searched refs:kRelu6 (Results 1 – 25 of 32) 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()
315 } else if (unary_op->type == OperatorType::kRelu6 || in Run()
333 case OperatorType::kRelu6: { in Run()
Dunfuse_activation_functions.cc45 case FusedActivationFunctionType::kRelu6: in Run()
Dreorder_elementwise_unary.cc39 case OperatorType::kRelu6: in IsElementwiseOperator()
Dpropagate_fake_quant_num_bits.cc115 case OperatorType::kRelu6: in DoesOpBlockBackwardPropagation()
Dquantize.cc64 type == OperatorType::kRelu1 || type == OperatorType::kRelu6 || in SupportsQuantization()
366 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()
Dtypes_test.cc204 {FusedActivationFunctionType::kRelu6, in TEST()
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h561 Ac == FusedActivationFunctionType::kRelu6 ||
674 } else if (Ac == FusedActivationFunctionType::kRelu6) {
695 } else if (Ac == FusedActivationFunctionType::kRelu6) {
710 } else if (Ac == FusedActivationFunctionType::kRelu6) {
Dtypes.h23 enum class FusedActivationFunctionType { kNone, kRelu6, kRelu1, kRelu }; enumerator
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h417 Ac == FusedActivationFunctionType::kRelu6 || in FullyConnected()
671 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
698 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
751 Ac == FusedActivationFunctionType::kRelu6 || in ConvAsGemm()
956 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1025 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAdd()
1064 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAddFivefold()
1119 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1273 Ac == FusedActivationFunctionType::kRelu6 || in AveragePool()
1364 Ac == FusedActivationFunctionType::kRelu6 || in MaxPool()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_float_test.cc88 FusedActivationFunctionType::kRelu6})); in TryTestOneDepthwiseConv()
Dcommon.h69 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
Dtypes.h25 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enumerator
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto52 kRelu6 = 3; enumerator
Ddnn.cc72 case ActivationMode::kRelu6: in ActivationModeString()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h71 case FusedActivationFunctionType::kRelu6: in GetActivationMinMax()
Dtypes.h31 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enumerator
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h341 Ac == FusedActivationFunctionType::kRelu6 || in Conv()
506 Ac == FusedActivationFunctionType::kRelu6 || in FullyConnected()
1176 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1232 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAdd()
1299 Ac == FusedActivationFunctionType::kRelu6 || in BroadcastAddFivefold()
1353 Ac == FusedActivationFunctionType::kRelu6 || in Add()
1523 Ac == FusedActivationFunctionType::kRelu6 || in AveragePool()
1614 Ac == FusedActivationFunctionType::kRelu6 || in MaxPool()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h76 kRelu6, enumerator
737 Relu6Operator() : Operator(OperatorType::kRelu6) {}
Ddump_graphviz.cc361 case FusedActivationFunctionType::kRelu6: in GetOpAttributes()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc288 macro_name(target_namespace, kRelu6); \

12