Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for pnum (0.13 sec)

  1. tensorflow/c/c_api.cc

                             int num_shapes) {
      std::vector<PartialTensorShape> shapes;
      shapes.reserve(num_shapes);
      for (int i = 0; i < num_shapes; ++i) {
        if (num_dims[i] < 0) {
          shapes.emplace_back();
        } else {
          shapes.emplace_back(ArraySlice<int64_t>(
              reinterpret_cast<const int64_t*>(dims[i]), num_dims[i]));
        }
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

    TF_ShapeAndTypeList* TF_NewShapeAndTypeList(int num_items) {
      TF_ShapeAndTypeList* result = new TF_ShapeAndTypeList;
      result->num_items = num_items;
      result->items = (num_items == 0) ? nullptr : new TF_ShapeAndType[num_items]();
      return result;
    }
    
    void TF_ShapeAndTypeListSetShape(TF_ShapeAndTypeList* shape_list, int index,
                                     const int64_t* dims, int num_dims) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

      /* we don't promise entries would be sorted */
      int num_entries =
          scandir(path, &dir_entries, RemoveSpecialDirectoryEntries, nullptr);
      if (num_entries < 0) {
        TF_SetStatusFromIOError(status, errno, path);
      } else {
        *entries = static_cast<char**>(
            plugin_memory_allocate(num_entries * sizeof((*entries)[0])));
        for (int i = 0; i < num_entries; i++) {
          (*entries)[i] = strdup(dir_entries[i]->d_name);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function.cc

            // artificial restriction and require that when num_opers=-1, such
            // nodes must have a single output.
            if (node->num_outputs() != 1) {
              return InvalidArgument(
                  "When `num_opers` is set to -1, nodes referenced in `inputs` "
                  "must have a single output. Node ",
                  node->name(), " has ", node->num_outputs(),
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  5. tensorflow/c/eager/c_api_unified_experimental_graph.cc

                               ArraySlice<const bool>(b.get(), num_values));
    
        return absl::OkStatus();
      }
      Status SetAttrShapeList(const char* attr_name, const int64_t** dims,
                              const int* num_dims, int num_values) override {
        std::vector<PartialTensorShape> shapes;
        shapes.reserve(num_values);
        for (int i = 0; i < num_values; ++i) {
          if (num_dims[i] < 0) {
            shapes.emplace_back();
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  6. tensorflow/c/eager/c_api.cc

        return -1;
      }
    
      int num_dims = -1;
      status->status = tensorflow::unwrap(h)->NumDims(&num_dims);
      return num_dims;
    }
    
    int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    
      int64_t num_elements = -1;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  7. tensorflow/c/eager/c_api_experimental.cc

    }
    
    TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype,
                                      const int64_t* dims, int num_dims,
                                      TF_Status* status) {
      std::vector<int64_t> dimvec(num_dims);
      for (int i = 0; i < num_dims; ++i) {
        dimvec[i] = static_cast<int64_t>(dims[i]);
      }
    
      if (ctx == nullptr) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  8. tensorflow/c/c_api_test.cc

      const int batch_size = 4;
      const int num_dims = 2;
      int64_t* dims = new int64_t[num_dims];
      int64_t num_elements = 1;
      dims[0] = batch_size;
      dims[1] = 1;
      for (int64_t i = 0; i < num_dims; ++i) {
        num_elements *= dims[i];
      }
      TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims,
                                       sizeof(TF_TString) * num_elements);
      delete[] dims;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top