Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 47 of 47 for stream_executor_ (0.35 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc

    #include "tensorflow/compiler/tf2xla/xla_compiler.h"
    #include "tensorflow/compiler/tf2xla/xla_helpers.h"
    #include "xla/client/client_library.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/lib/monitoring/cell_reader.h"
    #include "tensorflow/core/lib/monitoring/test_utils.h"
    #include "tensorflow/core/platform/env.h"
    #include "tensorflow/core/platform/test.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 23:59:33 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/device_compiler_test.cc

    #include "tensorflow/compiler/jit/tests/device_compiler_test_helper.h"
    #include "tensorflow/compiler/jit/xla_device_compiler_client.h"
    #include "xla/client/client_library.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/framework/fake_input.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/graph_to_functiondef.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 19.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_launch_util.h

    #include "tensorflow/compiler/jit/xla_tensor.h"
    #include "tensorflow/compiler/tf2xla/xla_compiler.h"
    #include "xla/client/local_client.h"
    #include "xla/pjrt/pjrt_client.h"
    #include "xla/service/shaped_buffer.h"
    #include "xla/stream_executor/device_memory_allocator.h"
    #include "tensorflow/core/framework/allocation_description.pb.h"
    #include "tensorflow/core/framework/tensor.h"
    #include "tensorflow/core/lib/core/status.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_device.h

      // `device`.
      static Status GetMetadataFromDevice(DeviceBase* device,
                                          const XlaDevice::Metadata** metadata);
    
      struct Options {
        // The StreamExecutor platform. Not owned. Must be non-null.
        se::Platform* platform = nullptr;
    
        // The device name's prefix (e.g., "/task:7")
        string device_name_prefix;
    
        // The name of the XLA device (e.g., "XLA_CPU")
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/BUILD

            "@llvm-project//llvm:Support",
            "@llvm-project//mlir:Pass",
            "@local_xla//xla:shape_util",
            "@local_xla//xla/client:client_library",
            "@local_xla//xla/stream_executor",
        ],
    )
    
    tf_cc_test(
        name = "convert_tf_quant_types_test",
        srcs = ["passes/bridge/convert_tf_quant_types_test.cc"],
        deps = [
            ":bridge_passes",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 28.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/xla_launch_util.cc

    #include "xla/pjrt/pjrt_future.h"
    #include "xla/pjrt/pjrt_stream_executor_client.h"
    #include "xla/pjrt/tracked_device_buffer.h"
    #include "xla/shape_util.h"
    #include "xla/status_macros.h"
    #include "xla/stream_executor/platform_manager.h"
    #include "tensorflow/core/common_runtime/dma_helper.h"
    #include "tensorflow/core/common_runtime/gpu/gpu_serving_device_selector.h"
    #include "tensorflow/core/common_runtime/gpu_device_context.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  7. RELEASE.md

      * Added function `experimental.extension_type.as_dict()`, which converts an instance of `tf.experimental.ExtensionType` to a `dict` representation.
    * `stream_executor`
      * Top level `stream_executor` directory has been deleted, users should use equivalent headers and targets under `compiler/xla/stream_executor`.
    * `tf.nn`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
Back to top