- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for create_stream (0.18 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
int64_t total = 0; executor->DeviceMemoryUsage(&free, &total); ASSERT_EQ(free, 45); ASSERT_EQ(total, 7); } TEST_F(StreamExecutorTest, CreateStream) { static bool stream_created = false; static bool stream_deleted = false; se_.create_stream = [](const SP_Device* const device, SP_Stream* stream, TF_Status* const status) -> void { *stream = new SP_Stream_st(14);
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_test_util.cc
se->host_memory_allocate = HostMemoryAllocate; se->host_memory_deallocate = HostMemoryDeallocate; se->get_allocator_stats = GetAllocatorStats; se->device_memory_usage = DeviceMemoryUsage; se->create_stream = CreateStream; se->destroy_stream = DestroyStream; se->create_stream_dependency = CreateStreamDependency; se->get_stream_status = GetStreamStatus; se->create_event = CreateEvent; se->destroy_event = DestroyEvent;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 21 20:09:00 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_internal.h
parent()->DeallocateStream(this); Destroy(); } absl::Status Create() { tensorflow::TF_StatusPtr c_status(TF_NewStatus()); stream_executor_->create_stream(device_, &stream_handle_, c_status.get()); absl::Status s = tensorflow::StatusFromTF_Status(c_status.get()); return s; } void Destroy() { if (stream_handle_ != nullptr) {
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.cc
TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, unified_memory_deallocate); } TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, device_memory_usage); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, create_stream); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, destroy_stream); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, create_stream_dependency); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, get_stream_status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.h
/*** STREAM CALLBACKS ***/ // Creates SP_Stream. This call should also allocate stream // resources on the underlying platform and initializes its // internals. void (*create_stream)(const SP_Device* device, SP_Stream* stream, TF_Status* status); // Destroys SP_Stream and deallocates any underlying resources.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
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; TF_ASSERT_OK(TensorShapeToXLAShape(DT_FLOAT, TensorShape({2, 2}), &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) -
staging/src/k8s.io/apimachinery/pkg/util/httpstream/spdy/connection.go
delete(c.streams, stream.Identifier()) } } c.streamLock.Unlock() } // CreateStream creates a new stream with the specified headers and registers // it with the connection. func (c *connection) CreateStream(headers http.Header) (httpstream.Stream, error) { stream, err := c.conn.CreateStream(headers, nil, false) if err != nil { return nil, err }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Jun 01 15:04:07 UTC 2021 - 6.7K bytes - Viewed (0) -
staging/src/k8s.io/apimachinery/pkg/util/httpstream/spdy/connection_test.go
} spdyConn, err := NewClientConnection(conn) if err != nil { t.Fatalf("client: error creating spdy connection: %v", err) } if _, err := spdyConn.CreateStream(http.Header{}); err != nil { t.Fatalf("client: error creating stream: %v", err) } spdyConn.Close() raw := spdyConn.(*connection).conn if err := raw.Wait(15 * time.Second); err != nil {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Mar 01 11:58:57 UTC 2023 - 7.9K bytes - Viewed (0) -
staging/src/k8s.io/apimachinery/pkg/util/httpstream/httpstream.go
} // Connection represents an upgraded HTTP connection. type Connection interface { // CreateStream creates a new Stream with the supplied headers. CreateStream(headers http.Header) (Stream, error) // Close resets all streams and closes the connection. Close() error // CloseChan returns a channel that is closed when the underlying connection is closed.
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Oct 23 22:33:38 UTC 2023 - 6.4K bytes - Viewed (0) -
pkg/kubelet/server/server_test.go
defer conn.Close() h := http.Header{} h.Set(api.StreamType, api.StreamTypeError) _, err = conn.CreateStream(h) require.NoError(t, err, "creating error stream") if test.stdin { h.Set(api.StreamType, api.StreamTypeStdin) stream, err := conn.CreateStream(h) require.NoError(t, err, "creating stdin stream") _, err = stream.Write([]byte(expectedStdin))
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 51.5K bytes - Viewed (0)