Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc

          data.data(), shape.element_type(), shape.dimensions(),
          /*byte_strides=*/std::nullopt,
          xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, nullptr,
          c_api_client->pjrt_c_client()->client->addressable_devices()[0]);
      CHECK_OK(buffer.status());
    
      return new PJRT_Buffer{std::move(*buffer), c_api_client->pjrt_c_client()};
    }
    
    TEST(TensorPjRtBufferUtilTest, GetPjRtCBufferFromTensorNoBuffer) {
    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