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