Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for create_stream (0.18 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top