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
|