Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_GetExecutedOpNames (0.24 sec)

  1. tensorflow/c/eager/c_api_experimental.cc

    TF_CAPI_EXPORT extern void TFE_TensorHandleGetStatus(TFE_TensorHandle* h,
                                                         TF_Status* status) {
      status->status = tensorflow::unwrap(h)->TensorHandleStatus();
    }
    
    void TFE_GetExecutedOpNames(TFE_Context* ctx, TF_Buffer* buf,
                                TF_Status* status) {
      const std::vector<std::string>& op_names =
          tensorflow::unwrap(ctx)->GetLoggedOpsTestonly();
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  2. tensorflow/c/eager/c_api_experimental.h

    // Get a comma-separated list of op names executed in graph functions dispatched
    // to `ctx`. This feature is currently only enabled for TFRT debug builds, for
    // performance and simplicity reasons.
    TF_CAPI_EXPORT extern void TFE_GetExecutedOpNames(TFE_Context* ctx,
                                                      TF_Buffer* buf,
                                                      TF_Status* status);
    
    // Set logical devices to the context's device manager.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test.cc

      // This capability of checking the executed op names is currently only enabled
      // for TFRT debug build, for performance and simplicity reasons.
      if (use_tfrt) {
        TF_Buffer* buf = TF_NewBuffer();
        TFE_GetExecutedOpNames(ctx, buf, status);
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    #ifndef NDEBUG
        if (enable_grappler)
          EXPECT_NE(strstr(static_cast<const char*>(buf->data), "tf.Mul"), nullptr);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top