Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_NewOp (0.39 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                               std::vector<TensorHandlePtr>* outputs,
                               TF_Status* status) const {
      if (op_ == nullptr) {
        TFE_ContextSetExecutorForThread(context, executor_.get());
        op_.reset(TFE_NewOp(context, operation_name, status));
        if (TF_GetCode(status) != TF_OK) return;
        TFE_OpSetDevice(op_.get(), device_.c_str(), status);
        if (TF_GetCode(status) != TF_OK) return;
      } else {
    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)
Back to top