Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SetStepId (0.16 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:
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Sep 26 22:40:32 GMT 2022
    - 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;
      }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
Back to top