Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for XLA (0.14 sec)

  1. tensorflow/c/BUILD

        ]),
    )
    
    # Check that c_api_no_xla does not depend on xla.
    check_deps(
        name = "c_api_no_xla_check_deps",
        disallowed_deps = ["//tensorflow/compiler/jit:xla_kernel_creator"],
        deps = [":c_api_no_xla"],
    )
    
    tf_cuda_library(
        name = "c_api_no_xla",
        srcs = [
            "c_api.cc",
            "c_api_function.cc",
        ],
        hdrs = [
            "c_api.h",
        ],
        copts = tf_copts(),
    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)
  2. tensorflow/c/experimental/next_pluggable_device/BUILD

            "@local_tsl//tsl/protobuf:error_codes_proto_impl_cc",
            "@local_xla//xla:shape_util",
            "@local_xla//xla/pjrt:pjrt_api",
            "@local_xla//xla/pjrt:pjrt_c_api_client",
            "@local_xla//xla/pjrt:tfrt_cpu_pjrt_client",
            "@local_xla//xla/pjrt/c:pjrt_c_api_cpu",
            "@local_xla//xla/pjrt/c:pjrt_c_api_hdrs",
            "@local_xla//xla/pjrt/c:pjrt_c_api_wrapper_impl",
        ],
    Plain Text
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 3.8K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/next_pluggable_device/c_api.cc

    #include "tensorflow/c/tf_tensor_internal.h"
    #include "tensorflow/compiler/jit/variable_info.h"
    #include "tensorflow/compiler/jit/variable_info_util.h"
    #include "xla/pjrt/c/pjrt_c_api.h"
    #include "xla/pjrt/c/pjrt_c_api_helpers.h"
    #include "xla/pjrt/pjrt_c_api_client.h"
    #include "xla/pjrt/pjrt_client.h"
    #include "tensorflow/core/common_runtime/next_pluggable_device/plugin_resource.h"
    #include "tensorflow/core/framework/op_kernel.h"
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
  4. .bazelrc

    test:rbe_cross_compile_linux_arm64 --config=rbe_cross_compile_base
    
    # RBE cross-compile configs for XLA Linux Aarch64
    build:rbe_cross_compile_linux_arm64_xla --config=cross_compile_linux_arm64_xla
    build:rbe_cross_compile_linux_arm64_xla --config=rbe_cross_compile_base_xla
    test:rbe_cross_compile_linux_arm64_xla --config=rbe_cross_compile_base_xla
    
    # END LINUX AARCH64 CROSS-COMPILE CONFIGS
    
    # START MACOS CROSS-COMPILE CONFIGS
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu May 02 19:34:20 GMT 2024
    - 52.8K bytes
    - Viewed (2)
  5. tensorflow/c/c_api_experimental.cc

        // These XLA flags are needed to trigger XLA properly from C (more generally
        // non-Python) clients. If this API is called again with `enable` set to
        // false, it is safe to keep these flag values as is.
        tensorflow::MarkForCompilationPassFlags* flags =
            tensorflow::GetMarkForCompilationPassFlags();
        flags->tf_xla_cpu_global_jit = true;
        flags->tf_xla_min_cluster_size = 1;
      } else {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/BUILD

            [],
            otherwise = [
                "@local_xla//xla/stream_executor/cuda:all_runtime",
                "@local_xla//xla/stream_executor/cuda:cuda_driver",
                "@local_xla//xla/stream_executor/cuda:cuda_platform",
                "@local_xla//xla/stream_executor/cuda:cudnn_plugin",
                "@local_xla//xla/stream_executor/cuda:cufft_plugin",
                "@local_xla//xla/stream_executor:cuda_platform",
            ],
        ),
    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)
  7. .gitignore

    .DS_Store
    .ipynb_checkpoints
    node_modules
    /.bazelrc.user
    /.tf_configure.bazelrc
    /xla_configure.bazelrc
    /bazel-*
    /bazel_pip
    /tools/python_bin_path.sh
    /tensorflow/tools/git/gen
    /pip_test
    /_python_build
    *.pyc
    __pycache__
    *.swp
    .vscode/
    cmake_build/
    tensorflow/contrib/cmake/_build/
    .idea/**
    /build/
    [Bb]uild/
    /build_output/
    /tensorflow/core/util/version_info.cc
    /tensorflow/python/framework/fast_tensor_util.cpp
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Wed Jan 31 22:28:59 GMT 2024
    - 934 bytes
    - Viewed (0)
  8. tensorflow/c/eager/BUILD

            "@com_google_absl//absl/container:flat_hash_map",
            "@com_google_absl//absl/memory",
            "@com_google_absl//absl/strings",
            "@com_google_absl//absl/time",
            "@local_xla//xla/tsl/c:tsl_status_internal",
            "@local_xla//xla/tsl/distributed_runtime/coordination:coordination_service_agent",
        ],
        alwayslink = 1,
    )
    
    cc_library(
        name = "c_api_experimental_reader",
        testonly = True,
    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. configure.py

            write_to_bazelrc(
                'build --copt="-DEIGEN_ALTIVEC_ENABLE_MMA_DYNAMIC_DISPATCH=1"')
    
      with_xla_support = environ_cp.get('TF_ENABLE_XLA', None)
      if with_xla_support is not None:
        write_to_bazelrc('build --define=with_xla_support=%s' %
                         ('true' if int(with_xla_support) else 'false'))
    
      set_action_env_var(
          environ_cp, 'TF_NEED_ROCM', 'ROCm', False, bazel_config_name='rocm')
    Python
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 18:25:36 GMT 2024
    - 53.8K bytes
    - Viewed (1)
  10. tensorflow/c/experimental/next_pluggable_device/c_api.h

    #include <cstdint>
    
    #include "tensorflow/c/c_api_macros.h"
    #include "tensorflow/c/kernels.h"
    #include "tensorflow/c/kernels_experimental.h"
    #include "tensorflow/c/tf_buffer.h"
    #include "tensorflow/c/tf_status.h"
    #include "xla/pjrt/c/pjrt_c_api.h"
    
    // --------------------------------------------------------------------------
    // C API for device. The API is under active development and eventually
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Dec 20 20:01:06 GMT 2023
    - 7.2K bytes
    - Viewed (0)
Back to top