- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for device_mgr_ (0.14 sec)
-
tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc
public: SavedConcreteFunctionLoadingTest() : device_mgr_(testing::CreateTestingDeviceMgr()), ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_; EagerContextPtr ctx_; }; class DummyCapture : public TensorHandleConvertible { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
/*graph=*/nullptr); context_->Ref(); device_mgr_ = CreateDeviceMgr(device_type_); if (!device_mgr_) return failure(); // Type of params_.device is DeviceBase* so store it as Device* to access // derived class method. device_ = device_mgr_->ListDevices().front(); params_.device = device_; params_.resource_manager = device_->resource_manager();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, flr, fld, shape_inference_graphs, has_outside_compilation); } private: std::unique_ptr<DeviceMgr> device_mgr_; std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; }; TEST_F(ExtractOutsideCompilationForFunctionTest, Basic) { // Build the XLA computation func. // "const0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
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/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)