Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ShallPinToThisDevice (0.19 sec)

  1. tensorflow/c/eager/c_api.cc

                                                  tensorflow::wrap(handles.data()),
                                                  handles.size(), &status, info_));
        return status.status;
      }
    
      absl::StatusOr<bool> ShallPinToThisDevice(
          const ImmediateExecutionOperation* op) override {
        TF_Status status;
        // Let this custom device choose the device to pin this op on if it
        // implements the pinning function.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
Back to top