| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/ |
| H A D | heartbeat.py | 33 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | mesh_util.py | 27 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | api.py | 27 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | dtensor_device.py | 31 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | tpu_util.py | 34 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | layout.py | 25 from tensorflow.python.framework import device as tf_device unknown
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | device_util.py | 21 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | mirrored_run.py | 30 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | one_device_strategy_test.py | 26 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | parameter_server_strategy_v2.py | 37 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | parameter_server_strategy_test.py | 41 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | collective_all_reduce_strategy.py | 43 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | parameter_server_strategy.py | 35 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | collective_all_reduce_strategy_test.py | 44 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | mirrored_strategy.py | 41 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | tpu_strategy_test.py | 37 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | input_lib.py | 41 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | mirrored_strategy_test.py | 46 from tensorflow.python.framework import device as tf_device unknown
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_system_metadata.py | 24 from tensorflow.python.framework import device as tf_device unknown
|
| H A D | tpu_embedding_v2.py | 32 from tensorflow.python.framework import device as tf_device unknown
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tensor_util.cc | 137 Device* tf_device; in GetTfDevice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | utils.cc | 47 tensorflow::Device* tf_device; in ConvertTfDeviceNameToTfrt() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | fused_batchnorm_reserve_space_test.cc | 111 string tf_device = absl::StrCat("/device:", test_device, ":0"); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | conversion_function.cc | 151 Device *tf_device; in TransferRuntimeFallbackToAnotherDevice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_device.cc | 50 namespace tf_device { namespace
|