Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TF_ASSERT_OK_AND_ASSIGN (0.42 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc 

      tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1});
  TF_ASSERT_OK_AND_ASSIGN(
      auto pjrt_client,
      xla::GetTfrtCpuClient(/*asynchronous=*/true, /*cpu_device_count=*/1));
  std::vector<int32_t> data(1, 0);
  xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1});
  TF_ASSERT_OK_AND_ASSIGN(
      auto buffer,
      pjrt_client->BufferFromHostBuffer(
    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