Searched defs:device_manager (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_utils.h | 37 se::Platform* device_manager, PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId()
41 const DeviceType& type, se::Platform* device_manager, in ExecutorForTfDeviceId()
52 se::Platform* device_manager, in CheckValidTfDeviceId()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.cc | 63 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
104 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
|
| D | kernel_fallback_compat_request_state.h | 106 const tensorflow::DeviceMgr& device_manager() const { in device_manager() function
|
| D | kernel_fallback_op_handler.cc | 193 const tensorflow::DeviceMgr& device_manager, in GetOrCreate()
|
| D | kernel_fallback_execute_compat.cc | 111 const tensorflow::DeviceMgr* device_manager, in SetUpKernelFallbackCompatRequestContext()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.h | 52 const DeviceMgr &device_manager() const { return device_manager_; } in device_manager() function
|
| D | op_kernel_runner.cc | 84 const tensorflow::DeviceMgr& device_manager, in Create()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 88 const DeviceMgr* device_manager; in LiftVariablesFromSession() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 501 const auto* device_manager = &src_fallback_request_state->device_manager(); in SetUpKernelFallbackCompatRequestContextForBatch() local
|