Home
last modified time | relevance | path

Searched refs:device_name_or_function (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops.py124 def __init__(self, device_name_or_function): argument
125 self._device_name_or_function = device_name_or_function
127 self.function = device_name_or_function
130 if isinstance(device_name_or_function, pydev.MergeDevice):
131 self.is_null_merge = device_name_or_function.is_null_merge
133 elif callable(device_name_or_function):
146 elif device_name_or_function is None:
153 self.raw_string = device_name_or_function
154 self.function = pydev.merge_device(device_name_or_function)
4703 def _add_device_to_stack(self, device_name_or_function, offset=0): argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-graph.pbtxt75 …argspec: "args=[\'self\', \'device_name_or_function\'], varargs=None, keywords=None, defaults=None"
H A Dtensorflow.pbtxt1129 argspec: "args=[\'device_name_or_function\'], varargs=None, keywords=None, defaults=None"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.-graph.pbtxt75 …argspec: "args=[\'self\', \'device_name_or_function\'], varargs=None, keywords=None, defaults=None"
H A Dtensorflow.__internal__.-func-graph.pbtxt156 …argspec: "args=[\'self\', \'device_name_or_function\'], varargs=None, keywords=None, defaults=None"