Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for const_op (0.14 sec)

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

        std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> const_op(
            TFE_NewOp(context, "Const", status), TFE_DeleteOp);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetDevice(const_op.get(), underlying_devices_[device_index].c_str(),
                        status);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetAttrTensor(const_op.get(), "value", tensor.get(), status);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

      // scheduled on that same threadpool, causing a deadlock in cases where the
      // caller of event_mgr->ThenExecute() blocks on the completion of the callback
      // (as in the case of ConstOp kernel creation on GPU, which involves copying a
      // CPU tensor to GPU).
      // Setting a larger thread pool does not help with the Swift caller, as we use
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
Back to top