- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 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_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/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/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) -
staging/src/k8s.io/apimachinery/pkg/util/httpstream/spdy/roundtripper_test.go
return } defer conn.Close() if resp.StatusCode != http.StatusSwitchingProtocols { t.Fatalf("expected http 101 switching protocols, got %d", resp.StatusCode) } stream, err := conn.CreateStream(http.Header{}) if err != nil { t.Fatalf("error creating client stream: %s", err) } n, err := stream.Write([]byte("hello")) if err != nil { t.Fatalf("error writing to stream: %s", err)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Oct 23 22:33:38 UTC 2023 - 32.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
// pool to avoid potential lifetime issues. TF_ASSIGN_OR_RETURN( stream_, backend->stream_executors()[device_ordinal_]->CreateStream()); TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "stream", &stream_, &need_new_device_context)); (*global_compute_streams_)[device_ordinal_] = stream_; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
staging/src/k8s.io/apiserver/pkg/util/proxy/streamtunnel_test.go
randomSize := 1024 * 1024 randomData := make([]byte, randomSize) _, err = rand.Read(randomData) require.NoError(t, err) var actual []byte go func() { clientStream, err := spdyClient.CreateStream(http.Header{}) require.NoError(t, err) _, err = io.Copy(clientStream, bytes.NewReader(randomData)) require.NoError(t, err) clientStream.Close() //nolint:errcheck }() select {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Mar 07 02:21:50 UTC 2024 - 22.7K bytes - Viewed (0)