Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CustomDevice (0.06 sec)

  1. tensorflow/c/eager/immediate_execution_context.h

    #include "tensorflow/core/platform/tstring.h"
    #include "tensorflow/core/protobuf/config.pb.h"
    #include "tensorflow/core/util/device_name_utils.h"
    
    namespace tensorflow {
    class EagerExecutor;
    class EagerContext;
    class CustomDevice;
    class CustomDeviceOpHandler;
    class Device;
    
    // LINT.IfChange
    // Note: Keep in sync with exported copy of enum in eager/c_api.h.
    enum ContextDevicePlacementPolicy {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

        return nullptr;
      }
      return const_cast<void*>(
          static_cast<const void*>(tensor->tensor_data().data()));
    }
    
    namespace tensorflow {
    namespace {
    class CustomDeviceAPI : public tensorflow::CustomDevice {
     public:
      CustomDeviceAPI(TFE_Context* context, TFE_CustomDevice device, void* info,
                      string name)
          : context_(context), device_(device), info_(info), name_(name) {}
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top