Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for custom (0.25 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. RELEASE.md

    *   `tf.saved_model`:
    
        *   SavedModels can now save custom gradients. Use the option
            `tf.saved_model.SaveOption(experimental_custom_gradients=True)` to
            enable this feature. The documentation in
            [Advanced autodiff](https://www.tensorflow.org/guide/advanced_autodiff#custom_gradients)
            has been updated.
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
  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)
  6. CONTRIBUTING.md

    [Google Python Style Guide](https://github.com/google/styleguide/blob/gh-pages/pyguide.md)
    
    Use `pylint` to check your Python changes. To install `pylint` and check a file
    with `pylint` against TensorFlow's custom style definition:
    
    ```bash
    pip install pylint
    pylint --rcfile=tensorflow/tools/ci_build/pylintrc myfile.py
    ```
    
    Note `pylint --rcfile=tensorflow/tools/ci_build/pylintrc` should run from the
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu Mar 21 11:45:51 GMT 2024
    - 15.6K bytes
    - Viewed (0)
Back to top