Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for memory_space (0.19 sec)

  1. tensorflow/c/experimental/stream_executor/stream_executor.cc

      }
    
      absl::Status Init() override { return absl::OkStatus(); }
    
      DeviceMemoryBase Allocate(uint64 size, int64_t memory_space) override {
        SP_DeviceMemoryBase mem = {SP_DEVICE_MEMORY_BASE_STRUCT_SIZE};
        stream_executor_->allocate(&device_, size, memory_space, &mem);
        absl::Status status = ValidateSPDeviceMemoryBase(mem);
        if (!status.ok()) {
          LOG(ERROR) << status.message();
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor_test.cc

      SP_TimerFns timer_fns_;
      std::unique_ptr<CPlatform> cplatform_;
    };
    
    TEST_F(StreamExecutorTest, Allocate) {
      se_.allocate = [](const SP_Device* const device, uint64_t size,
                        int64_t memory_space, SP_DeviceMemoryBase* const mem) {
        mem->struct_size = SP_DEVICE_MEMORY_BASE_STRUCT_SIZE;
        mem->opaque = malloc(size);
        mem->size = size;
      };
      se_.deallocate = [](const SP_Device* const device,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 19:54:04 UTC 2024
    - 26.5K bytes
    - Viewed (0)
Back to top