Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dfake_session.h72 std::unique_ptr<tensorflow::DeviceMgr> device_mgr_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.h70 std::unique_ptr<StaticDeviceMgr> device_mgr_; variable
Ddevice_resolver_local_test.cc43 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon53efc55c0111::DeviceResolverLocalTest
Dcollective_executor_mgr_test.cc63 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon021f738e0111::CollectiveExecutorMgrTest
Dprocess_function_library_runtime.h505 const DeviceMgr* const device_mgr_; variable
Dfunction_threadpool_test.cc197 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anonfa9f08480111::FunctionLibraryRuntimeTest
Dcollective_rma_local_test.cc62 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anoneda91a840111::CollectiveRemoteAccessLocalTest
/external/tensorflow/tensorflow/core/data/
Dstandalone.h122 std::unique_ptr<DeviceMgr> device_mgr_; variable
Dserialization_utils_test.cc87 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::data::__anon9819502d0111::TestContext
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_session.h116 const std::unique_ptr<DeviceMgr> device_mgr_; variable
Dgraph_mgr.h151 const DeviceMgr* device_mgr_; variable
Drpc_collective_executor_mgr_test.cc64 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::RpcCollectiveExecutorMgrTest
Dsession_mgr_test.cc58 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::SessionMgrTest
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dconstant_loading_test.cc46 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anoneee0963e0111::ConstantTest
Dsaved_variable_loading_test.cc56 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon69acc06c0111::SavedVariableLoadingTest
Dtf_concrete_function_loading_test.cc46 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon8716e6450111::SavedConcreteFunctionLoadingTest
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Drestore_ops_test.cc47 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon1d5c018d0111::RestoreOpsTest
Dvariable_ops_test.cc49 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon0dee37ef0111::VariableOpsTest
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h162 std::unique_ptr<DeviceMgr> device_mgr_; variable
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator_test.cc85 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::XlaKernelCreatorTest
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.h310 const DeviceMgr* device_mgr_; variable
/external/tensorflow/tensorflow/core/tfrt/utils/
Dgraph_partition_test.cc64 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::tfrt_stub::__anon2af3c4f00111::GraphPartitionTest
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc66 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anond84d0c860111::TestEnv
Dtensor_handle_test.cc135 StaticDeviceMgr* device_mgr_; member in tensorflow::PackedTensorHandleTest
366 StaticDeviceMgr* device_mgr_; member in tensorflow::RemoteTensorHandleTest
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h342 StaticDeviceMgr device_mgr_; variable

12