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()
373 ActivationFunctionType activation_type, in AddSubOpsAcceleratedModel()
399 ActivationFunctionType activation_type, in Init()
560 ActivationFunctionType activation_type, in HardSwishAddOpsAcceleratedModel()
582 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.py264 def generate_test_cases(ukernel, op_type, init_fn, activation_type, argument
/external/tensorflow/tensorflow/lite/kernels/
Dadd_test.cc34 ActivationFunctionType activation_type) { in BaseAddOpModel()
45 ActivationFunctionType activation_type) { in BaseAddOpModel()
83 ActivationFunctionType activation_type) in QuantizedAddOpModel()
Dl2norm_test.cc35 const ActivationFunctionType activation_type) { in L2NormOpModel()
Dsub_test.cc34 ActivationFunctionType activation_type) { in BaseSubOpModel()
78 ActivationFunctionType activation_type) in QuantizedSubOpModel()
Dpooling3d.cc348 #define TF_LITE_AVERAGE_POOL_3D(type, activation_type) \ in AverageEval() argument
388 #define TF_LITE_MAX_POOL_3D(type, activation_type) \ in MaxEval() argument
Ddiv_test.cc34 ActivationFunctionType activation_type) { in BaseDivOpModel()
Dmul_test.cc35 ActivationFunctionType activation_type) { in BaseMulOpModel()
/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.cc578 const Tensor& bias_data, const string& activation_type, in RunConv2DWithBiasAndActivation()
642 const string& activation_type, const std::string& padding, in RunConv2DWithBatchNormAndActivation()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h598 FusedActivationFunctionType activation_type; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1812 auto activation_type = static_cast<const nvinfer1::IActivationLayer*>(layer) in IsClipOrRelu() local