Searched refs:activation_func (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | mul_test.cc | 26 ActivationFunctionType activation_func) { in MulOpModel() argument
31 CreateMulOptions(builder_, activation_func).Union()); in MulOpModel()
60 void TestMulOutputImpl(ActivationFunctionType activation_func) { in TestMulOutputImpl() argument
64 /*output=*/{tensor_type, {2, 3}, -1.0f, 1.0f}, activation_func); in TestMulOutputImpl()
79 void TestLargeInputRangeImpl(ActivationFunctionType activation_func) { in TestLargeInputRangeImpl() argument
83 /*output=*/{tensor_type, {1, 2, 2, 3}, -1.0f, 1.0f}, activation_func); in TestLargeInputRangeImpl()
|
| D | arithmetic_test.cc | 78 const TensorData& output, ActivationFunctionType activation_func) in AddOpModel() argument
80 activation_func_(activation_func) {} in AddOpModel()
85 ActivationFunctionType activation_func) in AddOpModel() argument
87 activation_func_(activation_func) {} in AddOpModel()
100 void QuantizedTestsNoActivation(ActivationFunctionType activation_func) { in QuantizedTestsNoActivation() argument
109 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, activation_func); in QuantizedTestsNoActivation()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected_test.cc | 141 ActivationFunctionType activation_func = ActivationFunctionType_RELU, argument
185 CreateFullyConnectedOptions(builder_, activation_func,
|