Searched refs:tf_fn (Results 1 – 2 of 2) sorted by relevance
541 def _bitwise_binary_op(tf_fn, x1, x2): # pylint: disable=missing-function-docstring argument549 r = tf_fn(x1, x2)583 def _scalar(tf_fn, x, promote_to_float=False): argument601 return tf_fn(x)
478 def _reduce(tf_fn, argument545 return tf_fn(input_tensor=a, axis=axis, keepdims=keepdims)