Home
last modified time | relevance | path

Searched defs:tf_device (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/dtensor/python/
Dheartbeat.py33 from tensorflow.python.framework import device as tf_device unknown
Dmesh_util.py27 from tensorflow.python.framework import device as tf_device unknown
Dapi.py27 from tensorflow.python.framework import device as tf_device unknown
Ddtensor_device.py31 from tensorflow.python.framework import device as tf_device unknown
Dtpu_util.py34 from tensorflow.python.framework import device as tf_device unknown
Dlayout.py25 from tensorflow.python.framework import device as tf_device unknown
/external/tensorflow/tensorflow/python/distribute/
Ddevice_util.py21 from tensorflow.python.framework import device as tf_device unknown
Dmirrored_run.py30 from tensorflow.python.framework import device as tf_device unknown
Done_device_strategy_test.py26 from tensorflow.python.framework import device as tf_device unknown
Dparameter_server_strategy_v2.py37 from tensorflow.python.framework import device as tf_device unknown
Dparameter_server_strategy_test.py41 from tensorflow.python.framework import device as tf_device unknown
Dcollective_all_reduce_strategy.py43 from tensorflow.python.framework import device as tf_device unknown
Dcollective_all_reduce_strategy_test.py44 from tensorflow.python.framework import device as tf_device unknown
Dparameter_server_strategy.py35 from tensorflow.python.framework import device as tf_device unknown
Dmirrored_strategy.py41 from tensorflow.python.framework import device as tf_device unknown
Dtpu_strategy_test.py37 from tensorflow.python.framework import device as tf_device unknown
Dinput_lib.py41 from tensorflow.python.framework import device as tf_device unknown
Dmirrored_strategy_test.py46 from tensorflow.python.framework import device as tf_device unknown
/external/tensorflow/tensorflow/python/tpu/
Dtpu_system_metadata.py24 from tensorflow.python.framework import device as tf_device unknown
Dtpu_embedding_v2.py32 from tensorflow.python.framework import device as tf_device unknown
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtensor_util.cc137 Device* tf_device; in GetTfDevice() local
/external/tensorflow/tensorflow/core/tfrt/utils/
Dutils.cc47 tensorflow::Device* tf_device; in ConvertTfDeviceNameToTfrt() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfused_batchnorm_reserve_space_test.cc111 string tf_device = absl::StrCat("/device:", test_device, ":0"); in TEST() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Dconversion_function.cc151 Device *tf_device; in TransferRuntimeFallbackToAnotherDevice() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc50 namespace tf_device { namespace