Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for dim (0.03 sec)

  1. tensorflow/c/eager/c_api.cc

      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    
      int64_t dim = -1;
      status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim);
      return dim;
    }
    
    const char* TFE_TensorHandleDeviceName(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.h

    // Information about the shape of a Tensor and its type.
    struct TF_ShapeAndType {
      // Number of dimensions. -1 indicates unknown rank.
      int num_dims;
      // Array of dimensions. -1 indicates unknown dim.
      int64_t* dims;
      // The data type. May be 0 to denote unknown type.
      TF_DataType dtype;
    };
    
    typedef struct TF_ShapeAndType TF_ShapeAndType;
    
    // A list of TF_ShapeAndType elements..
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

          shape.dims = nullptr;
          continue;
        }
        shape.dims = new int64_t[shape.num_dims];
        for (size_t j = 0; j < shape.num_dims; ++j) {
          shape.dims[j] = c.Value(c.Dim(shape_handle, j));
        }
      }
      if (output_shapes != nullptr) *output_shapes = output_shapes_result;
    
      // TODO(bgogul): Set output_resource_shapes_and_types.
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device.cc

      ParallelTensor* t_released = t.release();
      TFE_CustomDeviceTensorHandleMethods handle_methods;
      handle_methods.num_dims = &ParallelTensorNumDims;
      handle_methods.dim = &ParallelTensorDim;
      handle_methods.deallocator = &ParallelTensorDeallocator;
      handle_methods.summarize = &ParallelTensorSummarize;
      return TensorHandlePtr(TFE_NewCustomDeviceTensorHandle(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.h

      // shape is explicitly requested where possible.
      int (*num_dims)(void* data, TF_Status* status);
    
      // Computes the axis length at `dim_index`.
      int64_t (*dim)(void* data, int dim_index, TF_Status* status);
    
      void (*deallocator)(void* data);
    
      // Summarizes the value of this tensor. The caller takes ownership of the
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
Back to top