Searched refs:f_types (Results 1 – 2 of 2) sorted by relevance
377 f_types = set()379 f_types.add(Callable[[Any], rt])381 self.new_symbols[f_name] = f_types385 def _resolve_typed_callable(self, f_types, arg_types, keyword_types): argument387 for t in f_types:
310 f_types = [default_gradient.get_zeros_dtype(x) for x in op.inputs]318 in_grads = functional_ops.symbolic_gradient(input=f_in, Tout=f_types, f=f)