Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for tf_library (0.15 sec)

  1. tensorflow/compiler/aot/BUILD

    # tfcompile.bzl correctly handles usage from outside of the package that it is
    # defined in.
    
    # A simple test of tf_library from a text protobuf, to enable benchmark_test.
    # This test uses an incompleted graph with a node that is not defined. The
    # compilation works because the undefined node is a feed node.
    tf_library(
        name = "test_graph_tfadd",
        testonly = 1,
        config = "test_graph_tfadd.config.pbtxt",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 16:13:05 UTC 2024
    - 11.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/aot/tests/BUILD

                # expand the "required by all tf_library targets" list in tfcompile.bzl.
                include_standard_runtime_deps = False,
                mlir_components = mlir_component,
                tags = [
                    "manual",
                    "no_mac",  # TODO(b/228273415)
                ],
            ),
            tf_library(
                name = "test_graph_tfadd_with_ckpt" + suffix,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 17.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/aot/tfcompile.bzl

    """Build macro that compiles a TensorFlow graph into a cc_library.
    
    To use from your BUILD file, add the following line to load the macro:
    
    load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
    
    Then call the macro like this:
    
    tf_library(
        name = "test_graph_tfmatmul",
        config = "test_graph_tfmatmul.config.pbtxt",
        cpp_class = "MatMulComp",
        graph = ":test_graph_tfmatmul.pb",
    )
    """
    
    load(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 19:18:08 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental_test.cc

      string lib_path =
          tensorflow::GetDataDependencyFilepath(tensorflow::io::JoinPath(
              "tensorflow", "c", "experimental", "stream_executor", "test",
              "test_pluggable_device.so"));
      TF_Library* lib = TF_LoadPluggableDeviceLibrary(lib_path.c_str(), status);
      TF_Code code = TF_GetCode(status);
      string status_msg(TF_Message(status));
      TF_DeleteStatus(status);
      ASSERT_EQ(TF_OK, code) << status_msg;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

    // for the first time.
    TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename,
                                              TF_Status* status) {
    #if defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
      status->status = tensorflow::errors::Unimplemented(
          "PluggableDevice plugin functionality is not supported on mobile");
      return nullptr;
    #else
      TF_Library* lib_handle = new TF_Library;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental.h

    //
    // On failure, returns nullptr and places an error status in status.
    TF_CAPI_EXPORT extern TF_Library* TF_LoadPluggableDeviceLibrary(
        const char* library_filename, TF_Status* status);
    
    // Frees the memory associated with the library handle.
    // Does NOT unload the library.
    TF_CAPI_EXPORT extern void TF_DeletePluggableDeviceLibraryHandle(
        TF_Library* lib_handle);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/aot/compile.cc

        TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::HloSnapshot> module,
                            computation.Snapshot());
        // Serialize the HloSnapshot deterministically so that all the outputs of a
        // tf_library genrule are deterministic.
        const size_t size = module->ByteSizeLong();
        auto serialized = std::make_unique<char[]>(size);
        TF_RET_CHECK(
            SerializeToBufferDeterministic(*module, serialized.get(), size));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 08:28:57 UTC 2024
    - 11.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/aot/tests/tfcompile_test.cc

    #include "xla/test.h"
    #include "tensorflow/core/platform/regexp.h"
    #include "tensorflow/core/platform/test.h"
    
    // The header files for the tests using mlir_bridge have the _mlir_bridge suffix
    // inherited from the tf_library target names.
    #if defined(ENABLE_MLIR_BRIDGE_TEST)
    #include "tensorflow/compiler/aot/tests/test_graph_tfadd_mlir_bridge.h"
    #include "tensorflow/compiler/aot/tests/test_graph_tfadd_with_ckpt_mlir_bridge.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 26.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/BUILD

    load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
    load("//tensorflow:tensorflow.bzl", "tf_cc_test")
    load("//tensorflow:tensorflow.default.bzl", "filegroup", "get_compatible_with_portable")
    load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
    
    # copybara:uncomment_end(google-only)
    package(
        # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
        default_visibility = ["//visibility:public"],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 35.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tfrt/BUILD

            "@llvm-project//mlir:TransformUtils",
        ],
    )
    
    cc_library(
        name = "constants",
        hdrs = ["constants.h"],
    )
    
    cc_library(
        name = "test_opkernels",
        testonly = True,
        srcs = ["test_opkernels.cc"],
        deps = ["//tensorflow/core:framework"],
        alwayslink = True,
    )
    
    cc_library(
        name = "backend_compiler",
        srcs = ["backend_compiler.cc"],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 19:04:21 UTC 2024
    - 23.6K bytes
    - Viewed (0)
Back to top