Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for custom (0.21 sec)

  1. tensorflow/c/eager/BUILD

            "//tensorflow/core:test_main",
            "@com_google_absl//absl/strings",
        ],
    )
    
    cc_library(
        name = "custom_device_testutil",
        testonly = True,
        srcs = ["custom_device_testutil.cc"],
        hdrs = ["custom_device_testutil.h"],
        visibility = ["//tensorflow:internal"],
        deps = [
            ":c_api",
            ":c_api_experimental",
            ":c_api_test_util",
    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)
  2. tensorflow/BUILD

    # Build a shared library (DLL) by cc_binary from tf_custom_op_library_additional_deps_impl,
    # it contains all object code from its dependencies.
    # This target is only used for parsing the symbols to be exported in tensorflow.dll.
    # Do NOT depend on it.
    tf_native_cc_binary(
        name = "tf_custom_op_library_additional_deps.dll",
        linkshared = 1,
        linkstatic = 1,
        deps = tf_custom_op_library_additional_deps_impl(),
    )
    
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 09 18:15:11 GMT 2024
    - 53.4K bytes
    - Viewed (8)
  3. tensorflow/c/experimental/grappler/BUILD

            "//tensorflow/core/grappler:grappler_item",
            "//tensorflow/core/grappler/costs:graph_properties",
            "//tensorflow/core/grappler/optimizers:custom_graph_optimizer",
            "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
            "//tensorflow/core/platform:logging",
            "//tensorflow/core/platform:status",
            "@com_google_absl//absl/status",
    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)
  4. tensorflow/c/experimental/gradients/BUILD

            ":not_differentiable",
            "//tensorflow/c/eager:abstract_context",
            "//tensorflow/c/eager:gradients_internal",
        ],
    )
    
    tf_cuda_cc_test(
        name = "custom_gradient_test",
        size = "small",
        srcs = [
            "custom_gradient_test.cc",
        ],
        args = ["--heap_check="],  # TODO(b/174752220): Remove
        tags = tf_cuda_tests_tags(),
        deps = [
            "//tensorflow/c:tf_status_helper",
    Plain Text
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Mon Apr 01 20:39:44 GMT 2024
    - 6.7K bytes
    - Viewed (0)
  5. tensorflow/c/BUILD

    load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
    load(
        "//tensorflow:tensorflow.bzl",
        "check_deps",
        "if_google",
        "if_not_mobile",
        "tf_cc_test",
        "tf_copts",
        "tf_cuda_library",
        "tf_custom_op_library",
        "tf_kernel_library",
    )
    load("//tensorflow:tensorflow.default.bzl", "filegroup", "tf_cuda_cc_test")
    load(
        "//tensorflow/core/tpu:build_defs.bzl",
        "if_libtpu_tf_status",
    )
    
    package(
    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)
Back to top