Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for RegisterLoggingDevice (0.21 sec)

  1. tensorflow/c/eager/custom_device_test.cc

      bool arrived = false;
      bool executed = false;
      RegisterLoggingDevice(context.get(), custom0,
                            /*strict_scope_placement=*/false, &arrived, &executed,
                            status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      RegisterLoggingDevice(context.get(), custom1,
                            /*strict_scope_placement=*/true, &arrived, &executed,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/custom_device_testutil.h

    #include <memory>
    
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/tf_status.h"
    
    void RegisterLoggingDevice(TFE_Context* context, const char* name,
                               bool strict_scope_placement, bool* arrived_flag,
                               bool* executed_flag, TF_Status* status);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 1.6K bytes
    - Viewed (0)
  3. tensorflow/c/eager/custom_device_testutil.cc

      }
      *(dev->executed_flag) = true;
    }
    
    void DeleteLoggingDevice(void* device_info) {
      delete reinterpret_cast<LoggingDevice*>(device_info);
    }
    
    }  // namespace
    
    void RegisterLoggingDevice(TFE_Context* context, const char* name,
                               bool strict_scope_placement, bool* arrived_flag,
                               bool* executed_flag, TF_Status* status) {
      TFE_CustomDevice custom_device;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
Back to top