Searched refs:apply_relu (Results 1 – 2 of 2) sorted by relevance
| /external/libtextclassifier/lang_id/common/ |
| D | embedding-network.cc | 66 void SparseReluProductPlusBias(bool apply_relu, in SparseReluProductPlusBias() argument
107 if (!apply_relu || (scale > 0)) { in SparseReluProductPlusBias()
131 if (!apply_relu || (scale > 0)) { in SparseReluProductPlusBias()
281 const bool apply_relu = i > 0; in ComputeFinalScores() local
283 apply_relu, layer_weights_[i], layer_bias_[i], *v_in, v_out); in ComputeFinalScores()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 672 side_input, biases, apply_relu): argument
703 if apply_relu:
857 def runTest(self, test_param, apply_relu): argument
922 activation_mode="Relu" if apply_relu else "None",
928 side_input_scale, side_input, biases, apply_relu)
939 for apply_relu in [True, False]:
941 self.runTest(test_param, apply_relu)
|