- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for SetStepId (0.19 sec)
-
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)