Searched refs:_to_tensor_or_tensor_spec (Results 1 – 1 of 1) sorted by relevance
2587 def _to_tensor_or_tensor_spec(self, x): member in FunctionSpec2592 args = [self._to_tensor_or_tensor_spec(x) for x in args]2593 kwargs = {kw: self._to_tensor_or_tensor_spec(x)2614 args[i] = self._to_tensor_or_tensor_spec(arg)2623 kwargs[kw] = self._to_tensor_or_tensor_spec(v)