Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SP_DeviceFns (0.36 sec)

  1. tensorflow/c/experimental/stream_executor/stream_executor.cc

      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    absl::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) {
      TF_VALIDATE_STRUCT_SIZE(SP_DeviceFns, device_fns, SP_DEVICE_FNS_STRUCT_SIZE);
      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor.h

      // Negative values are treated as "unset".
      double (*get_gflops)(const SP_Device* device);
    } SP_DeviceFns;
    
    #define SP_DEVICE_FNS_STRUCT_SIZE TF_OFFSET_OF_END(SP_DeviceFns, get_gflops)
    
    typedef struct SE_CreateDeviceFnsParams {
      size_t struct_size;
      void* ext;  // reserved for future use
    
      SP_DeviceFns* device_fns;  // output, to be filled by plugin
    } SE_CreateDeviceFnsParams;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 24 08:40:35 UTC 2022
    - 21.6K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/stream_executor/stream_executor_test_util.h

    constexpr int kDeviceCount = 2;
    constexpr char kDeviceName[] = "MY_DEVICE";
    constexpr char kDeviceType[] = "GPU";
    
    void PopulateDefaultStreamExecutor(SP_StreamExecutor* se);
    void PopulateDefaultDeviceFns(SP_DeviceFns* device_fns);
    void PopulateDefaultTimerFns(SP_TimerFns* timer_fns);
    void PopulateDefaultPlatform(SP_Platform* platform,
                                 SP_PlatformFns* platform_fns);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 17 01:32:30 UTC 2021
    - 1.9K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc

      se->synchronize_all_activity = SynchronizeAllActivity;
      se->host_callback = HostCallback;
      se->mem_zero = MemZero;
      se->memset = Memset;
      se->memset32 = Memset32;
    }
    
    void PopulateDefaultDeviceFns(SP_DeviceFns* device_fns) {
      *device_fns = {SP_DEVICE_FNS_STRUCT_SIZE};
    }
    
    /*** Functions for creating SP_TimerFns ***/
    uint64_t Nanoseconds(SP_Timer timer) { return timer->timer_id; }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 21 20:09:00 UTC 2023
    - 8.7K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/stream_executor/stream_executor_internal.h

                         void (*destroy_platform)(SP_Platform*),
                         SP_PlatformFns platform_fns,
                         void (*destroy_platform_fns)(SP_PlatformFns*),
                         SP_DeviceFns device_fns, SP_StreamExecutor stream_executor,
                         SP_TimerFns timer_fns);
      ~CPlatform() override;
    
      Id id() const override { return const_cast<int*>(&plugin_id_value_); }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 8K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/stream_executor/stream_executor_test.cc

            cplatform_->ExecutorForDevice(ordinal);
        TF_CHECK_OK(maybe_executor.status());
        return std::move(maybe_executor).value();
      }
      SP_Platform platform_;
      SP_PlatformFns platform_fns_;
      SP_DeviceFns device_fns_;
      SP_StreamExecutor se_;
      SP_TimerFns timer_fns_;
      std::unique_ptr<CPlatform> cplatform_;
    };
    
    TEST_F(StreamExecutorTest, Allocate) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 19:54:04 UTC 2024
    - 26.5K bytes
    - Viewed (0)
Back to top