Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for ExecutorForDevice (0.3 sec)

  1. tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc

      stream_executor::Platform* platform =
          stream_executor::PlatformManager::PlatformWithName("CUDA").value();
      stream_executor::StreamExecutor* executor =
          platform->ExecutorForDevice(0).value();
      TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream());
    
      se::DeviceMemoryBase gpu_dst{device_tensor.data(), 4 * sizeof(float)};
      xla::Shape shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor_internal.h

      bool ForceMemoryGrowth() const { return platform_.force_memory_growth; }
      absl::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
          int ordinal) const override;
      absl::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
      absl::StatusOr<StreamExecutor*> GetExecutor(
          const StreamExecutorConfig& config) override;
      absl::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 8K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/stream_executor/stream_executor_test.cc

      ASSERT_EQ(platform->Name(), test_util::kDeviceName);
      ASSERT_EQ(platform->VisibleDeviceCount(), test_util::kDeviceCount);
    
      absl::StatusOr<StreamExecutor*> maybe_executor =
          platform->ExecutorForDevice(0);
      TF_ASSERT_OK(maybe_executor.status());
    }
    
    TEST(StreamExecutor, NameNotSet) {
      auto plugin_init = [](SE_PlatformRegistrationParams* const params,
                            TF_Status* const status) -> void {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 19:54:04 UTC 2024
    - 26.5K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/stream_executor/stream_executor.cc

      // to be made const for it to work.
      internal::DeviceDescriptionBuilder builder;
      builder.set_name(name_);
      return builder.Build();
    }
    absl::StatusOr<StreamExecutor*> CPlatform::ExecutorForDevice(int ordinal) {
      stream_executor::StreamExecutorConfig config;
      config.ordinal = ordinal;
      return GetExecutor(config);
    }
    absl::StatusOr<StreamExecutor*> CPlatform::GetExecutor(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
Back to top