Home
last modified time | relevance | path

Searched defs:activation_type (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_nnapi_failure_handling_test.cc57 ActivationFunctionType activation_type, in AddSubOpsAcceleratedModel()
88 ActivationFunctionType activation_type, in Init()
Dnnapi_delegate_errno_test.cc54 ActivationFunctionType activation_type, in FloatAddOpModel()
73 const TensorData& output, ActivationFunctionType activation_type, in Init()
Dnnapi_delegate_device_selection_test.cc46 const TensorData& output, ActivationFunctionType activation_type, in Init()
376 ActivationFunctionType activation_type, in AddSubOpsAcceleratedModel()
402 ActivationFunctionType activation_type, in Init()
564 ActivationFunctionType activation_type, in HardSwishAddOpsAcceleratedModel()
586 const TensorData& output, ActivationFunctionType activation_type, in Init()
Dnnapi_delegate_test.cc156 ActivationFunctionType activation_type, in FloatAddOpModel()
164 ActivationFunctionType activation_type, in FloatAddOpModel()
183 const TensorData& output, ActivationFunctionType activation_type, in Init()
492 ActivationFunctionType activation_type) { in FloatMulOpModel()
1512 ActivationFunctionType activation_type) { in L2NormOpModel()
1666 ActivationFunctionType activation_type) { in FloatSubOpModel()
1700 ActivationFunctionType activation_type) { in FloatDivOpModel()
/external/XNNPACK/tools/
Dgenerate-vbinary-test.py257 def generate_test_cases(ukernel, op_type, activation_type, tester, batch_tile, isa): argument
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm_test.cc35 const ActivationFunctionType activation_type) { in L2NormOpModel()
Dadd_test.cc34 ActivationFunctionType activation_type) { in BaseAddOpModel()
69 ActivationFunctionType activation_type) in QuantizedAddOpModel()
Dsub_test.cc34 ActivationFunctionType activation_type) { in BaseSubOpModel()
76 ActivationFunctionType activation_type) in QuantizedSubOpModel()
Ddiv_test.cc34 ActivationFunctionType activation_type) { in BaseDivOpModel()
Dmul_test.cc35 ActivationFunctionType activation_type) { in BaseMulOpModel()
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddiv_test.cc32 ActivationFunctionType activation_type) { in BaseDivOpModel()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_benchmark_test.cc135 int out_depth, const string& activation_type, in Conv2DWithBiasAndActivation()
195 int out_depth, const string& activation_type, in Conv2DWithBatchNormAndActivation()
Dmatmul_op_test.cc121 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
Dconv_ops_test.cc635 const Tensor& bias_data, const string& activation_type, in RunConv2DWithBiasAndActivation()
699 const string& activation_type, const std::string& padding, in RunConv2DWithBatchNormAndActivation()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h769 FusedActivationFunctionType activation_type; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1797 auto activation_type = static_cast<const nvinfer1::IActivationLayer*>(layer) in IsClipOrRelu() local