- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for EagerContext (0.31 sec)
-
tensorflow/compiler/jit/get_compiler_ir.h
#include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/core/platform/statusor.h" namespace tensorflow { class ProcessFunctionLibraryRuntime; class Device; class Tensor; class TensorHandle; class EagerContext; enum class IrExportStage { HLO, HLO_NO_METADATA, HLO_SERIALIZED, OPTIMIZED_HLO, OPTIMIZED_HLO_SERIALIZED, OPTIMIZED_HLO_PROTO_SERIALIZED, OPTIMIZED_HLO_DOT };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops_test.cc
#include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { ImmediateTensorHandlePtr CreateScalarTensorHandle(EagerContext* context, float value) { AbstractTensorPtr tensor(context->CreateFloatScalar(value)); ImmediateTensorHandlePtr handle(context->CreateLocalHandle(tensor.get()));
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/internal/saved_model_api.cc
} else { std::unique_ptr<tensorflow::TFSavedModelAPI> saved_model; status->status = tensorflow::TFSavedModelAPI::Load( dirname, absl::nullopt, tensorflow::down_cast<tensorflow::EagerContext*>( tensorflow::unwrap(ctx)), &saved_model); result = std::move(saved_model); } if (!status->status.ok()) { return nullptr; } return tensorflow::wrap(result.release());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 04 00:45:47 UTC 2020 - 4.7K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/revived_types/variable.cc
new Variable(ctx, dtype, shape, std::move(name), std::move(handle))); return Status(); } if (!tensorflow::isa<EagerContext>(ctx)) { return errors::InvalidArgument( "Can only load distributed variables with EagerContext."); } EagerContext* eager_ctx = reinterpret_cast<EagerContext*>(ctx); std::vector<TensorHandle*> handles; for (const auto& device : component_devices) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 08 20:55:40 UTC 2020 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.h
#include "tensorflow/core/platform/types.h" namespace tensorflow { namespace testing { // Creates a DeviceMgr suitable for local tests. std::unique_ptr<StaticDeviceMgr> CreateTestingDeviceMgr(); // 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>.
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/mlir/tfr/integration/node_expansion_pass.cc
LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes"; // Get the FunctionDef and insert that into the context const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); auto& ctx = orig_op->EagerContext(); Fprint128 cache_key = orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); // Include soft placement policy in cache key since the placement strategy // can change and thus affect which kernel is picked.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_variable_loading_test.cc
options, "/job:localhost/replica:0/task:0", &devices)); device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); ctx_ = testing::CreateTestingEagerContext(device_mgr_.get()); } EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_; EagerContextPtr ctx_; }; // Sanity check that constructing a tensorflow::Variable from a SavedVariable
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/revived_types/flat_tensor_function.h
// FlatTensorFunction models a TF2 eager runtime view of a callable function, // taking + returning flat lists of tensors, including any captures. // Effectively, it is a thin wrapper around a FunctionDef owned by the // EagerContext, and any TensorHandle captures associated with the function. The // MakeCallOp method handles the logic of marshaling captures after the user // provided inputs automatically.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 22 21:03:41 UTC 2020 - 3.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/restore_ops_test.cc
} class RestoreOpsTest : public ::testing::Test { public: RestoreOpsTest() : device_mgr_(testing::CreateTestingDeviceMgr()), ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_; EagerContextPtr ctx_; }; // One way of obtaining the checkpointa checkpoint's tensor names is:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 4.2K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/constant_loading_test.cc
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)