- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 35 for device_mgr (0.24 sec)
-
tensorflow/c/experimental/saved_model/core/test_utils.cc
DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0")); } EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr) { return EagerContextPtr(new EagerContext( SessionOptions(), tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, /* async= */ false, device_mgr, /* device_mgr_owned= */ false, /* rendezvous= */ nullptr, /* cluster_flr= */ nullptr)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:13:32 UTC 2023 - 5.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.h
// Creates an EagerContext suitable for local tests. Does not take ownership // of `device_mgr`. EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr); // Converts a tensorflow::DatatypeSet to std::vector<DataType>. // This is useful for tests using GTest's ::testing::ValuesIn, since // DataTypeSet doesn't fullfill all the constraints of an STL-like iterable. std::vector<DataType> DataTypeSetToVector(DataTypeSet set);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.h
Device* GetDevice(const string& device_name); FunctionLibraryRuntime* flr() { return flr_; } private: FunctionLibraryRuntime* flr_; std::unique_ptr<DeviceMgr> device_mgr_; std::unique_ptr<FunctionLibraryDefinition> lib_def_; std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; }; } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/fake_session.h
const tensorflow::thread::ThreadPoolOptions& thread_pool_options) override; private: void InitVariables(); void BuildDeviceManager(); void Initialize(); std::unique_ptr<tensorflow::DeviceMgr> device_mgr_; bool initialized_ = false; }; } // namespace test_util } // namespace TF } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops_test.cc
} class VariableOpsTest : public ::testing::Test { public: VariableOpsTest() : device_mgr_(testing::CreateTestingDeviceMgr()), ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_; EagerContextPtr ctx_; }; // Sanity check for variable creation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 21 19:26:54 UTC 2020 - 3.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/constant_loading_test.cc
std::tuple<DataType, std::vector<int64_t>, bool>> { public: ConstantTest() : device_mgr_(testing::CreateTestingDeviceMgr()), ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_; EagerContextPtr ctx_; }; // Basic sanity check that roundtripping a Tensor->Tensorproto->Constant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc
#include "llvm/ADT/SmallSet.h" #include "llvm/ADT/StringRef.h" #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/utils/string_container_utils.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/framework/device.h" #include "tensorflow/core/framework/resource_var.h" namespace mlir { namespace tf_saved_model { std::string GetVariableName(TF::VarHandleOp var_handle_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables_test_pass.h
#include "mlir/Pass/PassManager.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/framework/resource_mgr.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/status.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 21 15:49:06 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc
#include "mlir/Pass/Pass.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.h" #include "tensorflow/compiler/mlir/tensorflow/utils/session_utils.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/framework/device_base.h" #include "tensorflow/core/framework/rendezvous.h" #include "tensorflow/core/framework/resource_var.h" #include "tensorflow/core/framework/tensor.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 19:14:56 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.h
mlir::PatternRewriter& rewriter_; tensorflow::OpOrArgLocNameMapper name_mapper_; tensorflow::XlaContext* context_; // Ref-counted. std::unique_ptr<tensorflow::StaticDeviceMgr> device_mgr_; tensorflow::Device* device_; // Owned by device_mgr_; std::unique_ptr<tensorflow::ScopedStepContainer> step_container_; std::unique_ptr<tensorflow::FunctionLibraryDefinition> flib_def_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 5K bytes - Viewed (0)