- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for EagerContext (0.26 sec)
-
tensorflow/compiler/jit/get_compiler_ir.cc
} return absl::OkStatus(); } absl::StatusOr<std::vector<XlaCompiler::Argument>> PrepareXlaCompilerArgs( FunctionLibraryRuntime* flr, const NameAttrList& function, EagerContext* context, Device* dev, absl::Span<const ArgShapeAndDType> input_arg_shape_and_dtype, absl::Span<const TensorHandle* const> input_handles, CompilerArgSource compiler_arg_source) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
auto r = tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>( new tensorflow::IntraProcessRendezvous(device_mgr.get())); tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( opts->session_options.options, static_cast<tensorflow::ContextDevicePlacementPolicy>( opts->device_placement_policy), opts->async, device_mgr.release(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
auto* gpu_options = config.mutable_gpu_options(); gpu_options->set_allow_growth(gpu_memory_allow_growth); (*config.mutable_device_count())["CPU"] = num_cpu_devices; // TODO(b/113217601): This is needed for EagerContext::runner_ to use a // threadpool, so that we avoid the possibility of running the runner_ in the // threadpool of GPU event mgr, as that can trigger more callbacks to be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
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/c/eager/immediate_execution_context.h
#include "tensorflow/core/platform/tstring.h" #include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { class EagerExecutor; class EagerContext; class CustomDevice; class CustomDeviceOpHandler; class Device; // LINT.IfChange // Note: Keep in sync with exported copy of enum in eager/c_api.h. enum ContextDevicePlacementPolicy {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 06 08:34:00 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
if (!s.ok()) { tsl::Set_TF_Status_from_Status(status, s); return "// error"; } const std::string& function_name = functiondef.signature().name(); EagerContext* cpp_context = ContextFromInterface(unwrap(tfe_context)); FunctionLibraryDefinition& flib_def = *cpp_context->FuncLibDef(); const tensorflow::FunctionDef* fdef = flib_def.Find(function_name); if (fdef == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
} void TFE_ContextDisableGraphCollection(TFE_Context* ctx) { tensorflow::unwrap(ctx)->SetShouldStoreGraphs(false); } uint64_t TFE_GetContextId(TFE_Context* ctx) { tensorflow::EagerContext* context = tensorflow::ContextFromInterface(tensorflow::unwrap(ctx)); return context->GetContextId(); } void TFE_MonitoringCounterCellIncrementBy(TFE_MonitoringCounterCell* cell,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 23:52:39 UTC 2024 - 35.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
// Aborts all ongoing collectives with the specified status. After abortion, // subsequent collectives will error with this status immediately. To reset the // collectives, create a new EagerContext. // // This is intended to be used when a peer failure is detected. TF_CAPI_EXPORT extern void TFE_AbortCollectiveOps(TFE_Context* ctx, TF_Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// Sets whether to use TFRT TF_CAPI_EXPORT extern void TFE_ContextOptionsSetTfrt(TFE_ContextOptions*, bool use_tfrt); // Returns the context_id from the EagerContext which is used by the // EagerService to maintain consistency between client and worker. The // context_id is initialized with a dummy value and is later set when the worker
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0)