Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 24 for readers (0.16 sec)

  1. tensorflow/c/checkpoint_reader.cc

      v2_reader_->Seek(kHeaderEntryKey);
      for (v2_reader_->Next(); v2_reader_->Valid(); v2_reader_->Next()) {
        if (filtered_keys.count(string(v2_reader_->key())) > 0) continue;
        CHECK(entry.ParseFromArray(v2_reader_->value().data(),
                                   v2_reader_->value().size()))
            << entry.InitializationErrorString();
        string key(v2_reader_->key());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
  2. tensorflow/c/checkpoint_reader.h

     private:
      // Uses "v2_reader_" to build "var name -> shape" and "var name -> data type"
      // maps; both owned by caller.
      // REQUIRES: "v2_reader_ != nullptr && v2_reader_.status().ok()".
      std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,
                std::unique_ptr<TensorSliceReader::VarToDataTypeMap> >
      BuildV2VarMaps();
    
      // Invariant: exactly one of "reader_" and "v2_reader_" is non-null.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 12 08:49:52 GMT 2023
    - 3.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/BUILD

    )
    
    filegroup(
        name = "device_headers",
        srcs = ["parallel_device.h"],
    )
    
    filegroup(
        name = "device_sources",
        srcs = ["parallel_device.cc"],
    )
    
    filegroup(
        name = "headers",
        srcs = [
            ":device_headers",
            ":lib_headers",
        ],
        visibility = ["//tensorflow/python:__pkg__"],
    )
    
    filegroup(
        name = "sources",
        srcs = [
            ":device_sources",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 01 20:19:06 GMT 2024
    - 4.8K bytes
    - Viewed (0)
  4. tensorflow/c/eager/tfe_monitoring_reader_internal.h

    limitations under the License.
    ==============================================================================*/
    
    #ifndef TENSORFLOW_C_EAGER_TFE_MONITORING_READER_INTERNAL_H_
    #define TENSORFLOW_C_EAGER_TFE_MONITORING_READER_INTERNAL_H_
    
    #include <memory>
    
    #include "tensorflow/core/lib/monitoring/cell_reader.h"
    
    struct TFE_MonitoringCounterReader {
      explicit TFE_MonitoringCounterReader(const char* name) {
        counter = std::make_unique<
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 1.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental_reader_test.cc

      auto* reader = TFE_MonitoringNewCounterReader(counter_name);
      IncrementCounter0(counter);
    
      int64_t actual = TFE_MonitoringReadCounter0(reader);
    
      CHECK_EQ(actual, 1);
    }
    
    TEST(CAPI, MonitoringCellReader1) {
      auto counter_name = "test/counter1";
      auto label_name = "test/label";
      auto* counter = CreateCounter1(counter_name, label_name);
      auto* reader = TFE_MonitoringNewCounterReader(counter_name);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 2.9K bytes
    - Viewed (0)
  6. tensorflow/c/BUILD

            "@local_tsl//tsl/platform:test",
            "@local_tsl//tsl/platform:test_main",
        ],
    )
    
    tf_cuda_library(
        name = "checkpoint_reader",
        srcs = ["checkpoint_reader.cc"],
        hdrs = ["checkpoint_reader.h"],
        visibility = ["//visibility:public"],
        deps = [
            ":tf_status",
            ":tf_status_helper",
            "//tensorflow/core:framework",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_experimental_reader.cc

      return result;
    }
    
    int64_t TFE_MonitoringReadCounter0(TFE_MonitoringCounterReader* cell_reader) {
      int64_t result = cell_reader->Read();
    
      return result;
    }
    
    int64_t TFE_MonitoringReadCounter1(TFE_MonitoringCounterReader* cell_reader,
                                       const char* label) {
      int64_t result = cell_reader->Read(label);
    
      return result;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 1.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/BUILD

        ],
    )
    
    filegroup(
        name = "headers",
        srcs = [
            "c_api.h",
            "c_api_experimental.h",
            "dlpack.h",
        ],
        visibility = ["//tensorflow:__subpackages__"],
    )
    
    filegroup(
        name = "pywrap_headers_monitoring_reader",
        srcs = [
            "c_api_experimental_reader.h",
            "tfe_monitoring_reader_internal.h",
        ],
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 33.3K bytes
    - Viewed (0)
  9. ci/official/containers/linux_arm64/setup.python.sh

      # set up symlink for devtoolset-10
      rm -f /dt10/usr/include/aarch64-linux-gnu/$f
      ln -s /usr/include/aarch64-linux-gnu/$f /dt10/usr/include/aarch64-linux-gnu/$f
    done
    popd
    
    # Python 3.10 include headers fix:
    # sysconfig.get_path('include') incorrectly points to /usr/local/include/python
    # map /usr/include/python3.10 to /usr/local/include/python3.10
    if [[ ! -f "/usr/local/include/$VERSION" ]]; then
    Shell Script
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Sep 29 00:26:34 GMT 2023
    - 2.1K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/grappler/BUILD

    cc_library(
        name = "grappler_hdrs",
        hdrs = ["grappler.h"],
        visibility = ["//tensorflow:internal"],
        deps = [
            "//tensorflow/c:c_api",
            "//tensorflow/c:c_api_macros",
            "//tensorflow/c:tf_status_headers",
        ],
    )
    
    cc_library(
        name = "grappler",
        srcs = ["grappler.cc"],
        hdrs = [
            "grappler.h",
            "grappler_internal.h",
        ],
        visibility = ["//tensorflow:internal"],
    Plain Text
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 2.1K bytes
    - Viewed (0)
Back to top