Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for TensorShape (0.22 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.cc

                                       const int num_dims) {
      if (num_dims > TensorShape::MaxDimensions()) {
        return errors::InvalidArgument("Value specified for `", attr_name, "` has ",
                                       num_dims,
                                       " dimensions which is over the limit of ",
                                       TensorShape::MaxDimensions(), ".");
      }
      TensorShapeProto proto;
      if (num_dims < 0) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  2. tensorflow/c/checkpoint_reader.cc

        TF_Status* out_status) const {
      Status status;
      if (reader_ != nullptr) {
        status = reader_->GetTensor(name, out_tensor);
      } else {
        tensorflow::DataType dtype;
        tensorflow::TensorShape shape;
        status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
        if (status.ok()) {
          out_tensor->reset(new Tensor(dtype, shape));
          status = v2_reader_->Lookup(name, out_tensor->get());
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

                        ForwardOperation* forward_op_) {
      if (num_dims > TensorShape::MaxDimensions()) {
        return errors::InvalidArgument("Value specified for `", attr_name, "` has ",
                                       num_dims,
                                       " dimensions which is over the limit of ",
                                       TensorShape::MaxDimensions(), ".");
      }
      TensorShapeProto proto;
      if (num_dims < 0) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

      auto dtype = static_cast<tensorflow::DataType>(data_type);
      DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype));
    
      tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({}));
      std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len);
    
      status->status = absl::OkStatus();
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_test.cc

      Status status;
      for (const std::vector<int64_t>& dims :
           std::vector<std::vector<int64_t>>{{n}, {1, n}, {n, 1}, {n / 2, 2}}) {
        // Create C++ Tensor
        Tensor src(tensorflow::DT_STRING, TensorShape(dims));
        for (int64_t i = 0; i < src.NumElements(); ++i) {
          src.flat<tstring>()(i) = data[i];
        }
        TF_Tensor* dst = TF_TensorFromTensor(src, &status);
        ASSERT_TRUE(status.ok()) << status.message();
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  6. tensorflow/c/eager/c_api.cc

          new TF_ManagedBuffer(data, len, deallocator, deallocator_arg,
                               /*owns_memory=*/false);
    
      tensorflow::Tensor t(static_cast<tensorflow::DataType>(dtype),
                           tensorflow::TensorShape(dimvec), buf);
      buf->Unref();
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(
          std::move(t), device, device, context));
    }
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  7. tensorflow/c/c_api.h

    // * int is used as an index into arrays.
    // * Deletion functions are safe to call on nullptr.
    //
    // Questions left to address:
    // * Might at some point need a way for callers to provide their own Env.
    // * Maybe add TF_TensorShape that encapsulates dimension info.
    //
    // Design decisions made:
    // * Backing store for tensor memory has an associated deallocation
    //   function.  This deallocation function will point to client code
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  8. tensorflow/c/c_api.cc

    }
    
    // Create an empty tensor of type 'dtype'. 'shape' can be arbitrary, but has to
    // result in a zero-sized tensor.
    static TF_Tensor* EmptyTensor(TF_DataType dtype,
                                  const tensorflow::TensorShape& shape) {
      static char empty;
      int64_t nelems = 1;
      std::vector<int64_t> dims;
      dims.reserve(shape.dims());
      for (int i = 0; i < shape.dims(); ++i) {
        dims.push_back(shape.dim_size(i));
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  9. RELEASE.md

    *   Adding TensorFlow-iOS CocoaPod for symmetry with tensorflow-android.
    *   Introduces base implementations of ClusterResolvers.
    *   Unify memory representations of TensorShape and PartialTensorShape. As a
        consequence, tensors now have a maximum of 254 dimensions, not 255.
    *   Changed references to LIBXSMM to use version 1.8.1.
    *   TensorFlow Debugger (tfdbg):
    Plain Text
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Apr 03 20:27:38 GMT 2024
    - 727.4K bytes
    - Viewed (8)
Back to top