Searched refs:activation_type (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | l2norm_test.cc | 30 const ActivationFunctionType activation_type) { in L2NormOpModel() argument
45 CreateL2NormOptions(builder_, activation_type).Union()); in L2NormOpModel()
|
| D | div_test.cc | 30 ActivationFunctionType activation_type) { in BaseDivOpModel() argument
35 CreateDivOptions(builder_, activation_type).Union()); in BaseDivOpModel()
|
| D | mul_test.cc | 30 ActivationFunctionType activation_type) { in BaseMulOpModel() argument
35 CreateMulOptions(builder_, activation_type).Union()); in BaseMulOpModel()
|
| D | add_test.cc | 30 ActivationFunctionType activation_type) { in BaseAddOpModel() argument
35 CreateAddOptions(builder_, activation_type).Union()); in BaseAddOpModel()
|
| D | sub_test.cc | 30 ActivationFunctionType activation_type) { in BaseSubOpModel() argument
35 CreateSubOptions(builder_, activation_type).Union()); in BaseSubOpModel()
|
| D | activations.cc | 375 params.activation_type = FusedActivationFunctionType::kRelu6; in QuantizedRelu6()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 93 ActivationFunctionType activation_type, in FloatAddOpModel() argument
99 CreateAddOptions(builder_, activation_type).Union()); in FloatAddOpModel()
165 ActivationFunctionType activation_type) { in FloatMulOpModel() argument
170 CreateMulOptions(builder_, activation_type).Union()); in FloatMulOpModel()
766 ActivationFunctionType activation_type) { in L2NormOpModel() argument
770 CreateL2NormOptions(builder_, activation_type).Union()); in L2NormOpModel()
839 ActivationFunctionType activation_type) { in FloatSubOpModel() argument
844 CreateMulOptions(builder_, activation_type).Union()); in FloatSubOpModel()
874 ActivationFunctionType activation_type) { in FloatDivOpModel() argument
879 CreateMulOptions(builder_, activation_type).Union()); in FloatDivOpModel()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 690 FusedActivationFunctionType activation_type; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 706 FusedActivationFunctionType activation_type; member
|