Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SetStepId (0.08 sec)

  1. 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)
  2. 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)
Back to top