- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ExecutorForDevice (0.3 sec)
-
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) -
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) -
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) -
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)