- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for stage (0.19 sec)
-
tensorflow/c/c_api.cc
*handle = buf; } } void TF_DeletePRunHandle(const char* handle) { delete[] handle; // TODO(suharshs): Free up any resources held by the partial run state. } void TF_SessionPRun(TF_Session* session, const char* handle, const TF_Output* inputs, TF_Tensor* const* input_values, int ninputs, const TF_Output* outputs,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
output_gradients, result, /*build_default_zeros_grads*/ false)); return absl::OkStatus(); } // Helper functions which delegate to `AbstractOperation`, update // the state of the ForwardOperation and call the tape as appropriate. // These APIs are mainly to facilitate testing and are subject to change. namespace internal { Status Reset(AbstractOperation* op_, const char* op,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
{ tensorflow::mutex_lock l(execution_mutex_); while (execution_state_ == ExecutionState::kIdle || execution_state_ == ExecutionState::kHasResult) { start_execute_.wait(l); } if (execution_state_ == ExecutionState::kShuttingDown) { return; } else if (execution_state_ == ExecutionState::kReadyToExecute) { // op_outputs_ may have been std::moved
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
absl::Status ResetPjRtClient(const DeviceType& device_type) { ResourceMgr* rmgr = tfrt_global::GetTFGlobalResourceMgr(); PjRtState* pjrt_state; TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(), kPjRtStateResourceName, &pjrt_state)); TF_RETURN_IF_ERROR(pjrt_state->MovePjRtClientToUnused(device_type)); return absl::OkStatus(); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TFE_Context* CreateContext(const std::string& serialized_server_def, bool isolate_session_state, int64_t init_timeout_in_ms) { TF_Status* status = TF_NewStatus(); TFE_ContextOptions* opts = TFE_NewContextOptions(); opts->session_options.options.config.set_isolate_session_state( isolate_session_state); TFE_ContextOptionsSetAsync(opts, static_cast<unsigned char>(false));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
TF_SetPayload(&s, tensorflow::CoordinationErrorPayloadKey().data(), error.SerializeAsString().c_str()); } *state_iter = std::move(s); ++state_iter; } status->status = absl::OkStatus(); } void TFE_WaitAtBarrier(TFE_Context* ctx, const char* barrier_id, int64_t barrier_timeout_in_ms, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.cc
downloaded_block = true; block->state = FetchState::FINISHED; } else { block->state = FetchState::ERROR; } block->cond_var.SignalAll(); return; case FetchState::FETCHING: block->cond_var.WaitWithTimeout(&block->mu, absl::Minutes(1)); if (block->state == FetchState::FINISHED) { return TF_SetStatus(status, TF_OK, "");
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 16 01:39:09 GMT 2020 - 11.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_DeleteContext(ctx); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status); } BENCHMARK(BM_InitOp); void BM_Execute(::testing::benchmark::State& state) { const int async = state.range(0); state.SetLabel(async ? "ExecuteAsync" : "Execute"); TF_Status* status = TF_NewStatus(); TFE_ContextOptions* opts = TFE_NewContextOptions();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
int32_t init_timeout_in_ms = 300000; TFE_Context* ctx_0 = CreateContext(serialized_server_def_0, /*isolate_session_state=*/false, init_timeout_in_ms); TFE_Context* ctx_1 = CreateContext(serialized_server_def_1, /*isolate_session_state=*/false, init_timeout_in_ms); // Remote device on `worker2`. const char remote_device[] = "/job:localhost/replica:0/task:2/device:CPU:0";
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
// Advance the fake timestamp so that "a" becomes stale via its first block. env->SetNowSeconds(now + 2); // The pruning thread periodically compares env->NowSeconds() with the oldest // block's timestamp to see if it should evict any files. At the current fake // timestamp of `now` + 2, file "a" is stale because its first block is stale, // but file "b" is not stale yet. Thus, once the pruning thread wakes up (in
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K bytes - Viewed (0)