Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 30 for int64_t (0.21 sec)

  1. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc

        buf.push_back(i);
      }
      // The fetcher just fetches slices of the buffer.
      auto fetcher = [&buf](const string& filename, size_t offset, size_t n,
                            char* buffer, TF_Status* status) -> int64_t {
        int64_t bytes_transferred;
        if (offset < buf.size()) {
          size_t bytes_to_copy = std::min<size_t>(buf.size() - offset, n);
          memcpy(buffer, buf.data() + offset, bytes_to_copy);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:57 GMT 2021
    - 23.2K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/filesystem/modular_filesystem.cc

    }
    
    Status ModularFileSystem::DeleteRecursively(const std::string& dirname,
                                                TransactionToken* token,
                                                int64_t* undeleted_files,
                                                int64_t* undeleted_dirs) {
      if (undeleted_files == nullptr || undeleted_dirs == nullptr)
        return errors::FailedPrecondition(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 23.1K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.cc

            block->state = FetchState::FETCHING;
            block->mu.Unlock();  // Release the lock while making the API call.
            block->data.clear();
            block->data.resize(block_size_, 0);
            int64_t bytes_transferred;
            bytes_transferred = block_fetcher_(key.first, key.second, block_size_,
                                               block->data.data(), status);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 16 01:39:09 GMT 2020
    - 11.1K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

      TapeVSpace vspace(ctx);
      std::vector<int64_t> target_tensor_ids = MakeTensorIDList(targets);
      std::vector<int64_t> source_tensor_ids = MakeTensorIDList(sources);
      tensorflow::gtl::FlatSet<int64_t> sources_set(source_tensor_ids.begin(),
                                                    source_tensor_ids.end());
      std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/tape.h

      // Map from tensor ID to how many references still exist for this tensor in
      // the tape.
      std::unordered_map<int64_t, int64_t> tensor_usage_counts;
    
      // Maps from op ID to how many output tensors of this op still need to have
      // their gradients computed.
      std::unordered_map<int64_t, int64_t> op_missing_tensor;
    };
    
    // If `persistent_tape` is true, op_tape is not changed and none of the
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        return absl::OkStatus();
      }
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override {
        PartialTensorShape shape;
        if (num_dims >= 0) {
          shape = PartialTensorShape(ArraySlice<int64_t>(
              reinterpret_cast<const int64_t*>(dims), num_dims));
        }
        op_->node_builder.Attr(attr_name, shape);
        return absl::OkStatus();
      }
    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)
  7. tensorflow/c/eager/dlpack.cc

    }
    
    // Checks whether the stride array matches the layout of compact, row-majored
    // data.
    bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr,
                                          int ndim) {
      bool valid = true;
      int64_t expected_stride = 1;
      for (int i = ndim - 1; i >= 0; --i) {
        // Empty tensors are always compact regardless of strides.
        if (shape_arr[i] == 0) return true;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test_util.cc

      return th;
    }
    
    TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx) {
      int64_t dims[] = {2, 2};
      double data[] = {1.0, 2.0, 3.0, 4.0};
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_DOUBLE, &dims[0],
                                            sizeof(dims) / sizeof(int64_t), status);
      memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  9. tensorflow/c/eager/c_api_unified_experimental_test.cc

      /* Want to test simple MatMul example:
        [[0,0],    *   [[0,0],    =   [[0,0],
         [0,0]]         [0,0]]         [0,0]]
      */
    
      // Build an abstract input tensor.
      int64_t dims[] = {2, 2};  // Matrices will be 2 x 2
      int num_dims = sizeof(dims) / sizeof(dims[0]);
    
      float vals[] = {0.0f, 0.0f, 0.0f, 0.0f};
      TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  10. tensorflow/c/eager/immediate_execution_context.h

      // Get a list of the names of functions that have been registered.
      virtual std::vector<string> ListFunctionNames() = 0;
    
      struct CacheStats {
        int64_t kernel_cache_size;
        int64_t device_cache_size;
        std::map<std::string, int64_t> func_kernel_cache_entries;
        int64_t local_rendezvous_cache_active_size;
      };
      virtual CacheStats GetCacheStats() = 0;
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 06 08:34:00 GMT 2023
    - 12.3K bytes
    - Viewed (0)
Back to top