Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetPjRtCBufferFromTensor (0.17 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc 

    #include "tensorflow/core/tfrt/common/pjrt_util.h"
#include "tsl/platform/errors.h"
#include "tsl/platform/statusor.h"

namespace tensorflow {

absl::StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor) {
  tensorflow::AsyncValueTensor* av_tensor =
      tensorflow::AsyncValueTensor::FromTensor(tensor);
  if (av_tensor == nullptr || av_tensor->GetBuffer() == nullptr) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Mon Oct 30 19:20:20 GMT 2023
    - 3.7K bytes
    - Viewed (0)
Back to top