- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 30 for device_mgr_ (0.26 sec)
-
tensorflow/c/eager/c_api.cc
&devices); if (!status->status.ok()) return nullptr; std::unique_ptr<tensorflow::DeviceMgr> device_mgr( new tensorflow::DynamicDeviceMgr(std::move(devices))); auto r = tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>( new tensorflow::IntraProcessRendezvous(device_mgr.get())); tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( opts->session_options.options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
#include "tensorflow/compiler/tf2xla/functionalize_control_flow.h" #include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/grappler/clusters/virtual_cluster.h" #include "tensorflow/core/grappler/grappler_item.h" #include "tensorflow/core/grappler/grappler_item_builder.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
#include "xla/client/xla_builder.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/common_runtime/process_function_library_runtime.h" #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/function.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
return CreateSessionInitFunc(module); } } } // namespace LogicalResult InitializeVariablesInSessionInitializer( ModuleOp module, tensorflow::Session* session) { const tensorflow::DeviceMgr* mgr = nullptr; auto status = session->LocalDeviceManager(&mgr); if (!status.ok()) { module->emitError( absl::StrCat("failed to fetch device manager: ", status.message())); return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/c/c_api_internal.h
#include "tensorflow/core/platform/mutex.h" #include "tensorflow/core/platform/status.h" #include "tensorflow/core/platform/types.h" #include "tensorflow/core/public/session.h" namespace tensorflow { class Device; class DeviceMgr; class ServerInterface; } // namespace tensorflow // Internal structures used by the C API. These are likely to change and should // not be depended on. struct TF_SessionOptions {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat May 13 00:49:12 UTC 2023 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
} // Returns ElementsAttr which has the value held by 'resource_tensor'. ElementsAttr GetTensorValueAsElementsAttr( TF::VarHandleOp var_handle_op, const tensorflow::Tensor& resource_tensor, const tensorflow::DeviceMgr* mgr, OpBuilder builder) { if (resource_tensor.dtype() != tensorflow::DT_RESOURCE) { return GetTensorValueAsElementsAttr(resource_tensor, builder); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
} std::unique_ptr<DeviceMgr> device_mgr = std::make_unique<StaticDeviceMgr>(std::move(devices)); const auto* config = &options.session_options->config; std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( new ProcessFunctionLibraryRuntime( device_mgr.get(), options.session_options->env, /*config=*/config, TF_GRAPH_DEF_VERSION, library,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
TF_CHECK_OK(DeviceFactory::AddDevices( session_options, "/job:localhost/replica:0/task:0", &devices)); OptimizerOptions opts; auto device_mgr = std::make_unique<StaticDeviceMgr>(std::move(devices)); auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( device_mgr.get(), Env::Default(), /*config=*/nullptr, TF_GRAPH_DEF_VERSION, lib_def.get(), opts, /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
#include "tensorflow/c/c_api_internal.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/c/tf_status_internal.h" #include "tensorflow/c/tf_tensor.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/common_runtime/eval_const_tensor.h" #include "tensorflow/core/common_runtime/graph_constructor.h" #include "tensorflow/core/common_runtime/shape_refiner.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
RELEASE.md
params after the creation of Estimator, leading to potentially undefined behavior. * Added None check for save_path in `saver.restore`. * Register devices under their legacy names in device_mgr to ease the transition to clusterspec-propagated configurations. * VectorExponential added to distributions. * Add a bitwise module with bitwise_and, bitwise_or, bitwise_xor, and invert functions.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)