const int64_t* new_dims, int num_new_dims) {
  tensorflow::TensorShape s;
  for (int i = 0; i < num_new_dims; ++i) {
    TF_RETURN_IF_ERROR(s.AddDimWithStatus(new_dims[i]));
  }
  return tensor_.BitcastFrom(from.tensor_, type, s);
}

Status TensorInterface::FromProto(const tensorflow::TensorProto& from) {
  bool success = tensor_.FromProto(from);

      } else {
        dims = output_buffer->dimensions();
      }
      TensorShape tensor_shape;
      for (int i = 0; i < dims.size(); ++i) {
        TF_RETURN_IF_ERROR(tensor_shape.AddDimWithStatus(dims[i]));
      }
      if (use_pjrt_tensor_buffer) {
        TF_ASSIGN_OR_RETURN(
            Tensor output_tensor,
            MakeTensorFromPjRtBuffer(