Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddevice_util.py144 ops.get_default_graph()._apply_device_functions(op) # pylint: disable=protected-access
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dimporter.py259 graph._apply_device_functions(new_op) # pylint: disable=protected-access
H A Dops.py3882 self._apply_device_functions(op)
4781 def _apply_device_functions(self, op): member in Graph
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py537 graph._apply_device_functions(fake_op) # pylint: disable=protected-access
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py866 graph._apply_device_functions(op)