Home
last modified time | relevance | path

Searched refs:GetActivation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfused_kernel_matcher.cc64 Operation *GetActivation(Value op) { in GetActivation() function
131 auto activation = GetActivation(bias_add); in matchAndRewrite()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval_test.cc496 TfLiteTensor* GetActivation() { in GetActivation() function in tflite::__anonb312a86f0111::QuantizedLstmParam
607 auto activation = one_parameter.GetActivation(); in TestOneFullyQuantizedLSTM()
726 TfLiteTensor* GetActivation() { in GetActivation() function in tflite::__anonb312a86f0111::HybridLstmParam
902 auto activation = one_parameter.GetActivation(); in TestOneHybridAsymmLSTM()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc606 RETURN_IF_ERROR(GetActivation(tflite_node, &activation)); in IsSupported()
678 absl::Status GetActivation(const TfLiteNode* tflite_node, in GetActivation() function in tflite::gpu::__anon48bc1b4d0111::ElementwiseOperationParser