- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SetStepId (0.1 sec)
-
tensorflow/c/eager/immediate_execution_operation.h
CancellationManager* cancellation_manager) = 0; // Returns the stack trace set by `SetStackTrace` if exists. virtual absl::optional<ManagedStackTrace> GetStackTrace() = 0; virtual void SetStepId(int64_t step_id) = 0; // For LLVM style RTTI. static bool classof(const AbstractOperation* ptr) { return ptr->getKind() == kEager || ptr->getKind() == kTfrt; } protected:
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// unwrap op_ and set step_id only if valid step id value was set. // Currently only required for non-TFRT use cases, e.g., EagerOp. if (step_id_.has_value()) { tensorflow::unwrap(op_.get())->SetStepId(step_id_.value()); } TFE_Execute(op_.get(), unwrapped_results.data(), &real_num_outputs, status); if (TF_GetCode(status) != TF_OK) { cancellation_manager_->StartCancel(); return; }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0)