- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 34 for shared_ptr (0.18 sec)
-
tensorflow/compiler/jit/xla_device.h
// If use_multiple_streams_, host to device transfers are performed using this // stream. std::shared_ptr<se::Stream> host_to_device_stream_ TF_GUARDED_BY(mu_); // If use_multiple_streams_, transfers between different devices are performed // using these streams. std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_ TF_GUARDED_BY(mu_); // See comments in options.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/revived_types/restored_resource.h
// the RevivedObjects struct (which owns all functions). One alternative would // be to have RevivedObjects store shared_ptr<TFConcreteFunction> instead, and // change RestoredResource's constructor take shared_ptr<TFConcreteFunction>. // To keep things simple, I've stuck to raw pointers for now. // // Params: // device - The device string associated with the SavedResource
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 23 04:49:47 UTC 2020 - 4.6K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.h
typedef std::map<Key, std::shared_ptr<Block>> BlockMap; /// Prune the cache by removing files with expired blocks. void Prune() ABSL_LOCKS_EXCLUDED(mu_); bool BlockNotStale(const std::shared_ptr<Block>& block) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Look up a Key in the block cache. std::shared_ptr<Block> Lookup(const Key& key) ABSL_LOCKS_EXCLUDED(mu_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 31 04:46:34 UTC 2020 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.h
// streams. // It is legal to reset the definition event of a tensor when overwriting the // tensor's value (at which point, it is effectively a new tensor once again.) void ResetDefinitionEvent(std::shared_ptr<se::Event> event, se::Stream* stream); // Refresh the status of streams_defined_on_. Return the first not-OK stream's // status or OK. Status RefreshStatusOfStreams();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
DeviceType compilation_device_type = DeviceType(DEVICE_GPU_XLA_JIT); se::Platform::Id platform_id = nullptr; auto xla_device_metadata = CreateXlaDeviceMetadata(compilation_device_type); std::shared_ptr<se::DeviceMemoryAllocator> custom_allocator; XlaPlatformInfo platform_info( compilation_device_type, platform_id, xla_device_metadata.get(), /*pjrt_device_metadata=*/nullptr, custom_allocator);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/cc/client/client_session.cc
class ClientSession::Impl { private: friend class ClientSession; Impl(Session* session, std::shared_ptr<Graph> graph) : session_(session), graph_(std::move(graph)) {} static SessionOptions MakeDefaultSessionOptions(const string& target); Status MaybeExtendGraph() const; std::unique_ptr<Session> session_; std::shared_ptr<Graph> graph_; mutable mutex mu_; mutable int last_num_graph_nodes_ TF_GUARDED_BY(mu_) = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:04:10 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
/*owned_args=*/{}, &non_donatable_input_indices)); EXPECT_EQ(exec_args.size(), 2); std::shared_ptr<xla::Literal> literal1 = *exec_args[0]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal( *literal1, xla::LiteralUtil::CreateR2<int32_t>({{1, 2, 3}}))); std::shared_ptr<xla::Literal> literal2 = *exec_args[1]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
/*owned_args=*/{}, &non_donatable_input_indices)); EXPECT_EQ(exec_args.size(), 2); std::shared_ptr<xla::Literal> literal1 = *exec_args[0]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal( *literal1, xla::LiteralUtil::CreateR2<int32_t>({{1, 2, 3}}))); std::shared_ptr<xla::Literal> literal2 = *exec_args[1]->ToLiteralSync(); EXPECT_TRUE(xla::LiteralTestUtil::Equal(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.cc
return; } stream->WaitFor(definition_event_.get()).IgnoreError(); streams_defined_on_.push_back(stream); } void XlaTensor::ResetDefinitionEvent(std::shared_ptr<se::Event> event, se::Stream* stream) { mutex_lock lock(mu_); definition_event_ = std::move(event); streams_defined_on_ = {stream}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.h
TF_FileStatistics base; int64_t generation_number; } GcsFileStat; typedef struct GCSFile { google::cloud::storage::Client gcs_client; // owned bool compose; absl::Mutex block_cache_lock; std::shared_ptr<RamFileBlockCache> file_block_cache ABSL_GUARDED_BY(block_cache_lock); uint64_t block_size; // Reads smaller than block_size will trigger a read // of block_size.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 31 04:37:41 UTC 2020 - 5.2K bytes - Viewed (0)