| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | fuse_activation_functions.cc | 34 ac_op->type != OperatorType::kRelu) { in Run()
84 } else if (ac_op->type == OperatorType::kRelu) { in Run()
85 op->fused_activation_function = FusedActivationFunctionType::kRelu; in Run()
|
| D | remove_trivial_quantized_activation_func.cc | 32 if (op->fused_activation_function != FusedActivationFunctionType::kRelu && in Run()
48 if (op->fused_activation_function == FusedActivationFunctionType::kRelu || in Run()
|
| D | unfuse_activation_functions.cc | 39 case FusedActivationFunctionType::kRelu: in Run()
|
| D | reorder_activation_functions.cc | 35 ac_op->type != OperatorType::kRelu) { in Run()
|
| D | propagate_fixed_sizes.cc | 1349 case OperatorType::kRelu: in Run()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| D | types.cc | 150 case FusedActivationFunctionType::kRelu: in Serialize()
167 return FusedActivationFunctionType::kRelu; in Deserialize()
|
| D | operator_test.cc | 104 CheckSimpleOperator<ReluOperator>("RELU", OperatorType::kRelu); in TEST_F()
343 op.fused_activation_function = FusedActivationFunctionType::kRelu; in TEST_F()
|
| D | types_test.cc | 171 {FusedActivationFunctionType::kRelu, in TEST()
|
| D | operator.cc | 852 new SimpleOperator<ReluOperator>("RELU", OperatorType::kRelu)); in BuildOperatorList()
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | depthwiseconv_float.h | 560 Ac == FusedActivationFunctionType::kRelu ||
670 if (Ac == FusedActivationFunctionType::kRelu) {
693 if (Ac == FusedActivationFunctionType::kRelu) {
708 if (Ac == FusedActivationFunctionType::kRelu) {
|
| D | types.h | 23 enum class FusedActivationFunctionType { kNone, kRelu6, kRelu1, kRelu }; enumerator
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | common.h | 63 case FusedActivationFunctionType::kRelu: in GetActivationMinMax()
|
| D | types.h | 22 enum class FusedActivationFunctionType : uint8 { kNone, kRelu6, kRelu1, kRelu }; enumerator
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 561 dnn::ActivationMode::kRelu, output_desc, &output_ptr, in launch()
599 bias_desc, bias_ptr, dnn::ActivationMode::kRelu, output_desc, in launch()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| D | model.h | 63 kRelu, enumerator
534 ReluOperator() : Operator(OperatorType::kRelu) {}
|
| D | dump_graphviz.cc | 230 case FusedActivationFunctionType::kRelu: in GetPropertiesForOperator()
|
| D | export_tensorflow.cc | 1630 } else if (src_op.type == OperatorType::kRelu) { in ConvertOperator()
|
| D | import_tensorflow.cc | 1509 op->fused_activation_function = FusedActivationFunctionType::kRelu; in ConvertSvdfOperator()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | fully_connected.cc | 171 macro_name(target_namespace, kRelu); \
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 62 case ActivationMode::kRelu: in ActivationModeString()
|
| D | dnn.h | 858 kRelu,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| D | reference_ops.h | 330 Ac == FusedActivationFunctionType::kRelu || in Conv()
548 Ac == FusedActivationFunctionType::kRelu || in FullyConnected()
839 Ac == FusedActivationFunctionType::kRelu || in Add()
1013 Ac == FusedActivationFunctionType::kRelu || in BroadcastAdd()
1815 Ac == FusedActivationFunctionType::kRelu || in AveragePool()
2038 Ac == FusedActivationFunctionType::kRelu || in MaxPool()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 625 Ac == FusedActivationFunctionType::kRelu || in FullyConnected()
922 Ac == FusedActivationFunctionType::kRelu || in Conv()
949 Ac == FusedActivationFunctionType::kRelu || in Conv()
1034 Ac == FusedActivationFunctionType::kRelu || in ConvAsGemm()
1412 Ac == FusedActivationFunctionType::kRelu || in Add()
1662 Ac == FusedActivationFunctionType::kRelu || in BroadcastAdd()
2845 Ac == FusedActivationFunctionType::kRelu || in AveragePool()
3044 Ac == FusedActivationFunctionType::kRelu || in MaxPool()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 837 case dnn::ActivationMode::kRelu: in ScopedActivationDescriptor()
2554 if (activation_mode != dnn::ActivationMode::kRelu) { in DoFusedConvolveImpl()
3915 case dnn::ActivationMode::kRelu: in DoActivate()
|