Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for BufferFromHostBuffer (0.22 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc 

      std::vector<int32_t> data(1, 0);
  xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1});

  auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer(
      data.data(), shape.element_type(), shape.dimensions(),
      /*byte_strides=*/std::nullopt,
      xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, nullptr,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Mon Oct 30 19:20:20 GMT 2023
    - 7.2K bytes
    - Viewed (0)
Back to top