Home
last modified time | relevance | path

Searched refs:activation_function (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/
H A Drnn_fc.cc55 ActivationFunction activation_function) { in GetActivationFunction() argument
56 switch (activation_function) { in GetActivationFunction()
71 ActivationFunction activation_function, in FullyConnectedLayer() argument
79 activation_function_(GetActivationFunction(activation_function)) { in FullyConnectedLayer()
H A Drnn_fc.h41 ActivationFunction activation_function,
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Doperator.cc110 auto activation_function = in WriteOptions() local
114 op.kheight, activation_function); in WriteOptions()
139 auto activation_function = in WriteOptions() local
142 op.stride_height, activation_function, in WriteOptions()
170 auto activation_function = in WriteOptions() local
174 op.depth_multiplier, activation_function, op.dilation_width_factor, in WriteOptions()
212 auto activation_function = in WriteOptions() local
214 return ::tflite::CreateAddOptions(*builder, activation_function); in WriteOptions()
270 auto activation_function = in WriteOptions() local
272 return ::tflite::CreateSubOptions(*builder, activation_function); in WriteOptions()
[all …]
H A Dtypes.cc242 int activation_function) { in Deserialize() argument
243 switch (::tflite::ActivationFunctionType(activation_function)) { in Deserialize()
H A Dtypes.h51 static FusedActivationFunctionType Deserialize(int activation_function);
H A Dexport_test.cc553 auto activation_function = in WriteOptions() local
556 op.stride_height, activation_function, in WriteOptions()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dremove_trivial_quantized_activation_func.cc62 FusedActivationFunctionType activation_function, in IsTrivialFusedActivationFunc() argument
66 switch (activation_function) { in IsTrivialFusedActivationFunc()
83 << static_cast<int>(activation_function); in IsTrivialFusedActivationFunc()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc28 ActivationFunctionType activation_function = ActivationFunctionType_NONE) in FullyConnectedOpModel() argument
52 CreateFullyConnectedOptions(builder_, activation_function, in FullyConnectedOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_svdf.cc140 std::string activation_function = in CreateNodes() local
142 (*svdf_node->mutable_attr())["ActivationFunction"].set_s(activation_function); in CreateNodes()